123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218 |
- package api
- import (
- "encoding/json"
- "errors"
- "time"
- "github.com/gorilla/websocket"
- jsoniter "github.com/json-iterator/go"
- "github.com/sirupsen/logrus"
- "github.com/spf13/viper"
- "gogs.daxia.dev/huanan/pkg.daxia.dev/event2"
- "gogs.daxia.dev/huanan/pkg.daxia.dev/webutils"
- "nn.daxia.dev/model"
- "nn.daxia.dev/online"
- )
- type Resp struct {
- Once string `json:"once"`
- Code int64 `json:"code"`
- Msg string `json:"msg"`
- Data interface{} `json:"data"`
- }
- type ReqData struct {
- Version string `json:"version"`
- Name string `json:"name"`
- Body string `json:"body"`
- Token string `json:"token"`
- Once string `json:"once"`
- }
- type PerformReq struct {
- UserModel model.User
- UniqueIndex int64
- Once string
- Name string
- Msg []byte
- Token string
- }
- type ConnectInfo struct {
- Bot uint `json:"bot"`
- }
- const CodeSuccess = 200
- const CodeFailed = -1
- func Perform(conn *websocket.Conn, message []byte, ctx map[string]interface{}) error {
- var err error
- reqData := ReqData{}
- err = json.Unmarshal(message, &reqData)
- if err != nil {
- logrus.Error(err)
- return err
- }
- userModel := ctx["userModel"].(model.User)
- performReq := PerformReq{
- Once: reqData.Once,
- UserModel: userModel,
- Name: reqData.Name,
- Msg: []byte(reqData.Body),
- Token: reqData.Token,
- UniqueIndex: ctx["unique_index"].(int64),
- }
- performReqData, err := json.Marshal(performReq)
- if err != nil {
- logrus.Error(err)
- return err
- }
- reqUrl := viper.GetString("bs.url") + "/api/bs/perform"
- respData, err := webutils.PostResp[interface{}](reqUrl, string(performReqData), time.Second*3)
- if err != nil {
- logrus.Error(err)
- return err
- }
- if respData.Code != webutils.CodeSuccess {
- logrus.Error(respData)
- return errors.New(respData.Msg)
- }
- return nil
- }
- func GamePingMessage(conn *websocket.Conn, message []byte, ctx map[string]interface{}) error {
- if ctx == nil {
- logrus.Error("ctx is nil")
- return nil
- }
- userModel := ctx["userModel"].(model.User)
- event2.MustFire(model.EventPing, online.EventPing{
- UserModel: &userModel,
- Conn: conn,
- })
- performReq := PerformReq{
- UserModel: userModel,
- Name: "ping",
- Msg: message,
- UniqueIndex: ctx["unique_index"].(int64),
- }
- performReqData, err := json.Marshal(performReq)
- if err != nil {
- logrus.Error(err)
- return err
- }
- reqUrl := viper.GetString("bs.url") + "/api/bs/perform"
- respData, err := webutils.PostResp[interface{}](reqUrl, string(performReqData), time.Second*3)
- if err != nil {
- logrus.Error(err)
- return err
- }
- if respData.Code != webutils.CodeSuccess {
- logrus.Error(respData)
- return errors.New(respData.Msg)
- }
- return nil
- }
- func GameConnect(conn *websocket.Conn, message []byte, ctx map[string]interface{}) error {
- logrus.Debugf("start GameConnect")
- connectInfo := ConnectInfo{}
- err := jsoniter.Unmarshal(message, &connectInfo)
- if err != nil {
- logrus.Errorf("Unmarshal err:%v", err)
- return err
- }
- userModel := ctx["userModel"].(model.User)
- event2.MustFire(model.EventConnect, online.EventConnect{
- UserModel: &userModel,
- Conn: conn,
- })
- logrus.Debug("MustFire EventConnect Finish")
- performReq := PerformReq{
- UserModel: userModel,
- Name: "connect",
- Msg: message,
- UniqueIndex: ctx["unique_index"].(int64),
- }
- performReqData, err := json.Marshal(performReq)
- if err != nil {
- logrus.Error(err)
- return err
- }
- reqUrl := viper.GetString("bs.url") + "/api/bs/perform"
- respData, err := webutils.PostResp[interface{}](reqUrl, string(performReqData), time.Second*3)
- if err != nil {
- logrus.Error(err)
- return err
- }
- logrus.Debugf("req %s Finish, resp:%v", reqUrl, respData)
- if respData.Code != webutils.CodeSuccess {
- logrus.Error(respData)
- return errors.New(respData.Msg)
- }
- return nil
- }
- func GameDisconnect(conn *websocket.Conn, message []byte, ctx map[string]interface{}) error {
- if ctx == nil {
- return nil
- }
- userModel := ctx["userModel"].(model.User)
- event2.MustFire(model.EventDisconnect, online.EventDisconnect{
- UserModel: &userModel,
- Conn: conn,
- })
- performReq := PerformReq{
- UserModel: userModel,
- Name: "disconnect",
- Msg: message,
- UniqueIndex: ctx["unique_index"].(int64),
- }
- performReqData, err := json.Marshal(performReq)
- if err != nil {
- logrus.Error(err)
- return err
- }
- reqUrl := viper.GetString("bs.url") + "/api/bs/perform"
- respData, err := webutils.PostResp[interface{}](reqUrl, string(performReqData), time.Second*3)
- if err != nil {
- logrus.Error(err)
- return err
- }
- if respData.Code != webutils.CodeSuccess {
- logrus.Error(respData)
- return errors.New(respData.Msg)
- }
- return nil
- }
|