MiFaZhan
126 lines
2.7 KiB
Go
126 lines
2.7 KiB
Go
// Package websocket
|
|
// @Link https://github.com/xiujiecn/xiu-admin
|
|
// @Copyright Copyright (c) 2025 LiXiujie
|
|
// @Author Lxj <li@example.com>
|
|
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
|
package websocket
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/gogf/gf/v2/os/gtime"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
)
|
|
|
|
// LoginController 用户登录
|
|
func LoginController(client *Client, req *request) {
|
|
l := &WsLoginDataReq{}
|
|
gconv.Struct(req.Data, &l)
|
|
userId := gconv.Uint64(l.UserID)
|
|
client.UserId = userId
|
|
// 用户登录
|
|
login := &login{
|
|
UserId: userId,
|
|
Client: client,
|
|
}
|
|
clientManager.Login <- login
|
|
|
|
client.SendMsg(&WResponse{
|
|
Event: Login,
|
|
Data: "success",
|
|
})
|
|
}
|
|
|
|
func IsAppController(client *Client) {
|
|
client.isApp = true
|
|
}
|
|
|
|
// JoinController 加入
|
|
func JoinController(client *Client, req *request) {
|
|
name := gconv.String(req.Data["name"])
|
|
|
|
if !client.tags.Contains(name) {
|
|
client.tags.Append(name)
|
|
for k, v := range clientManager.TagCallbackMap {
|
|
if strings.HasPrefix(name, k) {
|
|
if v != nil {
|
|
v(client, name)
|
|
}
|
|
}
|
|
}
|
|
AddTagCount(name)
|
|
}
|
|
client.SendMsg(&WResponse{
|
|
Event: Join,
|
|
Data: client.tags.Slice(),
|
|
})
|
|
}
|
|
|
|
// QuitController 退出
|
|
func QuitController(client *Client, req *request) {
|
|
name := gconv.String(req.Data["name"])
|
|
if client.tags.Contains(name) {
|
|
client.tags.RemoveValue(name)
|
|
DelTagCount(name)
|
|
}
|
|
client.SendMsg(&WResponse{
|
|
Event: Quit,
|
|
Data: client.tags.Slice(),
|
|
})
|
|
}
|
|
|
|
func PingController(client *Client) {
|
|
currentTime := uint64(gtime.Now().Unix())
|
|
client.Heartbeat(currentTime)
|
|
}
|
|
|
|
// JoinsController 加入
|
|
func JoinsController(client *Client, req *request) {
|
|
if req == nil || req.Data == nil {
|
|
fmt.Println("websocket.JoinsController req is nil or req.Data is nil")
|
|
return
|
|
}
|
|
// fmt.Println("websocket.JoinsController req.Data", req.Data)
|
|
names, ok := req.Data["names"]
|
|
if !ok || names == nil {
|
|
fmt.Println("websocket.JoinsController names is nil or names is empty")
|
|
return
|
|
}
|
|
// fmt.Println("websocket.JoinsController names", names)
|
|
for _, name := range gconv.Strings(names) {
|
|
if name == "" {
|
|
continue
|
|
}
|
|
if !client.tags.Contains(name) {
|
|
client.tags.Append(name)
|
|
for k, v := range clientManager.TagCallbackMap {
|
|
if strings.HasPrefix(name, k) {
|
|
if v != nil {
|
|
v(client, name)
|
|
}
|
|
}
|
|
}
|
|
AddTagCount(name)
|
|
}
|
|
}
|
|
client.SendMsg(&WResponse{
|
|
Event: Joins,
|
|
Data: client.tags.Slice(),
|
|
})
|
|
}
|
|
|
|
// QuitController 退出
|
|
func QuitsController(client *Client, req *request) {
|
|
for _, name := range gconv.Strings(req.Data["names"]) {
|
|
if client.tags.Contains(name) {
|
|
client.tags.RemoveValue(name)
|
|
DelTagCount(name)
|
|
}
|
|
}
|
|
client.SendMsg(&WResponse{
|
|
Event: Quits,
|
|
Data: client.tags.Slice(),
|
|
})
|
|
}
|