feat: add source field to ConnectMeta (#31)

* feat: add source field to ConnectMeta * feat: add source field to IConnectMeta --------- Co-authored-by: chenyuxin2 <chenyuxin2@kingsoft.com>
liumingmin · May 29, 2024 · 7c9cacc · 7c9cacc
1 parent 67ba5bb
commit 7c9cacc
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/ws/conn.go b/ws/conn.go
@@ -104,6 +104,7 @@ type ConnectionMeta struct {
 	UserId   string //userId
 	Typed    int    //客户端类型枚举
 	DeviceId string //设备ID
+	Source   string //defines where the connection comes from
 	Version  int    //版本
 	Charset  int    //客户端使用的字符集
 
@@ -112,7 +113,7 @@ type ConnectionMeta struct {
 }
 
 func (m *ConnectionMeta) BuildConnId() string {
-	return fmt.Sprintf("%v-%v-%v", m.UserId, m.Typed, m.DeviceId)
+	return fmt.Sprintf("%v-%v-%v-%v", m.UserId, m.Typed, m.DeviceId, m.Source)
 }
 
 func (c *Connection) Id() string {
@@ -131,6 +132,10 @@ func (c *Connection) DeviceId() string {
 	return c.meta.DeviceId
 }
 
+func (c *Connection) Source() string {
+	return c.meta.Source
+}
+
 func (c *Connection) Version() int {
 	return c.meta.Version
 }

diff --git a/ws/def.go b/ws/def.go
@@ -49,6 +49,7 @@ type IConnection interface {
 	UserId() string
 	Type() int
 	DeviceId() string
+	Source() string
 	Version() int
 	Charset() int
 	ClientIp() string

diff --git a/ws/wss_test.go b/ws/wss_test.go
@@ -151,6 +151,7 @@ func TestWssRequestResponse(t *testing.T) {
 			UserId:   r.URL.Query().Get("uid"),
 			Typed:    0,
 			DeviceId: "",
+			Source:   "",
 			Version:  0,
 			Charset:  0,
 		}
@@ -270,6 +271,7 @@ func TestWssRequestResponseWithTimeout(t *testing.T) {
 			UserId:   r.URL.Query().Get("uid"),
 			Typed:    0,
 			DeviceId: "",
+			Source:   "",
 			Version:  0,
 			Charset:  0,
 		}
@@ -399,6 +401,7 @@ func TestWssSendMessage(t *testing.T) {
 			UserId:   r.URL.Query().Get("uid"),
 			Typed:    gocast.ToInt(r.URL.Query().Get("typed")),
 			DeviceId: r.URL.Query().Get("deviceId"),
+			Source:   r.URL.Query().Get("source"),
 			Version:  gocast.ToInt(r.URL.Query().Get("version")),
 			Charset:  gocast.ToInt(r.URL.Query().Get("charset")),
 		}
@@ -535,6 +538,7 @@ func TestWssDialConnect(t *testing.T) {
 			UserId:   r.URL.Query().Get("uid"),
 			Typed:    0,
 			DeviceId: "",
+			Source:   "",
 			Version:  0,
 			Charset:  0,
 		}
@@ -655,6 +659,7 @@ func TestWssDialRetryConnect(t *testing.T) {
 			UserId:   r.URL.Query().Get("uid"),
 			Typed:    0,
 			DeviceId: "",
+			Source:   "",
 			Version:  0,
 			Charset:  0,
 		}
@@ -699,6 +704,7 @@ func TestAutoReDialConnect(t *testing.T) {
 			UserId:   r.URL.Query().Get("uid"),
 			Typed:    0,
 			DeviceId: "",
+			Source:   "",
 			Version:  0,
 			Charset:  0,
 		}