This commit is contained in:
孟帅
2023-03-13 17:00:46 +08:00
parent ab912d0ba6
commit 1acc6d17c4
51 changed files with 2777 additions and 1024 deletions

View File

@@ -0,0 +1,73 @@
package tcpserver
import (
"github.com/gogf/gf/v2/os/gtime"
"github.com/gogf/gf/v2/util/gconv"
"hotgo/internal/consts"
"hotgo/internal/dao"
"hotgo/internal/library/network/tcp"
"hotgo/internal/model/entity"
"hotgo/internal/model/input/msgin"
)
// onAuthSummary 获取授权信息
func (s *sTCPServer) onAuthSummary(args ...interface{}) {
var (
in *msgin.AuthSummary
client = args[1].(*tcp.ClientConn)
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)
return
}
if err := dao.SysServeLicense.Ctx(s.serv.Ctx).Where("appid = ?", client.Auth.AppId).Scan(&models); err != nil {
res.Code = 2
res.Message = err.Error()
s.serv.Write(client.Conn, res)
return
}
if models == nil {
res.Code = 3
res.Message = "授权信息不存在"
s.serv.Write(client.Conn, res)
return
}
if models.Status != consts.StatusEnabled {
res.Code = 4
res.Message = "授权已禁用,请联系管理员"
s.serv.Write(client.Conn, res)
return
}
if models.Group != client.Auth.Group {
res.Code = 5
res.Message = "你登录的授权分组未得到授权,请联系管理员"
s.serv.Write(client.Conn, res)
return
}
if models.EndAt.Before(gtime.Now()) {
res.Code = 6
res.Message = "授权已过期,请联系管理员"
s.serv.Write(client.Conn, res)
return
}
res.Data = new(msgin.AuthSummaryData)
res.Data.EndAt = models.EndAt
res.Data.Online = models.Online
s.serv.Write(client.Conn, res)
}

View File

@@ -0,0 +1,68 @@
package tcpserver
import (
"context"
"github.com/gogf/gf/v2/frame/g"
"hotgo/internal/library/network/tcp"
"hotgo/internal/service"
"hotgo/utility/simple"
)
type sTCPServer struct {
serv *tcp.Server
}
func init() {
service.RegisterTCPServer(newTCPServer())
}
func newTCPServer() *sTCPServer {
return &sTCPServer{}
}
// Start 启动服务
func (s *sTCPServer) Start(ctx context.Context) {
simple.SafeGo(ctx, func(ctx context.Context) {
g.Log().Debug(ctx, "TCPServer start..")
server, err := tcp.NewServer(&tcp.ServerConfig{
Name: "hotgo",
Addr: g.Cfg().MustGet(ctx, "tcpServe.address").String(),
})
if err != nil {
g.Log().Warningf(ctx, "TCPServer start fail%+v", err)
return
}
s.serv = server
// 消息队列路由
s.serv.RegisterQueueRouter(map[string]tcp.RouterHandler{
// ...
})
// 定时任务路由
s.serv.RegisterCronRouter(map[string]tcp.RouterHandler{
// ...
})
// 授权服务路由
s.serv.RegisterAuthRouter(map[string]tcp.RouterHandler{
"AuthSummary": s.onAuthSummary, // 获取授权信息
})
// 服务监听
if err := s.serv.Listen(); err != nil {
g.Log().Warningf(ctx, "TCPServer Listen err:%v", err)
}
})
}
// Stop 关闭服务
func (s *sTCPServer) Stop(ctx context.Context) {
if s.serv != nil {
s.serv.Close()
g.Log().Debug(ctx, "TCPServer stop..")
}
}