fusen/fusenapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
laodaming 2023-11-01 12:18:49 +08:00
parent b12a61b2f6
commit a6a01dae3d
1 changed files with 22 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
server/websocket/internal/logic/ws_statistics.go
View File

@ -16,12 +16,11 @@ const (
TYPE_CUR_UNITY_HANDLE_COUNT websocketStatType = "TYPE_CUR_UNITY_HANDLE_COUNT" //unity处理数 TYPE_CUR_UNITY_HANDLE_COUNT websocketStatType = "TYPE_CUR_UNITY_HANDLE_COUNT" //unity处理数
) )
type websocketStatItem struct { type websocketStatData struct {
UserId int64 `json:"user_id"` UserId int64 `json:"user_id"` //用户id
GuestId int64 `json:"guest_id"` GuestId int64 `json:"guest_id"` //游客id
Type websocketStatType `json:"type"` //类型 Type websocketStatType `json:"type"` //类型
Value int `json:"value"` //数值 Num int `json:"num"` //数值
UnityRenderLogId int64 `json:"unity_render_log_id"` //unity渲染的日志id
} }
type mapUserWsStatItem struct { type mapUserWsStatItem struct {
CurCombineCount int `json:"cur_combine_count"` //当前合图数 CurCombineCount int `json:"cur_combine_count"` //当前合图数
@ -34,16 +33,16 @@ var (
//用户连接统计 //用户连接统计
mapUserWsStat = sync.Map{} mapUserWsStat = sync.Map{}
//添加or减少连接的控制chan //添加or减少连接的控制chan
websocketStat = make(chan websocketStatItem, 1000) websocketStat = make(chan websocketStatData, 1000)
) )
// 累增ws连接数计数 // 累增ws连接数计数
func increaseWebsocketConnectCount(userId, guestId int64) { func increaseWebsocketConnectCount(userId, guestId int64) {
data := websocketStatItem{ data := websocketStatData{
UserId: userId, UserId: userId,
GuestId: guestId, GuestId: guestId,
Type: TYPE_CUR_CONNECT_COUNT, Type: TYPE_CUR_CONNECT_COUNT,
Value: 1, Num: 1,
} }
select { select {
case websocketStat <- data: case websocketStat <- data:
@ -56,11 +55,11 @@ func increaseWebsocketConnectCount(userId, guestId int64) {
// 减少ws连接数计数 // 减少ws连接数计数
func decreaseWebsocketConnectCount(userId, guestId int64) { func decreaseWebsocketConnectCount(userId, guestId int64) {
data := websocketStatItem{ data := websocketStatData{
UserId: userId, UserId: userId,
GuestId: guestId, GuestId: guestId,
Type: TYPE_CUR_CONNECT_COUNT, Type: TYPE_CUR_CONNECT_COUNT,
Value: -1, Num: -1,
} }
select { select {
case websocketStat <- data: case websocketStat <- data:
@ -73,11 +72,11 @@ func decreaseWebsocketConnectCount(userId, guestId int64) {
// 累增合图请求数计数 // 累增合图请求数计数
func increaseCombineRequestCount(userId, guestId int64) { func increaseCombineRequestCount(userId, guestId int64) {
data := websocketStatItem{ data := websocketStatData{
UserId: userId, UserId: userId,
GuestId: guestId, GuestId: guestId,
Type: TYPE_CUR_COMBINE_IMAGE_COUNT, Type: TYPE_CUR_COMBINE_IMAGE_COUNT,
Value: 1, Num: 1,
} }
select { select {
case websocketStat <- data: case websocketStat <- data:
@ -90,11 +89,11 @@ func increaseCombineRequestCount(userId, guestId int64) {
// 减少合图请求数计数 // 减少合图请求数计数
func decreaseCombineRequestCount(userId, guestId int64) { func decreaseCombineRequestCount(userId, guestId int64) {
data := websocketStatItem{ data := websocketStatData{
UserId: userId, UserId: userId,
GuestId: guestId, GuestId: guestId,
Type: TYPE_CUR_COMBINE_IMAGE_COUNT, Type: TYPE_CUR_COMBINE_IMAGE_COUNT,
Value: -1, Num: -1,
} }
select { select {
case websocketStat <- data: case websocketStat <- data:
@ -107,11 +106,11 @@ func decreaseCombineRequestCount(userId, guestId int64) {
// 累增unity请求数计数 // 累增unity请求数计数
func increaseUnityRequestCount(userId, guestId int64) { func increaseUnityRequestCount(userId, guestId int64) {
data := websocketStatItem{ data := websocketStatData{
UserId: userId, UserId: userId,
GuestId: guestId, GuestId: guestId,
Type: TYPE_CUR_UNITY_HANDLE_COUNT, Type: TYPE_CUR_UNITY_HANDLE_COUNT,
Value: 1, Num: 1,
} }
select { select {
case websocketStat <- data: case websocketStat <- data:
@ -124,11 +123,11 @@ func increaseUnityRequestCount(userId, guestId int64) {
// 减少unity请求数计数 // 减少unity请求数计数
func decreaseUnityRequestCount(userId, guestId int64) { func decreaseUnityRequestCount(userId, guestId int64) {
data := websocketStatItem{ data := websocketStatData{
UserId: userId, UserId: userId,
GuestId: guestId, GuestId: guestId,
Type: TYPE_CUR_UNITY_HANDLE_COUNT, Type: TYPE_CUR_UNITY_HANDLE_COUNT,
Value: -1, Num: -1,
} }
select { select {
case websocketStat <- data: case websocketStat <- data:
@ -161,7 +160,7 @@ func ConsumeWebsocketStatData(ctx context.Context) {
case TYPE_CUR_CONNECT_COUNT: //ws连接计数 case TYPE_CUR_CONNECT_COUNT: //ws连接计数
if !ok { if !ok {
mapUserWsStat.Store(key, mapUserWsStatItem{ mapUserWsStat.Store(key, mapUserWsStatItem{
CurWsConnectCount: data.Value, CurWsConnectCount: data.Num,
}) })
continue continue
} }
@ -170,7 +169,7 @@ func ConsumeWebsocketStatData(ctx context.Context) {
logx.Error("断言mapUserWsStatItem错误") logx.Error("断言mapUserWsStatItem错误")
continue continue
} }
stat.CurWsConnectCount += data.Value stat.CurWsConnectCount += data.Num
//没有连接就删除 //没有连接就删除
if stat.CurWsConnectCount <= 0 { if stat.CurWsConnectCount <= 0 {
mapUserWsStat.Delete(key) mapUserWsStat.Delete(key)
@ -189,7 +188,7 @@ func ConsumeWebsocketStatData(ctx context.Context) {
logx.Error("断言mapUserWsStatItem错误") logx.Error("断言mapUserWsStatItem错误")
continue continue
} }
stat.CurCombineCount += data.Value stat.CurCombineCount += data.Num
if stat.CurCombineCount < 0 { if stat.CurCombineCount < 0 {
stat.CurCombineCount = 0 stat.CurCombineCount = 0
} }
@ -205,7 +204,7 @@ func ConsumeWebsocketStatData(ctx context.Context) {
logx.Error("断言mapUserWsStatItem错误") logx.Error("断言mapUserWsStatItem错误")
continue continue
} }
stat.CurUnityHandleCount += data.Value stat.CurUnityHandleCount += data.Num
if stat.CurUnityHandleCount <= 0 { if stat.CurUnityHandleCount <= 0 {
stat.CurUnityHandleCount = 0 stat.CurUnityHandleCount = 0
} }