chunhao_admin/service/UserOnlineService.go

package service

import (
	"bufio"
	"designs/config"
	"designs/utils"
	"fmt"
	"math"
	"os"
	"path/filepath"
	"sort"
	"strconv"
	"strings"
	"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})
	//}

	userOnlineData := make(map[string]map[int][]logData)

	dateSlice := utils.GetTimeDayDateFormat(startTime, endTime)

	for _, v := range dateSlice {

		userOnlineData[v] = make(map[int][]logData)

		userOnlineData[v] = GetLocalActiveLogDetail(gid, pf, v)

	}

	//计算每天的数据，然后得出平均值
	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

	userOnlineData := make(map[int][]logData)
	userOnlineData = GetLocalActiveLogDetailRange(gid, pf, startTime, endTime)

	userLogMsg := make(map[int]int64)
	//分组后对于每个用户的数据进行处理，得到他们的具体在线时长
	for k, v := range userOnlineData {
		//aaa, _ := json.Marshal(v)
		//global.App.Log.Error(string(aaa))
		userLogMsg[k] = calculateUserOnlineTime(v)
	}
	userOnlineData = nil

	return userLogMsg, nil
}

func calculateUserOnlineTime(logData []logData) int64 {

	sort.Slice(logData, func(i, j int) bool {
		return logData[i].LogTime.Before(logData[j].LogTime)
	})

	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
}

func GetLocalActiveLogDetail(gid string, pf string, date string) map[int][]logData {
	var dir string
	if config.Get("app.local") == "local" {
		//url = "mongodb://localhost:27017"
		dir = "storage"
	} else {
		dir = "/www/wwwroot/chunhao_receive/storage"
	}

	dirPath := filepath.Join(dir, date)
	fileName := fmt.Sprintf("%s_%s.txt", gid, pf)
	filePath := filepath.Join(dirPath, fileName)

	file, _ := os.Open(filePath)

	defer file.Close() // 确保函数结束时关闭文件

	// 创建 Scanner 对象
	scanner := bufio.NewScanner(file)

	res := make(map[int][]logData)
	// 逐行读取文件内容
	lineNumber := 1
	for scanner.Scan() {
		line := scanner.Text() // 获取当前行内容
		lineNumber++

		//11887,2,2025-03-24 09:05:19
		lineSlice := strings.Split(line, ",")

		userId, _ := strconv.Atoi(lineSlice[0])
		types, _ := strconv.Atoi(lineSlice[1])
		// 解析时间
		t, _ := time.Parse("2006-01-02  15:04:05", lineSlice[2])

		res[userId] = append(res[userId], logData{t, types})
	}

	return res
}

func GetLocalActiveLogDetailRange(gid string, pf string, start string, end string) map[int][]logData {
	var dir string
	if config.Get("app.local") == "local" {
		//url = "mongodb://localhost:27017"
		dir = "storage"
	} else {
		dir = "/www/wwwroot/chunhao_receive/storage"
	}

	dateSlice := utils.GetTimeDayDateFormat(start, end)

	res := make(map[int][]logData)
	for _, v := range dateSlice {

		dirPath := filepath.Join(dir, v)
		fileName := fmt.Sprintf("%s_%s.txt", gid, pf)
		filePath := filepath.Join(dirPath, fileName)

		file, _ := os.Open(filePath)

		// 创建 Scanner 对象
		scanner := bufio.NewScanner(file)
		// 逐行读取文件内容
		lineNumber := 1
		for scanner.Scan() {
			line := scanner.Text() // 获取当前行内容
			lineNumber++

			//11887,2,2025-03-24 09:05:19
			lineSlice := strings.Split(line, ",")

			userId, _ := strconv.Atoi(lineSlice[0])
			types, _ := strconv.Atoi(lineSlice[1])
			// 解析时间
			t, _ := time.Parse("2006-01-02  15:04:05", lineSlice[2])

			res[userId] = append(res[userId], logData{t, types})
		}

		file.Close() // 确保函数结束时关闭文件
	}

	return res
}