chunhao_admin/service/UserOnlineService.go

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
}