This commit is contained in:
孟帅
2023-07-20 18:01:10 +08:00
parent 9113fc5297
commit 373d9627fb
492 changed files with 12170 additions and 6982 deletions

View File

@@ -1,76 +1,80 @@
// Package tcpserver
// @Link https://github.com/bufanyun/hotgo
// @Copyright Copyright (c) 2023 HotGo CLI
// @Author Ms <133814250@qq.com>
// @License https://github.com/bufanyun/hotgo/blob/master/LICENSE
package tcpserver
import (
"context"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/os/gtime"
"github.com/gogf/gf/v2/util/gconv"
"hotgo/api/servmsg"
"hotgo/internal/consts"
"hotgo/internal/dao"
"hotgo/internal/library/network/tcp"
"hotgo/internal/model/entity"
"hotgo/internal/model/input/msgin"
"hotgo/internal/model/input/servmsgin"
"hotgo/internal/service"
)
// OnAuthSummary 获取授权信息
func (s *sTCPServer) OnAuthSummary(ctx context.Context, args ...interface{}) {
func (s *sTCPServer) OnAuthSummary(ctx context.Context, req *servmsg.AuthSummaryReq) {
var (
in *msgin.AuthSummary
user = tcp.GetCtx(ctx)
res = new(msgin.ResponseAuthSummary)
conn = tcp.ConnFromCtx(ctx)
models *entity.SysServeLicense
res = new(servmsg.AuthSummaryRes)
)
if err := gconv.Scan(args, &in); err != nil {
res.Code = 1
res.Message = err.Error()
_ = s.serv.Reply(ctx, res)
if conn == nil {
g.Log().Warningf(ctx, "conn is nil.")
return
}
if user.Auth == nil {
res.Code = 2
res.Message = "登录信息获取失败,请重新登录"
_ = s.serv.Reply(ctx, res)
if conn.Auth == nil {
res.SetError(gerror.New("登录信息获取失败,请重新登录"))
conn.Send(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.Reply(ctx, res)
if err := dao.SysServeLicense.Ctx(ctx).Where("appid = ?", conn.Auth.AppId).Scan(&models); err != nil {
res.SetError(err)
conn.Send(ctx, res)
return
}
if models == nil {
res.Code = 4
res.Message = "授权信息不存在"
_ = s.serv.Reply(ctx, res)
res.SetError(gerror.New("授权信息不存在"))
conn.Send(ctx, res)
return
}
if models.Status != consts.StatusEnabled {
res.Code = 5
res.Message = "授权已禁用,请联系管理员"
_ = s.serv.Reply(ctx, res)
res.SetError(gerror.New("授权已禁用,请联系管理员"))
conn.Send(ctx, res)
return
}
if models.Group != user.Auth.Group {
res.Code = 6
res.Message = "你登录的授权分组未得到授权,请联系管理员"
_ = s.serv.Reply(ctx, res)
if models.Group != conn.Auth.Group {
res.SetError(gerror.New("你登录的授权分组未得到授权,请联系管理员"))
conn.Send(ctx, res)
return
}
if models.EndAt.Before(gtime.Now()) {
res.Code = 7
res.Message = "授权已过期,请联系管理员"
_ = s.serv.Reply(ctx, res)
res.SetError(gerror.New("授权已过期,请联系管理员"))
conn.Send(ctx, res)
return
}
res.Data = new(msgin.AuthSummaryData)
res.Data.EndAt = models.EndAt
res.Data.Online = models.Online
_ = s.serv.Reply(ctx, res)
data := new(servmsgin.AuthSummaryModel)
data.EndAt = models.EndAt
data.Online = service.TCPServer().Instance().GetAppIdOnline(models.Appid)
// 请填充你的授权数据
// ...
res.Data = data
conn.Send(ctx, res)
}