workTasks/userSign/main.go

package main

import (
	"context"
	"fmt"
	_ "github.com/gogf/gf/contrib/drivers/clickhouse/v2"
	_ "github.com/gogf/gf/contrib/drivers/mysql/v2"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gcron"
	"github.com/gogf/gf/v2/os/gctx"
	"time"
	"workTasks/userSign/userAnalysis"
	"workTasks/userSign/userTag"
)

func main() {
	ctx := gctx.New()
	runOnce(context.TODO())
	if e := userTag.BaseUserTag(); e != nil {
		g.Log().Errorf(ctx, "BaseUserTag 异常", e)
	}

	_, err := gcron.Add(gctx.New(), g.Cfg().MustGet(ctx, "runCron", "# 0 2 * * *").String(), func(ctx context.Context) {
		runOnce(context.TODO())
		if e := userTag.BaseUserTag(); e != nil {
			g.Log().Errorf(ctx, "BaseUserTag 异常", e)
		}
	}, "userSign")
	if err != nil {
		panic(err)
	}
	select {}
}

func runOnce(ctx context.Context) {
	manager := userAnalysis.NewManager()
	var allRes []*userAnalysis.AnalysisRes
	if err := manager.LoadMapping(); err != nil {
		g.Log().Errorf(ctx, "加载对应关系异常 %v", err)
	}
	//更新全量用户
	if len(manager.FullBaseUserId) > 0 {
		allRes = append(allRes, &userAnalysis.AnalysisRes{Name: "全量用户BaseUserId", Code: "fullBaseUserId", Data: manager.FullBaseUserId})
	}

	//订单分析
	if res, err := manager.TiDBOrderRange(ctx); err == nil && len(res) > 0 {
		allRes = append(allRes, res...)
	}

	//加载30天内取关用户
	manager.GetMgoUnSubUser(ctx, -30)

	//会员状态分析
	if res, err := manager.GetMgoUserRes(ctx); err == nil && len(res) > 0 {
		allRes = append(allRes, res...)
	}
	var (
		sourceArr    = []string{"pc", "wx", "app"}
		nowTime      = time.Now()
		unActiveTime = []int{30, 90}
	)
	//活跃用户分析
	for _, day := range unActiveTime {
		if day == 30 { //30天活跃用户各端
			for _, source := range sourceArr {
				if res, err := manager.ActiveUserLoad(ctx, nowTime.AddDate(0, 0, -day), source); err == nil && len(res) > 0 {
					allRes = append(allRes, &userAnalysis.AnalysisRes{Name: fmt.Sprintf("30天%s活跃用户", source), Code: fmt.Sprintf("active_%s_%d", source, 30), Data: res})
				}
			}
		}
		if res, err := manager.ActiveUserLoad(ctx, nowTime.AddDate(0, 0, -day), ""); err == nil && len(res) > 0 {
			allRes = append(allRes, &userAnalysis.AnalysisRes{Name: fmt.Sprintf("%d天活跃用户", day), Code: fmt.Sprintf("active_%d", day), Data: res})
		}
	}
	// 近7天访问详情页次数大于5
	var (
		dayLimit     = 7
		articleLimit = 5
	)
	if res, err := manager.ArticleVisit(ctx, nowTime.AddDate(0, 0, -dayLimit), articleLimit); err == nil && len(res) > 0 {
		allRes = append(allRes, &userAnalysis.AnalysisRes{Name: fmt.Sprintf("%d天内访问详情页次数大于%d次", dayLimit, articleLimit), Code: fmt.Sprintf("%dday_article_gt_%d", dayLimit, articleLimit), Data: res})
	}
	// 已访问过
	if res, err := manager.Visited(ctx, nowTime.AddDate(0, 0, -7)); err == nil && len(res) > 0 {
		allRes = append(allRes, &userAnalysis.AnalysisRes{Name: "访问过剑鱼产品", Code: "visited", Data: res, SaveOldData: true})
	}
	// 访问过指定页面

	if res, err := manager.GetVisitUserAnalysis(ctx, nowTime.AddDate(0, 0, -45)); err == nil && len(res) > 0 {
		for _, t := range res {
			allRes = append(allRes, t)
		}
	}

	for _, re := range allRes {
		re.UpdateTag(ctx)
	}
}