Ma/origin
1
0
Fork 0
mirror of https://github.com/duanhf2012/origin.git synced 2026-02-03 22:45:13 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

优化websocket

Browse Source 
优化蓝图
This commit is contained in:
boyce
2025-12-08 15:30:44 +08:00
parent 75660bdec0
commit 593abd263e
2 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
sysmodule/netmodule/wsmodule/WSModule.go
View File

@@ -119,7 +119,6 @@ func (ws *WSModule) recyclerReaderBytes([]byte) {
func (ws *WSModule) NewWSClient(conn *network.WSConn) network.Agent { func (ws *WSModule) NewWSClient(conn *network.WSConn) network.Agent {
ws.mapClientLocker.Lock() ws.mapClientLocker.Lock()
defer ws.mapClientLocker.Unlock() defer ws.mapClientLocker.Unlock()
pClient := &WSClient{wsConn: conn, id: primitive.NewObjectID().Hex()} pClient := &WSClient{wsConn: conn, id: primitive.NewObjectID().Hex()}
pClient.wsModule = ws pClient.wsModule = ws
ws.mapClient[pClient.id] = pClient ws.mapClient[pClient.id] = pClient
@@ -159,6 +158,22 @@ func (ws *WSModule) GetProcessor() processor.IRawProcessor {
return ws.process return ws.process
} }
func (ws *WSModule) GetClientHeader(clientId string,key string) string {
ws.mapClientLocker.Lock()
defer ws.mapClientLocker.Unlock()
pClient, ok := ws.mapClient[clientId]
if ok == false || pClient.wsConn == nil {
return ""
}
if pClient.wsConn.GetHeader() == nil {
log.Warn("clientId header is nil", log.String("clientId", clientId))
}
return pClient.wsConn.GetHeader().Get(key)
}
func (ws *WSModule) GetClientIp(clientId string) string { func (ws *WSModule) GetClientIp(clientId string) string {
ws.mapClientLocker.Lock() ws.mapClientLocker.Lock()
defer ws.mapClientLocker.Unlock() defer ws.mapClientLocker.Unlock()

4
util/blueprint/graph.go
View File

@@ -144,7 +144,7 @@ func (gc *graphConfig) GetNodeByID(nodeID string) *nodeConfig {
func (gr *Graph) GetAndCreateReturnPort() IPort { func (gr *Graph) GetAndCreateReturnPort() IPort {
p, ok := gr.globalVariables[ReturnVarial] p, ok := gr.globalVariables[ReturnVarial]
if ok { if ok && p != nil {
return p return p
} }
@@ -168,6 +168,8 @@ func (gr *Graph) Do(entranceID int64, args ...any) (Port_Array, error) {
if gr.globalVariables == nil { if gr.globalVariables == nil {
gr.globalVariables = map[string]IPort{} gr.globalVariables = map[string]IPort{}
}else {
gr.globalVariables[ReturnVarial] = nil
} }
err := entranceNode.Do(gr, args...) err := entranceNode.Do(gr, args...)