mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-10-12 04:53:47 +08:00
This commit is contained in:
@@ -4,71 +4,84 @@ import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"hotgo/internal/library/network/tcp"
|
||||
"hotgo/internal/model/input/msgin"
|
||||
"hotgo/internal/service"
|
||||
"hotgo/utility/simple"
|
||||
)
|
||||
|
||||
// tcp授权
|
||||
type sTCPAuth struct {
|
||||
client *tcp.Client
|
||||
// tcp客户端
|
||||
type sAuthClient struct {
|
||||
client *tcp.Client
|
||||
summary *msgin.AuthSummaryData
|
||||
}
|
||||
|
||||
func init() {
|
||||
service.RegisterTCPAuth(newTCPAuth())
|
||||
service.RegisterAuthClient(newAuthClient())
|
||||
}
|
||||
|
||||
func newTCPAuth() *sTCPAuth {
|
||||
return &sTCPAuth{}
|
||||
func newAuthClient() *sAuthClient {
|
||||
return &sAuthClient{}
|
||||
}
|
||||
|
||||
// Start 启动服务
|
||||
func (s *sTCPAuth) Start(ctx context.Context) {
|
||||
g.Log().Debug(ctx, "TCPAuth start..")
|
||||
func (s *sAuthClient) Start(ctx context.Context) {
|
||||
g.Log().Debug(ctx, "AuthClient start..")
|
||||
|
||||
config, err := service.SysConfig().GetLoadTCP(ctx)
|
||||
if err != nil {
|
||||
g.Log().Errorf(ctx, "AuthClient start fail:%+v", err)
|
||||
return
|
||||
}
|
||||
|
||||
if config.Client == nil || config.Client.Auth == nil {
|
||||
g.Log().Errorf(ctx, "AuthClient config is invalid")
|
||||
return
|
||||
}
|
||||
|
||||
simple.SafeGo(ctx, func(ctx context.Context) {
|
||||
client, err := tcp.NewClient(&tcp.ClientConfig{
|
||||
Addr: "127.0.0.1:8099",
|
||||
s.client, err = tcp.NewClient(&tcp.ClientConfig{
|
||||
Addr: config.Client.Auth.Address,
|
||||
Auth: &tcp.AuthMeta{
|
||||
Group: "auth",
|
||||
Name: "auth1",
|
||||
AppId: "mengshuai",
|
||||
SecretKey: "123456",
|
||||
Group: config.Client.Auth.Group,
|
||||
Name: config.Client.Auth.Name,
|
||||
AppId: config.Client.Auth.AppId,
|
||||
SecretKey: config.Client.Auth.SecretKey,
|
||||
},
|
||||
LoginEvent: s.onLoginEvent,
|
||||
CloseEvent: s.onCloseEvent,
|
||||
})
|
||||
if err != nil {
|
||||
g.Log().Infof(ctx, "TCPAuth NewClient fail:%+v", err)
|
||||
g.Log().Errorf(ctx, "AuthClient NewClient fail:%+v", err)
|
||||
return
|
||||
}
|
||||
|
||||
s.client = client
|
||||
|
||||
err = s.client.RegisterRouter(map[string]tcp.RouterHandler{
|
||||
// ...
|
||||
"ResponseAuthSummary": s.OnResponseAuthSummary,
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
g.Log().Infof(ctx, "TCPAuth RegisterRouter fail:%+v", err)
|
||||
g.Log().Errorf(ctx, "AuthClient RegisterRouter fail:%+v", err)
|
||||
return
|
||||
}
|
||||
|
||||
if err = s.client.Start(); err != nil {
|
||||
g.Log().Infof(ctx, "TCPAuth Start fail:%+v", err)
|
||||
g.Log().Errorf(ctx, "AuthClient Start fail:%+v", err)
|
||||
return
|
||||
}
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
// Stop 停止服务
|
||||
func (s *sTCPAuth) Stop(ctx context.Context) {
|
||||
func (s *sAuthClient) Stop(ctx context.Context) {
|
||||
if s.client != nil {
|
||||
s.client.Stop()
|
||||
g.Log().Debug(ctx, "TCPAuth stop..")
|
||||
g.Log().Debug(ctx, "AuthClient stop..")
|
||||
}
|
||||
}
|
||||
|
||||
// IsLogin 是否已登录认证
|
||||
func (s *sTCPAuth) IsLogin() bool {
|
||||
func (s *sAuthClient) IsLogin() bool {
|
||||
if s.client == nil {
|
||||
return false
|
||||
}
|
||||
@@ -76,11 +89,13 @@ func (s *sTCPAuth) IsLogin() bool {
|
||||
}
|
||||
|
||||
// onLoginEvent 登录认证成功事件
|
||||
func (s *sTCPAuth) onLoginEvent() {
|
||||
// ...
|
||||
func (s *sAuthClient) onLoginEvent() {
|
||||
|
||||
// 获取授权数据
|
||||
s.client.Send(s.client.Ctx, &msgin.AuthSummary{})
|
||||
}
|
||||
|
||||
// onCloseEvent 连接关闭回调事件
|
||||
func (s *sTCPAuth) onCloseEvent() {
|
||||
func (s *sAuthClient) onCloseEvent() {
|
||||
// ...
|
||||
}
|
||||
|
24
server/internal/logic/tcpclient/auth_handle.go
Normal file
24
server/internal/logic/tcpclient/auth_handle.go
Normal file
@@ -0,0 +1,24 @@
|
||||
package tcpclient
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/util/gconv"
|
||||
"hotgo/internal/model/input/msgin"
|
||||
)
|
||||
|
||||
// OnResponseAuthSummary 获取授权信息
|
||||
func (s *sAuthClient) OnResponseAuthSummary(ctx context.Context, args ...interface{}) {
|
||||
var in *msgin.ResponseAuthSummary
|
||||
if err := gconv.Scan(args[0], &in); err != nil {
|
||||
s.client.Logger.Warningf(ctx, "OnResponseAuthSummary Scan err:+v", err)
|
||||
return
|
||||
}
|
||||
|
||||
if err := in.GetError(); err != nil {
|
||||
s.client.Logger.Warningf(ctx, "OnResponseAuthSummary GetError :+v", err)
|
||||
return
|
||||
}
|
||||
|
||||
// 拿到授权的数据,可以是一些动态的功能、路由、权限控制等
|
||||
s.summary = in.Data
|
||||
}
|
1
server/internal/logic/tcpclient/client.go
Normal file
1
server/internal/logic/tcpclient/client.go
Normal file
@@ -0,0 +1 @@
|
||||
package tcpclient
|
102
server/internal/logic/tcpclient/cron.go
Normal file
102
server/internal/logic/tcpclient/cron.go
Normal file
@@ -0,0 +1,102 @@
|
||||
package tcpclient
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/frame/g"
|
||||
"hotgo/internal/library/network/tcp"
|
||||
"hotgo/internal/service"
|
||||
"hotgo/utility/simple"
|
||||
)
|
||||
|
||||
// tcp客户端
|
||||
type sCronClient struct {
|
||||
client *tcp.Client
|
||||
}
|
||||
|
||||
func init() {
|
||||
service.RegisterCronClient(newCronClient())
|
||||
}
|
||||
|
||||
func newCronClient() *sCronClient {
|
||||
return &sCronClient{}
|
||||
}
|
||||
|
||||
// Start 启动服务
|
||||
func (s *sCronClient) Start(ctx context.Context) {
|
||||
g.Log().Debug(ctx, "CronClient start..")
|
||||
|
||||
config, err := service.SysConfig().GetLoadTCP(ctx)
|
||||
if err != nil {
|
||||
g.Log().Errorf(ctx, "CronClient start fail:%+v", err)
|
||||
return
|
||||
}
|
||||
|
||||
if config.Client == nil || config.Client.Cron == nil {
|
||||
g.Log().Errorf(ctx, "CronClient config is invalid")
|
||||
return
|
||||
}
|
||||
|
||||
simple.SafeGo(ctx, func(ctx context.Context) {
|
||||
client, err := tcp.NewClient(&tcp.ClientConfig{
|
||||
Addr: config.Client.Cron.Address,
|
||||
Auth: &tcp.AuthMeta{
|
||||
Group: config.Client.Cron.Group,
|
||||
Name: config.Client.Cron.Name,
|
||||
AppId: config.Client.Cron.AppId,
|
||||
SecretKey: config.Client.Cron.SecretKey,
|
||||
},
|
||||
LoginEvent: s.onLoginEvent,
|
||||
CloseEvent: s.onCloseEvent,
|
||||
})
|
||||
if err != nil {
|
||||
g.Log().Errorf(ctx, "CronClient NewClient fail:%+v", err)
|
||||
return
|
||||
}
|
||||
|
||||
s.client = client
|
||||
|
||||
err = s.client.RegisterRouter(map[string]tcp.RouterHandler{
|
||||
"CronDelete": s.OnCronDelete, // 删除任务
|
||||
"CronEdit": s.OnCronEdit, // 编辑任务
|
||||
"CronStatus": s.OnCronStatus, // 修改任务状态
|
||||
"CronOnlineExec": s.OnCronOnlineExec, // 执行一次任务
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
g.Log().Errorf(ctx, "CronClient RegisterRouter fail:%+v", err)
|
||||
return
|
||||
}
|
||||
|
||||
if err = s.client.Start(); err != nil {
|
||||
g.Log().Errorf(ctx, "CronClient Start fail:%+v", err)
|
||||
return
|
||||
}
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
// Stop 停止服务
|
||||
func (s *sCronClient) Stop(ctx context.Context) {
|
||||
if s.client != nil {
|
||||
s.client.Stop()
|
||||
g.Log().Debug(ctx, "CronClient stop..")
|
||||
}
|
||||
}
|
||||
|
||||
// IsLogin 是否已登录认证
|
||||
func (s *sCronClient) IsLogin() bool {
|
||||
if s.client == nil {
|
||||
return false
|
||||
}
|
||||
return s.client.IsLogin
|
||||
}
|
||||
|
||||
// onLoginEvent 登录认证成功事件
|
||||
func (s *sCronClient) onLoginEvent() {
|
||||
// ...
|
||||
}
|
||||
|
||||
// onCloseEvent 连接关闭回调事件
|
||||
func (s *sCronClient) onCloseEvent() {
|
||||
// ...
|
||||
}
|
96
server/internal/logic/tcpclient/cron_handle.go
Normal file
96
server/internal/logic/tcpclient/cron_handle.go
Normal file
@@ -0,0 +1,96 @@
|
||||
package tcpclient
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/gogf/gf/v2/util/gconv"
|
||||
"hotgo/internal/model/input/msgin"
|
||||
"hotgo/internal/service"
|
||||
)
|
||||
|
||||
// OnCronDelete 删除任务
|
||||
func (s *sCronClient) OnCronDelete(ctx context.Context, args ...interface{}) {
|
||||
var (
|
||||
in *msgin.CronDelete
|
||||
res = new(msgin.ResponseCronDelete)
|
||||
)
|
||||
|
||||
if err := gconv.Scan(args[0], &in); err != nil {
|
||||
res.Code = 1
|
||||
res.Message = err.Error()
|
||||
s.client.Reply(ctx, res)
|
||||
return
|
||||
}
|
||||
|
||||
if err := service.SysCron().Delete(ctx, in.CronDeleteInp); err != nil {
|
||||
res.Code = 2
|
||||
res.Message = err.Error()
|
||||
}
|
||||
|
||||
s.client.Reply(ctx, res)
|
||||
}
|
||||
|
||||
// OnCronEdit 编辑任务
|
||||
func (s *sCronClient) OnCronEdit(ctx context.Context, args ...interface{}) {
|
||||
var (
|
||||
in *msgin.CronEdit
|
||||
res = new(msgin.ResponseCronEdit)
|
||||
)
|
||||
|
||||
if err := gconv.Scan(args[0], &in); err != nil {
|
||||
res.Code = 1
|
||||
res.Message = err.Error()
|
||||
s.client.Reply(ctx, res)
|
||||
return
|
||||
}
|
||||
|
||||
if err := service.SysCron().Edit(ctx, in.CronEditInp); err != nil {
|
||||
res.Code = 2
|
||||
res.Message = err.Error()
|
||||
}
|
||||
|
||||
s.client.Reply(ctx, res)
|
||||
}
|
||||
|
||||
// OnCronStatus 修改任务状态
|
||||
func (s *sCronClient) OnCronStatus(ctx context.Context, args ...interface{}) {
|
||||
var (
|
||||
in *msgin.CronStatus
|
||||
res = new(msgin.ResponseCronStatus)
|
||||
)
|
||||
|
||||
if err := gconv.Scan(args[0], &in); err != nil {
|
||||
res.Code = 1
|
||||
res.Message = err.Error()
|
||||
s.client.Reply(ctx, res)
|
||||
return
|
||||
}
|
||||
|
||||
if err := service.SysCron().Status(ctx, in.CronStatusInp); err != nil {
|
||||
res.Code = 2
|
||||
res.Message = err.Error()
|
||||
}
|
||||
|
||||
s.client.Reply(ctx, res)
|
||||
}
|
||||
|
||||
// OnCronOnlineExec 执行一次任务
|
||||
func (s *sCronClient) OnCronOnlineExec(ctx context.Context, args ...interface{}) {
|
||||
var (
|
||||
in *msgin.CronOnlineExec
|
||||
res = new(msgin.ResponseCronOnlineExec)
|
||||
)
|
||||
|
||||
if err := gconv.Scan(args[0], &in); err != nil {
|
||||
res.Code = 1
|
||||
res.Message = err.Error()
|
||||
s.client.Reply(ctx, res)
|
||||
return
|
||||
}
|
||||
|
||||
if err := service.SysCron().OnlineExec(ctx, in.OnlineExecInp); err != nil {
|
||||
res.Code = 1
|
||||
res.Message = err.Error()
|
||||
}
|
||||
|
||||
s.client.Reply(ctx, res)
|
||||
}
|
Reference in New Issue
Block a user