package service import ( "designs/global" "designs/model" "designs/utils" "math" "time" ) type logData struct { LogTime time.Time `json:"LogTime"` Type int `json:"Type"` } // 查询单日使用时长曲线 func UserOnlineSummaryByDay(gid string, pf string, startTime string, endTime string) (map[string]int, int, int, error) { var onlineData []model.UserOnline //从数据库中查询出所有的数据 query := global.App.DB.Table("user_online").Where("gid", gid).Where("pf", pf) if startTime != "" && endTime != "" { query = query.Where("logTime", ">=", startTime).Where("logTime", "<=", endTime) } err := query.Scan(&onlineData).Error if err != nil { return nil, 0, 0, err } //对数据进行分组 userOnlineData := make(map[string]map[int][]logData) dateSlice := utils.GetTimeDayDateFormat(startTime, endTime) for _, v := range dateSlice { userOnlineData[v] = make(map[int][]logData) } for _, v := range onlineData { date := v.LogTime.Format("2006-01-02") userOnlineData[date][v.UserId] = append(userOnlineData[date][v.UserId], logData{v.LogTime, v.Type}) } //计算每天的数据,然后得出平均值 var totalAvg int var total int var totalTime int64 daysAvg := make(map[string]int) for date, daysData := range userOnlineData { var userTotalTime int64 //分组后对于每个用户的数据进行处理,得到他们的具体在线时长 for _, v := range daysData { userTotalTime = userTotalTime + calculateUserOnlineTime(v) } //计算出平均值,并且格式化 if len(daysData) > 0 { avg := float64(userTotalTime) / float64(len(daysData)) daysAvg[date] = int(math.Round(avg)) } else { daysAvg[date] = 0 } //统计累加 total = total + len(daysData) totalTime = totalTime + userTotalTime } //统计平均值 if total <= 0 { totalAvg = 0 } else { avg := float64(totalTime) / float64(total) totalAvg = int(math.Round(avg)) } return daysAvg, total, totalAvg, nil } func UserOnlineSummary(gid string, pf string, date string, startTime string, endTime string) (map[int]int64, error) { var onlineData []model.UserOnline //从数据库中查询出所有的数据 query := global.App.DB.Table("user_online").Where("gid", gid).Where("pf", pf) if date != "" { query = query.Where("date", "=", date) } if startTime != "" && endTime != "" { query = query.Where("logTime", ">=", startTime).Where("logTime", "<=", endTime) } err := query.Scan(&onlineData).Error if err != nil { return nil, err } //对数据进行分组 userOnlineData := make(map[int][]logData) for _, v := range onlineData { value, exists := userOnlineData[v.UserId] if exists { userOnlineData[v.UserId] = append(value, logData{v.LogTime, v.Type}) } else { userOnlineData[v.UserId] = []logData{{v.LogTime, v.Type}} } } onlineData = nil userLogMsg := make(map[int]int64) //分组后对于每个用户的数据进行处理,得到他们的具体在线时长 for k, v := range userOnlineData { userLogMsg[k] = calculateUserOnlineTime(v) } userOnlineData = nil return userLogMsg, nil } func calculateUserOnlineTime(logData []logData) int64 { var lastLog int64 var isStart bool var onlineTimeTotal int64 for k, v := range logData { logTime := v.LogTime.Unix() //如果跟上一次的记录隔的时间超过6分钟,就认为是之前断线了,属于无效数据 if k > 0 && logData[k-1].LogTime.Unix()+60*6 < logTime { isStart = false continue } if v.Type == 1 && isStart == false { isStart = true lastLog = v.LogTime.Unix() continue } if v.Type == 1 && isStart == true { //logTime := v.LogTime.Unix() onlineTimeTotal = onlineTimeTotal + (logTime - lastLog) lastLog = logTime //如果下一次的心跳,间隔时间大于6分钟了,就可以认为,这次就算是结束了 if k+1 == len(logData) || logData[k+1].LogTime.Unix() > logTime+60*6 { isStart = false } continue } if v.Type == 2 && isStart == true { //logTime := v.LogTime.Unix() onlineTimeTotal = onlineTimeTotal + (logTime - lastLog) isStart = false continue } } return onlineTimeTotal }