All/Go/Wine-Server/handlers/manager/worker.go

package manager

import (
	"Wine-Server/utils"
	"Wine-Server/utils/tables"
	"errors"
	"github.com/gorilla/websocket"
)

func keepAlive(conn *websocket.Conn) {
	_ = conn.WriteJSON(utils.WsEvent("pon", nil))
}

func getSecrets(conn *websocket.Conn) {
	_ = conn.WriteJSON(utils.WsEvent("secrets", utils.JsonType{
		"public": utils.ClientPub, "private": utils.ClientPri,
	}))
}

func userLogin(conn *websocket.Conn, data any) (string, error) {
	var param loginParam
	err := utils.AnyTrans(data, &param)
	if err != nil {
		_ = conn.WriteJSON(utils.WsEvent("loginResult", utils.Fail("参数错误")))
		return "", err
	}
	var pass string
	pass, err = utils.Decrypt(param.Password)
	if err != nil {
		utils.Logger.Println(err)
		_ = conn.WriteJSON(utils.WsEvent("loginResult", utils.Fail("解密失败")))
		return "", err
	}
	manager := tables.ManagerTable{Account: param.Account, Password: utils.HashPassword(pass)}
	err = manager.Login()
	if err != nil {
		utils.Logger.Println(err)
		_ = conn.WriteJSON(utils.WsEvent("loginResult", utils.Fail("用户名或密码错误")))
		return "", err
	}
	var token string
	utils.ManagerLock.Lock()
	if _, exist := utils.ManagerWss[manager.Id]; exist {
		utils.ManagerLock.Unlock()
		_ = conn.WriteJSON(utils.WsEvent("loginResult", utils.Fail("账户已在别处登录")))
		return "", errors.New("login other where")
	}
	for {
		token = utils.Format("ManagerToken_%s", utils.RandomString(19, utils.AlphaNumPatten))
		exist, _ := utils.Redis.Exists(utils.WxPayCli, token).Result()
		if exist != 1 {
			utils.Redis.Set(utils.WxPayCli, token, manager.Id, utils.Duration(7*24*60*60))
			break
		}
	}
	utils.ManagerLock.Unlock()
	_ = conn.WriteJSON(utils.WsEvent("loginResult", utils.Success(utils.JsonType{"token": token, "super": manager.Super})))
	return manager.Id, nil
}

func getUserInfo(conn *websocket.Conn, manager *tables.ManagerTable) {
	_ = conn.WriteJSON(utils.WsEvent("userInfo", utils.JsonType{
		"id": manager.Id, "super": manager.Super, "name": manager.Name,
		"order": manager.Order, "income": manager.Income,
	}))
}

func userLogout(conn *websocket.Conn, data any) {
	utils.Redis.Del(utils.WxPayCli, data.(string))
	_ = conn.WriteJSON(utils.WsEvent("logoutResult", nil))
}