hongze_cygx/services/wx_user.go

package services

import (
	"context"
	"fmt"
	"hongze/hongze_cygx/models"
	"hongze/hongze_cygx/utils"
	"sort"
	"strconv"
	"strings"
	"time"
)

func UpdateWxUserLabel(cont context.Context) (err error) {
	defer func() {
		if err != nil {
			go utils.SendAlarmMsg("更新用户标签失败;Err:"+err.Error(), 2)
			go utils.SendEmail("更新用户标签失败"+"【"+utils.APPNAME+"】"+time.Now().Format(utils.FormatDateTime), ";Err:"+err.Error(), utils.EmailSendToUsers)
			utils.FileLog.Info("更新用户标签失败,Err:%s", err.Error())
		}
	}()
	listUser, err := models.GetUserRegisterList()
	if err != nil {
		fmt.Println("GetUserRegisterList Err", err)
		return err
	}
	var userIds string

	var interactionNum int //  互动量
	mapComapnyInteractionNum := make(map[int]int)

	//拉取策略平台用户同步的图表信息
	{
		var chartMobile string
		var chartItems []*models.CygxChartCollect
		for k, vUser := range listUser {
			if vUser.Mobile != "" {
				mobile := vUser.Mobile
				chartMobile += mobile + ","
				listChart, _, _ := GetChartCollectionByApi(mobile, 9999, 0)
				if len(listChart) > 0 {
					for _, vChart := range listChart {
						item := new(models.CygxChartCollect)
						item.ChartId = vChart.ChartId
						t1, _ := time.Parse("2006-01-02T15:04:05Z", vChart.CreateDate)
						item.CreateTime = t1
						item.Mobile = mobile
						chartItems = append(chartItems, item)
					}
				}
				fmt.Println(mobile, "长度", len(listChart), "条数", k)
			}
		}

		fmt.Println("总长度", len(chartItems))
		chartMobile = strings.TrimRight(chartMobile, ",")
		err = models.RemoveChartCollectByMobile(chartMobile)
		if err != nil {
			fmt.Println(" RemoveChartCollectByMobile Err", err)
		}
		_, err = models.AddCygxChartCollectList(chartItems)
		if err != nil {
			fmt.Println(" RemoveChartCollectByMobile Err", err)
		}

		//处理图表关注后的用户
		{
			fmt.Println("处理图表关注后的用户")
			listChartCollect, err := models.GetCygxChartCollectByMobileList()
			if err != nil {
				fmt.Println("GetCygxChartCollectByMobileList ，Err" + err.Error())
			}
			fmt.Println("长度", len(listChartCollect))
			for k, v := range listChartCollect {
				if v.Mobile != "" {
					user, err := models.GetWxUserItemByMobile(v.Mobile)
					if err != nil && err.Error() != utils.ErrNoRow() {
						fmt.Println("GetWxUserItemByUserId ，Err" + err.Error())
					}
					if user != nil {
						fmt.Println(user.RealName, k)
						err = models.UpdateCygxChartCollect(user)
						if err != nil {
							fmt.Println("UpdateCygxChartCollect ，Err" + err.Error())
						}
					}
				}
			}
		}
	}

	// 处理用户标签
	for _, vUser := range listUser {
		labels, err := models.GetCygxCompanyUserListSplit(strconv.Itoa(vUser.UserId))
		if err != nil {
			fmt.Println("GetCygxCompanyUserListSplit Err", err)
			return err
		}
		mapLabels := make(map[string]string)
		var labelsListStr []*models.UserLabel
		mapActivityLabels := make(map[string]string)
		labelsList := strings.Split(labels, ",")
		for _, vL := range labelsList {
			if mapLabels[vL] == "" && vL != "" {
				mapLabels[vL] = vL
				item := new(models.UserLabel)
				item.Label = vL
				labelsListStr = append(labelsListStr, item)
			}
		}
		dateTime := time.Now().AddDate(0, -3, 0).Format(utils.FormatDate)
		listArticlePv, err := models.GetArticlePvCount(vUser.Mobile, vUser.Email, dateTime)
		if err != nil && err.Error() != utils.ErrNoRow() {
			fmt.Println("GetArticlePvCount Err", err)
			return err
		}
		var articleIds string
		mapArticleIdLabelTotal := make(map[string]int)
		mapArticleIdPv := make(map[int]int)
		if len(listArticlePv) > 0 {
			for _, v := range listArticlePv {
				articleIds += strconv.Itoa(v.ArticleId) + ","
				mapArticleIdPv[v.ArticleId] = v.Pv
			}
			articleIds = strings.TrimRight(articleIds, ",")
			//产业、标的标签
			listLabelName, err := models.GetArticleGroupyIndustryName(articleIds)
			if err != nil && err.Error() != utils.ErrNoRow() {
				fmt.Println("GetArticleGroupyIndustryName Err", err)
				return err
			}
			if len(listLabelName) > 0 {
				for _, v := range listLabelName {
					labelsIndustryList := strings.Split(v.IndustryName, "/")
					labelsSubjectList := strings.Split(v.SubjectName, "/")
					//添加产业统计
					if len(labelsIndustryList) > 0 {
						for _, vIn := range labelsIndustryList {
							if mapActivityLabels[vIn] != "" {
								mapArticleIdLabelTotal[vIn] += mapArticleIdPv[v.ArticleId]
							} else {
								mapArticleIdLabelTotal[vIn] = mapArticleIdPv[v.ArticleId]
								mapActivityLabels[vIn] = vIn
							}
						}
					}
					if len(labelsSubjectList) > 0 {
						//添加标的统计
						for _, vIn := range labelsSubjectList {
							if mapActivityLabels[vIn] != "" {
								mapArticleIdLabelTotal[vIn] += mapArticleIdPv[v.ArticleId]
							} else {
								mapArticleIdLabelTotal[vIn] = mapArticleIdPv[v.ArticleId]
								mapActivityLabels[vIn] = vIn
							}
						}
					}
				}
			}
		}
		articleIds, err = models.GetCygxArticleCollectId(vUser.UserId, dateTime)
		if err != nil && err.Error() != utils.ErrNoRow() {
			fmt.Println("GetCygxArticleCollectId Err", err)
			return err
		}
		if articleIds != "" {
			//产业、标的标签
			listLabelName, err := models.GetArticleGroupyIndustryName(articleIds)
			if err != nil && err.Error() != utils.ErrNoRow() {
				fmt.Println("GetArticleGroupyIndustryName Err", err)
				return err
			}
			if len(listLabelName) > 0 {
				for _, v := range listLabelName {
					labelsIndustryList := strings.Split(v.IndustryName, "/")
					labelsSubjectList := strings.Split(v.SubjectName, "/")
					//添加产业统计
					if len(labelsIndustryList) > 0 {
						for _, vIn := range labelsIndustryList {
							if mapActivityLabels[vIn] != "" {
								mapArticleIdLabelTotal[vIn] += 1
							} else {
								mapArticleIdLabelTotal[vIn] = 1
								mapActivityLabels[vIn] = vIn
							}
						}
					}
					if len(labelsSubjectList) > 0 {
						//添加标的统计
						for _, vIn := range labelsSubjectList {
							if mapActivityLabels[vIn] != "" {
								mapArticleIdLabelTotal[vIn] += 1
							} else {
								mapArticleIdLabelTotal[vIn] = 1
								mapActivityLabels[vIn] = vIn
							}
						}
					}
				}
			}
		}
		//活动标签计算
		listActivityPv, err := models.GetActivitySignCount(vUser.Mobile, dateTime)
		if err != nil && err.Error() != utils.ErrNoRow() {
			fmt.Println("GetActivitySignCount Err", err)
			return err
		}
		if len(listActivityPv) > 0 {
			for _, v := range listActivityPv {
				labelsList := strings.Split(v.Label, "-")
				for _, v2 := range labelsList {
					labelsListSon := strings.Split(v2, "/")
					for _, v3 := range labelsListSon {
						if mapActivityLabels[v3] != "" {
							mapArticleIdLabelTotal[v3] += 1
						} else {
							mapArticleIdLabelTotal[v3] = 1
							mapActivityLabels[v3] = v3
						}
					}
				}
			}
		}
		//对标签数量进行排序
		type KvPair struct {
			Key string
			Val int
		}
		tmpList := make([]KvPair, 0)
		for k, v := range mapArticleIdLabelTotal {
			tmpList = append(tmpList, KvPair{Key: k, Val: v})
		}
		sort.Slice(tmpList, func(i, j int) bool {
			return tmpList[i].Val > tmpList[j].Val // 降序
		})
		for _, v := range tmpList {
			if v.Key != "" {
				mapLabels[v.Key] = v.Key
				item := new(models.UserLabel)
				item.Label = v.Key
				labelsListStr = append(labelsListStr, item)
			}
		}
		var labelNew string
		for _, v := range labelsListStr {
			labelNew += v.Label + ","
		}
		labelNew = strings.TrimRight(labelNew, ",")
		if labelNew != "" {
			err = models.UpdateUserLabel(labelNew, vUser.UserId)
			if err != nil {
				fmt.Println("UpdateUserLabel Err", err)
				return err
			}
		}
	}

	//处理用户、机构互动量数据
	{
		var itemsInteraction []*models.CygxUserInteractionNum
		mapUserInteraction := make(map[int]int)
		//获取已经处理记录的用户并记录切片
		listUserInteractionNum, err := models.GetCygxUserInteractionNumList()
		if err != nil && err.Error() != utils.ErrNoRow() {
			fmt.Println("GetCygxUserInteractionNumList Err", err)
			return err
		}
		for _, v := range listUserInteractionNum {
			mapUserInteraction[v.UserId] = v.UserId
		}

		//处理用户互动量数据
		for kUser, vUser := range listUser {
			userIds += strconv.Itoa(vUser.UserId) + ","
			//一次查询一百个避免用户过多
			if kUser%100 == 0 {
				userIds = strings.TrimRight(userIds, ",")
				userUserInteractionList, err := models.GetCygxCompanyUserUserInteraction(userIds)
				if err != nil {
					fmt.Println("GetCygxCompanyUserUserInteraction Err", err)
					return err
				}
				//处理用户的互动量
				if len(userUserInteractionList) > 0 {
					for _, vsplit := range userUserInteractionList {
						interactionNum = vsplit.HistoryNum + vsplit.CountNum + vsplit.IndustryFllowNum + vsplit.DepartmentFollowNum + vsplit.KeyWordNum + vsplit.OnLineNum + vsplit.OfficeNum + vsplit.ChartCountNum
						if interactionNum > 0 {
							//更改用户的互动量
							err = models.UpdateUserInteractionNum(interactionNum, int(vsplit.UserId))
							if err != nil {
								fmt.Println("UpdateUserInteractionNum Err", err)
								return err
							}
							mapComapnyInteractionNum[vsplit.CompanyId] += interactionNum

							itemInteraction := new(models.CygxUserInteractionNum)
							itemInteraction.UserId = int(vsplit.UserId)
							itemInteraction.ArticleCountNum = vsplit.HistoryNum
							itemInteraction.ArticleCountNum = vsplit.CountNum
							itemInteraction.ChartCountNum = vsplit.ChartCountNum
							itemInteraction.IndustryFllowNum = vsplit.IndustryFllowNum
							itemInteraction.DepartmentFollowNum = vsplit.DepartmentFollowNum
							itemInteraction.KeyWordNum = vsplit.KeyWordNum
							itemInteraction.ActivityOnLineNum = vsplit.OnLineNum
							itemInteraction.ActivityOfficeNum = vsplit.OfficeNum
							itemInteraction.ActivityNum = vsplit.OnLineNum + vsplit.OfficeNum
							itemInteraction.CreateTime = time.Now()
							itemInteraction.ModifyTime = time.Now()
							//如果存在就修改不存在就新增
							if mapUserInteraction[int(vsplit.UserId)] == 0 {
								itemsInteraction = append(itemsInteraction, itemInteraction)
							} else {
								err = models.UpdateCygxUserInteractionNum(itemInteraction)
							}
						}
					}
				}
				userIds = ""
			}
		}

		userIds = strings.TrimRight(userIds, ",")
		if userIds != "" {
			userUserInteractionList, err := models.GetCygxCompanyUserUserInteraction(userIds)
			if err != nil {
				fmt.Println("GetCygxCompanyUserUserInteraction Err", err)
				return err
			}
			fmt.Println("处理用户剩余的")
			//处理用户的互动量
			if len(userUserInteractionList) > 0 {
				for _, vsplit := range userUserInteractionList {
					interactionNum = vsplit.HistoryNum + vsplit.CountNum + vsplit.IndustryFllowNum + vsplit.DepartmentFollowNum + vsplit.KeyWordNum + vsplit.OnLineNum + vsplit.OfficeNum + vsplit.ChartCountNum
					if interactionNum > 0 {
						err = models.UpdateUserInteractionNum(interactionNum, int(vsplit.UserId))
						if err != nil {
							fmt.Println("UpdateUserInteractionNum Err", err)
							return err
						}
						mapComapnyInteractionNum[vsplit.CompanyId] += interactionNum

						itemInteraction := new(models.CygxUserInteractionNum)
						itemInteraction.UserId = int(vsplit.UserId)
						itemInteraction.ArticleCountNum = vsplit.HistoryNum
						itemInteraction.ArticleCountNum = vsplit.CountNum
						itemInteraction.ChartCountNum = vsplit.ChartCountNum
						itemInteraction.IndustryFllowNum = vsplit.IndustryFllowNum
						itemInteraction.DepartmentFollowNum = vsplit.DepartmentFollowNum
						itemInteraction.KeyWordNum = vsplit.KeyWordNum
						itemInteraction.ActivityOnLineNum = vsplit.OnLineNum
						itemInteraction.ActivityOfficeNum = vsplit.OfficeNum
						itemInteraction.ActivityNum = vsplit.OnLineNum + vsplit.OfficeNum
						itemInteraction.CreateTime = time.Now()
						itemInteraction.ModifyTime = time.Now()
						//如果存在就修改不存在就新增
						if mapUserInteraction[int(vsplit.UserId)] == 0 {
							itemsInteraction = append(itemsInteraction, itemInteraction)
						} else {
							err = models.UpdateCygxUserInteractionNum(itemInteraction)
						}
					}
				}
			}
		}

		//修改机构互动量信息
		if len(mapComapnyInteractionNum) > 0 {
			for k, v := range mapComapnyInteractionNum {
				err = models.UpdateComapanyInteractionNum(v, k)
				fmt.Println(k, "修改", v)
			}
		}
		// 批量添加用户互动量信息
		if len(itemsInteraction) > 0 {
			_, err = models.AddCygxUserInteractionNumList(itemsInteraction)
			if err != nil {
				fmt.Println(" RemoveChartCollectByMobile Err", err)
			}
		}
	}
	return
}