blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-12-24 08:07:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: replace LongConn with ClientConn interface and simplify message handling

Browse Source 
(cherry picked from commit 9da7db2ac25e93a937a3f4dc6d80771c248f60be)
This commit is contained in:
withchao 2025-12-19 15:53:17 +08:00
parent e23d901c1f
commit ee10c3d35a
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/msggateway/client.go
View File

@ -68,6 +68,7 @@ type Client struct {
UserID string `json:"userID"` UserID string `json:"userID"`
IsBackground bool `json:"isBackground"` IsBackground bool `json:"isBackground"`
SDKType string `json:"sdkType"` SDKType string `json:"sdkType"`
SDKVersion string `json:"sdkVersion"`
Encoder Encoder Encoder Encoder
ctx *UserConnContext ctx *UserConnContext
longConnServer LongConnServer longConnServer LongConnServer
@ -95,6 +96,7 @@ func (c *Client) ResetClient(ctx *UserConnContext, conn ClientConn, longConnServ
c.closedErr = nil c.closedErr = nil
c.token = ctx.GetToken() c.token = ctx.GetToken()
c.SDKType = ctx.GetSDKType() c.SDKType = ctx.GetSDKType()
c.SDKVersion = ctx.GetSDKVersion()
c.hbCtx, c.hbCancel = context.WithCancel(c.ctx) c.hbCtx, c.hbCancel = context.WithCancel(c.ctx)
c.subLock = new(sync.Mutex) c.subLock = new(sync.Mutex)
if c.subUserIDs != nil { if c.subUserIDs != nil {

11
internal/msggateway/context.go
View File

@ -38,6 +38,7 @@ type UserConnContextInfo struct {
SDKType string `json:"sdkType"` SDKType string `json:"sdkType"`
SendResponse bool `json:"sendResponse"` SendResponse bool `json:"sendResponse"`
Background bool `json:"background"` Background bool `json:"background"`
SDKVersion string `json:"sdkVersion"`
} }
type UserConnContext struct { type UserConnContext struct {
@ -74,6 +75,8 @@ func (c *UserConnContext) Value(key any) any {
return c.GetPlatformID() return c.GetPlatformID()
case constant.RemoteAddr: case constant.RemoteAddr:
return c.RemoteAddr return c.RemoteAddr
case SDKVersion:
return c.info.SDKVersion
default: default:
return "" return ""
} }
@ -117,6 +120,7 @@ func (c *UserConnContext) parseByQuery(query url.Values, header http.Header) err
OperationID: query.Get(OperationID), OperationID: query.Get(OperationID),
Compression: query.Get(Compression), Compression: query.Get(Compression),
SDKType: query.Get(SDKType), SDKType: query.Get(SDKType),
SDKVersion: query.Get(SDKVersion),
} }
platformID, err := strconv.Atoi(query.Get(PlatformID)) platformID, err := strconv.Atoi(query.Get(PlatformID))
if err != nil { if err != nil {
@ -246,6 +250,13 @@ func (c *UserConnContext) GetSDKType() string {
} }
} }
func (c *UserConnContext) GetSDKVersion() string {
if c == nil || c.info == nil {
return ""
}
return c.info.SDKVersion
}
func (c *UserConnContext) ShouldSendResp() bool { func (c *UserConnContext) ShouldSendResp() bool {
return c != nil && c.info != nil && c.info.SendResponse return c != nil && c.info != nil && c.info.SendResponse
} }