This commit is contained in:
孟帅
2023-05-10 23:54:50 +08:00
parent bbe655a4d8
commit 49a96750bf
314 changed files with 15138 additions and 6244 deletions

View File

@@ -1,6 +1,7 @@
package tcpserver
import (
"context"
"github.com/gogf/gf/v2/os/gtime"
"github.com/gogf/gf/v2/util/gconv"
"hotgo/internal/consts"
@@ -10,64 +11,66 @@ import (
"hotgo/internal/model/input/msgin"
)
// onAuthSummary 获取授权信息
func (s *sTCPServer) onAuthSummary(args ...interface{}) {
// OnAuthSummary 获取授权信息
func (s *sTCPServer) OnAuthSummary(ctx context.Context, args ...interface{}) {
var (
in *msgin.AuthSummary
client = args[1].(*tcp.ClientConn)
user = tcp.GetCtx(ctx)
res = new(msgin.ResponseAuthSummary)
models *entity.SysServeLicense
)
if err := gconv.Scan(args, &in); err != nil {
s.serv.Logger.Infof(s.serv.Ctx, "onAuthSummary message Scan failed:%+v, args:%+v", err, args)
return
}
if client.Auth == nil {
res.Code = 1
res.Message = "登录信息获取失败,请重新登录"
s.serv.Write(client.Conn, res)
res.Message = err.Error()
s.serv.Reply(ctx, res)
return
}
if err := dao.SysServeLicense.Ctx(s.serv.Ctx).Where("appid = ?", client.Auth.AppId).Scan(&models); err != nil {
if user.Auth == nil {
res.Code = 2
res.Message = "登录信息获取失败,请重新登录"
s.serv.Reply(ctx, res)
return
}
if err := dao.SysServeLicense.Ctx(ctx).Where("appid = ?", user.Auth.AppId).Scan(&models); err != nil {
res.Code = 3
res.Message = err.Error()
s.serv.Write(client.Conn, res)
s.serv.Reply(ctx, res)
return
}
if models == nil {
res.Code = 3
res.Code = 4
res.Message = "授权信息不存在"
s.serv.Write(client.Conn, res)
s.serv.Reply(ctx, res)
return
}
if models.Status != consts.StatusEnabled {
res.Code = 4
res.Code = 5
res.Message = "授权已禁用,请联系管理员"
s.serv.Write(client.Conn, res)
s.serv.Reply(ctx, res)
return
}
if models.Group != client.Auth.Group {
res.Code = 5
if models.Group != user.Auth.Group {
res.Code = 6
res.Message = "你登录的授权分组未得到授权,请联系管理员"
s.serv.Write(client.Conn, res)
s.serv.Reply(ctx, res)
return
}
if models.EndAt.Before(gtime.Now()) {
res.Code = 6
res.Code = 7
res.Message = "授权已过期,请联系管理员"
s.serv.Write(client.Conn, res)
s.serv.Reply(ctx, res)
return
}
res.Data = new(msgin.AuthSummaryData)
res.Data.EndAt = models.EndAt
res.Data.Online = models.Online
s.serv.Write(client.Conn, res)
s.serv.Reply(ctx, res)
}

View File

@@ -0,0 +1,117 @@
package tcpserver
import (
"context"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/util/gconv"
"hotgo/internal/consts"
"hotgo/internal/model/input/msgin"
)
// CronDelete 删除任务
func (s *sTCPServer) CronDelete(ctx context.Context, in *msgin.CronDelete) (err error) {
clients := s.serv.GetGroupClients(consts.TCPClientGroupCron)
if len(clients) == 0 {
err = gerror.New("没有在线的定时任务服务")
return
}
for _, client := range clients {
res, err := s.serv.RpcRequest(ctx, client, in)
if err != nil {
return err
}
var resp = new(msgin.ResponseCronDelete)
if err = gconv.Scan(res, &resp); err != nil {
return err
}
if err = resp.GetError(); err != nil {
return err
}
}
return
}
// CronEdit 编辑任务
func (s *sTCPServer) CronEdit(ctx context.Context, in *msgin.CronEdit) (err error) {
clients := s.serv.GetGroupClients(consts.TCPClientGroupCron)
if len(clients) == 0 {
err = gerror.New("没有在线的定时任务服务")
return
}
for _, client := range clients {
res, err := s.serv.RpcRequest(ctx, client, in)
if err != nil {
return err
}
var resp = new(msgin.ResponseCronEdit)
if err = gconv.Scan(res, &resp); err != nil {
return err
}
if err = resp.GetError(); err != nil {
return err
}
}
return
}
// CronStatus 修改任务状态
func (s *sTCPServer) CronStatus(ctx context.Context, in *msgin.CronStatus) (err error) {
clients := s.serv.GetGroupClients(consts.TCPClientGroupCron)
if len(clients) == 0 {
err = gerror.New("没有在线的定时任务服务")
return
}
for _, client := range clients {
res, err := s.serv.RpcRequest(ctx, client, in)
if err != nil {
return err
}
var resp = new(msgin.ResponseCronStatus)
if err = gconv.Scan(res, &resp); err != nil {
return err
}
if err = resp.GetError(); err != nil {
return err
}
}
return
}
// CronOnlineExec 执行一次任务
func (s *sTCPServer) CronOnlineExec(ctx context.Context, in *msgin.CronOnlineExec) (err error) {
clients := s.serv.GetGroupClients(consts.TCPClientGroupCron)
if len(clients) == 0 {
err = gerror.New("没有在线的定时任务服务")
return
}
for _, client := range clients {
res, err := s.serv.RpcRequest(ctx, client, in)
if err != nil {
return err
}
var resp = new(msgin.ResponseCronOnlineExec)
if err = gconv.Scan(res, &resp); err != nil {
return err
}
if err = resp.GetError(); err != nil {
return err
}
}
return
}

View File

@@ -27,7 +27,7 @@ func (s *sTCPServer) Start(ctx context.Context) {
server, err := tcp.NewServer(&tcp.ServerConfig{
Name: "hotgo",
Addr: g.Cfg().MustGet(ctx, "tcpServe.address").String(),
Addr: g.Cfg().MustGet(ctx, "tcp.server.address").String(),
})
if err != nil {
@@ -49,7 +49,7 @@ func (s *sTCPServer) Start(ctx context.Context) {
// 授权服务路由
s.serv.RegisterAuthRouter(map[string]tcp.RouterHandler{
"AuthSummary": s.onAuthSummary, // 获取授权信息
"AuthSummary": s.OnAuthSummary, // 获取授权信息
})
// 服务监听