124 lines
3.6 KiB
Go
124 lines
3.6 KiB
Go
package datacenter
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"91porn-server/app/appg"
|
|
"91porn-server/common"
|
|
"91porn-server/common/log"
|
|
"91porn-server/models/v/usermod"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type OnlineStatData struct {
|
|
UserId int64 `bson:"userId" json:"userId"` // 用户Id
|
|
DistrictCode string `json:"districtCode" bson:"districtCode"` // 渠道码
|
|
WatchCount int64 `json:"watchCount,omitempty" bson:"watchCount"` // 观看次数
|
|
WatchTime int64 `json:"watchTime,omitempty" bson:"watchTime"` // 观看时长
|
|
AdClick int64 `json:"adClick,omitempty" bson:"adClick"` // 广告点击
|
|
AppClick int64 `json:"appClick,omitempty" bson:"appClick"` // APP点击
|
|
RequestCount int64 `json:"requestCount,omitempty" bson:"requestCount"` // 请求次数
|
|
RegisterAt time.Time `json:"registerAt,omitempty" bson:"registerAt"` // 注册时间(只统计最近一个月注册的渠道用户?)
|
|
StatIndex int64 `json:"statIndex" bson:"statIndex"` // 统计数据更新时钟
|
|
SendIndex int64 `json:"sendIndex" bson:"sendIndex"` // 上次发送时钟
|
|
SendAt time.Time `json:"sendAt" bson:"sendAt"` // 上次发送时间
|
|
}
|
|
|
|
// OnlineStat 统计用户在线时间和接口访问次数
|
|
func OnlineStat(ctx *gin.Context) {
|
|
now := time.Now()
|
|
var statInfo OnlineStatData
|
|
|
|
// 查询用户
|
|
userId, err := common.GetUID(ctx)
|
|
if userId <= 0 || err != nil {
|
|
return
|
|
}
|
|
|
|
userOnlineKey := fmt.Sprintf("mdsmOnline:%s:%d", now.Format("20060102"), userId)
|
|
onlineValue, err := appg.Redis.Get(userOnlineKey)
|
|
if err != nil {
|
|
log.Error("OnlineStat", log.E(err))
|
|
return
|
|
}
|
|
|
|
if onlineValue != nil && *onlineValue != "" {
|
|
err = json.Unmarshal([]byte(*onlineValue), &statInfo)
|
|
if err != nil {
|
|
log.Error("OnlineStat", log.E(err))
|
|
return
|
|
}
|
|
|
|
} else {
|
|
user, serr := usermod.FindUserByUID(userId)
|
|
if serr != nil {
|
|
return
|
|
}
|
|
|
|
statInfo = OnlineStatData{
|
|
UserId: int64(userId),
|
|
DistrictCode: user.DistrictCode,
|
|
RegisterAt: user.CreatedAt,
|
|
WatchCount: 0,
|
|
WatchTime: 0,
|
|
AdClick: 0,
|
|
AppClick: 0,
|
|
RequestCount: 0,
|
|
StatIndex: 0,
|
|
SendIndex: 0,
|
|
SendAt: now,
|
|
}
|
|
}
|
|
|
|
// 统计数据
|
|
StatApiHandle(ctx, &statInfo)
|
|
|
|
// 更新数据库埋点规则 () 后期只统计渠道用户
|
|
if (statInfo.StatIndex < 2) || //前1次直接发送
|
|
(statInfo.StatIndex-statInfo.SendIndex) > 50 || //每30次合并发送
|
|
statInfo.SendAt.Before(time.Now().Add(-time.Second*40)) { // 超过40秒
|
|
|
|
// 清理数据
|
|
statInfo.WatchCount = 0
|
|
statInfo.AdClick = 0
|
|
statInfo.AppClick = 0
|
|
statInfo.RequestCount = 0
|
|
statInfo.SendAt = now
|
|
statInfo.SendIndex = statInfo.StatIndex
|
|
}
|
|
|
|
marshalStr, err := json.Marshal(statInfo)
|
|
if err != nil {
|
|
log.Error("OnlineStat", log.E(err))
|
|
return
|
|
}
|
|
_ = appg.Redis.Set(userOnlineKey, marshalStr, 24*time.Hour)
|
|
}
|
|
|
|
// StatApiHandle 统计Api和关键接口次数
|
|
func StatApiHandle(ctx *gin.Context, statInfo *OnlineStatData) {
|
|
statInfo.StatIndex += 1
|
|
statInfo.RequestCount += 1
|
|
|
|
// 播放接口
|
|
if strings.HasPrefix(ctx.Request.URL.Path, "/api/app/vid/h5/m3u8/") ||
|
|
strings.HasPrefix(ctx.Request.URL.Path, "/api/app/vid/m3u8/") ||
|
|
strings.HasPrefix(ctx.Request.URL.Path, "/api/app/vid/h5/light/m3u8/") {
|
|
statInfo.WatchCount += 1
|
|
statInfo.StatIndex += 1
|
|
}
|
|
|
|
// 广告点击
|
|
if strings.Contains(ctx.Request.URL.Path, "/api/app/ads/click") ||
|
|
strings.Contains(ctx.Request.URL.Path, "/api/app/recreation/click") {
|
|
statInfo.AdClick += 1
|
|
statInfo.StatIndex += 1
|
|
}
|
|
|
|
// 其他关键监控指标
|
|
}
|