96 lines
2.5 KiB
Go
96 lines
2.5 KiB
Go
package cmd
|
||
|
||
import (
|
||
"context"
|
||
"github.com/ayflying/utility_go/s3"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/net/ghttp"
|
||
"github.com/gogf/gf/v2/os/gcfg"
|
||
"github.com/gogf/gf/v2/os/gcmd"
|
||
"os"
|
||
"time"
|
||
)
|
||
|
||
type serverCfg struct {
|
||
Name string `json:"name" dc:"服务名"`
|
||
Address string `json:"address" dc:"服务地址"`
|
||
Prod bool `json:"prod" dc:"是否生产环境"`
|
||
}
|
||
|
||
type UpdateReq struct {
|
||
File *ghttp.UploadFile `json:"file" binding:"required" dc:"文件"`
|
||
FileUrl string `json:"file_url" dc:"文件地址"`
|
||
}
|
||
|
||
var (
|
||
Update = gcmd.Command{
|
||
Name: "update",
|
||
Usage: "update",
|
||
Brief: "更新系统",
|
||
Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
|
||
|
||
g.Log().Info(ctx, "准备上传更新文件")
|
||
//加载编辑配置文件
|
||
g.Cfg("hack").GetAdapter().(*gcfg.AdapterFile).SetFileName("hack/config.yaml")
|
||
getFileName, err := g.Cfg("hack").Get(ctx, "gfcli.build.name")
|
||
|
||
var list []*serverCfg
|
||
serverList := g.Cfg().MustGet(ctx, "server_list")
|
||
serverList.Scan(&list)
|
||
|
||
//如果有p或者prod参数，则删除prod字段为true的服务
|
||
if parser.GetOpt("a").IsNil() {
|
||
var temp []*serverCfg
|
||
for _, v := range list {
|
||
if v.Prod == false {
|
||
temp = append(temp, v)
|
||
}
|
||
}
|
||
list = temp
|
||
} else {
|
||
g.Dump("升级", parser.GetOpt("a"))
|
||
}
|
||
|
||
g.Dump("需要更新的服务器", list)
|
||
|
||
filename := "linux_amd64/shining_u_server"
|
||
obj, err := os.Open(filename)
|
||
ff, err := obj.Stat()
|
||
var s3Mod *s3.Mod
|
||
//切换s3目标
|
||
updateServerS3Name, _ := g.Config().Get(ctx, "update_server_s3_name")
|
||
if updateServerS3Name != nil {
|
||
s3Mod = s3.New(updateServerS3Name.String())
|
||
} else {
|
||
s3Mod = s3.New("default")
|
||
}
|
||
bucketName := s3Mod.GetCfg().BucketName
|
||
|
||
_, err = s3Mod.PutObject(obj, getFileName.String(), bucketName, ff.Size())
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
//上传当前文件
|
||
url, err := s3Mod.GetFileUrl(getFileName.String(), bucketName)
|
||
g.Log().Debugf(ctx, "下载地址:%v", url)
|
||
|
||
//循环服务器，推送更新
|
||
for _, v := range list {
|
||
address := v.Address
|
||
g.Log().Debugf(ctx, "准备同步服务器:%v,url=%v", v.Name, address+"/callback/update")
|
||
get, err := g.Client().Discovery(nil).Timeout(time.Minute*1).Post(ctx, address+"/callback/update", &UpdateReq{
|
||
FileUrl: url.String(),
|
||
})
|
||
if err != nil {
|
||
g.Log().Error(ctx, err)
|
||
}
|
||
defer get.Close()
|
||
g.Log().Debugf(ctx, "同步服务器:%v,完成=%v", v.Name, address)
|
||
}
|
||
|
||
return
|
||
},
|
||
}
|
||
)
|