This commit is contained in:
laodaming 2023-10-31 17:53:49 +08:00
parent 21f06c6e39
commit 8434bd1d95

View File

@ -94,12 +94,10 @@ func ConsumeWebsocketStatData(ctx context.Context) {
case TYPE_CONNECT_COUNT: //ws连接计数
statData, ok := mapUserWsStat.Load(key)
if !ok {
if data.Value <= 0 {
continue
}
mapUserWsStat.Store(key, mapUserWsStatItem{
CurWsConnectCount: data.Value,
})
continue
}
stat, ok := statData.(mapUserWsStatItem)
if !ok {
@ -107,22 +105,12 @@ func ConsumeWebsocketStatData(ctx context.Context) {
continue
}
stat.CurWsConnectCount += data.Value
if stat.CurWsConnectCount <= 0 { //小于等于0则移除
mapUserWsStat.Delete(key)
} else {
mapUserWsStat.Store(key, stat)
}
mapUserWsStat.Store(key, stat)
case TYPE_COMBINE_IMAGE_REQUEST_COUNT: //请求算法合图计数
statData, ok := mapUserWsStat.Load(key)
//不存在
if !ok {
//数据是减少的
if data.Value <= 0 {
continue
}
mapUserWsStat.Store(key, mapUserWsStatItem{
CurCombineCount: data.Value,
})
continue
}
//存在
if stat, ok := statData.(mapUserWsStatItem); ok {