mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-02-03 02:54:41 +08:00
81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
// 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"
|
|
"hotgo/api/servmsg"
|
|
"hotgo/internal/consts"
|
|
"hotgo/internal/dao"
|
|
"hotgo/internal/library/network/tcp"
|
|
"hotgo/internal/model/entity"
|
|
"hotgo/internal/model/input/servmsgin"
|
|
"hotgo/internal/service"
|
|
)
|
|
|
|
// OnAuthSummary 获取授权信息
|
|
func (s *sTCPServer) OnAuthSummary(ctx context.Context, req *servmsg.AuthSummaryReq) {
|
|
var (
|
|
conn = tcp.ConnFromCtx(ctx)
|
|
models *entity.SysServeLicense
|
|
res = new(servmsg.AuthSummaryRes)
|
|
)
|
|
|
|
if conn == nil {
|
|
g.Log().Warningf(ctx, "conn is nil.")
|
|
return
|
|
}
|
|
|
|
if conn.Auth == nil {
|
|
res.SetError(gerror.New("登录信息获取失败，请重新登录"))
|
|
_ = conn.Send(ctx, res)
|
|
return
|
|
}
|
|
|
|
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.SetError(gerror.New("授权信息不存在"))
|
|
_ = conn.Send(ctx, res)
|
|
return
|
|
}
|
|
|
|
if models.Status != consts.StatusEnabled {
|
|
res.SetError(gerror.New("授权已禁用，请联系管理员"))
|
|
_ = conn.Send(ctx, res)
|
|
return
|
|
}
|
|
|
|
if models.Group != conn.Auth.Group {
|
|
res.SetError(gerror.New("你登录的授权分组未得到授权，请联系管理员"))
|
|
_ = conn.Send(ctx, res)
|
|
return
|
|
}
|
|
|
|
if models.EndAt.Before(gtime.Now()) {
|
|
res.SetError(gerror.New("授权已过期，请联系管理员"))
|
|
_ = conn.Send(ctx, res)
|
|
return
|
|
}
|
|
|
|
data := new(servmsgin.AuthSummaryModel)
|
|
data.EndAt = models.EndAt
|
|
data.Online = service.TCPServer().Instance().GetAppIdOnline(models.Appid)
|
|
|
|
// 请填充你的授权数据
|
|
// ...
|
|
|
|
res.Data = data
|
|
_ = conn.Send(ctx, res)
|
|
}
|