11 | {
12 | "GO-OPS-X-APPID":"你的appid",
13 | "GO-OPS-X-SIGNATURE":"签名值",
14 | "GO-OPS-X-TIMESTAMP":"时间戳",
15 | "GO-OPS-X-NONCE":"请求唯一id"
16 | }
17 |
18 | 使用sha1作为签名算法,seckey作为签名使用key,依次把apikey、nonce、timestame、body 写入签名内容中,最后对签名结果进行base64转码
20 |Golang
21 |
22 | func GetSign(apikey string, seckey string, nonce, timestamp string, body []byte) (r string) {
23 | key := []byte(seckey)
24 | mac := hmac.New(sha1.New, key)
25 | mac.Write([]byte(apikey))
26 | mac.Write([]byte(nonce))
27 | mac.Write([]byte(timestamp))
28 | mac.Write(body)
29 | r = base64.StdEncoding.EncodeToString(mac.Sum(nil))
30 | return
31 | }
32 |
33 | `
34 | )
35 |
--------------------------------------------------------------------------------
/internal/consts/error_code.go:
--------------------------------------------------------------------------------
1 | package consts
2 |
3 | const (
4 | API_CODE_INVALID_APPID = 1001 // 无效的APPID
5 |
6 | API_CODE_NODE_NOT_FOUND = 2001 // 节点不存在
7 |
8 | API_CODE_NODE_CONNECT_ERR = 2001 // 连接节点失败
9 |
10 | API_CODE_NODE_NOT_RUN = 2003 // 节点没有在运行
11 | )
12 |
--------------------------------------------------------------------------------
/internal/controller/agent.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | v1 "go-ops/pkg/api/v1"
6 | )
7 |
8 | var Agent *agent = new(agent)
9 |
10 | type agent struct{}
11 |
12 | func (self *agent) Create(ctx context.Context, req *v1.AddAgentReq) (res *v1.AgentInfoRes, err error) {
13 | return
14 | }
15 |
16 | func (self *agent) Update(ctx context.Context, req *v1.UpdateAgentReq) (res *v1.AgentInfoRes, err error) {
17 | return
18 | }
19 |
20 | func (self *agent) Query(ctx context.Context, req *v1.QueryAgentStatusReq) (res *v1.QueryAgentStatusRes, err error) {
21 | return
22 | }
23 |
--------------------------------------------------------------------------------
/internal/controller/app.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/service"
6 | v1 "go-ops/pkg/api/v1"
7 | )
8 |
9 | var App *app = new(app)
10 |
11 | type app struct{}
12 |
13 | func (self *app) Create(ctx context.Context, req *v1.AddAppReq) (res *v1.AddAppRes, err error) {
14 | return service.App().Create(ctx, req)
15 | }
16 |
17 | func (self *app) Update(ctx context.Context, req *v1.UpdateAppReq) (res *v1.AddAppRes, err error) {
18 | return service.App().Update(ctx, req)
19 | }
20 |
21 | func (self *app) Query(ctx context.Context, req *v1.QueryAppReq) (res *v1.QueryAppRes, err error) {
22 | return service.App().Query(ctx, req)
23 | }
24 |
25 | func (self *app) SingleQuery(ctx context.Context, req *v1.QuerySingleAppReq) (res *v1.AddAppRes, err error) {
26 | return service.App().SingleQuery(ctx, req)
27 | }
28 |
29 | func (self *app) Delete(ctx context.Context, req *v1.DeleteAppReq) (res v1.DeleteRes, err error) {
30 | err = service.App().Delete(ctx, req)
31 | if err != nil {
32 | res = v1.DeleteRes("删除失败")
33 | return
34 | }
35 | res = v1.DeleteRes("删除成功")
36 | return
37 | }
38 |
--------------------------------------------------------------------------------
/internal/controller/auth.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/service"
6 | v1 "go-ops/pkg/api/v1"
7 | )
8 |
9 | type authController struct{}
10 |
11 | var Auth = authController{}
12 |
13 | func (c *authController) Login(ctx context.Context, req *v1.AuthLoginReq) (res *v1.AuthLoginRes, err error) {
14 | res = &v1.AuthLoginRes{}
15 | res.Token, res.Expire = service.Auth().LoginHandler(ctx)
16 | return
17 | }
18 |
19 | func (c *authController) RefreshToken(ctx context.Context, req *v1.AuthRefreshTokenReq) (res *v1.AuthRefreshTokenRes, err error) {
20 | res = &v1.AuthRefreshTokenRes{}
21 | res.Token, res.Expire = service.Auth().RefreshHandler(ctx)
22 | return
23 | }
24 |
25 | func (c *authController) Logout(ctx context.Context, req *v1.AuthLogoutReq) (res *v1.AuthLogoutRes, err error) {
26 | service.Auth().LogoutHandler(ctx)
27 | return
28 | }
29 |
--------------------------------------------------------------------------------
/internal/controller/check.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/service"
6 | v1 "go-ops/pkg/api/v1"
7 |
8 | "github.com/gogf/gf/v2/os/glog"
9 | )
10 |
11 | var CheckItem *checkitem = new(checkitem)
12 |
13 | type checkitem struct{}
14 |
15 | func (self *checkitem) Create(ctx context.Context, req *v1.AddCheckItemReq) (res *v1.CheckItemRes, err error) {
16 | res, err = service.Checkitem().Create(ctx, req)
17 | if err != nil {
18 | glog.Errorf(ctx, "create checkitem err:%v", err)
19 | return
20 | }
21 | return
22 | }
23 |
24 | func (self *checkitem) Update(ctx context.Context, req *v1.UpdateCheckItemReq) (res *v1.CheckItemRes, err error) {
25 | res, err = service.Checkitem().Update(ctx, req)
26 | if err != nil {
27 | glog.Errorf(ctx, "update checkitem err:%v", err)
28 | return
29 | }
30 | return
31 | }
32 |
33 | func (self *checkitem) Query(ctx context.Context, req *v1.QueryCheckItemReq) (res *v1.QueryCheckItemRes, err error) {
34 | res, err = service.Checkitem().Query(ctx, req)
35 | if err != nil {
36 | glog.Errorf(ctx, "query checkitem err:%v", err)
37 | return
38 | }
39 | return
40 | }
41 |
42 | func (self *checkitem) Delete(ctx context.Context, req *v1.DeleteCheckItemReq) (res v1.DeleteRes, err error) {
43 | err = service.Checkitem().Delete(ctx, req)
44 | if err != nil {
45 | glog.Errorf(ctx, "delete checkitem err:%v", err)
46 | res = v1.DeleteRes("删除失败")
47 | return
48 | }
49 | res = v1.DeleteRes("删除成功")
50 | return
51 | }
52 |
--------------------------------------------------------------------------------
/internal/controller/check_tpl.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | v1 "go-ops/pkg/api/v1"
6 | )
7 |
8 | var CheckTpl *checktpl = new(checktpl)
9 |
10 | type checktpl struct{}
11 |
12 | func (self *checktpl) Create(ctx context.Context, req *v1.AddCheckTplReq) (res *v1.CheckTplItemRes, err error) {
13 | return
14 | }
15 |
16 | func (self *checktpl) Update(ctx context.Context, req *v1.UpdateCheckTplReq) (res *v1.CheckTplItemRes, err error) {
17 | return
18 | }
19 |
20 | func (self *checktpl) Query(ctx context.Context, req *v1.QueryCheckTplReq) (res *v1.CheckTplItemRes, err error) {
21 | return
22 | }
23 |
24 | func (self *checktpl) Delete(ctx context.Context, req *v1.DeleteCheckTplReq) (res v1.DeleteRes, err error) {
25 | return
26 | }
27 |
--------------------------------------------------------------------------------
/internal/controller/download_file.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/model"
6 | "go-ops/internal/peer"
7 | "go-ops/internal/service"
8 | v1 "go-ops/pkg/api/v1"
9 | )
10 |
11 | var DownloadFileTask = &downloadFileTask{}
12 |
13 | type downloadFileTask struct {
14 | }
15 |
16 | func (self *downloadFileTask) Create(ctx context.Context, req *v1.DownloadFileReq) (res *v1.DownloadfileRes, err error) {
17 |
18 | createFunc := func(peerid string, job *model.DownloadFileJob) error {
19 | err = peer.SendMsgAsync(peer.GetOspPeer().PNet, job, peerid, "")
20 | return err
21 | }
22 | taskid, err := service.Task().CreateFileDownload(ctx, req, createFunc)
23 | if err != nil {
24 | return
25 | }
26 | res = new(v1.DownloadfileRes)
27 | res.Taskid = taskid
28 | return
29 | }
30 |
31 | func (self *downloadFileTask) Details(ctx context.Context, req *v1.DownloadFileDetailsReq) (res *v1.DownloadfileRes, err error) {
32 | return service.Task().GetFileDownloadTaskInfo(ctx, req.Taskid)
33 | }
34 |
--------------------------------------------------------------------------------
/internal/controller/middleware.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "go-ops/internal/service"
5 | "go-ops/pkg/util"
6 | "net/http"
7 |
8 | "github.com/gogf/gf/v2/net/ghttp"
9 | "github.com/gogf/gf/v2/os/glog"
10 | )
11 |
12 | const (
13 | GoOpsHeaderAppId = "GO-OPS-X-APPID"
14 | GoOpsHeaderSignature = "GO-OPS-X-SIGNATURE"
15 | GoOpsHeaderTimestamp = "GO-OPS-X-TIMESTAMP"
16 | GoOpsHeaderNonce = "GO-OPS-X-NONCE"
17 | GoOpsHeaderToken = "GO-OPS-X-TOKEN"
18 | )
19 |
20 | func MiddlewareGetApp(r *ghttp.Request) {
21 | appid := r.GetHeader(GoOpsHeaderAppId)
22 | if appid == "" {
23 | glog.Error(r.GetCtx(), "appid is empty")
24 | r.Response.WriteStatus(http.StatusForbidden)
25 | return
26 | }
27 |
28 | app, err := service.App().GetApp(r.GetCtx(), appid)
29 | if err != nil {
30 | glog.Errorf(r.GetCtx(), "get app:%s err:%v", appid, err)
31 | r.Response.WriteStatus(http.StatusForbidden)
32 | return
33 | }
34 |
35 | signature := r.GetHeader(GoOpsHeaderSignature)
36 | timestamp := r.GetHeader(GoOpsHeaderTimestamp)
37 | nonce := r.GetHeader(GoOpsHeaderNonce)
38 | body := r.GetBody()
39 | sign := util.GetSign(app.ApiKey, app.SecKey, nonce, timestamp, body)
40 |
41 | if sign != signature {
42 | glog.Errorf(r.GetCtx(), "sign is not match, req sign:%s server sign:%s", signature, sign)
43 | r.Response.WriteStatus(http.StatusForbidden)
44 | return
45 | }
46 |
47 | r.Middleware.Next()
48 |
49 | }
50 |
51 | func AuthUser(r *ghttp.Request) {
52 |
53 | appid := r.GetHeader(GoOpsHeaderAppId)
54 | if appid != "" {
55 | // 如果是app
56 | MiddlewareGetApp(r)
57 | } else {
58 | service.Auth().MiddlewareFunc()(r)
59 | r.Middleware.Next()
60 | }
61 |
62 | }
63 |
--------------------------------------------------------------------------------
/internal/controller/plugin.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/service"
6 | v1 "go-ops/pkg/api/v1"
7 | )
8 |
9 | var Plugin *plugin = new(plugin)
10 |
11 | type plugin struct{}
12 |
13 | func (self *plugin) Create(ctx context.Context, req *v1.AddPluginReq) (res *v1.PluginItemRes, err error) {
14 | return service.Plugin().Create(ctx, req)
15 | }
16 |
17 | func (self *plugin) Update(ctx context.Context, req *v1.UpdatePluginReq) (res *v1.PluginItemRes, err error) {
18 | return service.Plugin().Update(ctx, req)
19 | }
20 |
21 | func (self *plugin) Query(ctx context.Context, req *v1.QueryPluginReq) (res *v1.QueryPluginRes, err error) {
22 | return service.Plugin().Query(ctx, req)
23 | }
24 |
25 | func (self *plugin) Delete(ctx context.Context, req *v1.DeletePluginReq) (res v1.DeleteRes, err error) {
26 | err = service.Plugin().Delete(ctx, req)
27 | if err != nil {
28 | res = v1.DeleteRes("删除失败")
29 | return
30 | }
31 | res = v1.DeleteRes("删除成功")
32 | return
33 | }
34 |
--------------------------------------------------------------------------------
/internal/controller/script.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/service"
6 | v1 "go-ops/pkg/api/v1"
7 |
8 | "github.com/gogf/gf/v2/os/glog"
9 | )
10 |
11 | var Script *script = new(script)
12 |
13 | type script struct{}
14 |
15 | func (self *script) Create(ctx context.Context, req *v1.AddScriptReq) (res *v1.ScriptItemRes, err error) {
16 |
17 | res, err = service.Script().Create(ctx, req)
18 | if err != nil {
19 | glog.Errorf(ctx, "create script err:%v", err)
20 | }
21 | return
22 | }
23 |
24 | func (self *script) Update(ctx context.Context, req *v1.UpdateScriptReq) (res *v1.ScriptItemRes, err error) {
25 | res, err = service.Script().Update(ctx, req)
26 | if err != nil {
27 | glog.Errorf(ctx, "update script err:%v", err)
28 | }
29 | return
30 | }
31 |
32 | func (self *script) Query(ctx context.Context, req *v1.ScriptQueryReq) (res *v1.ScriptInfoRes, err error) {
33 | res, err = service.Script().Query(ctx, req)
34 | if err != nil {
35 | glog.Errorf(ctx, "query script err:%v", err)
36 | }
37 | return
38 | }
39 |
40 | func (self *script) Delete(ctx context.Context, req *v1.DeleteScriptReq) (res v1.DeleteScriptRes, err error) {
41 | err = service.Script().Delete(ctx, req)
42 | if err != nil {
43 | glog.Errorf(ctx, "delete script err:%v", err)
44 | res = v1.DeleteScriptRes("删除失败")
45 | return
46 | }
47 | res = v1.DeleteScriptRes("删除成功")
48 | return
49 | }
50 |
--------------------------------------------------------------------------------
/internal/controller/task.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/service"
6 | v1 "go-ops/pkg/api/v1"
7 | )
8 |
9 | var Task *task = new(task)
10 |
11 | type task struct{}
12 |
13 | func (self *task) Query(ctx context.Context, req *v1.TaskQueryReq) (res *v1.TaskInfoRes, err error) {
14 | return service.Task().QueryTask(ctx, req)
15 | }
16 |
17 | func (self *task) Get(ctx context.Context, req *v1.TaskInfoReq) (res v1.TaskDetailRes, err error) {
18 | taskinfo, err := service.Task().GetTaskInfo(ctx, req)
19 | if err != nil {
20 | return
21 | }
22 | res = v1.TaskDetailRes(*taskinfo)
23 | return
24 | }
25 |
--------------------------------------------------------------------------------
/internal/controller/task_cron.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/service"
6 | v1 "go-ops/pkg/api/v1"
7 |
8 | "github.com/gogf/gf/v2/os/glog"
9 | )
10 |
11 | var TaskCron *taskcron = new(taskcron)
12 |
13 | type taskcron struct{}
14 |
15 | func (self *taskcron) Create(ctx context.Context, req *v1.AddCronTaskReq) (res *v1.CronTaskItemRes, err error) {
16 | res, err = service.TaskCron().Create(ctx, req)
17 | if err != nil {
18 | glog.Errorf(ctx, "create task cron err:%v", err)
19 | }
20 | return
21 | }
22 |
23 | func (self *taskcron) Update(ctx context.Context, req *v1.UpdateCronTaskReq) (res *v1.CronTaskItemRes, err error) {
24 | return service.TaskCron().Update(ctx, req)
25 | }
26 |
27 | func (self *taskcron) Delete(ctx context.Context, req *v1.DeleteCronTaskReq) (res *v1.DeleteTaskPresetRes, err error) {
28 |
29 | res = new(v1.DeleteTaskPresetRes)
30 | err = service.TaskCron().Delete(ctx, req)
31 | if err != nil {
32 | res.Message = err.Error()
33 | glog.Errorf(ctx, "delete task cron err:%v", err)
34 | return
35 | }
36 | res.Message = "DELETED SUCCESS!"
37 | return
38 | }
39 |
40 | func (self *taskcron) Query(ctx context.Context, req *v1.QueryCronTaskReq) (res *v1.QueryCronTaskRes, err error) {
41 | res, err = service.TaskCron().Query(ctx, req)
42 | if err != nil {
43 | glog.Errorf(ctx, "query task cron err:%v", err)
44 | }
45 | return
46 | }
47 |
48 | func (self *taskcron) Start(ctx context.Context, req *v1.StartCronTaskReq) (res v1.CronTaskOpRes, err error) {
49 | return
50 | }
51 |
52 | func (self *taskcron) Stop(ctx context.Context, req *v1.StopCronTaskReq) (res v1.CronTaskOpRes, err error) {
53 | return
54 | }
55 |
--------------------------------------------------------------------------------
/internal/controller/task_preset.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/service"
6 | v1 "go-ops/pkg/api/v1"
7 | )
8 |
9 | var TaskPreset *taskPreset = new(taskPreset)
10 |
11 | type taskPreset struct{}
12 |
13 | func (self *taskPreset) Create(ctx context.Context, req *v1.AddTaskPresetReq) (res *v1.TaskPresetItemRes, err error) {
14 | return service.TaskPreset().Create(ctx, req)
15 | }
16 |
17 | func (self *taskPreset) Update(ctx context.Context, req *v1.UpdateTaskPresetReq) (res *v1.TaskPresetItemRes, err error) {
18 | return service.TaskPreset().Update(ctx, req)
19 | }
20 |
21 | func (self *taskPreset) Delete(ctx context.Context, req *v1.DeleteTaskPresetReq) (res *v1.DeleteTaskPresetRes, err error) {
22 | err = service.TaskPreset().Delete(ctx, req)
23 | res = new(v1.DeleteTaskPresetRes)
24 | if err != nil {
25 | res.Message = err.Error()
26 | return
27 | }
28 | res.Message = "DELETED SUCCESS!"
29 | return
30 | }
31 |
32 | func (self *taskPreset) Query(ctx context.Context, req *v1.QueryTaskPresetReq) (res *v1.QueryTaskPresetRes, err error) {
33 | return service.TaskPreset().Query(ctx, req)
34 | }
35 |
--------------------------------------------------------------------------------
/internal/controller/user.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/service"
6 | v1 "go-ops/pkg/api/v1"
7 |
8 | "github.com/gogf/gf/v2/util/gconv"
9 | )
10 |
11 | type user struct{}
12 |
13 | var User = user{}
14 |
15 | func (self *user) CurrentInfo(ctx context.Context, req *v1.CurrentUserInfoReq) (res *v1.UserInfoRes, erro error) {
16 |
17 | id := service.Auth().GetIdentity(ctx)
18 |
19 | uid := gconv.String(id)
20 |
21 | user, err := service.User().Get(ctx, uid)
22 | if err != nil {
23 | return
24 | }
25 |
26 | res = &v1.UserInfoRes{
27 | Uid: uid,
28 | Username: user.Username,
29 | Email: user.Email,
30 | Avatar: user.Avatar,
31 | Phone: user.Phone,
32 | }
33 | return
34 |
35 | }
36 |
--------------------------------------------------------------------------------
/internal/controller/vm.go:
--------------------------------------------------------------------------------
1 | package controller
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/model"
6 | "go-ops/internal/peer"
7 | "go-ops/internal/service"
8 | v1 "go-ops/pkg/api/v1"
9 | "strings"
10 | "time"
11 | )
12 |
13 | var VM *vm = new(vm)
14 |
15 | type vm struct{}
16 |
17 | func (self *vm) Create(ctx context.Context, req *v1.AddVmReq) (res *v1.VmItemRes, err error) {
18 | return service.VM().Create(ctx, req)
19 | }
20 |
21 | func (self *vm) Update(ctx context.Context, req *v1.UpdateVmReq) (res *v1.VmItemRes, err error) {
22 | return service.VM().Update(ctx, req)
23 | }
24 |
25 | func (self *vm) Query(ctx context.Context, req *v1.QueryVmReq) (res *v1.QueryVmRes, err error) {
26 | res, err = service.VM().Query(ctx, req)
27 | if err != nil {
28 | return
29 | }
30 | if len(res.List) > 0 {
31 | return
32 | }
33 |
34 | req.Hostname = strings.TrimSpace(req.Hostname)
35 |
36 | if req.Hostname == "" {
37 | return
38 | }
39 |
40 | // 没有查找到主机信息，需要到网络中查找
41 |
42 | err = peer.SendMsgBroadCast(peer.GetOspPeer().PNet, &model.GetPeerInfo{HostName: req.Hostname})
43 | if err != nil {
44 | return
45 | }
46 |
47 | for i := 0; i < 30; i++ {
48 | res, err = service.VM().Query(ctx, req)
49 | if err != nil {
50 | return
51 | }
52 | if len(res.List) > 0 {
53 | return
54 | }
55 | time.Sleep(time.Second * 2)
56 | }
57 | return
58 | }
59 |
60 | func (self *vm) Delete(ctx context.Context, req *v1.DeleteVmReq) (res v1.DeleteRes, err error) {
61 | err = service.VM().Delete(ctx, req)
62 | if err != nil {
63 | res = v1.DeleteRes("删除失败")
64 | return
65 | }
66 | res = v1.DeleteRes("删除成功")
67 | return
68 | }
69 |
--------------------------------------------------------------------------------
/internal/model/agent_info.go:
--------------------------------------------------------------------------------
1 | package model
2 |
3 | type AgentInfo struct {
4 | Name string `json:"name" dc:"agent 名称"`
5 | Version string `json:"version" dc:"版本信息"`
6 | UrlAddress string `json:"urlAddress" dc:"下载地址"`
7 | Status string `json:"status" dc:"agent状态"`
8 | Timeout int `json:"timeout" dc:"执行命令超时时间"`
9 | }
10 |
11 | type AgentDetails struct {
12 | AgentInfo
13 | Details string `json:"details" dc:"agent 运行详细信息"`
14 | }
15 |
--------------------------------------------------------------------------------
/internal/model/download_file.go:
--------------------------------------------------------------------------------
1 | package model
2 |
3 | import (
4 | "go-ops/pkg/message"
5 | "go-ops/pkg/util"
6 | "reflect"
7 | )
8 |
9 | type DownloadFileInfo struct {
10 | Filename string `json:"filename" dc:"文件名称"` // 文件名称
11 | Address string `json:"address" dc:"文件的下载url地址"` // 文件下载地址
12 | Path string `json:"path" dc:"文件的保存路径"` // 文件保存路径
13 | AutoCreatePath bool `json:"auto_create_path" dc:"是否自动创建文件夹"` // 自动创建文件夹
14 | Replace bool `json:"replace" dc:"如果文件已经存在,是否替换文件"` // 是否替换
15 | }
16 |
17 | type DownloadFileRes struct {
18 | Filename string `json:"filename" dc:"文件名"` //
19 | Code ResCode `json:"code" dc:"代码"`
20 | Msg string `json:"msg"`
21 | }
22 |
23 | type DownloadFileJobRes struct {
24 | PeerId string `json:"peerid" dc:"节点id"`
25 | Jobid string `json:"jobid" dc:"任务id"` // 任务id
26 | *DownloadFileRes
27 | }
28 |
29 | type DownloadFileJob struct {
30 | Jobid string `json:"jobid"` // 任务id
31 | *DownloadFileInfo
32 | }
33 |
34 | func init() {
35 | message.RegisterMessage(&message.MessageMeta{
36 | Type: reflect.TypeOf((*DownloadFileJob)(nil)).Elem(),
37 | ID: uint32(util.StringHash("model.DownloadFileJob")),
38 | })
39 | message.RegisterMessage(&message.MessageMeta{
40 | Type: reflect.TypeOf((*DownloadFileJobRes)(nil)).Elem(),
41 | ID: uint32(util.StringHash("model.DownloadFileJobRes")),
42 | })
43 | }
44 |
--------------------------------------------------------------------------------
/internal/model/entity/agent_info.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | import (
8 | "github.com/gogf/gf/v2/os/gtime"
9 | )
10 |
11 | // AgentInfo is the golang structure for table agent_info.
12 | type AgentInfo struct {
13 | Id int `json:"id" ` //
14 | Peerid string `json:"peerid" ` // 节点id
15 | Name string `json:"name" ` // agent名称
16 | ExpectedStatus string `json:"expectedStatus" ` // 期望状态
17 | Status string `json:"status" ` //
18 | IsDefault int `json:"isDefault" ` //
19 | Timeout int `json:"timeout" ` // 启动超时时间
20 | Created *gtime.Time `json:"created" ` //
21 | Updated *gtime.Time `json:"updated" ` //
22 | Version string `json:"version" ` // 版本信息
23 | }
24 |
--------------------------------------------------------------------------------
/internal/model/entity/app.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | import (
8 | "github.com/gogf/gf/v2/os/gtime"
9 | )
10 |
11 | // App is the golang structure for table app.
12 | type App struct {
13 | Id int `json:"id" ` //
14 | Appid string `json:"appid" ` //
15 | ApiKey string `json:"apiKey" ` //
16 | SecKey string `json:"secKey" ` //
17 | Owner string `json:"owner" ` //
18 | Name string `json:"name" ` // 应用名
19 | Status int `json:"status" ` // 1启用 0 禁用
20 | OwnerUid string `json:"ownerUid" ` // 拥有者uid
21 | CreatedAt *gtime.Time `json:"createdAt" ` //
22 | UpdatedAt *gtime.Time `json:"updatedAt" ` //
23 | }
24 |
--------------------------------------------------------------------------------
/internal/model/entity/check_item.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | import (
8 | "github.com/gogf/gf/v2/os/gtime"
9 | )
10 |
11 | // CheckItem is the golang structure for table check_item.
12 | type CheckItem struct {
13 | Id int `json:"id" ` //
14 | CheckItemId string `json:"checkItemId" ` //
15 | Name string `json:"name" ` // 检查项名称
16 | Desc string `json:"desc" ` // 检查项描述
17 | Type string `json:"type" ` //
18 | Content string `json:"content" ` // 检查项内容
19 | Creater string `json:"creater" ` // 创建人
20 | Updater string `json:"updater" ` // 更新人
21 | Created *gtime.Time `json:"created" ` //
22 | Updated *gtime.Time `json:"updated" ` //
23 | Weight float64 `json:"weight" ` // 权重
24 | WaitTime int `json:"waitTime" ` // 超时时间
25 | Cmd string `json:"cmd" ` // 脚本执行解释器
26 | }
27 |
--------------------------------------------------------------------------------
/internal/model/entity/check_tpl.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | import (
8 | "github.com/gogf/gf/v2/os/gtime"
9 | )
10 |
11 | // CheckTpl is the golang structure for table check_tpl.
12 | type CheckTpl struct {
13 | Id int `json:"id" ` //
14 | Tid string `json:"tid" ` // 唯一id
15 | Name string `json:"name" ` // 名称
16 | Description string `json:"description" ` // 描述
17 | Type string `json:"type" ` // 类型
18 | Creater string `json:"creater" ` // 创建人
19 | Updater string `json:"updater" ` // 更新人
20 | Created *gtime.Time `json:"created" ` //
21 | Updated *gtime.Time `json:"updated" ` //
22 | }
23 |
--------------------------------------------------------------------------------
/internal/model/entity/check_tpl_detail.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | // CheckTplDetail is the golang structure for table check_tpl_detail.
8 | type CheckTplDetail struct {
9 | Id int `json:"id" ` //
10 | Tid string `json:"tid" ` // 模板id
11 | Cid string `json:"cid" ` // 检查项id
12 | Sort int `json:"sort" ` // 排序
13 | Weight float64 `json:"weight" ` // 权重
14 | }
15 |
--------------------------------------------------------------------------------
/internal/model/entity/cron_task.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | import (
8 | "github.com/gogf/gf/v2/os/gtime"
9 | )
10 |
11 | // CronTask is the golang structure for table cron_task.
12 | type CronTask struct {
13 | Id int `json:"id" ` //
14 | CronUid string `json:"cronUid" ` //
15 | Name string `json:"name" ` // 定时任务名称
16 | Content string `json:"content" ` // 定时任务内容
17 | CronExpr string `json:"cronExpr" ` // 定时任务表达式
18 | LastrunTime *gtime.Time `json:"lastrunTime" ` // 最后执行时间
19 | NextrunTime *gtime.Time `json:"nextrunTime" ` //
20 | Status string `json:"status" ` // 状态 running stopped
21 | Creater string `json:"creater" ` // 创建人
22 | Updater string `json:"updater" ` // 更新人
23 | Created *gtime.Time `json:"created" ` //
24 | Updated *gtime.Time `json:"updated" ` //
25 | Type string `json:"type" ` // 任务类型
26 | }
27 |
--------------------------------------------------------------------------------
/internal/model/entity/plugin.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | import (
8 | "github.com/gogf/gf/v2/os/gtime"
9 | )
10 |
11 | // Plugin is the golang structure for table plugin.
12 | type Plugin struct {
13 | Id int `json:"id" ` //
14 | Uuid string `json:"uuid" ` // 插件uuid
15 | Name string `json:"name" ` // 插件名
16 | PackageName string `json:"packageName" ` // 包名
17 | Os string `json:"os" ` // 操作系统
18 | Arch string `json:"arch" ` // 架构
19 | Md5 string `json:"md5" ` // 包md5名称
20 | Creater string `json:"creater" ` // 创建人
21 | Updater string `json:"updater" ` // 更新人
22 | Created *gtime.Time `json:"created" ` //
23 | Updated *gtime.Time `json:"updated" ` //
24 | }
25 |
--------------------------------------------------------------------------------
/internal/model/entity/script.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | import (
8 | "github.com/gogf/gf/v2/os/gtime"
9 | )
10 |
11 | // Script is the golang structure for table script.
12 | type Script struct {
13 | Id int `json:"id" ` //
14 | Name string `json:"name" ` // 命令名称
15 | Content string `json:"content" ` // 脚本内容
16 | Args string `json:"args" ` // 参数信息
17 | Desc string `json:"desc" ` // 描述信息
18 | Type string `json:"type" ` // 脚本类型shell或者powershell
19 | OwnerType string `json:"ownerType" ` // 命令拥有者类型
20 | OwnerUid string `json:"ownerUid" ` // 拥有者uid
21 | Created *gtime.Time `json:"created" ` //
22 | Updated *gtime.Time `json:"updated" ` //
23 | ScriptUid string `json:"scriptUid" ` //
24 | WaitTime int `json:"waitTime" ` // 脚本超时时间
25 | Cmd string `json:"cmd" ` // 脚本解释器
26 | Ext string `json:"ext" ` // 脚本文件扩展名
27 | }
28 |
--------------------------------------------------------------------------------
/internal/model/entity/task.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | import (
8 | "github.com/gogf/gf/v2/os/gtime"
9 | )
10 |
11 | // Task is the golang structure for table task.
12 | type Task struct {
13 | Id int `json:"id" ` //
14 | TaskId string `json:"taskId" ` // 任务id
15 | Type string `json:"type" ` // 任务类型
16 | Content string `json:"content" ` // 任务内容
17 | Name string `json:"name" ` // 任务名称
18 | Reqid string `json:"reqid" ` // 请求id
19 | ParentId string `json:"parentId" ` //
20 | Status string `json:"status" ` //
21 | Created *gtime.Time `json:"created" ` //
22 | Updated *gtime.Time `json:"updated" ` //
23 | Creater string `json:"creater" ` // 创建人
24 | }
25 |
--------------------------------------------------------------------------------
/internal/model/entity/task_preset.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | import (
8 | "github.com/gogf/gf/v2/os/gtime"
9 | )
10 |
11 | // TaskPreset is the golang structure for table task_preset.
12 | type TaskPreset struct {
13 | Id int `json:"id" ` //
14 | Uuid string `json:"uuid" ` //
15 | Name string `json:"name" ` // 任务名
16 | Type string `json:"type" ` // 任务类型
17 | Content string `json:"content" ` // 任务内容
18 | Peers string `json:"peers" ` // 主机列表
19 | Creater string `json:"creater" ` // 创建人
20 | Created *gtime.Time `json:"created" ` // 创建时间
21 | Updated *gtime.Time `json:"updated" ` // 更新时间
22 | Updater string `json:"updater" ` // 更新人
23 | }
24 |
--------------------------------------------------------------------------------
/internal/model/entity/user.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | import (
8 | "github.com/gogf/gf/v2/os/gtime"
9 | )
10 |
11 | // User is the golang structure for table user.
12 | type User struct {
13 | Id int `json:"id" ` //
14 | Uid string `json:"uid" ` //
15 | Username string `json:"username" ` //
16 | Passwd string `json:"passwd" ` //
17 | Email string `json:"email" ` //
18 | Created *gtime.Time `json:"created" ` //
19 | Updated *gtime.Time `json:"updated" ` //
20 | Phone string `json:"phone" ` //
21 | Avatar string `json:"avatar" ` // 头像
22 | Status string `json:"status" ` // 状态 1 启用 2 禁用
23 | }
24 |
--------------------------------------------------------------------------------
/internal/model/entity/vm.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package entity
6 |
7 | import (
8 | "github.com/gogf/gf/v2/os/gtime"
9 | )
10 |
11 | // Vm is the golang structure for table vm.
12 | type Vm struct {
13 | Id int `json:"id" ` //
14 | Uuid string `json:"uuid" ` // 主机uuid
15 | Name string `json:"name" ` //
16 | Hostname string `json:"hostname" ` //
17 | OsType string `json:"osType" ` // 操作系统类型
18 | OsInfo string `json:"osInfo" ` // 操作系统信息
19 | Hosttype string `json:"hosttype" ` // 主机类型
20 | Networktype string `json:"networktype" ` // 网络类型
21 | PrivateIp string `json:"privateIp" ` //
22 | PublicIp string `json:"publicIp" ` //
23 | Created *gtime.Time `json:"created" ` //
24 | Updated *gtime.Time `json:"updated" ` //
25 | Creater string `json:"creater" ` // 创建人
26 | Address string `json:"address" ` //
27 | PeerId string `json:"peerId" ` //
28 | }
29 |
--------------------------------------------------------------------------------
/internal/model/fileinfo.go:
--------------------------------------------------------------------------------
1 | package model
2 |
3 | import (
4 | "go-ops/pkg/message"
5 | "go-ops/pkg/util"
6 | "reflect"
7 | "time"
8 | )
9 |
10 | type FileInfo struct {
11 | Name string `json:"name" dc:"文件名称"`
12 | Type string `json:"type" dc:"文件类型, dir 表示文件夹"`
13 | Size int64 `json:"size" dc:"大小"`
14 | Path string `json:"path" dc:"路径"`
15 | ModTime time.Time `json:"mtime" dc:"修改时间"`
16 | }
17 |
18 | type PeerListFileInfo struct {
19 | Peerid string `json:"peerid" dc:"节点id"`
20 | Path string `json:"path" dc:"文件夹路径"`
21 | }
22 |
23 | type PeerListFileInfoRes struct {
24 | Peerid string `json:"peerid" dc:"节点id"`
25 | Path string `json:"path" dc:"文件夹路径"`
26 | List []*FileInfo `json:"list" dc:"文件列表"`
27 | Error string `json:"error" dc:"报错信息"`
28 | }
29 |
30 | type PeerNewDir struct {
31 | Peerid string `json:"peerid" dc:"节点id"`
32 | Path string `json:"path" dc:"文件夹路径, 如果不存在会递归创建"`
33 | }
34 |
35 | type PeerMoveFile struct {
36 | Peerid string `json:"peerid" dc:"节点id"`
37 | Src string `json:"src" dc:"源文件"`
38 | Dst string `json:"dst" dc:"目标文件"`
39 | }
40 |
41 | type PeerDeleteFile struct {
42 | Peerid string `json:"peerid" dc:"节点id"`
43 | Path string `json:"path" dc:"文件路径"`
44 | }
45 |
46 | func init() {
47 | message.RegisterMessage(&message.MessageMeta{
48 | Type: reflect.TypeOf((*PeerListFileInfo)(nil)).Elem(),
49 | ID: uint32(util.StringHash("model.PeerListFileInfo")),
50 | })
51 |
52 | message.RegisterMessage(&message.MessageMeta{
53 | Type: reflect.TypeOf((*PeerListFileInfoRes)(nil)).Elem(),
54 | ID: uint32(util.StringHash("model.PeerListFileInfoRes")),
55 | })
56 |
57 | message.RegisterMessage(&message.MessageMeta{
58 | Type: reflect.TypeOf((*PeerNewDir)(nil)).Elem(),
59 | ID: uint32(util.StringHash("model.PeerNewDir")),
60 | })
61 |
62 | message.RegisterMessage(&message.MessageMeta{
63 | Type: reflect.TypeOf((*PeerMoveFile)(nil)).Elem(),
64 | ID: uint32(util.StringHash("model.PeerMoveFile")),
65 | })
66 |
67 | message.RegisterMessage(&message.MessageMeta{
68 | Type: reflect.TypeOf((*PeerDeleteFile)(nil)).Elem(),
69 | ID: uint32(util.StringHash("model.PeerDeleteFile")),
70 | })
71 |
72 | }
73 |
--------------------------------------------------------------------------------
/internal/model/peer.go:
--------------------------------------------------------------------------------
1 | package model
2 |
3 | import (
4 | "go-ops/pkg/message"
5 | "go-ops/pkg/util"
6 | "reflect"
7 | )
8 |
9 | type PeerInfo struct {
10 | PeerId string `json:"peerId" ` // 节点uuid
11 | HostName string `json:"hostname"` // 节点hostname
12 | Name string `json:"name"` // 节点名称
13 | Address string `json:"address"` // 节点地址
14 | PublicIp string `json:"publicIp"` // 节点ip
15 | Os string `json:"os"`
16 | Arch string `json:"arch"`
17 | }
18 |
19 | type GetPeerInfo struct {
20 | HostName string `json:"hostname"` // 节点hostname
21 | }
22 |
23 | func init() {
24 | message.RegisterMessage(&message.MessageMeta{
25 | Type: reflect.TypeOf((*PeerInfo)(nil)).Elem(),
26 | ID: uint32(util.StringHash("model.PeerInfo")),
27 | })
28 |
29 | message.RegisterMessage(&message.MessageMeta{
30 | Type: reflect.TypeOf((*GetPeerInfo)(nil)).Elem(),
31 | ID: uint32(util.StringHash("model.GetPeerInfo")),
32 | })
33 |
34 | }
35 |
--------------------------------------------------------------------------------
/internal/packed/packed.go:
--------------------------------------------------------------------------------
1 | package packed
2 |
--------------------------------------------------------------------------------
/internal/peer/handlermsg.go:
--------------------------------------------------------------------------------
1 | package peer
2 |
--------------------------------------------------------------------------------
/internal/peer/osp_peer.go:
--------------------------------------------------------------------------------
1 | package peer
2 |
3 | import (
4 | "context"
5 | "encoding/json"
6 | "fmt"
7 | "go-ops/internal/model"
8 | "go-ops/internal/service"
9 |
10 | "github.com/luxingwen/pnet"
11 | "github.com/luxingwen/pnet/config"
12 | "github.com/luxingwen/pnet/log"
13 | )
14 |
15 | type OspPeer struct {
16 | *pnet.PNet
17 | }
18 |
19 | func (self *OspPeer) HandlerFunc(msg interface{}, msgID []byte, srcID, rpath string, pn *pnet.PNet) {
20 | switch v := msg.(type) {
21 | case *model.ResponseResCmd:
22 | fmt.Println("res cmd:", v.Jobid)
23 | err := service.Task().UpdateSubScriptTask(context.Background(), v)
24 | if err != nil {
25 | fmt.Println("update err:", err)
26 | }
27 | case *model.DownloadFileJobRes:
28 | err := service.Task().UpdateDownloadFileTask(context.Background(), v)
29 | if err != nil {
30 | fmt.Println("update err:", err)
31 | }
32 | case *model.PeerInfo:
33 | err := service.VM().Check(context.Background(), v)
34 | if err != nil {
35 | fmt.Println("check vm err:", err)
36 | }
37 | default:
38 | b, _ := json.Marshal(v)
39 | log.Error("msg handler not found,msg: %s", string(b))
40 | }
41 | }
42 |
43 | var ospPeer *OspPeer
44 |
45 | func InitOspPeer(id string, conf *config.Config) (err error) {
46 |
47 | ospPeer = &OspPeer{}
48 |
49 | pn, err := NewPnet(id, conf, ospPeer.HandlerFunc)
50 | if err != nil {
51 | return
52 | }
53 | ospPeer.PNet = pn
54 | err = pn.Start()
55 | if err != nil {
56 | return
57 | }
58 | return
59 | }
60 |
61 | func GetOspPeer() *OspPeer {
62 | if ospPeer == nil {
63 | panic("osp peer is nil")
64 | }
65 | return ospPeer
66 | }
67 |
--------------------------------------------------------------------------------
/internal/peer/peer.go:
--------------------------------------------------------------------------------
1 | package peer
2 |
3 | import (
4 | "go-ops/pkg/message"
5 | "time"
6 |
7 | "github.com/luxingwen/pnet/log"
8 | "github.com/luxingwen/pnet/protos"
9 |
10 | "github.com/luxingwen/pnet/config"
11 |
12 | "github.com/luxingwen/pnet"
13 | "github.com/luxingwen/pnet/node"
14 | )
15 |
16 | var defaultPN *pnet.PNet
17 |
18 | type HandlerPNetMsg func(msg interface{}, msgID []byte, srcID, rpath string, pn *pnet.PNet)
19 |
20 | func NewPnet(id string, conf *config.Config, h HandlerPNetMsg) (pn *pnet.PNet, err error) {
21 | pn, err = pnet.NewPNet(id, conf)
22 | if err != nil {
23 | return
24 | }
25 |
26 | pn.ApplyMiddleware(node.BytesReceived{func(msg, msgID []byte, srcID, rpath string, remoteNode *node.RemoteNode) ([]byte, bool) {
27 | v, err := message.JSONCodec.Decode(msg)
28 | if err != nil {
29 | log.Error("json decode err:", err)
30 | return nil, true
31 | }
32 |
33 | h(v, msgID, srcID, rpath, pn)
34 | return nil, true
35 | }, 0})
36 |
37 | pn.ApplyMiddleware(node.LocalNodeStarted{func(lc *node.LocalNode) bool {
38 | lc.SetReady(true)
39 | return true
40 | }, 0})
41 |
42 | return
43 | }
44 |
45 | // 发送异步消息
46 | func SendMsgAsync(pn *pnet.PNet, msg interface{}, srcID, rpath string) (err error) {
47 | data, err := message.JSONCodec.Encode(msg)
48 | if err != nil {
49 | log.Error("json encode err:", err)
50 | return
51 | }
52 | _, err = pn.SendBytesRelayAsync(data, srcID)
53 | return
54 | }
55 |
56 | // 发送同步消息
57 | func SendMsgSync(pn *pnet.PNet, msg interface{}, srcID, rpath string) (r []byte, err error) {
58 | return SendMsgSyncWithTimeout(pn, msg, srcID, rpath, 0)
59 | }
60 |
61 | // 发送同步消息
62 | func SendMsgSyncWithTimeout(pn *pnet.PNet, msg interface{}, srcID, rpath string, timeout time.Duration) (r []byte, err error) {
63 | data, err := message.JSONCodec.Encode(msg)
64 | if err != nil {
65 | log.Error("json encode err:", err)
66 | return
67 | }
68 | r, _, err = pn.SendBytesRelaySyncWithTimeout(data, srcID, timeout)
69 | return
70 | }
71 |
72 | // 回复消息
73 | func SendMsgReplay(pn *pnet.PNet, msg interface{}, msgID []byte, srcID, rpath string) (err error) {
74 | data, err := message.JSONCodec.Encode(msg)
75 | if err != nil {
76 | log.Error("json encode err:", err)
77 | return
78 | }
79 | _, err = pn.SendBytesRelayReply(msgID, data, srcID)
80 | return
81 | }
82 |
83 | func SendMsgBroadCast(pn *pnet.PNet, msg interface{}) (err error) {
84 | data, err := message.JSONCodec.Encode(msg)
85 | if err != nil {
86 | log.Error("json encode err:", err)
87 | return
88 | }
89 |
90 | _, err = pn.SendMessageAsync(pn.GetLocalNode().NewBraodcastMessage(data), protos.BROADCAST)
91 | return
92 | }
93 |
--------------------------------------------------------------------------------
/internal/service/auth.go:
--------------------------------------------------------------------------------
1 | package service
2 |
3 | import (
4 | "context"
5 | v1 "go-ops/pkg/api/v1"
6 | "time"
7 |
8 | "github.com/gogf/gf/v2/frame/g"
9 |
10 | jwt "github.com/gogf/gf-jwt/v2"
11 | )
12 |
13 | var authService *jwt.GfJWTMiddleware
14 |
15 | func Auth() *jwt.GfJWTMiddleware {
16 | return authService
17 | }
18 |
19 | var jwtKey string = "Go-ops-apiserver-jwt-key"
20 |
21 | func init() {
22 | auth := jwt.New(&jwt.GfJWTMiddleware{
23 | Realm: "Go-ops apiserver",
24 | Key: []byte(jwtKey),
25 | Timeout: time.Minute * 10,
26 | MaxRefresh: time.Minute * 20,
27 | IdentityKey: "uid",
28 | TokenLookup: "header: GO-OPS-X-TOKEN, query: token, cookie: jwt",
29 | TokenHeadName: "Bearer",
30 | TimeFunc: time.Now,
31 | Authenticator: Authenticator,
32 | Unauthorized: Unauthorized,
33 | PayloadFunc: PayloadFunc,
34 | IdentityHandler: IdentityHandler,
35 | })
36 | authService = auth
37 | }
38 |
39 | // PayloadFunc is a callback function that will be called during login.
40 | // Using this function it is possible to add additional payload data to the webtoken.
41 | // The data is then made available during requests via c.Get("JWT_PAYLOAD").
42 | // Note that the payload is not encrypted.
43 | // The attributes mentioned on jwt.io can't be used as keys for the map.
44 | // Optional, by default no additional data will be set.
45 | func PayloadFunc(data interface{}) jwt.MapClaims {
46 | claims := jwt.MapClaims{}
47 | params := data.(map[string]interface{})
48 | if len(params) > 0 {
49 | for k, v := range params {
50 | claims[k] = v
51 | }
52 | }
53 | return claims
54 | }
55 |
56 | // IdentityHandler get the identity from JWT and set the identity for every request
57 | // Using this function, by r.GetParam("id") get identity
58 | func IdentityHandler(ctx context.Context) interface{} {
59 | claims := jwt.ExtractClaims(ctx)
60 | return claims[authService.IdentityKey]
61 | }
62 |
63 | // Unauthorized is used to define customized Unauthorized callback function.
64 | func Unauthorized(ctx context.Context, code int, message string) {
65 |
66 | r := g.RequestFromCtx(ctx)
67 |
68 | r.Response.WriteStatus(code, g.Map{
69 | "code": code,
70 | "message": message,
71 | })
72 | r.ExitAll()
73 | }
74 |
75 | // Authenticator is used to validate login parameters.
76 | // It must return user data as user identifier, it will be stored in Claim Array.
77 | // if your identityKey is 'id', your user data must have 'id'
78 | // Check error (e) to determine the appropriate error message.
79 | func Authenticator(ctx context.Context) (interface{}, error) {
80 | var (
81 | r = g.RequestFromCtx(ctx)
82 | in v1.AuthLoginReq
83 | )
84 | if err := r.Parse(&in); err != nil {
85 | return "", err
86 | }
87 |
88 | user, err := User().SignIn(ctx, &in)
89 |
90 | if err != nil {
91 | return nil, err
92 | }
93 |
94 | if user != nil {
95 | return g.Map{
96 | "uid": user.Uid,
97 | "username": user.Username,
98 | }, nil
99 | }
100 |
101 | return nil, jwt.ErrFailedAuthentication
102 | }
103 |
--------------------------------------------------------------------------------
/internal/service/checktpl.go:
--------------------------------------------------------------------------------
1 | package service
2 |
3 | import (
4 | "context"
5 | "errors"
6 | "go-ops/internal/model/entity"
7 | "go-ops/internal/service/internal/dao"
8 | v1 "go-ops/pkg/api/v1"
9 |
10 | "github.com/gogf/gf/v2/database/gdb"
11 | "github.com/gogf/gf/v2/util/guid"
12 | )
13 |
14 | type sCheckTpl struct {
15 | }
16 |
17 | var (
18 | insCheckTpl = sCheckTpl{}
19 | )
20 |
21 | func CheckTpl() *sCheckTpl {
22 | return &insCheckTpl
23 | }
24 |
25 | func (selef *sCheckTpl) Create(ctx context.Context, req *v1.AddCheckTplReq) (err error) {
26 |
27 | checktpl := &entity.CheckTpl{
28 | Tid: guid.S(),
29 | Name: req.Name,
30 | Description: req.Desc,
31 | Type: req.Type,
32 | }
33 |
34 | checktplDetails := make([]*entity.CheckTplDetail, 0)
35 |
36 | for _, item := range req.Items {
37 |
38 | checktplDetails = append(checktplDetails, &entity.CheckTplDetail{
39 | Tid: checktpl.Tid,
40 | Cid: item.ItemId,
41 | Weight: item.Weight,
42 | })
43 |
44 | }
45 |
46 | err = dao.CheckTpl.Transaction(ctx, func(ctx context.Context, tx *gdb.TX) error {
47 | _, err = dao.CheckTpl.Ctx(ctx).Data(checktpl).Insert()
48 |
49 | if err != nil {
50 | return err
51 | }
52 |
53 | _, err = dao.CheckTplDetail.Ctx(ctx).Data(checktplDetails).Insert()
54 | return err
55 | })
56 | return
57 | }
58 |
59 | func (self *sCheckTpl) Update(ctx context.Context, req *v1.UpdateCheckTplReq) (err error) {
60 | var checkTpl *entity.CheckTpl
61 |
62 | err = dao.CheckTpl.Ctx(ctx).Where("tid = ?", req.Tid).Scan(&checkTpl)
63 | if err != nil {
64 | return
65 | }
66 |
67 | if checkTpl == nil {
68 | err = errors.New("not found:" + req.Tid)
69 | return
70 | }
71 |
72 | checktplDetails := make([]*entity.CheckTplDetail, 0)
73 |
74 | for _, item := range req.Items {
75 |
76 | checktplDetails = append(checktplDetails, &entity.CheckTplDetail{
77 | Tid: checkTpl.Tid,
78 | Cid: item.ItemId,
79 | Weight: item.Weight,
80 | })
81 |
82 | }
83 |
84 | dao.CheckTplDetail.Transaction(ctx, func(ctx context.Context, tx *gdb.TX) error {
85 | _, err = dao.CheckTplDetail.Ctx(ctx).Where("tid = ?", req.Tid).Delete()
86 | if err != nil {
87 | return err
88 | }
89 |
90 | _, err = dao.CheckTplDetail.Ctx(ctx).Data(checktplDetails).Insert()
91 | if err != nil {
92 | return err
93 | }
94 | // _, err = dao.CheckTpl.Ctx(ctx).Data(checktplDetails).Insert()
95 | // if err != nil {
96 | // return err
97 | // }
98 |
99 | return err
100 | })
101 |
102 | return
103 | }
104 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/agent_info.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalAgentInfoDao is internal type for wrapping internal DAO implements.
12 | type internalAgentInfoDao = *internal.AgentInfoDao
13 |
14 | // agentInfoDao is the data access object for table agent_info.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type agentInfoDao struct {
17 | internalAgentInfoDao
18 | }
19 |
20 | var (
21 | // AgentInfo is globally public accessible object for table agent_info operations.
22 | AgentInfo = agentInfoDao{
23 | internal.NewAgentInfoDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/app.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalAppDao is internal type for wrapping internal DAO implements.
12 | type internalAppDao = *internal.AppDao
13 |
14 | // appDao is the data access object for table app.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type appDao struct {
17 | internalAppDao
18 | }
19 |
20 | var (
21 | // App is globally public accessible object for table app operations.
22 | App = appDao{
23 | internal.NewAppDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/check_item.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalCheckItemDao is internal type for wrapping internal DAO implements.
12 | type internalCheckItemDao = *internal.CheckItemDao
13 |
14 | // checkItemDao is the data access object for table check_item.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type checkItemDao struct {
17 | internalCheckItemDao
18 | }
19 |
20 | var (
21 | // CheckItem is globally public accessible object for table check_item operations.
22 | CheckItem = checkItemDao{
23 | internal.NewCheckItemDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/check_tpl.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalCheckTplDao is internal type for wrapping internal DAO implements.
12 | type internalCheckTplDao = *internal.CheckTplDao
13 |
14 | // checkTplDao is the data access object for table check_tpl.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type checkTplDao struct {
17 | internalCheckTplDao
18 | }
19 |
20 | var (
21 | // CheckTpl is globally public accessible object for table check_tpl operations.
22 | CheckTpl = checkTplDao{
23 | internal.NewCheckTplDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/check_tpl_detail.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalCheckTplDetailDao is internal type for wrapping internal DAO implements.
12 | type internalCheckTplDetailDao = *internal.CheckTplDetailDao
13 |
14 | // checkTplDetailDao is the data access object for table check_tpl_detail.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type checkTplDetailDao struct {
17 | internalCheckTplDetailDao
18 | }
19 |
20 | var (
21 | // CheckTplDetail is globally public accessible object for table check_tpl_detail operations.
22 | CheckTplDetail = checkTplDetailDao{
23 | internal.NewCheckTplDetailDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/cron_task.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalCronTaskDao is internal type for wrapping internal DAO implements.
12 | type internalCronTaskDao = *internal.CronTaskDao
13 |
14 | // cronTaskDao is the data access object for table cron_task.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type cronTaskDao struct {
17 | internalCronTaskDao
18 | }
19 |
20 | var (
21 | // CronTask is globally public accessible object for table cron_task operations.
22 | CronTask = cronTaskDao{
23 | internal.NewCronTaskDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/internal/app.go:
--------------------------------------------------------------------------------
1 | // ==========================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // ==========================================================================
4 |
5 | package internal
6 |
7 | import (
8 | "context"
9 | "github.com/gogf/gf/v2/database/gdb"
10 | "github.com/gogf/gf/v2/frame/g"
11 | )
12 |
13 | // AppDao is the data access object for table app.
14 | type AppDao struct {
15 | table string // table is the underlying table name of the DAO.
16 | group string // group is the database configuration group name of current DAO.
17 | columns AppColumns // columns contains all the column names of Table for convenient usage.
18 | }
19 |
20 | // AppColumns defines and stores column names for table app.
21 | type AppColumns struct {
22 | Id string //
23 | Appid string //
24 | ApiKey string //
25 | SecKey string //
26 | Owner string //
27 | Name string // 应用名
28 | Status string // 1启用 0 禁用
29 | OwnerUid string // 拥有者uid
30 | CreatedAt string //
31 | UpdatedAt string //
32 | }
33 |
34 | // appColumns holds the columns for table app.
35 | var appColumns = AppColumns{
36 | Id: "id",
37 | Appid: "appid",
38 | ApiKey: "api_key",
39 | SecKey: "sec_key",
40 | Owner: "owner",
41 | Name: "name",
42 | Status: "status",
43 | OwnerUid: "owner_uid",
44 | CreatedAt: "created_at",
45 | UpdatedAt: "updated_at",
46 | }
47 |
48 | // NewAppDao creates and returns a new DAO object for table data access.
49 | func NewAppDao() *AppDao {
50 | return &AppDao{
51 | group: "default",
52 | table: "app",
53 | columns: appColumns,
54 | }
55 | }
56 |
57 | // DB retrieves and returns the underlying raw database management object of current DAO.
58 | func (dao *AppDao) DB() gdb.DB {
59 | return g.DB(dao.group)
60 | }
61 |
62 | // Table returns the table name of current dao.
63 | func (dao *AppDao) Table() string {
64 | return dao.table
65 | }
66 |
67 | // Columns returns all column names of current dao.
68 | func (dao *AppDao) Columns() AppColumns {
69 | return dao.columns
70 | }
71 |
72 | // Group returns the configuration group name of database of current dao.
73 | func (dao *AppDao) Group() string {
74 | return dao.group
75 | }
76 |
77 | // Ctx creates and returns the Model for current DAO, It automatically sets the context for current operation.
78 | func (dao *AppDao) Ctx(ctx context.Context) *gdb.Model {
79 | return dao.DB().Model(dao.table).Safe().Ctx(ctx)
80 | }
81 |
82 | // Transaction wraps the transaction logic using function f.
83 | // It rollbacks the transaction and returns the error from function f if it returns non-nil error.
84 | // It commits the transaction and returns nil if function f returns nil.
85 | //
86 | // Note that, you should not Commit or Rollback the transaction in function f
87 | // as it is automatically handled by this function.
88 | func (dao *AppDao) Transaction(ctx context.Context, f func(ctx context.Context, tx *gdb.TX) error) (err error) {
89 | return dao.Ctx(ctx).Transaction(ctx, f)
90 | }
91 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/internal/check_tpl.go:
--------------------------------------------------------------------------------
1 | // ==========================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // ==========================================================================
4 |
5 | package internal
6 |
7 | import (
8 | "context"
9 | "github.com/gogf/gf/v2/database/gdb"
10 | "github.com/gogf/gf/v2/frame/g"
11 | )
12 |
13 | // CheckTplDao is the data access object for table check_tpl.
14 | type CheckTplDao struct {
15 | table string // table is the underlying table name of the DAO.
16 | group string // group is the database configuration group name of current DAO.
17 | columns CheckTplColumns // columns contains all the column names of Table for convenient usage.
18 | }
19 |
20 | // CheckTplColumns defines and stores column names for table check_tpl.
21 | type CheckTplColumns struct {
22 | Id string //
23 | Tid string // 唯一id
24 | Name string // 名称
25 | Description string // 描述
26 | Type string // 类型
27 | Creater string // 创建人
28 | Updater string // 更新人
29 | Created string //
30 | Updated string //
31 | }
32 |
33 | // checkTplColumns holds the columns for table check_tpl.
34 | var checkTplColumns = CheckTplColumns{
35 | Id: "id",
36 | Tid: "tid",
37 | Name: "name",
38 | Description: "description",
39 | Type: "type",
40 | Creater: "creater",
41 | Updater: "updater",
42 | Created: "created",
43 | Updated: "updated",
44 | }
45 |
46 | // NewCheckTplDao creates and returns a new DAO object for table data access.
47 | func NewCheckTplDao() *CheckTplDao {
48 | return &CheckTplDao{
49 | group: "default",
50 | table: "check_tpl",
51 | columns: checkTplColumns,
52 | }
53 | }
54 |
55 | // DB retrieves and returns the underlying raw database management object of current DAO.
56 | func (dao *CheckTplDao) DB() gdb.DB {
57 | return g.DB(dao.group)
58 | }
59 |
60 | // Table returns the table name of current dao.
61 | func (dao *CheckTplDao) Table() string {
62 | return dao.table
63 | }
64 |
65 | // Columns returns all column names of current dao.
66 | func (dao *CheckTplDao) Columns() CheckTplColumns {
67 | return dao.columns
68 | }
69 |
70 | // Group returns the configuration group name of database of current dao.
71 | func (dao *CheckTplDao) Group() string {
72 | return dao.group
73 | }
74 |
75 | // Ctx creates and returns the Model for current DAO, It automatically sets the context for current operation.
76 | func (dao *CheckTplDao) Ctx(ctx context.Context) *gdb.Model {
77 | return dao.DB().Model(dao.table).Safe().Ctx(ctx)
78 | }
79 |
80 | // Transaction wraps the transaction logic using function f.
81 | // It rollbacks the transaction and returns the error from function f if it returns non-nil error.
82 | // It commits the transaction and returns nil if function f returns nil.
83 | //
84 | // Note that, you should not Commit or Rollback the transaction in function f
85 | // as it is automatically handled by this function.
86 | func (dao *CheckTplDao) Transaction(ctx context.Context, f func(ctx context.Context, tx *gdb.TX) error) (err error) {
87 | return dao.Ctx(ctx).Transaction(ctx, f)
88 | }
89 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/internal/check_tpl_detail.go:
--------------------------------------------------------------------------------
1 | // ==========================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // ==========================================================================
4 |
5 | package internal
6 |
7 | import (
8 | "context"
9 | "github.com/gogf/gf/v2/database/gdb"
10 | "github.com/gogf/gf/v2/frame/g"
11 | )
12 |
13 | // CheckTplDetailDao is the data access object for table check_tpl_detail.
14 | type CheckTplDetailDao struct {
15 | table string // table is the underlying table name of the DAO.
16 | group string // group is the database configuration group name of current DAO.
17 | columns CheckTplDetailColumns // columns contains all the column names of Table for convenient usage.
18 | }
19 |
20 | // CheckTplDetailColumns defines and stores column names for table check_tpl_detail.
21 | type CheckTplDetailColumns struct {
22 | Id string //
23 | Tid string // 模板id
24 | Cid string // 检查项id
25 | Sort string // 排序
26 | Weight string // 权重
27 | }
28 |
29 | // checkTplDetailColumns holds the columns for table check_tpl_detail.
30 | var checkTplDetailColumns = CheckTplDetailColumns{
31 | Id: "id",
32 | Tid: "tid",
33 | Cid: "cid",
34 | Sort: "sort",
35 | Weight: "weight",
36 | }
37 |
38 | // NewCheckTplDetailDao creates and returns a new DAO object for table data access.
39 | func NewCheckTplDetailDao() *CheckTplDetailDao {
40 | return &CheckTplDetailDao{
41 | group: "default",
42 | table: "check_tpl_detail",
43 | columns: checkTplDetailColumns,
44 | }
45 | }
46 |
47 | // DB retrieves and returns the underlying raw database management object of current DAO.
48 | func (dao *CheckTplDetailDao) DB() gdb.DB {
49 | return g.DB(dao.group)
50 | }
51 |
52 | // Table returns the table name of current dao.
53 | func (dao *CheckTplDetailDao) Table() string {
54 | return dao.table
55 | }
56 |
57 | // Columns returns all column names of current dao.
58 | func (dao *CheckTplDetailDao) Columns() CheckTplDetailColumns {
59 | return dao.columns
60 | }
61 |
62 | // Group returns the configuration group name of database of current dao.
63 | func (dao *CheckTplDetailDao) Group() string {
64 | return dao.group
65 | }
66 |
67 | // Ctx creates and returns the Model for current DAO, It automatically sets the context for current operation.
68 | func (dao *CheckTplDetailDao) Ctx(ctx context.Context) *gdb.Model {
69 | return dao.DB().Model(dao.table).Safe().Ctx(ctx)
70 | }
71 |
72 | // Transaction wraps the transaction logic using function f.
73 | // It rollbacks the transaction and returns the error from function f if it returns non-nil error.
74 | // It commits the transaction and returns nil if function f returns nil.
75 | //
76 | // Note that, you should not Commit or Rollback the transaction in function f
77 | // as it is automatically handled by this function.
78 | func (dao *CheckTplDetailDao) Transaction(ctx context.Context, f func(ctx context.Context, tx *gdb.TX) error) (err error) {
79 | return dao.Ctx(ctx).Transaction(ctx, f)
80 | }
81 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/internal/plugin.go:
--------------------------------------------------------------------------------
1 | // ==========================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // ==========================================================================
4 |
5 | package internal
6 |
7 | import (
8 | "context"
9 | "github.com/gogf/gf/v2/database/gdb"
10 | "github.com/gogf/gf/v2/frame/g"
11 | )
12 |
13 | // PluginDao is the data access object for table plugin.
14 | type PluginDao struct {
15 | table string // table is the underlying table name of the DAO.
16 | group string // group is the database configuration group name of current DAO.
17 | columns PluginColumns // columns contains all the column names of Table for convenient usage.
18 | }
19 |
20 | // PluginColumns defines and stores column names for table plugin.
21 | type PluginColumns struct {
22 | Id string //
23 | Uuid string // 插件uuid
24 | Name string // 插件名
25 | PackageName string // 包名
26 | Os string // 操作系统
27 | Arch string // 架构
28 | Md5 string // 包md5名称
29 | Creater string // 创建人
30 | Updater string // 更新人
31 | Created string //
32 | Updated string //
33 | }
34 |
35 | // pluginColumns holds the columns for table plugin.
36 | var pluginColumns = PluginColumns{
37 | Id: "id",
38 | Uuid: "uuid",
39 | Name: "name",
40 | PackageName: "package_name",
41 | Os: "os",
42 | Arch: "arch",
43 | Md5: "md5",
44 | Creater: "creater",
45 | Updater: "updater",
46 | Created: "created",
47 | Updated: "updated",
48 | }
49 |
50 | // NewPluginDao creates and returns a new DAO object for table data access.
51 | func NewPluginDao() *PluginDao {
52 | return &PluginDao{
53 | group: "default",
54 | table: "plugin",
55 | columns: pluginColumns,
56 | }
57 | }
58 |
59 | // DB retrieves and returns the underlying raw database management object of current DAO.
60 | func (dao *PluginDao) DB() gdb.DB {
61 | return g.DB(dao.group)
62 | }
63 |
64 | // Table returns the table name of current dao.
65 | func (dao *PluginDao) Table() string {
66 | return dao.table
67 | }
68 |
69 | // Columns returns all column names of current dao.
70 | func (dao *PluginDao) Columns() PluginColumns {
71 | return dao.columns
72 | }
73 |
74 | // Group returns the configuration group name of database of current dao.
75 | func (dao *PluginDao) Group() string {
76 | return dao.group
77 | }
78 |
79 | // Ctx creates and returns the Model for current DAO, It automatically sets the context for current operation.
80 | func (dao *PluginDao) Ctx(ctx context.Context) *gdb.Model {
81 | return dao.DB().Model(dao.table).Safe().Ctx(ctx)
82 | }
83 |
84 | // Transaction wraps the transaction logic using function f.
85 | // It rollbacks the transaction and returns the error from function f if it returns non-nil error.
86 | // It commits the transaction and returns nil if function f returns nil.
87 | //
88 | // Note that, you should not Commit or Rollback the transaction in function f
89 | // as it is automatically handled by this function.
90 | func (dao *PluginDao) Transaction(ctx context.Context, f func(ctx context.Context, tx *gdb.TX) error) (err error) {
91 | return dao.Ctx(ctx).Transaction(ctx, f)
92 | }
93 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/internal/task.go:
--------------------------------------------------------------------------------
1 | // ==========================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // ==========================================================================
4 |
5 | package internal
6 |
7 | import (
8 | "context"
9 | "github.com/gogf/gf/v2/database/gdb"
10 | "github.com/gogf/gf/v2/frame/g"
11 | )
12 |
13 | // TaskDao is the data access object for table task.
14 | type TaskDao struct {
15 | table string // table is the underlying table name of the DAO.
16 | group string // group is the database configuration group name of current DAO.
17 | columns TaskColumns // columns contains all the column names of Table for convenient usage.
18 | }
19 |
20 | // TaskColumns defines and stores column names for table task.
21 | type TaskColumns struct {
22 | Id string //
23 | TaskId string // 任务id
24 | Type string // 任务类型
25 | Content string // 任务内容
26 | Name string // 任务名称
27 | Reqid string // 请求id
28 | ParentId string //
29 | Status string //
30 | Created string //
31 | Updated string //
32 | Creater string // 创建人
33 | }
34 |
35 | // taskColumns holds the columns for table task.
36 | var taskColumns = TaskColumns{
37 | Id: "id",
38 | TaskId: "task_id",
39 | Type: "type",
40 | Content: "content",
41 | Name: "name",
42 | Reqid: "reqid",
43 | ParentId: "parent_id",
44 | Status: "status",
45 | Created: "created",
46 | Updated: "updated",
47 | Creater: "creater",
48 | }
49 |
50 | // NewTaskDao creates and returns a new DAO object for table data access.
51 | func NewTaskDao() *TaskDao {
52 | return &TaskDao{
53 | group: "default",
54 | table: "task",
55 | columns: taskColumns,
56 | }
57 | }
58 |
59 | // DB retrieves and returns the underlying raw database management object of current DAO.
60 | func (dao *TaskDao) DB() gdb.DB {
61 | return g.DB(dao.group)
62 | }
63 |
64 | // Table returns the table name of current dao.
65 | func (dao *TaskDao) Table() string {
66 | return dao.table
67 | }
68 |
69 | // Columns returns all column names of current dao.
70 | func (dao *TaskDao) Columns() TaskColumns {
71 | return dao.columns
72 | }
73 |
74 | // Group returns the configuration group name of database of current dao.
75 | func (dao *TaskDao) Group() string {
76 | return dao.group
77 | }
78 |
79 | // Ctx creates and returns the Model for current DAO, It automatically sets the context for current operation.
80 | func (dao *TaskDao) Ctx(ctx context.Context) *gdb.Model {
81 | return dao.DB().Model(dao.table).Safe().Ctx(ctx)
82 | }
83 |
84 | // Transaction wraps the transaction logic using function f.
85 | // It rollbacks the transaction and returns the error from function f if it returns non-nil error.
86 | // It commits the transaction and returns nil if function f returns nil.
87 | //
88 | // Note that, you should not Commit or Rollback the transaction in function f
89 | // as it is automatically handled by this function.
90 | func (dao *TaskDao) Transaction(ctx context.Context, f func(ctx context.Context, tx *gdb.TX) error) (err error) {
91 | return dao.Ctx(ctx).Transaction(ctx, f)
92 | }
93 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/internal/task_preset.go:
--------------------------------------------------------------------------------
1 | // ==========================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // ==========================================================================
4 |
5 | package internal
6 |
7 | import (
8 | "context"
9 | "github.com/gogf/gf/v2/database/gdb"
10 | "github.com/gogf/gf/v2/frame/g"
11 | )
12 |
13 | // TaskPresetDao is the data access object for table task_preset.
14 | type TaskPresetDao struct {
15 | table string // table is the underlying table name of the DAO.
16 | group string // group is the database configuration group name of current DAO.
17 | columns TaskPresetColumns // columns contains all the column names of Table for convenient usage.
18 | }
19 |
20 | // TaskPresetColumns defines and stores column names for table task_preset.
21 | type TaskPresetColumns struct {
22 | Id string //
23 | Uuid string //
24 | Name string // 任务名
25 | Type string // 任务类型
26 | Content string // 任务内容
27 | Peers string // 主机列表
28 | Creater string // 创建人
29 | Created string // 创建时间
30 | Updated string // 更新时间
31 | Updater string // 更新人
32 | }
33 |
34 | // taskPresetColumns holds the columns for table task_preset.
35 | var taskPresetColumns = TaskPresetColumns{
36 | Id: "id",
37 | Uuid: "uuid",
38 | Name: "name",
39 | Type: "type",
40 | Content: "content",
41 | Peers: "peers",
42 | Creater: "creater",
43 | Created: "created",
44 | Updated: "updated",
45 | Updater: "updater",
46 | }
47 |
48 | // NewTaskPresetDao creates and returns a new DAO object for table data access.
49 | func NewTaskPresetDao() *TaskPresetDao {
50 | return &TaskPresetDao{
51 | group: "default",
52 | table: "task_preset",
53 | columns: taskPresetColumns,
54 | }
55 | }
56 |
57 | // DB retrieves and returns the underlying raw database management object of current DAO.
58 | func (dao *TaskPresetDao) DB() gdb.DB {
59 | return g.DB(dao.group)
60 | }
61 |
62 | // Table returns the table name of current dao.
63 | func (dao *TaskPresetDao) Table() string {
64 | return dao.table
65 | }
66 |
67 | // Columns returns all column names of current dao.
68 | func (dao *TaskPresetDao) Columns() TaskPresetColumns {
69 | return dao.columns
70 | }
71 |
72 | // Group returns the configuration group name of database of current dao.
73 | func (dao *TaskPresetDao) Group() string {
74 | return dao.group
75 | }
76 |
77 | // Ctx creates and returns the Model for current DAO, It automatically sets the context for current operation.
78 | func (dao *TaskPresetDao) Ctx(ctx context.Context) *gdb.Model {
79 | return dao.DB().Model(dao.table).Safe().Ctx(ctx)
80 | }
81 |
82 | // Transaction wraps the transaction logic using function f.
83 | // It rollbacks the transaction and returns the error from function f if it returns non-nil error.
84 | // It commits the transaction and returns nil if function f returns nil.
85 | //
86 | // Note that, you should not Commit or Rollback the transaction in function f
87 | // as it is automatically handled by this function.
88 | func (dao *TaskPresetDao) Transaction(ctx context.Context, f func(ctx context.Context, tx *gdb.TX) error) (err error) {
89 | return dao.Ctx(ctx).Transaction(ctx, f)
90 | }
91 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/internal/user.go:
--------------------------------------------------------------------------------
1 | // ==========================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // ==========================================================================
4 |
5 | package internal
6 |
7 | import (
8 | "context"
9 | "github.com/gogf/gf/v2/database/gdb"
10 | "github.com/gogf/gf/v2/frame/g"
11 | )
12 |
13 | // UserDao is the data access object for table user.
14 | type UserDao struct {
15 | table string // table is the underlying table name of the DAO.
16 | group string // group is the database configuration group name of current DAO.
17 | columns UserColumns // columns contains all the column names of Table for convenient usage.
18 | }
19 |
20 | // UserColumns defines and stores column names for table user.
21 | type UserColumns struct {
22 | Id string //
23 | Uid string //
24 | Username string //
25 | Passwd string //
26 | Email string //
27 | Created string //
28 | Updated string //
29 | Phone string //
30 | Avatar string // 头像
31 | Status string // 状态 1 启用 2 禁用
32 | }
33 |
34 | // userColumns holds the columns for table user.
35 | var userColumns = UserColumns{
36 | Id: "id",
37 | Uid: "uid",
38 | Username: "username",
39 | Passwd: "passwd",
40 | Email: "email",
41 | Created: "created",
42 | Updated: "updated",
43 | Phone: "phone",
44 | Avatar: "avatar",
45 | Status: "status",
46 | }
47 |
48 | // NewUserDao creates and returns a new DAO object for table data access.
49 | func NewUserDao() *UserDao {
50 | return &UserDao{
51 | group: "default",
52 | table: "user",
53 | columns: userColumns,
54 | }
55 | }
56 |
57 | // DB retrieves and returns the underlying raw database management object of current DAO.
58 | func (dao *UserDao) DB() gdb.DB {
59 | return g.DB(dao.group)
60 | }
61 |
62 | // Table returns the table name of current dao.
63 | func (dao *UserDao) Table() string {
64 | return dao.table
65 | }
66 |
67 | // Columns returns all column names of current dao.
68 | func (dao *UserDao) Columns() UserColumns {
69 | return dao.columns
70 | }
71 |
72 | // Group returns the configuration group name of database of current dao.
73 | func (dao *UserDao) Group() string {
74 | return dao.group
75 | }
76 |
77 | // Ctx creates and returns the Model for current DAO, It automatically sets the context for current operation.
78 | func (dao *UserDao) Ctx(ctx context.Context) *gdb.Model {
79 | return dao.DB().Model(dao.table).Safe().Ctx(ctx)
80 | }
81 |
82 | // Transaction wraps the transaction logic using function f.
83 | // It rollbacks the transaction and returns the error from function f if it returns non-nil error.
84 | // It commits the transaction and returns nil if function f returns nil.
85 | //
86 | // Note that, you should not Commit or Rollback the transaction in function f
87 | // as it is automatically handled by this function.
88 | func (dao *UserDao) Transaction(ctx context.Context, f func(ctx context.Context, tx *gdb.TX) error) (err error) {
89 | return dao.Ctx(ctx).Transaction(ctx, f)
90 | }
91 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/plugin.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalPluginDao is internal type for wrapping internal DAO implements.
12 | type internalPluginDao = *internal.PluginDao
13 |
14 | // pluginDao is the data access object for table plugin.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type pluginDao struct {
17 | internalPluginDao
18 | }
19 |
20 | var (
21 | // Plugin is globally public accessible object for table plugin operations.
22 | Plugin = pluginDao{
23 | internal.NewPluginDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/script.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalScriptDao is internal type for wrapping internal DAO implements.
12 | type internalScriptDao = *internal.ScriptDao
13 |
14 | // scriptDao is the data access object for table script.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type scriptDao struct {
17 | internalScriptDao
18 | }
19 |
20 | var (
21 | // Script is globally public accessible object for table script operations.
22 | Script = scriptDao{
23 | internal.NewScriptDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/task.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalTaskDao is internal type for wrapping internal DAO implements.
12 | type internalTaskDao = *internal.TaskDao
13 |
14 | // taskDao is the data access object for table task.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type taskDao struct {
17 | internalTaskDao
18 | }
19 |
20 | var (
21 | // Task is globally public accessible object for table task operations.
22 | Task = taskDao{
23 | internal.NewTaskDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/task_preset.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalTaskPresetDao is internal type for wrapping internal DAO implements.
12 | type internalTaskPresetDao = *internal.TaskPresetDao
13 |
14 | // taskPresetDao is the data access object for table task_preset.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type taskPresetDao struct {
17 | internalTaskPresetDao
18 | }
19 |
20 | var (
21 | // TaskPreset is globally public accessible object for table task_preset operations.
22 | TaskPreset = taskPresetDao{
23 | internal.NewTaskPresetDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/user.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalUserDao is internal type for wrapping internal DAO implements.
12 | type internalUserDao = *internal.UserDao
13 |
14 | // userDao is the data access object for table user.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type userDao struct {
17 | internalUserDao
18 | }
19 |
20 | var (
21 | // User is globally public accessible object for table user operations.
22 | User = userDao{
23 | internal.NewUserDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/dao/vm.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // This is auto-generated by GoFrame CLI tool only once. Fill this file as you wish.
3 | // =================================================================================
4 |
5 | package dao
6 |
7 | import (
8 | "go-ops/internal/service/internal/dao/internal"
9 | )
10 |
11 | // internalVmDao is internal type for wrapping internal DAO implements.
12 | type internalVmDao = *internal.VmDao
13 |
14 | // vmDao is the data access object for table vm.
15 | // You can define custom methods on it to extend its functionality as you wish.
16 | type vmDao struct {
17 | internalVmDao
18 | }
19 |
20 | var (
21 | // Vm is globally public accessible object for table vm operations.
22 | Vm = vmDao{
23 | internal.NewVmDao(),
24 | }
25 | )
26 |
27 | // Fill with you ideas below.
28 |
--------------------------------------------------------------------------------
/internal/service/internal/do/agent_info.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | "github.com/gogf/gf/v2/os/gtime"
10 | )
11 |
12 | // AgentInfo is the golang structure of table agent_info for DAO operations like Where/Data.
13 | type AgentInfo struct {
14 | g.Meta `orm:"table:agent_info, do:true"`
15 | Id interface{} //
16 | Peerid interface{} // 节点id
17 | Name interface{} // agent名称
18 | ExpectedStatus interface{} // 期望状态
19 | Status interface{} //
20 | IsDefault interface{} //
21 | Timeout interface{} // 启动超时时间
22 | Created *gtime.Time //
23 | Updated *gtime.Time //
24 | Version interface{} // 版本信息
25 | }
26 |
--------------------------------------------------------------------------------
/internal/service/internal/do/app.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | "github.com/gogf/gf/v2/os/gtime"
10 | )
11 |
12 | // App is the golang structure of table app for DAO operations like Where/Data.
13 | type App struct {
14 | g.Meta `orm:"table:app, do:true"`
15 | Id interface{} //
16 | Appid interface{} //
17 | ApiKey interface{} //
18 | SecKey interface{} //
19 | Owner interface{} //
20 | Name interface{} // 应用名
21 | Status interface{} // 1启用 0 禁用
22 | OwnerUid interface{} // 拥有者uid
23 | CreatedAt *gtime.Time //
24 | UpdatedAt *gtime.Time //
25 | }
26 |
--------------------------------------------------------------------------------
/internal/service/internal/do/check_item.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | "github.com/gogf/gf/v2/os/gtime"
10 | )
11 |
12 | // CheckItem is the golang structure of table check_item for DAO operations like Where/Data.
13 | type CheckItem struct {
14 | g.Meta `orm:"table:check_item, do:true"`
15 | Id interface{} //
16 | CheckItemId interface{} //
17 | Name interface{} // 检查项名称
18 | Desc interface{} // 检查项描述
19 | Type interface{} //
20 | Content interface{} // 检查项内容
21 | Creater interface{} // 创建人
22 | Updater interface{} // 更新人
23 | Created *gtime.Time //
24 | Updated *gtime.Time //
25 | Weight interface{} // 权重
26 | WaitTime interface{} // 超时时间
27 | Cmd interface{} // 脚本执行解释器
28 | }
29 |
--------------------------------------------------------------------------------
/internal/service/internal/do/check_tpl.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | "github.com/gogf/gf/v2/os/gtime"
10 | )
11 |
12 | // CheckTpl is the golang structure of table check_tpl for DAO operations like Where/Data.
13 | type CheckTpl struct {
14 | g.Meta `orm:"table:check_tpl, do:true"`
15 | Id interface{} //
16 | Tid interface{} // 唯一id
17 | Name interface{} // 名称
18 | Description interface{} // 描述
19 | Type interface{} // 类型
20 | Creater interface{} // 创建人
21 | Updater interface{} // 更新人
22 | Created *gtime.Time //
23 | Updated *gtime.Time //
24 | }
25 |
--------------------------------------------------------------------------------
/internal/service/internal/do/check_tpl_detail.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | )
10 |
11 | // CheckTplDetail is the golang structure of table check_tpl_detail for DAO operations like Where/Data.
12 | type CheckTplDetail struct {
13 | g.Meta `orm:"table:check_tpl_detail, do:true"`
14 | Id interface{} //
15 | Tid interface{} // 模板id
16 | Cid interface{} // 检查项id
17 | Sort interface{} // 排序
18 | Weight interface{} // 权重
19 | }
20 |
--------------------------------------------------------------------------------
/internal/service/internal/do/cron_task.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | "github.com/gogf/gf/v2/os/gtime"
10 | )
11 |
12 | // CronTask is the golang structure of table cron_task for DAO operations like Where/Data.
13 | type CronTask struct {
14 | g.Meta `orm:"table:cron_task, do:true"`
15 | Id interface{} //
16 | CronUid interface{} //
17 | Name interface{} // 定时任务名称
18 | Content interface{} // 定时任务内容
19 | CronExpr interface{} // 定时任务表达式
20 | LastrunTime *gtime.Time // 最后执行时间
21 | NextrunTime *gtime.Time //
22 | Status interface{} // 状态 running stopped
23 | Creater interface{} // 创建人
24 | Updater interface{} // 更新人
25 | Created *gtime.Time //
26 | Updated *gtime.Time //
27 | Type interface{} // 任务类型
28 | }
29 |
--------------------------------------------------------------------------------
/internal/service/internal/do/plugin.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | "github.com/gogf/gf/v2/os/gtime"
10 | )
11 |
12 | // Plugin is the golang structure of table plugin for DAO operations like Where/Data.
13 | type Plugin struct {
14 | g.Meta `orm:"table:plugin, do:true"`
15 | Id interface{} //
16 | Uuid interface{} // 插件uuid
17 | Name interface{} // 插件名
18 | PackageName interface{} // 包名
19 | Os interface{} // 操作系统
20 | Arch interface{} // 架构
21 | Md5 interface{} // 包md5名称
22 | Creater interface{} // 创建人
23 | Updater interface{} // 更新人
24 | Created *gtime.Time //
25 | Updated *gtime.Time //
26 | }
27 |
--------------------------------------------------------------------------------
/internal/service/internal/do/script.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | "github.com/gogf/gf/v2/os/gtime"
10 | )
11 |
12 | // Script is the golang structure of table script for DAO operations like Where/Data.
13 | type Script struct {
14 | g.Meta `orm:"table:script, do:true"`
15 | Id interface{} //
16 | Name interface{} // 命令名称
17 | Content interface{} // 脚本内容
18 | Args interface{} // 参数信息
19 | Desc interface{} // 描述信息
20 | Type interface{} // 脚本类型shell或者powershell
21 | OwnerType interface{} // 命令拥有者类型
22 | OwnerUid interface{} // 拥有者uid
23 | Created *gtime.Time //
24 | Updated *gtime.Time //
25 | ScriptUid interface{} //
26 | WaitTime interface{} // 脚本超时时间
27 | Cmd interface{} // 脚本解释器
28 | Ext interface{} // 脚本文件扩展名
29 | }
30 |
--------------------------------------------------------------------------------
/internal/service/internal/do/task.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | "github.com/gogf/gf/v2/os/gtime"
10 | )
11 |
12 | // Task is the golang structure of table task for DAO operations like Where/Data.
13 | type Task struct {
14 | g.Meta `orm:"table:task, do:true"`
15 | Id interface{} //
16 | TaskId interface{} // 任务id
17 | Type interface{} // 任务类型
18 | Content interface{} // 任务内容
19 | Name interface{} // 任务名称
20 | Reqid interface{} // 请求id
21 | ParentId interface{} //
22 | Status interface{} //
23 | Created *gtime.Time //
24 | Updated *gtime.Time //
25 | Creater interface{} // 创建人
26 | }
27 |
--------------------------------------------------------------------------------
/internal/service/internal/do/task_preset.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | "github.com/gogf/gf/v2/os/gtime"
10 | )
11 |
12 | // TaskPreset is the golang structure of table task_preset for DAO operations like Where/Data.
13 | type TaskPreset struct {
14 | g.Meta `orm:"table:task_preset, do:true"`
15 | Id interface{} //
16 | Uuid interface{} //
17 | Name interface{} // 任务名
18 | Type interface{} // 任务类型
19 | Content interface{} // 任务内容
20 | Peers interface{} // 主机列表
21 | Creater interface{} // 创建人
22 | Created *gtime.Time // 创建时间
23 | Updated *gtime.Time // 更新时间
24 | Updater interface{} // 更新人
25 | }
26 |
--------------------------------------------------------------------------------
/internal/service/internal/do/user.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | "github.com/gogf/gf/v2/os/gtime"
10 | )
11 |
12 | // User is the golang structure of table user for DAO operations like Where/Data.
13 | type User struct {
14 | g.Meta `orm:"table:user, do:true"`
15 | Id interface{} //
16 | Uid interface{} //
17 | Username interface{} //
18 | Passwd interface{} //
19 | Email interface{} //
20 | Created *gtime.Time //
21 | Updated *gtime.Time //
22 | Phone interface{} //
23 | Avatar interface{} // 头像
24 | Status interface{} // 状态 1 启用 2 禁用
25 | }
26 |
--------------------------------------------------------------------------------
/internal/service/internal/do/vm.go:
--------------------------------------------------------------------------------
1 | // =================================================================================
2 | // Code generated by GoFrame CLI tool. DO NOT EDIT.
3 | // =================================================================================
4 |
5 | package do
6 |
7 | import (
8 | "github.com/gogf/gf/v2/frame/g"
9 | "github.com/gogf/gf/v2/os/gtime"
10 | )
11 |
12 | // Vm is the golang structure of table vm for DAO operations like Where/Data.
13 | type Vm struct {
14 | g.Meta `orm:"table:vm, do:true"`
15 | Id interface{} //
16 | Uuid interface{} // 主机uuid
17 | Name interface{} //
18 | Hostname interface{} //
19 | OsType interface{} // 操作系统类型
20 | OsInfo interface{} // 操作系统信息
21 | Hosttype interface{} // 主机类型
22 | Networktype interface{} // 网络类型
23 | PrivateIp interface{} //
24 | PublicIp interface{} //
25 | Created *gtime.Time //
26 | Updated *gtime.Time //
27 | Creater interface{} // 创建人
28 | Address interface{} //
29 | PeerId interface{} //
30 | }
31 |
--------------------------------------------------------------------------------
/internal/service/user.go:
--------------------------------------------------------------------------------
1 | package service
2 |
3 | import (
4 | "context"
5 |
6 | "go-ops/internal/model/entity"
7 | "go-ops/internal/service/internal/dao"
8 | "go-ops/internal/service/internal/do"
9 | v1 "go-ops/pkg/api/v1"
10 |
11 | "github.com/gogf/gf/v2/errors/gerror"
12 | )
13 |
14 | type (
15 | // sUser is service struct of module User.
16 | sUser struct{}
17 | )
18 |
19 | var (
20 | // insUser is the instance of service User.
21 | insUser = sUser{}
22 | )
23 |
24 | // User returns the interface of User service.
25 | func User() *sUser {
26 | return &insUser
27 | }
28 |
29 | func (s *sUser) Get(ctx context.Context, uid string) (r *entity.User, err error) {
30 | err = dao.User.Ctx(ctx).Where("uid = ?", uid).Scan(&r)
31 | if err != nil {
32 | return
33 | }
34 | if r == nil {
35 | return
36 | }
37 | return
38 | }
39 |
40 | func (s *sUser) IsUsernameAvailable(ctx context.Context, username string) (bool, error) {
41 | count, err := dao.User.Ctx(ctx).Where(do.User{
42 | Username: username,
43 | }).Count()
44 | if err != nil {
45 | return false, err
46 | }
47 | return count == 0, nil
48 | }
49 |
50 | // SignIn creates session for given user account.
51 | func (s *sUser) SignIn(ctx context.Context, in *v1.AuthLoginReq) (r *entity.User, err error) {
52 | var user *entity.User
53 | err = dao.User.Ctx(ctx).Where(do.User{
54 | Username: in.Username,
55 | Passwd: in.Passwd,
56 | }).Scan(&user)
57 | if err != nil {
58 | return nil, err
59 | }
60 | if user == nil {
61 | return nil, gerror.New(`username or Password not correct`)
62 | }
63 | return user, nil
64 | }
65 |
--------------------------------------------------------------------------------
/manifest/config/config.yaml:
--------------------------------------------------------------------------------
1 | server:
2 | address: ":8000"
3 | openapiPath: "/api.json"
4 | swaggerPath: "/swagger"
5 |
6 | logger:
7 | level : "all"
8 | stdout: true
9 |
10 |
11 |
12 |
--------------------------------------------------------------------------------
/manifest/deploy/kustomize/base/deployment.yaml:
--------------------------------------------------------------------------------
1 | apiVersion: apps/v1
2 | kind: Deployment
3 | metadata:
4 | name: template-single
5 | labels:
6 | app: template-single
7 | spec:
8 | replicas: 1
9 | selector:
10 | matchLabels:
11 | app: template-single
12 | template:
13 | metadata:
14 | labels:
15 | app: template-single
16 | spec:
17 | containers:
18 | - name : main
19 | image: template-single
20 | imagePullPolicy: Always
21 |
22 |
--------------------------------------------------------------------------------
/manifest/deploy/kustomize/base/kustomization.yaml:
--------------------------------------------------------------------------------
1 | apiVersion: kustomize.config.k8s.io/v1beta1
2 | kind: Kustomization
3 | resources:
4 | - deployment.yaml
5 | - service.yaml
6 |
7 |
8 |
9 |
--------------------------------------------------------------------------------
/manifest/deploy/kustomize/base/service.yaml:
--------------------------------------------------------------------------------
1 | apiVersion: v1
2 | kind: Service
3 | metadata:
4 | name: template-single
5 | spec:
6 | ports:
7 | - port: 80
8 | protocol: TCP
9 | targetPort: 8000
10 | selector:
11 | app: template-single
12 |
13 |
--------------------------------------------------------------------------------
/manifest/deploy/kustomize/overlays/develop/configmap.yaml:
--------------------------------------------------------------------------------
1 | apiVersion: v1
2 | kind: ConfigMap
3 | metadata:
4 | name: template-single-configmap
5 | data:
6 | config.yaml: |
7 | server:
8 | address: ":8000"
9 | openapiPath: "/api.json"
10 | swaggerPath: "/swagger"
11 |
12 | logger:
13 | level : "all"
14 | stdout: true
15 |
--------------------------------------------------------------------------------
/manifest/deploy/kustomize/overlays/develop/deployment.yaml:
--------------------------------------------------------------------------------
1 | apiVersion: apps/v1
2 | kind: Deployment
3 | metadata:
4 | name: template-single
5 | spec:
6 | template:
7 | spec:
8 | containers:
9 | - name : main
10 | image: template-single:develop
--------------------------------------------------------------------------------
/manifest/deploy/kustomize/overlays/develop/kustomization.yaml:
--------------------------------------------------------------------------------
1 | apiVersion: kustomize.config.k8s.io/v1beta1
2 | kind: Kustomization
3 |
4 | resources:
5 | - ../../base
6 | - configmap.yaml
7 |
8 | patchesStrategicMerge:
9 | - deployment.yaml
10 |
11 | namespace: default
12 |
13 |
14 |
15 |
--------------------------------------------------------------------------------
/manifest/docker/Dockerfile:
--------------------------------------------------------------------------------
1 | FROM loads/alpine:3.8
2 |
3 | ###############################################################################
4 | # INSTALLATION
5 | ###############################################################################
6 |
7 | ENV WORKDIR /app
8 | ADD resource $WORKDIR/
9 | ADD ./temp/linux_amd64/main $WORKDIR/main
10 | RUN chmod +x $WORKDIR/main
11 |
12 | ###############################################################################
13 | # START
14 | ###############################################################################
15 | WORKDIR $WORKDIR
16 | CMD ./main
17 |
--------------------------------------------------------------------------------
/manifest/docker/docker.sh:
--------------------------------------------------------------------------------
1 | #!/bin/bash
2 |
3 | # This shell is executed before docker build.
4 |
5 |
6 |
7 |
8 |
9 |
--------------------------------------------------------------------------------
/pkg/agent/action/agent_manager_control.go:
--------------------------------------------------------------------------------
1 | package action
2 |
3 | import (
4 | "context"
5 | "path/filepath"
6 |
7 | "go-ops/internal/model"
8 | "go-ops/pkg/agent/script/cmd"
9 | "go-ops/pkg/errors"
10 | "go-ops/pkg/system"
11 |
12 | "github.com/toolkits/file"
13 | )
14 |
15 | func control(workdir, args string, timeout int) (stdout string, err error) {
16 | controlScript := getControlName()
17 | cmdRunner := system.NewExecCmdRunner(context.Background())
18 |
19 | command := cmd.BuildCommand("." + string(filepath.Separator) + controlScript)
20 | command.Args = append(command.Args, args)
21 |
22 | command.Timeout = timeout
23 | command.WorkingDir = workdir
24 |
25 | stdout, stderr, exitCode, _, err := cmdRunner.RunComplexCommand(command)
26 | if stderr != "" {
27 | err = errors.WrapErrorf(err, "stderr:%s \nexitCode:%d", stderr, exitCode)
28 | return
29 | }
30 |
31 | return
32 | }
33 |
34 | func (self *agentManager) workDir(agentName string) (r string) {
35 | return self.agentDir + string(filepath.Separator) + agentName
36 | }
37 |
38 | func (self *agentManager) Start(a *model.AgentInfo) (err error) {
39 | _, err = control(self.workDir(a.Name), "start", a.Timeout)
40 | return
41 | }
42 |
43 | func (self *agentManager) Stop(a *model.AgentInfo) (err error) {
44 | _, err = control(self.workDir(a.Name), "stop", a.Timeout)
45 | return
46 | }
47 |
48 | func (self *agentManager) Status(a *model.AgentInfo) (s string, err error) {
49 | s, err = control(self.workDir(a.Name), "status", a.Timeout)
50 | if err != nil {
51 | return
52 | }
53 | return
54 | }
55 |
56 | func (self *agentManager) Version(a *model.AgentInfo) (s string, err error) {
57 | s, err = control(self.workDir(a.Name), "status", a.Timeout)
58 | if err != nil {
59 | return
60 | }
61 | return
62 | }
63 |
64 | func (self *agentManager) Delete(a *model.AgentInfo) (err error) {
65 | _, err = control(self.workDir(a.Name), "delete", a.Timeout)
66 | if err != nil {
67 | return
68 | }
69 |
70 | return
71 | }
72 |
73 | func (self *agentManager) ControlScriptCheck(a *model.AgentInfo) bool {
74 | if file.IsExist(self.workDir(a.Name) + string(filepath.Separator) + getControlName()) {
75 | return true
76 | }
77 | return false
78 | }
79 |
--------------------------------------------------------------------------------
/pkg/agent/action/agent_manager_control_unix.go:
--------------------------------------------------------------------------------
1 | package action
2 |
3 | func getControlName() string {
4 | return "control"
5 | }
6 |
--------------------------------------------------------------------------------
/pkg/agent/action/agent_manager_control_windows.go:
--------------------------------------------------------------------------------
1 | package action
2 |
3 | import (
4 | "os/exec"
5 | "path/filepath"
6 | )
7 |
8 | func getControlName() string {
9 | return "control.ps1"
10 | }
11 |
12 | func control(workdir, args string) (err error) {
13 | controlScript := getControlName()
14 |
15 | cmd := exec.Command("powershell", "-File", "."+string(filepath.Separator)+controlScript, args)
16 | cmd.Dir = workdir
17 | return cmd.Start()
18 | }
19 |
--------------------------------------------------------------------------------
/pkg/agent/action/file_disk.go:
--------------------------------------------------------------------------------
1 | package action
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/model"
6 | "io/ioutil"
7 | "os"
8 | "path/filepath"
9 | )
10 |
11 | type fileDisk struct {
12 | }
13 |
14 | func FileDisk() *fileDisk {
15 | return &fileDisk{}
16 | }
17 |
18 | func (self *fileDisk) GetDirInfo(ctx context.Context, spath string) (r []*model.FileInfo, err error) {
19 | infos, err := ioutil.ReadDir(spath)
20 | if err != nil {
21 | return
22 | }
23 | for _, item := range infos {
24 | fi := &model.FileInfo{
25 | Name: item.Name(),
26 | ModTime: item.ModTime(),
27 | Size: item.Size(),
28 | Path: spath,
29 | }
30 | if item.IsDir() {
31 | fi.Type = "dir"
32 | } else {
33 | fi.Path = filepath.Join(spath, item.Name())
34 | }
35 | r = append(r, fi)
36 | }
37 | return
38 | }
39 |
40 | func (self *fileDisk) Remove(ctx context.Context, filename string) (err error) {
41 | err = os.Remove(filename)
42 | return
43 | }
44 |
45 | func (self *fileDisk) Move(ctx context.Context, src, dest string) (err error) {
46 | err = os.Rename(src, dest)
47 | return
48 | }
49 |
50 | func (self *fileDisk) CreateDir(ctx context.Context, spath string) (r []*model.FileInfo, err error) {
51 |
52 | if err = os.MkdirAll(spath, os.ModePerm); err != nil {
53 | return
54 | }
55 |
56 | return self.GetDirInfo(ctx, spath)
57 | }
58 |
--------------------------------------------------------------------------------
/pkg/agent/action/runner.go:
--------------------------------------------------------------------------------
1 | package action
2 |
--------------------------------------------------------------------------------
/pkg/agent/agent_filemanager.go:
--------------------------------------------------------------------------------
1 | package agent
2 |
3 | import (
4 | "context"
5 | "fmt"
6 | "go-ops/internal/model"
7 | "go-ops/internal/peer"
8 | "go-ops/pkg/agent/action"
9 | "path/filepath"
10 |
11 | "github.com/luxingwen/pnet"
12 | )
13 |
14 | func (self *OspAgent) ListFileInfo(req *model.PeerListFileInfo, srcId string, msgID []byte, rpath string, pn *pnet.PNet) {
15 | ctx := context.Background()
16 | fileInfos, err := action.FileDisk().GetDirInfo(ctx, req.Path)
17 | if err != nil {
18 | err = peer.SendMsgReplay(pn, &model.PeerListFileInfoRes{List: fileInfos, Path: req.Path, Error: err.Error()}, msgID, srcId, rpath)
19 | return
20 | }
21 |
22 | err = peer.SendMsgReplay(pn, &model.PeerListFileInfoRes{List: fileInfos, Path: req.Path}, msgID, srcId, rpath)
23 | if err != nil {
24 | fmt.Println("send msg replay err:", err)
25 | }
26 | }
27 |
28 | func (self *OspAgent) MoveFile(req *model.PeerMoveFile, srcId string, msgID []byte, rpath string, pn *pnet.PNet) {
29 | ctx := context.Background()
30 | err := action.FileDisk().Move(ctx, req.Src, req.Dst)
31 | if err != nil {
32 | err = peer.SendMsgReplay(pn, &model.PeerListFileInfoRes{Error: err.Error()}, msgID, srcId, rpath)
33 |
34 | return
35 | }
36 |
37 | bpath := filepath.Clean(req.Dst)
38 |
39 | fileInfos, err := action.FileDisk().GetDirInfo(ctx, bpath)
40 | if err != nil {
41 | err = peer.SendMsgReplay(pn, &model.PeerListFileInfoRes{Error: err.Error()}, msgID, srcId, rpath)
42 |
43 | return
44 | }
45 |
46 | err = peer.SendMsgReplay(pn, &model.PeerListFileInfoRes{List: fileInfos, Path: bpath}, msgID, srcId, rpath)
47 | if err != nil {
48 | fmt.Println("send msg replay err:", err)
49 | }
50 | }
51 |
52 | func (self *OspAgent) CreateDir(req *model.PeerNewDir, srcId string, msgID []byte, rpath string, pn *pnet.PNet) {
53 | ctx := context.Background()
54 | fileInfos, err := action.FileDisk().CreateDir(ctx, req.Path)
55 | if err != nil {
56 | err = peer.SendMsgReplay(pn, &model.PeerListFileInfoRes{Error: err.Error()}, msgID, srcId, rpath)
57 | return
58 | }
59 |
60 | err = peer.SendMsgReplay(pn, &model.PeerListFileInfoRes{List: fileInfos, Path: req.Path}, msgID, srcId, rpath)
61 | if err != nil {
62 | fmt.Println("send msg replay err:", err)
63 | }
64 | }
65 |
66 | func (self *OspAgent) RemoveFile(req *model.PeerDeleteFile, srcId string, msgID []byte, rpath string, pn *pnet.PNet) {
67 | ctx := context.Background()
68 | err := action.FileDisk().Remove(ctx, req.Path)
69 | if err != nil {
70 | err = peer.SendMsgReplay(pn, &model.PeerListFileInfoRes{Error: err.Error()}, msgID, srcId, rpath)
71 | fmt.Println("remove file err:", err)
72 | return
73 | }
74 |
75 | bpath := filepath.Dir(req.Path)
76 |
77 | fileInfos, err := action.FileDisk().GetDirInfo(ctx, bpath)
78 | if err != nil {
79 | fmt.Println("获取文件夹失败:", err)
80 | err = peer.SendMsgReplay(pn, &model.PeerListFileInfoRes{Error: err.Error()}, msgID, srcId, rpath)
81 | return
82 | }
83 |
84 | err = peer.SendMsgReplay(pn, &model.PeerListFileInfoRes{List: fileInfos, Path: bpath}, msgID, srcId, rpath)
85 | if err != nil {
86 | fmt.Println("send msg replay err:", err)
87 | }
88 | }
89 |
--------------------------------------------------------------------------------
/pkg/agent/cmd.go:
--------------------------------------------------------------------------------
1 | package agent
2 |
3 | import (
4 | "encoding/json"
5 | "errors"
6 | "go-ops/internal/peer"
7 | "io/ioutil"
8 | "os"
9 | "time"
10 |
11 | "github.com/luxingwen/pnet/config"
12 | "github.com/luxingwen/pnet/node"
13 | "github.com/shirou/gopsutil/v3/host"
14 |
15 | agentConf "go-ops/pkg/agent/config"
16 |
17 | "github.com/google/uuid"
18 |
19 | log "go-ops/pkg/logger"
20 | )
21 |
22 | func Main() (err error) {
23 | b, err := ioutil.ReadFile("config.json")
24 | if err != nil {
25 | log.Error("read config.json err:", err)
26 | return
27 | }
28 |
29 | var cfg agentConf.Config
30 | err = json.Unmarshal(b, &cfg)
31 | if err != nil {
32 | log.Error("Unmarshal err:", err)
33 | return
34 | }
35 |
36 | opsAgent := NewOspAgent(&cfg)
37 |
38 | name, err := os.Hostname()
39 | if err != nil {
40 | log.Error("Unmarshal err:", err)
41 | return err
42 | }
43 |
44 | conf := &config.Config{
45 | Port: uint16(cfg.Port),
46 | Name: name,
47 | }
48 |
49 | id := ""
50 | hid, err := host.HostID()
51 | if err != nil {
52 | id = uuid.New().String()
53 | } else {
54 | id = hid
55 | }
56 |
57 | id = id + "@agent"
58 |
59 | p, err := peer.NewPnet(id, conf, opsAgent.HandlerFunc)
60 | if err != nil {
61 | log.Error("new peer err:", err)
62 | return
63 | }
64 |
65 | err = p.ApplyMiddleware(node.RemoteNodeReady{func(rmoteNode *node.RemoteNode) bool {
66 | peer.SendMsgBroadCast(p, opsAgent.GetPeerInfo(p))
67 | return true
68 | }, 0})
69 |
70 | if err != nil {
71 | log.Error("apply middle ware err:", err)
72 | }
73 |
74 | err = p.Start()
75 | if err != nil {
76 | log.Error("peer start err:", err)
77 | return
78 | }
79 |
80 | if len(cfg.Bootlist) == 0 {
81 | log.Error("need join remote peer")
82 | return errors.New("need join remote peer")
83 | }
84 | addr := cfg.Bootlist[0]
85 |
86 | var remoteNode *node.RemoteNode
87 |
88 | for {
89 | remoteNode, err = p.Join(addr)
90 | if err != nil {
91 | log.Errorf("join remote perr: %s err:%v", addr, err)
92 | time.Sleep(time.Second)
93 | continue
94 | }
95 | break
96 | }
97 |
98 | for {
99 | time.Sleep(time.Second)
100 | if remoteNode.IsStopped() {
101 | remoteNode1, err := p.Join(remoteNode.Addr)
102 | if err != nil {
103 | log.Errorf("join remote perr: %s err:%v", addr, err)
104 | } else if remoteNode1 != nil {
105 | remoteNode = remoteNode1
106 | }
107 | }
108 | }
109 |
110 | }
111 |
--------------------------------------------------------------------------------
/pkg/agent/cmdrunner/cmd_runner.go:
--------------------------------------------------------------------------------
1 | package cmdrunner
2 |
3 | import (
4 | "go-ops/internal/model"
5 | ospsys "go-ops/pkg/system"
6 | )
7 |
8 | type CmdResult struct {
9 | Stdout []byte
10 | Stderr []byte
11 |
12 | ExitStatus int
13 | ResCode model.ResCode
14 | }
15 |
16 | type CmdRunner interface {
17 | RunCommand(jobId string, cmd ospsys.Command) (*CmdResult, error)
18 | }
19 |
--------------------------------------------------------------------------------
/pkg/agent/config/config.go:
--------------------------------------------------------------------------------
1 | package config
2 |
3 | type Config struct {
4 | Port uint32 `json:"port"`
5 | TaskPath string `json:"taskPath"`
6 | Bootlist []string `json:"bootlist"`
7 | }
8 |
--------------------------------------------------------------------------------
/pkg/agent/handlermsg.go:
--------------------------------------------------------------------------------
1 | package agent
2 |
3 | import (
4 | "encoding/json"
5 | "go-ops/internal/model"
6 |
7 | "github.com/luxingwen/pnet"
8 | "github.com/luxingwen/pnet/log"
9 | )
10 |
11 | func (ospAgent *OspAgent) HandlerFunc(msg interface{}, msgID []byte, srcID, rpath string, pn *pnet.PNet) {
12 | switch v := msg.(type) {
13 | case *model.ScriptJob:
14 | ospAgent.CreateScriptTask(v, srcID, msgID, rpath, pn)
15 | case *model.ScriptJobCancel:
16 | ospAgent.CancelcriptTask(v.Jobid, srcID, msgID, rpath, pn)
17 | case *model.GetTaskInfo:
18 | ospAgent.GetTaskInfo(v.TaskId, srcID, msgID, rpath, pn)
19 | case *model.DownloadFileJob:
20 | ospAgent.DownloadFile(v, srcID, msgID, rpath, pn)
21 | case *model.PeerListFileInfo:
22 | ospAgent.ListFileInfo(v, srcID, msgID, rpath, pn)
23 | case *model.PeerMoveFile:
24 | ospAgent.MoveFile(v, srcID, msgID, rpath, pn)
25 | case *model.PeerNewDir:
26 | ospAgent.CreateDir(v, srcID, msgID, rpath, pn)
27 | case *model.PeerDeleteFile:
28 | ospAgent.RemoveFile(v, srcID, msgID, rpath, pn)
29 | case *model.GetPeerInfo:
30 | ospAgent.FineAgentByHostname(v, srcID, msgID, rpath, pn)
31 | default:
32 | b, _ := json.Marshal(v)
33 | log.Error("msg handler not found,msg: %s", string(b))
34 | }
35 | }
36 |
--------------------------------------------------------------------------------
/pkg/agent/script/cmd/build_cmd_unix.go:
--------------------------------------------------------------------------------
1 | package cmd
2 |
3 | import (
4 | ospenv "go-ops/pkg/agent/script/pathenv"
5 | ospsys "go-ops/pkg/system"
6 | )
7 |
8 | func BuildCommand(path string) ospsys.Command {
9 | return ospsys.Command{
10 | Name: path,
11 | Env: map[string]string{
12 | "PATH": ospenv.Path(),
13 | },
14 | }
15 | }
16 |
--------------------------------------------------------------------------------
/pkg/agent/script/cmd/build_cmd_windows.go:
--------------------------------------------------------------------------------
1 | package cmd
2 |
3 | import (
4 | ospenv "go-ops/pkg/agent/script/pathenv"
5 | ospsys "go-ops/pkg/system"
6 | )
7 |
8 | func BuildCommand(path string) ospsys.Command {
9 | return ospsys.Command{
10 | Name: "powershell",
11 | Args: []string{path},
12 | Env: map[string]string{
13 | "PATH": ospenv.Path(),
14 | },
15 | }
16 | }
17 |
--------------------------------------------------------------------------------
/pkg/agent/script/content_script.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | import (
4 | "go-ops/internal/model"
5 | "go-ops/pkg/agent/cmdrunner"
6 | "os"
7 |
8 | "go-ops/pkg/agent/script/cmd"
9 | "path"
10 | "strings"
11 | )
12 |
13 | type ContentScript struct {
14 | GenericScript
15 | cmd string
16 | input string
17 | ext string
18 | }
19 |
20 | func NewContentScript(
21 | runner cmdrunner.CmdRunner,
22 | jobid string,
23 | path string,
24 | cmd string,
25 | content string,
26 | env map[string]string,
27 | timeout int,
28 | input string,
29 | user string,
30 | args []string,
31 | ext string,
32 | ) ContentScript {
33 |
34 | if cmd == "" {
35 | cmd = Cmder
36 | }
37 |
38 | s := ContentScript{cmd: cmd, input: input, ext: ext}
39 | s.GenericScript.runner = runner
40 | s.GenericScript.path = path
41 | s.GenericScript.content = content
42 | s.GenericScript.env = env
43 | s.GenericScript.jobid = jobid
44 | s.GenericScript.timeout = timeout
45 | s.GenericScript.user = user
46 | s.GenericScript.args = args
47 |
48 | return s
49 | }
50 |
51 | func (s ContentScript) Run() (r model.ResCmd) {
52 |
53 | if s.path == "" {
54 | s.path = ScriptPath
55 | }
56 | runpath := path.Join(s.path, s.jobid+s.ext)
57 | err := s.ensureContainingDir(runpath)
58 |
59 | if err != nil {
60 | return s.getResCmd(nil, err)
61 | }
62 |
63 | f, err := os.OpenFile(runpath, fileOpenFlag, fileOpenPerm)
64 | if err != nil {
65 | return
66 | }
67 | defer f.Close()
68 |
69 | _, err = f.WriteString(s.content)
70 | if err != nil {
71 | return
72 | }
73 | err = f.Close()
74 | if err != nil {
75 | return
76 | }
77 |
78 | cmdstr, args := getCmdArgs(s.cmd)
79 |
80 | command := cmd.BuildCommand(cmdstr)
81 |
82 | command.Args = append(command.Args, args...)
83 | command.Args = append(command.Args, runpath)
84 | command.Args = append(command.Args, s.args...)
85 | command.Timeout = s.timeout
86 | command.User = s.user
87 | command.WorkingDir = s.path
88 |
89 | if s.input != "" {
90 | command.Stdin = strings.NewReader(s.input)
91 | }
92 |
93 | for key, val := range s.env {
94 | command.Env[key] = val
95 | }
96 |
97 | res, err := s.runner.RunCommand(s.jobid, command)
98 | return s.getResCmd(res, err)
99 | }
100 |
--------------------------------------------------------------------------------
/pkg/agent/script/content_script_test.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | import (
4 | "context"
5 | "fmt"
6 | "go-ops/internal/model"
7 |
8 | "testing"
9 |
10 | "github.com/gogf/gf/v2/util/guid"
11 | )
12 |
13 | func testGetContentScriptJob(content, spath string) model.ScriptJob {
14 | s := model.Script{
15 | Content: content,
16 | Path: spath,
17 | ExecWay: model.ExecContent,
18 | Ext: ".sh",
19 | }
20 | s1 := model.ScriptJob{
21 | Jobid: guid.S(),
22 | Script: s,
23 | }
24 | return s1
25 | }
26 |
27 | func TestContentGenericScript(t *testing.T) {
28 | // s := testGetScriptJob("ls", "")
29 | ctx, _ := context.WithCancel(context.Background())
30 | // res := NewJobScriptProvider(ctx, s).Run()
31 |
32 | s := testGetContentScriptJob("echo $HOME", "")
33 | res := NewJobScriptProvider(ctx, s).Run()
34 | fmt.Println("res:", res)
35 | }
36 |
--------------------------------------------------------------------------------
/pkg/agent/script/ext_unix.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | const ScriptExt = ""
4 |
5 | const Cmder = "/bin/sh"
6 |
7 | const ScriptPath = "/tmp/go-ops/script"
8 |
--------------------------------------------------------------------------------
/pkg/agent/script/ext_windows.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | const ScriptExt = ".ps1"
4 |
5 | const Cmder = "powershell"
6 |
7 | const ScriptPath = "/tmp/go-ops/script"
8 |
--------------------------------------------------------------------------------
/pkg/agent/script/generic_script.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | import (
4 | "os"
5 | "path/filepath"
6 |
7 | "github.com/charlievieth/fs"
8 |
9 | "go-ops/internal/model"
10 | "go-ops/pkg/agent/cmdrunner"
11 | "go-ops/pkg/agent/script/cmd"
12 | )
13 |
14 | const (
15 | fileOpenFlag int = os.O_RDWR | os.O_CREATE | os.O_TRUNC
16 | fileOpenPerm os.FileMode = os.FileMode(0777)
17 | )
18 |
19 | type ExecType int
20 |
21 | const (
22 | ExecCmd ExecType = iota // 直接命令执行
23 | ExecContent // 需要解释器 内容保存在一个文件脚本里面执行
24 | ExecScript // 需要解释器 脚本就在本地
25 | ExecURL // 脚本从云端下载
26 | )
27 |
28 | type GenericScript struct {
29 | runner cmdrunner.CmdRunner
30 | path string
31 | content string
32 | args []string
33 | jobid string
34 | env map[string]string
35 | timeout int
36 | user string
37 | }
38 |
39 | func NewScript(
40 | runner cmdrunner.CmdRunner,
41 | jobid string,
42 | path string,
43 | content string,
44 | env map[string]string,
45 | timeout int,
46 | user string,
47 | args []string,
48 | ) GenericScript {
49 | return GenericScript{
50 | runner: runner,
51 | path: path,
52 | content: content,
53 | env: env,
54 | jobid: jobid,
55 | timeout: timeout,
56 | user: user,
57 | args: args,
58 | }
59 | }
60 |
61 | func (s GenericScript) Path() string { return s.path }
62 |
63 | func (s GenericScript) Run() (r model.ResCmd) {
64 | cmdstr, args := getCmdArgs(s.content)
65 | command := cmd.BuildCommand(cmdstr)
66 | for key, val := range s.env {
67 | command.Env[key] = val
68 | }
69 |
70 | command.Args = append(command.Args, args...)
71 | command.Timeout = s.timeout
72 | command.User = s.user
73 | command.WorkingDir = s.path
74 |
75 | res, err := s.runner.RunCommand(s.jobid, command)
76 | return s.getResCmd(res, err)
77 | }
78 |
79 | func (s GenericScript) getResCmd(res *cmdrunner.CmdResult, err error) (r model.ResCmd) {
80 | if err != nil {
81 | r.Code = model.CodeNotRun
82 | r.Err = err.Error()
83 | r.ExitCode = -1
84 | return
85 | }
86 | r = model.ResCmd{
87 | Stdout: string(res.Stdout),
88 | Stderr: string(res.Stderr),
89 | ExitCode: res.ExitStatus,
90 | }
91 |
92 | if res.ResCode != "" {
93 | r.Code = res.ResCode
94 | return
95 | }
96 |
97 | if r.ExitCode == 0 {
98 | r.Code = model.CodeSuccess
99 | } else {
100 | r.Code = model.CodeFailed
101 | }
102 | return
103 | }
104 |
105 | func (s GenericScript) ensureContainingDir(fullLogFilename string) error {
106 | dir, _ := filepath.Split(fullLogFilename)
107 | return fs.MkdirAll(dir, os.FileMode(0750))
108 | }
109 |
--------------------------------------------------------------------------------
/pkg/agent/script/generic_script_test.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | import (
4 | "context"
5 | "fmt"
6 | "go-ops/internal/model"
7 |
8 | "testing"
9 |
10 | "github.com/gogf/gf/v2/util/guid"
11 | )
12 |
13 | func testGetScriptJob(content, spath string) model.ScriptJob {
14 | s := model.Script{
15 | Content: content,
16 | Path: spath,
17 | }
18 | s1 := model.ScriptJob{
19 | Jobid: guid.S(),
20 | Script: s,
21 | }
22 | return s1
23 | }
24 |
25 | func TestGenericScript(t *testing.T) {
26 | s := testGetScriptJob("ls", "")
27 | ctx, _ := context.WithCancel(context.Background())
28 | res := NewJobScriptProvider(ctx, s).Run()
29 |
30 | fmt.Println("res0:", res)
31 |
32 | s = testGetScriptJob("pwd", "")
33 | res = NewJobScriptProvider(ctx, s).Run()
34 | fmt.Println("res1:", res)
35 |
36 | s = testGetScriptJob("pwd", "/home/lewis/go/src/go-ops/pkg/agent/script")
37 | res = NewJobScriptProvider(ctx, s).Run()
38 | fmt.Println("res2:", res)
39 |
40 | s = testGetScriptJob("ls", "/home/lewis/go/src/go-ops/pkg/agent/script")
41 | res = NewJobScriptProvider(ctx, s).Run()
42 | fmt.Println("res3:", res)
43 |
44 | s = testGetScriptJob("echo $HOME", "")
45 | res = NewJobScriptProvider(ctx, s).Run()
46 | fmt.Println("res4:", res)
47 | }
48 |
--------------------------------------------------------------------------------
/pkg/agent/script/job_script.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/model"
6 | "go-ops/pkg/agent/cmdrunner"
7 | ospsys "go-ops/pkg/system"
8 | "os"
9 | "strings"
10 |
11 | log "go-ops/pkg/logger"
12 | )
13 |
14 | type JobScriptProvider struct {
15 | cmdRunner cmdrunner.CmdRunner
16 | scripter Script
17 | }
18 |
19 | func NewJobScriptProvider(
20 | ctx context.Context,
21 | scriptJob model.ScriptJob,
22 | ) JobScriptProvider {
23 | path := scriptJob.Script.Path
24 | if path == "" {
25 | path = ScriptPath + "/" + scriptJob.Jobid
26 | }
27 | runer := cmdrunner.NewScriptCmdRunner(ospsys.NewExecCmdRunner(ctx), ScriptPath)
28 | p := JobScriptProvider{
29 | cmdRunner: runer,
30 | }
31 |
32 | // 如果文件得扩展名不是以.结尾
33 | if scriptJob.Script.Ext != "" && !strings.HasPrefix(scriptJob.Script.Ext, ".") {
34 | scriptJob.Script.Ext = "." + scriptJob.Script.Ext
35 | }
36 |
37 | if scriptJob.Script.Ext == "" {
38 | scriptJob.Script.Ext = ScriptExt
39 | }
40 |
41 | log.Infof("[%s] new script job, execway: %v, waitime:%d", scriptJob.Jobid, scriptJob.Script.ExecWay, scriptJob.Script.Timeout)
42 |
43 | switch scriptJob.Script.ExecWay {
44 | case model.ExecCmd:
45 | p.scripter = NewScript(runer, scriptJob.Jobid, path, scriptJob.Script.Content, scriptJob.Script.Env, scriptJob.Script.Timeout, scriptJob.Script.User, scriptJob.Script.Args)
46 | case model.ExecContent:
47 | p.scripter = NewContentScript(runer, scriptJob.Jobid, path, scriptJob.Script.Cmd, scriptJob.Script.Content, scriptJob.Script.Env,
48 | scriptJob.Script.Timeout, scriptJob.Script.Input, scriptJob.Script.User, scriptJob.Script.Args, scriptJob.Script.Ext)
49 | case model.ExecScriptName:
50 | p.scripter = NewNameScript(runer, scriptJob.Jobid, path, scriptJob.Script.Cmd, scriptJob.Script.Content, scriptJob.Script.Env, scriptJob.Script.Timeout, scriptJob.Script.Input, scriptJob.Script.User, scriptJob.Script.Args)
51 | case model.ExecURL:
52 | p.scripter = NewUrlScript(runer, scriptJob.Jobid, path, scriptJob.Script.Cmd, scriptJob.Script.Content, scriptJob.Script.Env, scriptJob.Script.Timeout, scriptJob.Script.User, scriptJob.Script.Args, scriptJob.Script.Input)
53 |
54 | }
55 | return p
56 | }
57 |
58 | func (p JobScriptProvider) Run() model.ResCmd {
59 | return p.scripter.Run()
60 | }
61 |
62 | func getCmdArgs(s string) (cmd string, args []string) {
63 | s = os.ExpandEnv(s)
64 | list := strings.Fields(s)
65 | if len(list) <= 0 {
66 | return
67 | }
68 | if len(list) == 1 {
69 | cmd = list[0]
70 | return
71 | }
72 |
73 | cmd = list[0]
74 | args = list[1:]
75 | return
76 |
77 | }
78 |
--------------------------------------------------------------------------------
/pkg/agent/script/name_script.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | import (
4 | "go-ops/internal/model"
5 | "go-ops/pkg/agent/cmdrunner"
6 | "go-ops/pkg/agent/script/cmd"
7 | "strings"
8 | )
9 |
10 | type NameScript struct {
11 | GenericScript
12 | cmd string
13 | input string
14 | }
15 |
16 | func NewNameScript(
17 | runner cmdrunner.CmdRunner,
18 | jobid string,
19 | path string,
20 | cmd string,
21 | content string,
22 | env map[string]string,
23 | timeout int,
24 | input string,
25 | user string,
26 | args []string,
27 | ) NameScript {
28 |
29 | if cmd == "" {
30 | cmd = Cmder
31 | }
32 |
33 | s := NameScript{cmd: cmd, input: input}
34 | s.GenericScript.runner = runner
35 | s.GenericScript.path = path
36 | s.GenericScript.content = content
37 | s.GenericScript.env = env
38 | s.GenericScript.jobid = jobid
39 | s.GenericScript.timeout = timeout
40 | s.GenericScript.args = args
41 | return s
42 | }
43 |
44 | func (s NameScript) Run() (r model.ResCmd) {
45 |
46 | cmdstr, args := getCmdArgs(s.cmd)
47 | command := cmd.BuildCommand(cmdstr)
48 | command.Args = append(command.Args, args...)
49 | command.Args = append(command.Args, s.content)
50 | command.Args = append(command.Args, s.args...)
51 | command.Timeout = s.timeout
52 | command.User = s.user
53 | command.WorkingDir = s.path
54 |
55 | if s.input != "" {
56 | command.Stdin = strings.NewReader(s.input)
57 | }
58 |
59 | for key, val := range s.env {
60 | command.Env[key] = val
61 | }
62 |
63 | res, err := s.runner.RunCommand(s.jobid, command)
64 | return s.getResCmd(res, err)
65 | }
66 |
--------------------------------------------------------------------------------
/pkg/agent/script/pack_script.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | import (
4 | "context"
5 | "go-ops/internal/model"
6 | "go-ops/pkg/agent/action"
7 | "go-ops/pkg/agent/cmdrunner"
8 | "go-ops/pkg/agent/script/cmd"
9 | "path"
10 | "path/filepath"
11 | "strings"
12 | )
13 |
14 | // 压缩包脚本，脚本放在一个压缩包内
15 | type PackScript struct {
16 | GenericScript
17 | cmd string
18 | input string
19 | fileMd5 string
20 | }
21 |
22 | func NewPackScript(
23 | runner cmdrunner.CmdRunner,
24 | jobid string,
25 | path string,
26 | cmd string,
27 | content string,
28 | env map[string]string,
29 | timeout int,
30 | input string,
31 | user string,
32 | args []string,
33 | filemd5 string,
34 | ) PackScript {
35 |
36 | if cmd == "" {
37 | cmd = Cmder
38 | }
39 |
40 | s := PackScript{cmd: cmd, input: input, fileMd5: filemd5}
41 | s.GenericScript.runner = runner
42 | s.GenericScript.path = path
43 | s.GenericScript.content = content
44 | s.GenericScript.env = env
45 | s.GenericScript.jobid = jobid
46 | s.GenericScript.timeout = timeout
47 | s.GenericScript.user = user
48 | s.GenericScript.args = args
49 |
50 | return s
51 | }
52 |
53 | func (s PackScript) Run() (r model.ResCmd) {
54 |
55 | if s.path == "" {
56 | s.path = ScriptPath
57 | }
58 | runpath := path.Join(s.path, s.jobid)
59 | err := s.ensureContainingDir(runpath)
60 |
61 | if err != nil {
62 | return s.getResCmd(nil, err)
63 | }
64 |
65 | filename := filepath.Join(runpath, filepath.Base(s.content))
66 |
67 | ctx := context.Background()
68 |
69 | err = action.Download(ctx, filename, s.content)
70 |
71 | if err != nil {
72 | return s.getResCmd(nil, err)
73 | }
74 |
75 | err = action.CheckFileMd5(filename, s.fileMd5)
76 | if err != nil {
77 | return s.getResCmd(nil, err)
78 | }
79 |
80 | err = action.Untar(filename)
81 | if err != nil {
82 | return s.getResCmd(nil, err)
83 | }
84 |
85 | cmdstr, args := getCmdArgs(s.cmd)
86 |
87 | command := cmd.BuildCommand(cmdstr)
88 |
89 | command.Args = append(command.Args, args...)
90 | command.Args = append(command.Args, runpath)
91 | command.Args = append(command.Args, s.args...)
92 | command.Timeout = s.timeout
93 | command.User = s.user
94 | command.WorkingDir = s.path
95 |
96 | if s.input != "" {
97 | command.Stdin = strings.NewReader(s.input)
98 | }
99 |
100 | for key, val := range s.env {
101 | command.Env[key] = val
102 | }
103 |
104 | res, err := s.runner.RunCommand(s.jobid, command)
105 | return s.getResCmd(res, err)
106 | }
107 |
--------------------------------------------------------------------------------
/pkg/agent/script/pathenv/pathenv.go:
--------------------------------------------------------------------------------
1 | package pathenv
2 |
3 | import "os"
4 |
5 | func Path() string { return os.Getenv("PATH") }
6 |
--------------------------------------------------------------------------------
/pkg/agent/script/plugin.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | import (
4 | "context"
5 | "encoding/json"
6 | "go-ops/internal/model"
7 | "time"
8 | )
9 |
10 | type ScriptPlugin struct {
11 | GenericScript
12 | cmd string
13 | input string
14 | }
15 |
16 | func NewScriptPlugin(
17 | jobid string,
18 | path string,
19 | cmd string,
20 | content string,
21 | env map[string]string,
22 | timeout int,
23 | input string,
24 | user string,
25 | args []string,
26 | ) ScriptPlugin {
27 |
28 | if cmd == "" {
29 | cmd = Cmder
30 | }
31 |
32 | s := ScriptPlugin{cmd: cmd, input: input}
33 | s.GenericScript.path = path
34 | s.GenericScript.content = content
35 | s.GenericScript.env = env
36 | s.GenericScript.jobid = jobid
37 | s.GenericScript.timeout = timeout
38 | s.GenericScript.user = user
39 | s.GenericScript.args = args
40 |
41 | return s
42 | }
43 |
44 | func (s ScriptPlugin) Run() (r model.ResCmd) {
45 |
46 | opsAgentPlugin := new(AgentPlugin)
47 |
48 | err := json.Unmarshal([]byte(s.content), opsAgentPlugin)
49 | if err != nil {
50 | r.Err = err.Error()
51 | return
52 | }
53 |
54 | opsAgentPlugin.Args = s.args
55 | opsAgentPlugin.Cmd = s.cmd
56 | opsAgentPlugin.Timeout = time.Duration(s.timeout * int(time.Second))
57 |
58 | res, err := opsAgentPlugin.Run(context.Background(), []byte(s.input))
59 |
60 | if err != nil {
61 | r.Err = err.Error()
62 | return
63 | }
64 |
65 | r.Res = res
66 |
67 | return
68 | }
69 |
--------------------------------------------------------------------------------
/pkg/agent/script/script_interface.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | import "go-ops/internal/model"
4 |
5 | type Script interface {
6 | Path() string
7 | Run() model.ResCmd
8 | }
9 |
10 | //counterfeiter:generate . CancellableScript
11 |
12 | type CancellableScript interface {
13 | Script
14 | Cancel() error
15 | }
16 |
--------------------------------------------------------------------------------
/pkg/agent/script/url_script.go:
--------------------------------------------------------------------------------
1 | package script
2 |
3 | import (
4 | "go-ops/internal/model"
5 | "go-ops/pkg/agent/cmdrunner"
6 | "go-ops/pkg/agent/script/cmd"
7 | "io"
8 | "net/http"
9 | "os"
10 | "path"
11 | "strings"
12 | )
13 |
14 | type UrlScript struct {
15 | GenericScript
16 | cmd string
17 | input string
18 | }
19 |
20 | func NewUrlScript(
21 | runner cmdrunner.CmdRunner,
22 | jobid string,
23 | path string,
24 | cmd string,
25 | content string,
26 | env map[string]string,
27 | timeout int,
28 | user string,
29 | args []string,
30 | input string,
31 | ) UrlScript {
32 |
33 | if cmd == "" {
34 | cmd = Cmder
35 | }
36 |
37 | s := UrlScript{cmd: cmd, input: input}
38 | s.GenericScript.runner = runner
39 | s.GenericScript.path = path
40 | s.GenericScript.content = content
41 | s.GenericScript.env = env
42 | s.GenericScript.jobid = jobid
43 | s.GenericScript.timeout = timeout
44 | s.GenericScript.user = user
45 | s.GenericScript.args = args
46 | return s
47 | }
48 |
49 | func (s UrlScript) Run() (r model.ResCmd) {
50 |
51 | err := s.downloadFile()
52 | if err != nil {
53 | return s.getResCmd(nil, err)
54 | }
55 | cmdstr, args := getCmdArgs(s.cmd)
56 | command := cmd.BuildCommand(cmdstr)
57 | command.Args = append(command.Args, args...)
58 |
59 | filename := path.Base(s.content)
60 |
61 | command.Args = append(command.Args, filename)
62 | command.Args = append(command.Args, s.args...)
63 | command.Timeout = s.timeout
64 | command.User = s.user
65 | command.WorkingDir = s.path
66 |
67 | for key, val := range s.env {
68 | command.Env[key] = val
69 | }
70 |
71 | if s.input != "" {
72 | command.Stdin = strings.NewReader(s.input)
73 | }
74 |
75 | res, err := s.runner.RunCommand(s.jobid, command)
76 | return s.getResCmd(res, err)
77 | }
78 |
79 | func (s UrlScript) downloadFile() (err error) {
80 | savePath := s.path
81 | if savePath == ScriptPath {
82 | savePath = path.Join(s.path, s.jobid)
83 | }
84 |
85 | filename := path.Base(s.content)
86 |
87 | savePath = path.Join(savePath, filename)
88 |
89 | err = s.ensureContainingDir(savePath)
90 | if err != nil {
91 | return
92 | }
93 |
94 | f, err := os.OpenFile(savePath, fileOpenFlag, fileOpenPerm)
95 | if err != nil {
96 | return
97 | }
98 | defer f.Close()
99 |
100 | res, err := http.Get(s.content)
101 | if err != nil {
102 | return
103 | }
104 | defer res.Body.Close()
105 |
106 | _, err = io.Copy(f, res.Body)
107 | return
108 | }
109 |
--------------------------------------------------------------------------------
/pkg/agent/task/async_task_service.go:
--------------------------------------------------------------------------------
1 | package task
2 |
3 | import (
4 | "github.com/panjf2000/ants/v2"
5 | )
6 |
7 | type asyncTaskService struct {
8 | currentTasks map[string]Task
9 | taskChan chan Task
10 | taskSem chan func()
11 | taskPool *ants.PoolWithFunc
12 | }
13 |
14 | func NewAsyncTaskService() (service Service) {
15 | s := asyncTaskService{
16 | currentTasks: make(map[string]Task),
17 | taskChan: make(chan Task),
18 | taskSem: make(chan func()),
19 | }
20 |
21 | p, _ := ants.NewPoolWithFunc(100, s.execTask)
22 |
23 | s.taskPool = p
24 | go s.processTasks()
25 | go s.processSemFuncs()
26 |
27 | return s
28 | }
29 |
30 | func (service asyncTaskService) CreateTask(
31 | id string,
32 | req interface{},
33 | taskFunc Func,
34 | cancelFunc CancelFunc,
35 | endFunc EndFunc,
36 | ) Task {
37 | return Task{
38 | ID: id,
39 | Req: req,
40 | State: StateRunning,
41 | Func: taskFunc,
42 | CancelFunc: cancelFunc,
43 | EndFunc: endFunc,
44 | }
45 | }
46 |
47 | func (service asyncTaskService) StartTask(task Task) {
48 | taskChan := make(chan Task)
49 |
50 | service.taskSem <- func() {
51 | service.currentTasks[task.ID] = task
52 | taskChan <- task
53 | }
54 |
55 | recordedTask := <-taskChan
56 | service.taskChan <- recordedTask
57 | }
58 |
59 | func (service asyncTaskService) FindTaskWithID(id string) (Task, bool) {
60 | taskChan := make(chan Task)
61 | foundChan := make(chan bool)
62 |
63 | service.taskSem <- func() {
64 | task, found := service.currentTasks[id]
65 | taskChan <- task
66 | foundChan <- found
67 | }
68 |
69 | return <-taskChan, <-foundChan
70 | }
71 |
72 | func (service asyncTaskService) processSemFuncs() {
73 |
74 | for {
75 | do := <-service.taskSem
76 | do()
77 | }
78 | }
79 |
80 | func (service asyncTaskService) processTasks() {
81 |
82 | for {
83 | task := <-service.taskChan
84 | service.taskPool.Invoke(task)
85 |
86 | }
87 | }
88 |
89 | func (service asyncTaskService) execTask(val interface{}) {
90 | task := val.(Task)
91 | value, err := task.Func()
92 | if err != nil {
93 | task.Error = err
94 | task.State = StateFailed
95 | } else {
96 | task.Value = value
97 | task.State = StateDone
98 | }
99 |
100 | if task.EndFunc != nil {
101 | task.EndFunc(task)
102 | }
103 |
104 | task.Func = nil
105 | task.CancelFunc = nil
106 | task.EndFunc = nil
107 |
108 | service.taskSem <- func() {
109 | service.currentTasks[task.ID] = task
110 | }
111 |
112 | }
113 |
114 | func (service asyncTaskService) Stop() {
115 | for _, item := range service.currentTasks {
116 | if item.State == StateRunning {
117 | item.Cancel()
118 | }
119 | }
120 | }
121 |
--------------------------------------------------------------------------------
/pkg/agent/task/service_interface.go:
--------------------------------------------------------------------------------
1 | package task
2 |
3 | type Service interface {
4 | CreateTask(string, interface{}, Func, CancelFunc, EndFunc) Task
5 | StartTask(Task)
6 | FindTaskWithID(string) (Task, bool)
7 | }
8 |
--------------------------------------------------------------------------------
/pkg/agent/task/task.go:
--------------------------------------------------------------------------------
1 | package task
2 |
3 | type Func func() (value interface{}, err error)
4 |
5 | type CancelFunc func(task Task) error
6 |
7 | type EndFunc func(task Task)
8 |
9 | type State string
10 |
11 | const (
12 | StateRunning State = "running"
13 | StateDone State = "done"
14 | StateFailed State = "failed"
15 | )
16 |
17 | type Task struct {
18 | ID string
19 | Req interface{} // 请求参数
20 | State State
21 | Value interface{}
22 | Error error
23 |
24 | Func Func
25 | CancelFunc CancelFunc
26 | EndFunc EndFunc
27 | }
28 |
29 | func (t Task) Cancel() error {
30 | if t.CancelFunc != nil {
31 | return t.CancelFunc(t)
32 | }
33 | return nil
34 | }
35 |
36 | type StateValue struct {
37 | AgentTaskID string `json:"agent_task_id"`
38 | State State `json:"state"`
39 | }
40 |
--------------------------------------------------------------------------------
/pkg/agent/task/task_manager_interface.go:
--------------------------------------------------------------------------------
1 | package task
2 |
3 | type Info struct {
4 | TaskID string
5 | Method string
6 | Payload []byte
7 | }
8 |
9 | type ManagerProvider interface {
10 | NewManager(string) Manager
11 | }
12 |
13 | type Manager interface {
14 | GetInfos() ([]Info, error)
15 | AddInfo(taskInfo Info) error
16 | RemoveInfo(taskID string) error
17 | }
18 |
--------------------------------------------------------------------------------
/pkg/agent/test/test.go:
--------------------------------------------------------------------------------
1 | package main
2 |
3 | import (
4 | "context"
5 | "fmt"
6 | "go-ops/internal/model"
7 | "go-ops/pkg/agent"
8 | "go-ops/pkg/agent/script"
9 | "go-ops/pkg/agent/task"
10 | )
11 |
12 | func main() {
13 |
14 | s := model.Script{
15 | Content: "ping baidu.com",
16 | }
17 | s1 := model.ScriptJob{
18 | Jobid: "1111",
19 | Script: s,
20 | }
21 |
22 | ctx, cancel := context.WithCancel(context.Background())
23 | startFunc := func() (r interface{}, err error) {
24 |
25 | res := script.NewJobScriptProvider(ctx, s1).Run()
26 | r = &model.ResponseResCmd{
27 | Jobid: s1.Jobid,
28 | ResCmd: res,
29 | }
30 |
31 | return
32 | }
33 |
34 | endFunc := func(t task.Task) {
35 |
36 | fmt.Println("end -> ")
37 |
38 | }
39 |
40 | c := func(t task.Task) error {
41 | fmt.Println("取消")
42 | cancel()
43 | return nil
44 | }
45 |
46 | ospAgent := agent.NewOspAgent("./")
47 | t := ospAgent.CreateTask(s1.Jobid, s1, startFunc, c, endFunc)
48 | ospAgent.StartTask(t)
49 | // ospAgent.StartTask(t)
50 |
51 | select {}
52 |
53 | }
54 |
--------------------------------------------------------------------------------
/pkg/agent/test/test2.go:
--------------------------------------------------------------------------------
1 | package main
2 |
3 | import (
4 | "fmt"
5 | "os"
6 | "os/exec"
7 | )
8 |
9 | func main() {
10 |
11 | res, err := exec.Command("echo", "$HOME").CombinedOutput()
12 | if err != nil {
13 | return
14 | }
15 |
16 | s := os.ExpandEnv("echo hello")
17 | fmt.Println("s:", s)
18 |
19 | fmt.Println(string(res))
20 | }
21 |
--------------------------------------------------------------------------------
/pkg/agent/test/test3.go:
--------------------------------------------------------------------------------
1 | package main
2 |
3 | import (
4 | "context"
5 | "fmt"
6 | "go-ops/internal/model"
7 | "go-ops/pkg/agent/script"
8 | "os"
9 | "os/signal"
10 | "syscall"
11 | )
12 |
13 | func main() {
14 | s := model.Script{
15 | Content: "ping baidu.com",
16 | }
17 | s1 := model.ScriptJob{
18 | Jobid: "1111",
19 | Script: s,
20 | }
21 | fmt.Println("tttt->")
22 |
23 | ctx, cancel := context.WithCancel(context.Background())
24 | go handlesig(func() {
25 | cancel()
26 | })
27 | res := script.NewJobScriptProvider(ctx, s1).Run()
28 |
29 | fmt.Println(res)
30 |
31 | }
32 |
33 | func handlesig(f func()) {
34 | c := make(chan os.Signal, 1)
35 | signal.Notify(
36 | c,
37 | syscall.SIGINT,
38 | syscall.SIGTERM,
39 | syscall.SIGKILL,
40 | syscall.SIGQUIT,
41 | )
42 | sig := <-c
43 | fmt.Println("sig:", sig)
44 | f()
45 |
46 | }
47 |
--------------------------------------------------------------------------------
/pkg/api/v1/agent_info.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
3 | import (
4 | "github.com/gogf/gf/v2/frame/g"
5 | )
6 |
7 | type AddAgentReq struct {
8 | g.Meta `path:"/peer/agent/add" tags:"Agent管理" method:"post" summary:"节点添加agent"`
9 | List []*AgentInfo `json:"list"`
10 | }
11 |
12 | type AgentInfo struct {
13 | Peerid string `json:"peerid" dc:"节点id, 空表示默认版本,所有节点将使用这个版本" ` // 节点id
14 | Name string `json:"name" dc:"agent 名称" ` // agent名称
15 | Version string `json:"version" dc:"agent 版本"`
16 | ExpectedStatus string `json:"expectedStatus" dc:"期望状态 running,stopped,deleted" ` // 期望状态
17 | DownloadUrl string `json:"downloadUrl" dc:"下载地址"` // 下载地址
18 | Status string `json:"status" dc:"agent 当前状态" ` // 当前agent状态
19 | IsDefault int `json:"isDefault" dc:"是否默认安装 0-否 1-是" ` //
20 | Timeout int `json:"timeout" dc:"agent 启动超时时间" ` // 启动超时时间
21 | }
22 |
23 | type AgentInfoRes struct {
24 | List []*AgentInfo `json:"list"`
25 | }
26 |
27 | type UpdateAgentReq struct {
28 | g.Meta `path:"/peer/agent/update" tags:"Agent管理" method:"post" summary:"节点更新agent"`
29 | List []*AgentInfo `json:"list"`
30 | }
31 |
32 | type QueryAgentStatusReq struct {
33 | g.Meta `path:"/peer/agent/status" tags:"Agent管理" method:"post" summary:"查询agent状态"`
34 | List []*QueryAgentStatus `json:"list"`
35 | }
36 |
37 | type QueryAgentStatus struct {
38 | Peerid string `json:"peerid" dc:"节点id"`
39 | Agents []string `json:"agents" dc:"需要查询的agent名称列表"`
40 | }
41 |
42 | type PeerAgentStatus struct {
43 | Peerid string `json:"peerid" dc:"节点id"`
44 | Agents []*AgentInfo
45 | }
46 |
47 | type QueryAgentStatusRes struct {
48 | List []*PeerAgentStatus
49 | }
50 |
--------------------------------------------------------------------------------
/pkg/api/v1/app.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
3 | import (
4 | "go-ops/internal/model/entity"
5 |
6 | "github.com/gogf/gf/v2/frame/g"
7 | )
8 |
9 | type AddAppReq struct {
10 | g.Meta `path:"/v1/m/app/create" tags:"App管理" method:"post" summary:"创建一个app"`
11 | Name string `json:"name" dc:"应用名"` // 应用名
12 | Owner string `json:"owner" dc:"拥有者"` // 拥有者
13 | }
14 |
15 | type AddAppRes struct {
16 | Appid string `json:"appid" dc:"appid" ` //
17 | ApiKey string `json:"apiKey" ` //
18 | SecKey string `json:"secKey" ` //
19 | Owner string `json:"owner" dc:"拥有者" ` //
20 | Name string `json:"name" dc:"应用名" ` // 应用名
21 | Status int `json:"status" dc:"状态 1启用 0 禁用" ` // 1启用 0 禁用
22 | OwnerUid string `json:"ownerUid" dc:"拥有者uid" ` // 拥有者uid
23 | }
24 |
25 | type UpdateAppReq struct {
26 | g.Meta `path:"/v1/m/app/update" tags:"App管理" method:"post" summary:"更新一个app"`
27 | Appid string `json:"appid" dc:"appid" ` //
28 | Name string `json:"name" dc:"应用名"` // 应用名
29 | Owner string `json:"owner" dc:"拥有者"` // 拥有者
30 | Status int `json:"status" dc:"状态 1启用 0 禁用" ` // 1启用 0 禁用
31 | }
32 |
33 | type QueryAppReq struct {
34 | g.Meta `path:"/v1/m/app/query" tags:"App管理" method:"post" summary:"查询app"`
35 | Name string `json:"name" dc:"应用名"` // 应用名
36 | Owner string `json:"owner" dc:"拥有者"` // 拥有者
37 | PageReq
38 | }
39 |
40 | type QuerySingleAppReq struct {
41 | g.Meta `path:"/v1/m/app/single_query" tags:"App管理" method:"post" summary:"通过id查询app"`
42 | AppId string `json:"appid" dc:"appid"` // 应用名
43 | }
44 |
45 | type QueryAppRes struct {
46 | Page
47 | List []*entity.App `json:"list"`
48 | }
49 |
50 | type DeleteAppReq struct {
51 | g.Meta `path:"/v1/m/app/delete" tags:"App管理" method:"post" summary:"删除app"`
52 | Appids []string `json:"appids" dc:"app id 列表"`
53 | }
54 |
--------------------------------------------------------------------------------
/pkg/api/v1/base.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
3 | type DeleteRes string
4 |
5 | type Page struct {
6 | Total int `json:"total" dc:"总数"`
7 | PageNum int `json:"pageNum" dc:"第几页"`
8 | PageSize int `json:"pageSize" dc:"每页的数量"`
9 | PageTotal int `json:"pageTotal" dc:"总共多少页"`
10 | }
11 |
12 | type PageReq struct {
13 | PageNum int `json:"pageNum" dc:"第几页" d:"1" v:"min:0#分页号码错误"`
14 | PageSize int `json:"pageSize" dc:"每页的数量" d:"10" v:"max:50#分页数量最大50条"`
15 | }
16 |
17 | type KVString struct {
18 | Key string `json:"key"`
19 | Value string `json:"value"`
20 | Desc string `json:"desc"`
21 | }
22 |
--------------------------------------------------------------------------------
/pkg/api/v1/check.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
3 | import "github.com/gogf/gf/v2/frame/g"
4 |
5 | type AddCheckItemReq struct {
6 | g.Meta `path:"/check/item/add" tags:"巡检管理" method:"post" summary:"添加巡检项"`
7 | Name string `json:"name" dc:"检查项名称" ` // 检查项名称
8 | Desc string `json:"desc" dc:"检查项描述" ` // 检查项描述
9 | Type string `json:"type" dc:"类型" ` //
10 | Content string `json:"content" dc:"内容" ` // 检查项内容
11 | WaitTime int `json:"waitTime" dc:"脚本超时时间"`
12 | Cmd string `json:"cmd" dc:"脚本解释器"`
13 | Weight float64 `json:"weight" 权重`
14 | }
15 |
16 | type CheckItemRes struct {
17 | CheckItemId string `json:"checkItemId" dc:"检查项id"`
18 | Name string `json:"name" dc:"检查项名称" ` // 检查项名称
19 | Desc string `json:"desc" dc:"检查项描述" ` // 检查项描述
20 | Type string `json:"type" dc:"类型" ` //
21 | Content string `json:"content" dc:"内容" ` // 检查项内容
22 | WaitTime int `json:"waitTime" dc:"脚本超时时间"`
23 | Cmd string `json:"cmd" dc:"脚本解释器"`
24 | Weight float64 `json:"weight" 权重`
25 | }
26 |
27 | type UpdateCheckItemReq struct {
28 | g.Meta `path:"/check/item/update" tags:"巡检管理" method:"post" summary:"更新检查项"`
29 | CheckItemRes
30 | }
31 |
32 | type QueryCheckItemReq struct {
33 | g.Meta `path:"/check/item/query" tags:"巡检管理" method:"post" summary:"查询检查项"`
34 | CheckItemId string `json:"checkItemId" dc:"检查项id"`
35 | Name string `json:"name" dc:"检查项名称" ` // 检查项名称
36 | Type string `json:"type" dc:"类型" ` //
37 | PageReq
38 | }
39 |
40 | type QueryCheckItemRes struct {
41 | Page
42 | List []*CheckItemRes `json:"list"`
43 | }
44 |
45 | type DeleteCheckItemReq struct {
46 | g.Meta `path:"/check/item/delete" tags:"巡检管理" method:"post" summary:"删除检查项"`
47 | CheckItemIds []string `json:"checkItemIds" dc:"检查项id列表"`
48 | }
49 |
50 | type CheckTplItemInfo struct {
51 | ItemId string `json"itemId" dc:"巡检项id"`
52 | Weight float64 `json:"weight" dc:"权重"`
53 | }
54 |
55 | type CheckTplItem struct {
56 | Tid string `json:"tid"`
57 | Name string `json:"name" dc:"检查项名称" ` // 检查项名称
58 | Desc string `json:"desc" dc:"检查项描述" ` // 检查项描述
59 | Type string `json:"type" dc:"类型" ` //
60 | Items []*CheckTplItemInfo `json:"items" dc:"items"` //
61 | }
62 |
63 | type CheckTplItemRes *CheckTplItem
64 |
65 | type AddCheckTplReq struct {
66 | g.Meta `path:"/check/tpl/create" tags:"巡检管理" method:"post" summary:"添加巡检模版"`
67 | CheckTplItem
68 | }
69 |
70 | type UpdateCheckTplReq struct {
71 | g.Meta `path:"/check/tpl/update" tags:"巡检管理" method:"post" summary:"更新巡检模版"`
72 | CheckTplItem
73 | }
74 |
75 | type DeleteCheckTplReq struct {
76 | g.Meta `path:"/check/tpl/delete" tags:"巡检管理" method:"post" summary:"删除巡检模版"`
77 | Tids []string `json:"tids"`
78 | }
79 |
80 | type QueryCheckTplReq struct {
81 | g.Meta `path:"/check/tpl/query" tags:"巡检管理" method:"post" summary:"查询巡检模版"`
82 | Tid string `json:"tid"`
83 | Name string `json:"name" dc:"检查项名称" ` // 检查项名称
84 | PageReq
85 | }
86 |
--------------------------------------------------------------------------------
/pkg/api/v1/download_file.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
3 | import (
4 | "go-ops/internal/model"
5 |
6 | "github.com/gogf/gf/v2/frame/g"
7 | )
8 |
9 | type DownloadFileReq struct {
10 | g.Meta `path:"/peer/downloadfile" tags:"文件分发" method:"post" summary:"创建文件分发任务"`
11 | Name string `json:"name" dc:"任务名称"`
12 | Creater string `json:"creater" dc:"创建人"`
13 | DownloadFileInfo
14 | }
15 |
16 | type DownloadFileInfo struct {
17 | Peers []string `json:"peers" dc:"节点列表"`
18 | Files []*model.DownloadFileInfo `json:"files" dc:""`
19 | }
20 |
21 | type DownloadfileRes struct {
22 | Taskid string `json:"taskid" dc:"任务id"`
23 | Status string `json:"status" dc:"状态（doing, failed, done）"`
24 | List []*DownloadfileItem `json:"list" dc:"任务列表详情"`
25 | }
26 |
27 | type DownloadfileItem struct {
28 | Status string `json:"status" dc:"状态（doing, failed, done）`
29 | model.DownloadFileJobRes
30 | }
31 |
32 | type DownloadFileDetailsReq struct {
33 | g.Meta `path:"/peer/downloadfile/details" tags:"文件分发" method:"post" summary:"文件分发任务详情"`
34 | Taskid string `json:"taskid"`
35 | }
36 |
--------------------------------------------------------------------------------
/pkg/api/v1/fileinfo.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
--------------------------------------------------------------------------------
/pkg/api/v1/plugin.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
3 | import (
4 | "go-ops/internal/model/entity"
5 |
6 | "github.com/gogf/gf/v2/frame/g"
7 | )
8 |
9 | type AddPluginReq struct {
10 | g.Meta `path:"/v1/m/plugin/create" tags:"插件" method:"post" summary:"添加插件"`
11 | Name string `json:"name" dc:"插件名" ` // 插件名
12 | PackageName string `json:"packageName" dc:"包名"` // 包名
13 | Os string `json:"os" dc:"操作系统" ` // 操作系统
14 | Arch string `json:"arch" dc:"架构" ` // 架构
15 | Md5 string `json:"md5" dc:"包md5名称" ` // 包md5名称
16 | Creater string `json:"creater" dc:"创建人"`
17 | }
18 |
19 | type PluginItemRes struct {
20 | Uuid string `json:"uuid" dc:"插件uuid"`
21 | Name string `json:"name" dc:"插件名" ` // 插件名
22 | PackageName string `json:"packageName" dc:"包名"` // 包名
23 | Os string `json:"os" dc:"操作系统" ` // 操作系统
24 | Arch string `json:"arch" dc:"架构" ` // 架构
25 | Md5 string `json:"md5" dc:"包md5名称" ` // 包md5名称
26 | Creater string `json:"creater" dc:"创建人"`
27 | Updater string `json:"updater" dc:"更新人"`
28 | Created string `json:"created" dc:"创建时间"`
29 | Updated string `json:"updated" dc:"更新时间"`
30 | }
31 |
32 | type UpdatePluginReq struct {
33 | g.Meta `path:"/v1/m/plugin/update" tags:"插件" method:"post" summary:"更新插件"`
34 | PluginItemRes
35 | }
36 |
37 | type DeletePluginReq struct {
38 | g.Meta `path:"/v1/m/plugin/delete" tags:"插件" method:"post" summary:"删除插件"`
39 | Uuids []string `json:"uuids" dc:"插件uuid列表"`
40 | }
41 |
42 | type QueryPluginReq struct {
43 | g.Meta `path:"/v1/m/plugin/query" tags:"插件" method:"post" summary:"查询插件"`
44 | PageReq
45 | PluginItemRes
46 | }
47 |
48 | type QueryPluginRes struct {
49 | Page
50 | List []*entity.Plugin `json:"list"`
51 | }
52 |
--------------------------------------------------------------------------------
/pkg/api/v1/pmanager.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
3 | import (
4 | "go-ops/internal/model"
5 |
6 | "github.com/gogf/gf/v2/frame/g"
7 | "github.com/luxingwen/pnet/protos"
8 | "github.com/luxingwen/pnet/stat"
9 | )
10 |
11 | type NodeReq struct {
12 | g.Meta `path:"/peer/nodes" tags:"节点管理" method:"post" summary:"获取节点连接信息"`
13 | NodeId string `json:"nodeid" dc:"节点id,空表示当前节点"`
14 | }
15 |
16 | type NodeRes struct {
17 | Nodes []*protos.Node `json:"" dc:"节点列表"`
18 | }
19 |
20 | type NodeConnectReq struct {
21 | g.Meta `path:"/peer/node/connect" tags:"节点管理" method:"post" summary:"连接节点"`
22 | NodeId string `json:"nodeid" dc:"节点id"`
23 | RemoteAddr string `json:"remoteAddr" dc:"远程节点连接地址"`
24 | }
25 |
26 | type NodeOpRes struct {
27 | Msg string `json:"msg"`
28 | }
29 |
30 | type NodeStopReq struct {
31 | g.Meta `path:"/peer/node/stop" tags:"节点管理" method:"post" summary:"停止节点连接"`
32 | NodeId string `json:"nodeid"`
33 | RemoteId string `json:"remoteId"`
34 | }
35 |
36 | type NodeStatReq struct {
37 | g.Meta `path:"/peer/node/stat" tags:"节点管理" method:"post" summary:"获取节点状态"`
38 | NodeId string `json:"nodeid"`
39 | }
40 |
41 | type NodeStatRes struct {
42 | *stat.NodeStat
43 | }
44 |
45 | type NodeFileListReq struct {
46 | g.Meta `path:"/peer/node/files" tags:"节点管理" method:"post" summary:"获取节点文件夹信息"`
47 | NodeId string `json:"nodeid"`
48 | Path string `json:"path" dc:"路径"`
49 | }
50 |
51 | type NodeFileListRes struct {
52 | Files []*model.FileInfo `json:"files"`
53 | }
54 |
55 | type NodeFileCreateDirReq struct {
56 | g.Meta `path:"/peer/node/files/createDir" tags:"节点管理" method:"post" summary:"创建节点文件夹"`
57 | NodeId string `json:"nodeid"`
58 | Path string `json:"path" dc:"文件夹路径"`
59 | }
60 |
61 | type NodeFileDeleteReq struct {
62 | g.Meta `path:"/peer/node/files/delete" tags:"节点管理" method:"post" summary:"删除节点文件"`
63 | NodeId string `json:"nodeid"`
64 | Path string `json:"path" dc:"文件夹(文件)路径"`
65 | }
66 |
67 | type NodeFileMoveReq struct {
68 | g.Meta `path:"/peer/node/files/move" tags:"节点管理" method:"post" summary:"移动节点文件"`
69 | NodeId string `json:"nodeid"`
70 | Src string `json:"src" dc:"源文件夹(文件)路径"`
71 | Dst string `json:"dst" dc:"目标文件夹(文件)路径"`
72 | }
73 |
--------------------------------------------------------------------------------
/pkg/api/v1/script.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
3 | import (
4 | "go-ops/internal/model/entity"
5 |
6 | "github.com/gogf/gf/v2/frame/g"
7 | )
8 |
9 | type AddScriptReq struct {
10 | g.Meta `path:"/v1/m/script/add" tags:"脚本库管理" method:"post" summary:"创建一个脚本"`
11 | Name string `json:"name" dc:"脚本名称" ` // 命令名称
12 | Content string `json:"content" dc:"脚本内容" ` // 脚本内容
13 | Args []*KVString `json:"args" dc:"参数信息" ` // 参数信息
14 | Desc string `json:"desc" dc:"描述信息" ` // 描述信息
15 | Type string `json:"type" dc:"脚本类型shell或者powershell" ` // 脚本类型shell或者powershell
16 | Creater string `json:"creater" dc:"创建人"`
17 | WaitTime int `json:"waitTime" dc:"脚本超时时间"`
18 | Cmd string `json:"cmd" dc:"脚本解释器"`
19 | Ext string `json:"ext" dc:"脚本文件扩展名"`
20 | }
21 |
22 | type ScriptQueryReq struct {
23 | g.Meta `path:"/v1/m/script/query" tags:"脚本库管理" method:"post" summary:"查询脚本库信息"`
24 | Name string `json:"name" dc:"脚本名称" `
25 | Type string `json:"type" dc:"脚本类型shell或者powershell"`
26 | ScriptId string `json:"scriptId" dc:"脚本id"`
27 | PageReq
28 | }
29 |
30 | type ScriptInfoRes struct {
31 | Page
32 | List []*entity.Script `json:"list"`
33 | }
34 |
35 | type UpdateScriptReq struct {
36 | g.Meta `path:"/v1/m/script/update" tags:"脚本库管理" method:"post" summary:"更新脚本信息"`
37 | ScriptId string `json:"scriptId" dc:"脚本id"`
38 | Name string `json:"name" dc:"脚本名称" ` // 命令名称
39 | Content string `json:"content" dc:"脚本内容" ` // 脚本内容
40 | Args []*KVString `json:"args" dc:"参数信息" ` // 参数信息
41 | Desc string `json:"desc" dc:"描述信息" ` // 描述信息
42 | Type string `json:"type" dc:"脚本类型shell或者powershell" ` // 脚本类型shell或者powershell
43 | Updater string `json:"updater" dc:"更新人"`
44 | WaitTime int `json:"waitTime" dc:"脚本超时时间"`
45 | Cmd string `json:"cmd" dc:"脚本解释器"`
46 | Ext string `json:"ext" dc:"脚本文件扩展名"`
47 | }
48 |
49 | type ScriptItemRes struct {
50 | ScriptId string `json:"scriptId" dc:"脚本id"`
51 | Name string `json:"name" dc:"脚本名称" ` // 命令名称
52 | Content string `json:"content" dc:"脚本内容" ` // 脚本内容
53 | Args []*KVString `json:"args" dc:"参数信息" ` // 参数信息
54 | Desc string `json:"desc" dc:"描述信息" ` // 描述信息
55 | Type string `json:"type" dc:"脚本类型shell或者powershell" ` // 脚本类型shell或者powershell
56 | Creater string `json:"creater" dc:"创建人"`
57 | Updater string `json:"updater" dc:"更新人"`
58 | WaitTime int `json:"waitTime" dc:"脚本超时时间"`
59 | Cmd string `json:"cmd" dc:"脚本解释器"`
60 | Ext string `json:"ext" dc:"脚本文件扩展名"`
61 | }
62 |
63 | type DeleteScriptReq struct {
64 | g.Meta `path:"/v1/m/script/delete" tags:"脚本库管理" method:"post" summary:"删除脚本信息"`
65 | ScriptIds []string `json:"scriptIds" dc:"脚本id 列表"`
66 | }
67 |
68 | type DeleteScriptRes string
69 |
--------------------------------------------------------------------------------
/pkg/api/v1/script_task.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
3 | import (
4 | "go-ops/internal/model"
5 |
6 | "github.com/gogf/gf/v2/frame/g"
7 | )
8 |
9 | type ScriptTaskReq struct {
10 | g.Meta `path:"/script/async" tags:"脚本任务" method:"post" summary:"脚本异步执行"`
11 | ScriptTask
12 | }
13 |
14 | type ScriptTask struct {
15 | Name string `json:"name" dc:"脚本任务名"`
16 | Creater string `json:"creater" dc:"创建者"`
17 | Peers []string `json:"peers" dc:"节点id列表"`
18 | Content model.Script `json:"content" dc:"脚本内容信息"`
19 | }
20 |
21 | type ScriptTaskRes struct {
22 | TaskId string `json:"taskid" dc:"任务id"`
23 | }
24 |
25 | type ScriptRes struct {
26 | List []*model.ResponseResCmd `json:"list" dc:"脚本任务详情列表"`
27 | }
28 |
29 | type ScriptTaskSyncReq struct {
30 | g.Meta `path:"/script/sync" tags:"脚本任务" method:"post" summary:"脚本同步执行"`
31 | ScriptTask
32 | }
33 |
34 | type ScriptTaskCancelReq struct {
35 | g.Meta `path:"/script/cancel" tags:"脚本任务" method:"post" summary:"取消脚本运行"`
36 | Tasks []ScriptTaskCancel `json:"tasks"`
37 | }
38 |
39 | type ScriptTaskCancel struct {
40 | PeerId string `json:"peerid" dc:"节点id"`
41 | Jobid string `json:"jobid" dc:"任务id"`
42 | Msg string `json:"msg"`
43 | }
44 |
45 | type ScriptTaskCancelRes struct {
46 | List []*ScriptTaskCancel `json:"list"`
47 | }
48 |
49 | type PeerScriptTaskInfoReq struct {
50 | g.Meta `path:"/script/peer/taskinfo" tags:"脚本任务" method:"post" summary:"远程节点上的脚本任务信息"`
51 | PeerId string `json:"peerid" dc:"节点id"`
52 | TaskId string `json:"taskid" dc:"任务id"`
53 | }
54 |
55 | type ScriptTaskInfoRes struct {
56 | PeerId string `json:"peerid"`
57 | *model.TaskInfo
58 | }
59 |
60 | type ScriptTaskExecItem struct {
61 | *model.ResponseResCmd
62 | Status string `json:"status"`
63 | }
64 |
65 | type ScriptTaskExecRes struct {
66 | TaskId string `json:"taskid" dc:"任务id"`
67 | Status string `json:"status"`
68 | List []*ScriptTaskExecItem `json:"list"`
69 | }
70 |
71 | type ScriptTaskInfoReq struct {
72 | g.Meta `path:"/script/taskinfo" tags:"脚本任务" method:"post" summary:"脚本任务信息"`
73 | TaskId string `json:"taskid"`
74 | }
75 |
--------------------------------------------------------------------------------
/pkg/api/v1/user.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
3 | import (
4 | "time"
5 |
6 | "github.com/gogf/gf/v2/frame/g"
7 | )
8 |
9 | type AuthLoginReq struct {
10 | g.Meta `path:"/user/login" method:"post" tags:"用户" summary:"登录"`
11 | Username string `json:"username" dc:"username"`
12 | Passwd string `json:"passwd"`
13 | }
14 |
15 | type AuthLoginRes struct {
16 | Token string `json:"token"`
17 | Expire time.Time `json:"expire"`
18 | }
19 |
20 | type AuthRefreshTokenReq struct {
21 | g.Meta `path:"/user/refresh_token" method:"post" tags:"用户" summary:"token续期"`
22 | }
23 |
24 | type AuthRefreshTokenRes struct {
25 | Token string `json:"token"`
26 | Expire time.Time `json:"expire"`
27 | }
28 |
29 | type AuthLogoutReq struct {
30 | g.Meta `path:"/user/logout" method:"post" tags:"用户" summary:"登出"`
31 | }
32 |
33 | type AuthLogoutRes struct {
34 | }
35 |
36 | type CurrentUserInfoReq struct {
37 | g.Meta `path:"/user/info" method:"get" tags:"用户" summary:"用户信息"`
38 | }
39 |
40 | type UserInfoRes struct {
41 | Uid string `json:"uid" dc:"uid"`
42 | Username string `json:"username" dc:"username"`
43 | Email string `json:"email"`
44 | Phone string `json:"phone"`
45 | Avatar string `json:"avatar"`
46 | }
47 |
--------------------------------------------------------------------------------
/pkg/api/v1/vm.go:
--------------------------------------------------------------------------------
1 | package v1
2 |
3 | import (
4 | "go-ops/internal/model/entity"
5 |
6 | "github.com/gogf/gf/v2/frame/g"
7 | )
8 |
9 | type AddVmReq struct {
10 | g.Meta `path:"/v1/m/vm/add" tags:"主机节点" method:"post" summary:"添加主机信息"`
11 | Name string `json:"name"`
12 | Hostname string `json:"hostname" dc:"主机hostname"`
13 | PublicIp string `json:"publicIp"`
14 | Uuid string `json:"uuid"`
15 | Os string `json:"os"`
16 | Creater string `json:"creater"`
17 | }
18 |
19 | type VmItemRes struct {
20 | PeerId string `json:"peerId"`
21 | Name string `json:"name"`
22 | Hostname string `json:"hostname" dc:"主机hostname"`
23 | PublicIp string `json:"publicIp"`
24 | Uuid string `json:"uuid"`
25 | Os string `json:"os"`
26 | Creater string `json:"creater"`
27 | Updater string `json:"updater"`
28 | }
29 |
30 | type UpdateVmReq struct {
31 | g.Meta `path:"/v1/m/vm/update" tags:"主机节点" method:"post" summary:"更新主机信息"`
32 | Name string `json:"name"`
33 | Uuid string `json:"uuid"`
34 | Os string `json:"os"`
35 | Updater string `json:"updater"`
36 | }
37 |
38 | type DeleteVmReq struct {
39 | g.Meta `path:"/v1/m/vm/delete" tags:"主机节点" method:"post" summary:"删除主机信息"`
40 | Uuids string `json:"uuids"`
41 | }
42 |
43 | type QueryVmReq struct {
44 | g.Meta `path:"/v1/m/vm/query" tags:"主机节点" method:"post" summary:"查询主机节点"`
45 | Name string `json:"name"`
46 | Uuid string `json:"uuid"`
47 | Hostname string `json:"hostname"`
48 | PeerId string `json:"peerId"`
49 | PageReq
50 | }
51 |
52 | type QueryVmRes struct {
53 | Page
54 | List []*entity.Vm `json:"list"`
55 | }
56 |
--------------------------------------------------------------------------------
/pkg/consistenthash/consistenhash.go:
--------------------------------------------------------------------------------
1 | package consistenthash
2 |
3 | /*
4 | Copyright 2013 Google Inc.
5 |
6 | Licensed under the Apache License, Version 2.0 (the "License");
7 | you may not use this file except in compliance with the License.
8 | You may obtain a copy of the License at
9 |
10 | http://www.apache.org/licenses/LICENSE-2.0
11 |
12 | Unless required by applicable law or agreed to in writing, software
13 | distributed under the License is distributed on an "AS IS" BASIS,
14 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | See the License for the specific language governing permissions and
16 | limitations under the License.
17 | */
18 |
19 | // Package consistenthash provides an implementation of a ring hash.
20 |
21 | import (
22 | "hash/crc32"
23 | "sort"
24 | "strconv"
25 | )
26 |
27 | type Hash func(data []byte) uint32
28 |
29 | type Map struct {
30 | hash Hash
31 | replicas int
32 | keys []int // Sorted
33 | hashMap map[int]string
34 | }
35 |
36 | func New(replicas int, fn Hash) *Map {
37 | m := &Map{
38 | replicas: replicas,
39 | hash: fn,
40 | hashMap: make(map[int]string),
41 | }
42 | if m.hash == nil {
43 | m.hash = crc32.ChecksumIEEE
44 | }
45 | return m
46 | }
47 |
48 | // Returns true if there are no items available.
49 | func (m *Map) IsEmpty() bool {
50 | return len(m.keys) == 0
51 | }
52 |
53 | // Adds some keys to the hash.
54 | func (m *Map) Add(keys ...string) {
55 | for _, key := range keys {
56 | for i := 0; i < m.replicas; i++ {
57 | hash := int(m.hash([]byte(strconv.Itoa(i) + key)))
58 | if m.hashMap[hash] == "" {
59 | m.keys = append(m.keys, hash)
60 | m.hashMap[hash] = key
61 | }
62 | }
63 | }
64 | sort.Ints(m.keys)
65 | }
66 |
67 | // Gets the closest item in the hash to the provided key.
68 | func (m *Map) Get(key string) string {
69 | if m.IsEmpty() {
70 | return ""
71 | }
72 |
73 | hash := int(m.hash([]byte(key)))
74 |
75 | // Binary search for appropriate replica.
76 | idx := sort.Search(len(m.keys), func(i int) bool { return m.keys[i] >= hash })
77 |
78 | // Means we have cycled back to the first replica.
79 | if idx == len(m.keys) {
80 | idx = 0
81 | }
82 |
83 | return m.hashMap[m.keys[idx]]
84 | }
85 |
--------------------------------------------------------------------------------
/pkg/daemon/attr_unix.go:
--------------------------------------------------------------------------------
1 | package daemon
2 |
3 | import "syscall"
4 |
5 | func NewSysProcAttr() *syscall.SysProcAttr {
6 | return &syscall.SysProcAttr{
7 | Chroot: "/",
8 | Setsid:true,
9 | }
10 | }
--------------------------------------------------------------------------------
/pkg/daemon/attr_windows.go:
--------------------------------------------------------------------------------
1 | package daemon
2 |
3 |
4 | import "syscall"
5 |
6 | func NewSysProcAttr() *syscall.SysProcAttr {
7 | return &syscall.SysProcAttr{
8 | HideWindow: true,
9 | }
10 | }
--------------------------------------------------------------------------------
/pkg/daemon/daemon.go:
--------------------------------------------------------------------------------
1 | package daemon
2 |
3 | import (
4 | "fmt"
5 | "log"
6 | "os"
7 | "os/exec"
8 | "strconv"
9 | "time"
10 | "syscall"
11 | )
12 |
13 | const ENV_NAME = "LXW_DAEMON_IDX"
14 |
15 | //运行时调用background的次数
16 | var runIdx int = 0
17 |
18 | type Daemon struct {
19 | pid int
20 | cpid int
21 | }
22 |
23 | func NewDaemon() *Daemon {
24 | return &Daemon{}
25 | }
26 |
27 | func (self *Daemon) Start() {
28 | Background(true)
29 |
30 | for {
31 | cmd, err := Background(false)
32 | if err != nil {
33 | log.Println("start proc err:", err)
34 | time.Sleep(time.Second)
35 | continue
36 |
37 | }
38 |
39 | // self.cpid = cmd.Process.Pid
40 | err = cmd.Wait()
41 | if err != nil {
42 | log.Println("sub procss exit err:", err)
43 | time.Sleep(time.Second)
44 | }
45 | }
46 | }
47 |
48 | // auto Start 是否自动重启
49 | func Background(isExit bool) (cmd *exec.Cmd, err error) {
50 |
51 | var runIdx int = 0
52 |
53 | envIdx, err := strconv.Atoi(os.Getenv(ENV_NAME))
54 | if err != nil {
55 | envIdx = 0
56 | }
57 | if runIdx > envIdx { //子进程, 退出
58 | return nil, err
59 | }
60 |
61 | //设置子进程环境变量
62 | env := os.Environ()
63 | env = append(env, fmt.Sprintf("%s=%d", ENV_NAME, runIdx))
64 |
65 | //启动子进程
66 | cmd, err = startProc(os.Args, env)
67 | if err != nil {
68 | log.Println(os.Getpid(), "启动子进程失败:", err)
69 | return nil, err
70 | }
71 |
72 | if isExit {
73 | syscall.Umask(0x00);
74 | os.Exit(0)
75 | }
76 |
77 | return cmd, nil
78 | }
79 |
80 | func startProc(args, env []string) (*exec.Cmd, error) {
81 | cmd := &exec.Cmd{
82 | Path: args[0],
83 | Args: args,
84 | Env: env,
85 | SysProcAttr: NewSysProcAttr(),
86 | }
87 |
88 | err := cmd.Start()
89 | if err != nil {
90 | return nil, err
91 | }
92 | return cmd, nil
93 | }
94 |
--------------------------------------------------------------------------------
/pkg/daemon/test/main.go:
--------------------------------------------------------------------------------
1 | package main
2 |
3 | import (
4 | "fmt"
5 | "io"
6 | "log"
7 | )
8 |
9 | func main() {
10 | rp, _ := io.Pipe()
11 |
12 | data := make([]byte, 1024)
13 | for {
14 | _, err := rp.Read(data)
15 | if err != nil {
16 | log.Fatal(err)
17 | }
18 | fmt.Println("data:", string(data))
19 | }
20 | }
21 |
--------------------------------------------------------------------------------
/pkg/dcron/driver.go:
--------------------------------------------------------------------------------
1 | package dcron
2 |
3 | import "time"
4 |
5 | //Driver is a driver interface
6 | type Driver interface {
7 | // Ping is check dirver is valid
8 | Ping() error
9 | SetHeartBeat(nodeID string)
10 | SetTimeout(timeout time.Duration)
11 | GetServiceNodeList(ServiceName string) ([]string, error)
12 | RegisterServiceNode(ServiceName string) (string, error)
13 | }
14 |
--------------------------------------------------------------------------------
/pkg/dcron/job_warpper.go:
--------------------------------------------------------------------------------
1 | package dcron
2 |
3 | import "github.com/robfig/cron/v3"
4 |
5 | type Job interface {
6 | Run()
7 | }
8 |
9 | //JobWarpper is a job warpper
10 | type JobWarpper struct {
11 | ID cron.EntryID
12 | Dcron *Dcron
13 | Name string
14 | CronStr string
15 | Func func()
16 | Job Job
17 | }
18 |
19 | //Run is run job
20 | func (job JobWarpper) Run() {
21 | //如果该任务分配给了这个节点 则允许执行
22 | if job.Dcron.allowThisNodeRun(job.Name) {
23 | if job.Func != nil {
24 | job.Func()
25 | }
26 | if job.Job != nil {
27 | job.Job.Run()
28 | }
29 | }
30 | }
31 |
--------------------------------------------------------------------------------
/pkg/dcron/node_pool.go:
--------------------------------------------------------------------------------
1 | package dcron
2 |
3 | import (
4 | "go-ops/pkg/consistenthash"
5 |
6 | "sync"
7 | "time"
8 | )
9 |
10 | //NodePool is a node pool
11 | type NodePool struct {
12 | serviceName string
13 | NodeID string
14 |
15 | mu sync.Mutex
16 | nodes *consistenthash.Map
17 |
18 | Driver Driver
19 | hashReplicas int
20 | hashFn consistenthash.Hash
21 | updateDuration time.Duration
22 |
23 | dcron *Dcron
24 | }
25 |
26 | func newNodePool(serverName string, driver Driver, dcron *Dcron, updateDuration time.Duration, hashReplicas int) *NodePool {
27 |
28 | err := driver.Ping()
29 | if err != nil {
30 | panic(err)
31 | }
32 |
33 | nodePool := &NodePool{
34 | Driver: driver,
35 | serviceName: serverName,
36 | dcron: dcron,
37 | hashReplicas: hashReplicas,
38 | updateDuration: updateDuration,
39 | }
40 | return nodePool
41 | }
42 |
43 | func (np *NodePool) StartPool() error {
44 | var err error
45 | np.Driver.SetTimeout(np.updateDuration)
46 | np.NodeID, err = np.Driver.RegisterServiceNode(np.serviceName)
47 | if err != nil {
48 | return err
49 | }
50 | np.Driver.SetHeartBeat(np.NodeID)
51 |
52 | err = np.updatePool()
53 | if err != nil {
54 | return err
55 | }
56 |
57 | go np.tickerUpdatePool()
58 | return nil
59 | }
60 |
61 | func (np *NodePool) updatePool() error {
62 | np.mu.Lock()
63 | defer np.mu.Unlock()
64 | nodes, err := np.Driver.GetServiceNodeList(np.serviceName)
65 | if err != nil {
66 | return err
67 | }
68 | np.nodes = consistenthash.New(np.hashReplicas, np.hashFn)
69 | for _, node := range nodes {
70 | np.nodes.Add(node)
71 | }
72 | return nil
73 | }
74 | func (np *NodePool) tickerUpdatePool() {
75 | tickers := time.NewTicker(np.updateDuration)
76 | for range tickers.C {
77 | if np.dcron.isRun {
78 | err := np.updatePool()
79 | if err != nil {
80 | np.dcron.err("update node pool error %+v", err)
81 | }
82 | } else {
83 | tickers.Stop()
84 | return
85 | }
86 | }
87 | }
88 |
89 | //PickNodeByJobName : 使用一致性hash算法根据任务名获取一个执行节点
90 | func (np *NodePool) PickNodeByJobName(jobName string) string {
91 | np.mu.Lock()
92 | defer np.mu.Unlock()
93 | if np.nodes.IsEmpty() {
94 | return ""
95 | }
96 | return np.nodes.Get(jobName)
97 | }
98 |
--------------------------------------------------------------------------------
/pkg/dcron/option.go:
--------------------------------------------------------------------------------
1 | package dcron
2 |
3 | import (
4 | "time"
5 |
6 | "github.com/robfig/cron/v3"
7 | )
8 |
9 | type Option func(*Dcron)
10 |
11 | // WithLogger both set dcron and cron logger.
12 | func WithLogger(logger interface{ Printf(string, ...interface{}) }) Option {
13 | return func(dcron *Dcron) {
14 | //set dcron logger
15 | dcron.logger = logger
16 | //set cron logger
17 | f := cron.WithLogger(cron.PrintfLogger(logger))
18 | dcron.crOptions = append(dcron.crOptions, f)
19 | }
20 | }
21 |
22 | // WithNodeUpdateDuration set node update duration
23 | func WithNodeUpdateDuration(d time.Duration) Option {
24 | return func(dcron *Dcron) {
25 | dcron.nodeUpdateDuration = d
26 | }
27 | }
28 |
29 | // WithHashReplicas set hashReplicas
30 | func WithHashReplicas(d int) Option {
31 | return func(dcron *Dcron) {
32 | dcron.hashReplicas = d
33 | }
34 | }
35 |
36 | //CronOptionLocation is warp cron with location
37 | func CronOptionLocation(loc *time.Location) Option {
38 | return func(dcron *Dcron) {
39 | f := cron.WithLocation(loc)
40 | dcron.crOptions = append(dcron.crOptions, f)
41 | }
42 | }
43 |
44 | //CronOptionSeconds is warp cron with seconds
45 | func CronOptionSeconds() Option {
46 | return func(dcron *Dcron) {
47 | f := cron.WithSeconds()
48 | dcron.crOptions = append(dcron.crOptions, f)
49 | }
50 | }
51 |
52 | // CronOptionParser is warp cron with schedules.
53 | func CronOptionParser(p cron.ScheduleParser) Option {
54 | return func(dcron *Dcron) {
55 | f := cron.WithParser(p)
56 | dcron.crOptions = append(dcron.crOptions, f)
57 | }
58 | }
59 |
60 | // CronOptionChain is Warp cron with chain
61 | func CronOptionChain(wrappers ...cron.JobWrapper) Option {
62 | return func(dcron *Dcron) {
63 | f := cron.WithChain(wrappers...)
64 | dcron.crOptions = append(dcron.crOptions, f)
65 | }
66 | }
67 |
--------------------------------------------------------------------------------
/pkg/errors/error.go:
--------------------------------------------------------------------------------
1 | package errors
2 |
3 | import (
4 | "errors"
5 | "fmt"
6 | )
7 |
8 | type ComplexError struct {
9 | Err error
10 | Cause error
11 | }
12 |
13 | func (e ComplexError) Error() string {
14 | return fmt.Sprintf("%s: %s", e.Err.Error(), e.Cause.Error())
15 | }
16 |
17 | func Error(msg string) error {
18 | return errors.New(msg)
19 | }
20 |
21 | func Errorf(msg string, args ...interface{}) error {
22 | return fmt.Errorf(msg, args...)
23 | }
24 |
25 | func WrapError(cause error, msg string) error {
26 | return WrapComplexError(cause, Error(msg))
27 | }
28 |
29 | func WrapErrorf(cause error, msg string, args ...interface{}) error {
30 | return WrapComplexError(cause, Errorf(msg, args...))
31 | }
32 |
33 | func WrapComplexError(cause, err error) error {
34 | if cause == nil {
35 | cause = Error("