mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-09 03:50:26 +08:00
优化节点到API节点连接管理
This commit is contained in:
@@ -72,6 +72,7 @@ func (this *NodeTaskDAO) CreateNodeTask(tx *dbs.Tx, role string, clusterId int64
|
|||||||
"isDone": 0,
|
"isDone": 0,
|
||||||
"isOk": 0,
|
"isOk": 0,
|
||||||
"error": "",
|
"error": "",
|
||||||
|
"isNotified": 0,
|
||||||
})
|
})
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -291,6 +292,7 @@ func (this *NodeTaskDAO) FindAllDoingNodeIds(tx *dbs.Tx, role string) ([]int64,
|
|||||||
Attr("role", role).
|
Attr("role", role).
|
||||||
Gt("nodeId", 0).
|
Gt("nodeId", 0).
|
||||||
Attr("isDone", false).
|
Attr("isDone", false).
|
||||||
|
Attr("isNotified", 0).
|
||||||
FindAll()
|
FindAll()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|||||||
@@ -170,6 +170,12 @@ func (this *NSNodeService) NsNodeStream(server pb.NSNodeService_NsNodeStreamServ
|
|||||||
}
|
}
|
||||||
nodeLocker.Unlock()
|
nodeLocker.Unlock()
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
nodeLocker.Lock()
|
||||||
|
delete(requestChanMap, nodeId)
|
||||||
|
nodeLocker.Unlock()
|
||||||
|
}()
|
||||||
|
|
||||||
// 发送请求
|
// 发送请求
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
for {
|
||||||
|
|||||||
@@ -134,6 +134,12 @@ func (this *NodeService) NodeStream(server pb.NodeService_NodeStreamServer) erro
|
|||||||
}
|
}
|
||||||
nodeLocker.Unlock()
|
nodeLocker.Unlock()
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
nodeLocker.Lock()
|
||||||
|
delete(requestChanMap, nodeId)
|
||||||
|
nodeLocker.Unlock()
|
||||||
|
}()
|
||||||
|
|
||||||
// 发送请求
|
// 发送请求
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
for {
|
||||||
|
|||||||
Reference in New Issue
Block a user