105 lines
2.8 KiB
Go
105 lines
2.8 KiB
Go
// Package transform ...
|
|
//
|
|
// Description : transform ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 2021-11-30 4:04 下午
|
|
package transform
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
|
|
"github.com/ddliu/go-httpclient"
|
|
)
|
|
|
|
// NewApolloMigrate 从apollo迁移数据
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 4:06 下午 2021/11/30
|
|
func NewApolloMigrate(getAllApolloConfigURL string, appID string, env string, namespace string, cookie map[string]string, header map[string]string) ITransform {
|
|
if cookie == nil {
|
|
cookie = make(map[string]string)
|
|
header = make(map[string]string)
|
|
}
|
|
return &ApolloMigrate{
|
|
getDataURL: getAllApolloConfigURL,
|
|
cookie: cookie,
|
|
header: header,
|
|
appID: appID,
|
|
}
|
|
}
|
|
|
|
// ApolloMigrate 从apollo迁移数据
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 4:05 下午 2021/11/30
|
|
type ApolloMigrate struct {
|
|
getDataURL string
|
|
cookie map[string]string
|
|
header map[string]string
|
|
appID string
|
|
}
|
|
|
|
// GetNamespace 迁移之后，存储到哪一个命名空间
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 4:17 下午 2021/11/30
|
|
func (a *ApolloMigrate) GetNamespace() string {
|
|
return a.appID
|
|
}
|
|
|
|
// GetSourceData ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 4:18 下午 2021/11/30
|
|
func (a *ApolloMigrate) GetSourceData() (map[string]string, error) {
|
|
var (
|
|
err error
|
|
resp *httpclient.Response
|
|
responseData []byte
|
|
)
|
|
hc := httpclient.NewHttpClient()
|
|
cookieList := make([]*http.Cookie, 0)
|
|
for name, val := range a.cookie {
|
|
cookieList = append(cookieList, &http.Cookie{Name: name, Value: val})
|
|
}
|
|
resp, err = hc.WithHeaders(a.header).WithCookie(cookieList...).Get(a.getDataURL)
|
|
if nil != err {
|
|
return make(map[string]string), errors.New("apollo接口请求失败 : " + err.Error())
|
|
}
|
|
responseData, err = ioutil.ReadAll(resp.Body)
|
|
if nil != err {
|
|
return make(map[string]string), errors.New("apollo 响应数据读取失败 : " + err.Error())
|
|
}
|
|
var cfgList []ApolloData
|
|
if err = json.Unmarshal(responseData, &cfgList); nil != err {
|
|
return make(map[string]string), errors.New("apollo 数据解析失败 : " + err.Error())
|
|
}
|
|
if len(cfgList) == 0 {
|
|
return make(map[string]string), errors.New("当前apollo配置读取到的数据为空")
|
|
}
|
|
var realCfg map[string]string
|
|
if err = json.Unmarshal([]byte(cfgList[0].Configurations), &realCfg); nil != err {
|
|
return make(map[string]string), errors.New("真实配置文件解析失败 ")
|
|
}
|
|
return realCfg, nil
|
|
}
|
|
|
|
// MigrateKeyFailCallback ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 4:58 下午 2021/11/30
|
|
func (a *ApolloMigrate) MigrateKeyFailCallback(namespace string, key string, val string, err error) {
|
|
fmt.Println(namespace, key, val, err)
|
|
}
|