昔念
caa5fc37b9
refactor(common/rpc): 修改服务引用从blazing到config模块 将RPC服务中的blservice引用替换为config服务， 统一使用config.NewServerService()进行服务器信息获取。 feat(blazing): 实现新的会话生成机制 - 添加Gensession方法，基于accountID、UUID生成唯一会话标识 - 会话ID由accountID(4字节) + UUID(16字节)组成，编码为十六进制字符串 - 更新登录控制器使用新的会话生成方式 fix(pet_info): 添加宠物信息空值检查 在切换宠物背包仓库时，当宠物信息查询结果为空时， 返回系统错误避免空指针异常。
102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
package middleware
|
|
|
|
import (
|
|
"blazing/cool"
|
|
"blazing/modules/base/config"
|
|
"blazing/modules/config/service"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/net/ghttp"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
"github.com/golang-jwt/jwt/v4"
|
|
"github.com/lxzan/gws"
|
|
)
|
|
|
|
func MiddlewareCORS(r *ghttp.Request) {
|
|
r.Response.CORSDefault()
|
|
corsOptions := r.Response.DefaultCORSOptions()
|
|
corsOptions.AllowDomain = []string{"*", "localhost", "tauri.localhost"}
|
|
if !r.Response.CORSAllowedOrigin(corsOptions) {
|
|
r.Response.WriteStatus(http.StatusForbidden)
|
|
return
|
|
}
|
|
r.Response.CORS(corsOptions)
|
|
if r.Response.Request.Method == "OPTIONS" {
|
|
r.Response.WriteStatus(http.StatusOK)
|
|
return
|
|
}
|
|
// fmt.Println(r.Response.Header())
|
|
//g.Dump(r.Response.Server.SetConfig(gtt))
|
|
//r.Response.Header().Del("Server") // 删除Server头
|
|
// r.Response.Header().Set("Server", "blazing")
|
|
r.Middleware.Next()
|
|
}
|
|
func init() {
|
|
if config.Config.Middleware.Authority.Enable {
|
|
g.Server().BindMiddleware("/admin/*/open/*", BaseAuthorityMiddlewareOpen)
|
|
g.Server().BindMiddleware("/admin/*/comm/*", BaseAuthorityMiddlewareComm)
|
|
g.Server().BindMiddleware("/admin/*", BaseAuthorityMiddleware)
|
|
// g.Server().BindMiddleware("/*", AutoI18n)
|
|
g.Server().BindMiddleware("/*", MiddlewareCORS)
|
|
|
|
}
|
|
if config.Config.Middleware.Log.Enable {
|
|
g.Server().BindMiddleware("/admin/*", BaseLog)
|
|
}
|
|
|
|
g.Server().BindHandler("/server/*", func(r *ghttp.Request) {
|
|
|
|
tt := new(ServerHandler)
|
|
id := gconv.Uint16(r.URL.Query().Get("id"))
|
|
tt.ServerList = service.NewServerService().GetServerID(id)
|
|
tt.isinstall = gconv.Uint32(r.URL.Query().Get("isinstall"))
|
|
|
|
upgrader := gws.NewUpgrader(tt, &gws.ServerOption{
|
|
|
|
Authorize: func(rt *http.Request, session gws.SessionStorage) bool {
|
|
|
|
tokenString := r.URL.Query().Get("Authorization")
|
|
token, err := jwt.ParseWithClaims(tokenString, &cool.Claims{}, func(token *jwt.Token) (interface{}, error) {
|
|
|
|
return []byte(config.Config.Jwt.Secret), nil
|
|
})
|
|
if err != nil {
|
|
return false
|
|
}
|
|
if !token.Valid {
|
|
return false
|
|
}
|
|
admin := token.Claims.(*cool.Claims)
|
|
if admin.UserId != 10001 {
|
|
return false
|
|
}
|
|
|
|
// var name = r.URL.Query().Get("name")
|
|
// if name == "" {
|
|
// return false
|
|
// }
|
|
// t, _ := service.NewBaseSysUserService().Person(admin.UserID)
|
|
|
|
//Loger.Debug(context.TODO(), t.Mimi)
|
|
// session.Store("name", t.Mimi)
|
|
//session.Store("key", r.Header.Get("Sec-WebSocket-Key"))
|
|
return true
|
|
},
|
|
})
|
|
|
|
socket, err := upgrader.Upgrade(r.Response.Writer, r.Request)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
// ants.Submit(func() {
|
|
// socket.ReadLoop()
|
|
// })
|
|
// ants.Submit(func() { socket.ReadLoop() })
|
|
go socket.ReadLoop()
|
|
|
|
})
|
|
}
|