mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-11-10 11:13:44 +08:00
发布v2.8.4版本,更新内容请查看:https://github.com/bufanyun/hotgo/tree/v2.0/docs/guide-zh-CN/addon-version-upgrade.md
This commit is contained in:
@@ -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)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user