87 lines
1.8 KiB
Go
87 lines
1.8 KiB
Go
|
package logic
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/zeromicro/go-zero/core/logx"
|
||
|
|
)
|
||
|
|
|
||
|
|
// 统计信息
|
||
|
|
var (
|
||
|
|
//当前ws连接数
|
||
|
|
currentWebsocketConnectCount = 0
|
||
|
|
//当前合图进行的请求总数
|
||
|
|
currentRequestCombineApiCount = 0
|
||
|
|
//添加or减少连接的控制chan
|
||
|
|
websocketStat = make(chan websocketStatItem, 20)
|
||
|
|
)
|
||
|
|
|
||
|
|
type websocketStatType string
|
||
|
|
|
||
|
|
const (
|
||
|
|
TYPE_CONNECT_COUNT websocketStatType = "connect_count" //ws连接数
|
||
|
|
TYPE_COMBINE_IMAGE_REQUEST_COUNT websocketStatType = "combine_count"
|
||
|
|
)
|
||
|
|
|
||
|
|
type websocketStatItem struct {
|
||
|
|
Type websocketStatType `json:"type"` //类型
|
||
|
|
Value int `json:"value"` //数值
|
||
|
|
}
|
||
|
|
|
||
|
|
// 累增ws连接数计数
|
||
|
|
func increaseWebsocketConnectCount() {
|
||
|
|
websocketStat <- websocketStatItem{
|
||
|
|
Type: TYPE_CONNECT_COUNT,
|
||
|
|
Value: 1,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// 减少ws连接数计数
|
||
|
|
func decreaseWebsocketConnectCount() {
|
||
|
|
websocketStat <- websocketStatItem{
|
||
|
|
Type: TYPE_CONNECT_COUNT,
|
||
|
|
Value: -1,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// 累增合图请求数计数
|
||
|
|
func increaseCombineRequestCount() {
|
||
|
|
websocketStat <- websocketStatItem{
|
||
|
|
Type: TYPE_COMBINE_IMAGE_REQUEST_COUNT,
|
||
|
|
Value: 1,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// 减少合图请求数计数
|
||
|
|
func decreaseCombineRequestCount() {
|
||
|
|
websocketStat <- websocketStatItem{
|
||
|
|
Type: TYPE_COMBINE_IMAGE_REQUEST_COUNT,
|
||
|
|
Value: -1,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// 消费数据
|
||
|
|
func ConsumeWebsocketStatData(ctx context.Context) {
|
||
|
|
defer func() {
|
||
|
|
if err := recover(); err != nil {
|
||
|
|
logx.Error("ConsumeWebsocketStatData panic:", err)
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
go func() {
|
||
|
|
select {
|
||
|
|
case <-ctx.Done():
|
||
|
|
panic("ConsumeWebsocketStatData ctx deadline")
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case data := <-websocketStat:
|
||
|
|
switch data.Type {
|
||
|
|
case TYPE_CONNECT_COUNT: //ws连接计数
|
||
|
|
currentWebsocketConnectCount += data.Value
|
||
|
|
case TYPE_COMBINE_IMAGE_REQUEST_COUNT: //请求算法合图计数
|
||
|
|
currentRequestCombineApiCount += data.Value
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|