/
sync.go
114 lines (98 loc) 路 3.32 KB
/
sync.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package remote
import (
"encoding/json"
"fmt"
"net/url"
"path"
"github.com/apolloconfig/agollo/v4/component/log"
"github.com/apolloconfig/agollo/v4/constant"
"github.com/apolloconfig/agollo/v4/env/config"
"github.com/apolloconfig/agollo/v4/extension"
"github.com/apolloconfig/agollo/v4/protocol/http"
"github.com/apolloconfig/agollo/v4/utils"
)
// CreateSyncApolloConfig 鍒涘缓鍚屾鑾峰彇 Apollo 閰嶇疆
func CreateSyncApolloConfig() ApolloConfig {
a := &syncApolloConfig{}
a.remoteApollo = a
return a
}
type syncApolloConfig struct {
AbsApolloConfig
}
func (*syncApolloConfig) GetNotifyURLSuffix(notifications string, config config.AppConfig) string {
return ""
}
func (*syncApolloConfig) GetSyncURI(config config.AppConfig, namespaceName string) string {
return fmt.Sprintf("configfiles/json/%s/%s/%s?&ip=%s&label=%s",
url.QueryEscape(config.AppID),
url.QueryEscape(config.Cluster),
url.QueryEscape(namespaceName),
utils.GetInternal(),
url.QueryEscape(config.Label))
}
func (*syncApolloConfig) CallBack(namespace string) http.CallBack {
return http.CallBack{
SuccessCallBack: processJSONFiles,
NotModifyCallBack: touchApolloConfigCache,
Namespace: namespace,
}
}
func processJSONFiles(b []byte, callback http.CallBack) (o interface{}, err error) {
apolloConfig := &config.ApolloConfig{}
apolloConfig.NamespaceName = callback.Namespace
configurations := make(map[string]interface{}, 0)
apolloConfig.Configurations = configurations
err = json.Unmarshal(b, &apolloConfig.Configurations)
if utils.IsNotNil(err) {
return nil, err
}
parser := extension.GetFormatParser(constant.ConfigFileFormat(path.Ext(apolloConfig.NamespaceName)))
if parser == nil {
parser = extension.GetFormatParser(constant.DEFAULT)
}
if parser == nil {
return apolloConfig, nil
}
content, ok := configurations[defaultContentKey]
if !ok {
content = string(b)
}
m, err := parser.Parse(content)
if err != nil {
log.Debugf("GetContent fail ! error: %v", err)
}
if len(m) > 0 {
apolloConfig.Configurations = m
}
return apolloConfig, nil
}
func (a *syncApolloConfig) Sync(appConfigFunc func() config.AppConfig) []*config.ApolloConfig {
appConfig := appConfigFunc()
configs := make([]*config.ApolloConfig, 0, 8)
config.SplitNamespaces(appConfig.NamespaceName, func(namespace string) {
apolloConfig := a.SyncWithNamespace(namespace, appConfigFunc)
if apolloConfig != nil {
configs = append(configs, apolloConfig)
return
}
configs = append(configs, loadBackupConfig(appConfig.NamespaceName, appConfig)...)
})
return configs
}