Initial commit

tool/yc/main.go

@@ -0,0 +1,25 @@
+package main
+
+import (
+	"stat/common/log"
+	"stat/tool/yc/model"
+	"stat/tool/yc/service"
+	"time"
+)
+
+func initMongoDB() error {
+
+	if err := model.InitMongoDB(); err != nil {
+		log.Error("InitMongoDB err", log.E(err))
+		return err
+	}
+
+	return nil
+}
+
+func main() {
+	initMongoDB()
+	time.Local, _ = time.LoadLocation("Asia/Shanghai")
+	_ = service.SyncUserInfo()
+
+}

tool/yc/model/init.go

@@ -0,0 +1,25 @@
+package model
+
+import (
+	"fmt"
+	"stat/common/log"
+	"stat/common/mongos"
+	"stat/models"
+)
+
+//  InitMongoDB 初始化同步数据库
+func InitMongoDB() error {
+
+	statDB := mongos.NewManager("mongodb://stat:7u9CUAEE1Nv2tSvz@192.53.118.36:47017,192.53.173.209:47017,139.162.65.62:47017,172.105.237.106:47017,172.105.189.212:47017/stat?replicaSet=MDstat&maxPoolSize=200&connectTimeout=30&authSource=stat")
+	if err := statDB.Connect(); err != nil {
+		log.Error("mgoDB connect err", log.E(err))
+		return err
+	}
+	mongos.MgoManager = statDB
+
+	// 初始化数据库索引
+	models.InitIndex()
+	fmt.Println("init")
+
+	return nil
+}

tool/yc/service/service.go

@@ -0,0 +1,83 @@
+package service
+
+import (
+	"context"
+	"stat/common/log"
+	"stat/common/utils"
+	"stat/models/syncvip"
+	"time"
+
+	"go.mongodb.org/mongo-driver/bson"
+)
+
+func SyncUserInfo() error {
+	ctx := context.Background()
+
+	end := utils.BeginOfToday()
+	start := end.AddDate(0, 0, -1)
+
+	registerMap := make(map[time.Time]int64)
+
+	filter := bson.M{
+		syncvip.FN.SyncVip.AppId:     888,
+		syncvip.FN.SyncVip.SuccessAt: bson.M{"$gte": start, "$lt": end},
+	}
+
+	VipList, err := syncvip.FindManySyncVip(ctx, filter)
+	if err != nil {
+		log.Error("ChannelChannelStatMethod FindManyVisitInfo ", log.E(err))
+		return err
+	}
+	//fmt.Println(len(VipList))
+
+	for _, orderinfo := range VipList {
+		registerTime := utils.BeginWithTime(orderinfo.RegisterTime)
+		if registerTime.Before(start.AddDate(0, 0, -20)) {
+			continue
+		}
+		if money, ok := registerMap[registerTime]; ok {
+			registerMap[registerTime] = money + orderinfo.Money
+		} else {
+			registerMap[registerTime] = orderinfo.Money
+		}
+	}
+
+	//for key, value := range registerMap {
+	//	fmt.Println(key, "充值金额：", fmt.Sprintf("%02f", float64(value)/100))
+	//}
+
+	return nil
+}
+
+// func SyncUserInfo() error {
+// 	ctx := context.Background()
+
+// 	end := utils.BeginOfToday().AddDate(0, 0, -15)
+// 	start := end.AddDate(0, 0, -1)
+
+// 	registerMap := make(map[time.Time]int64)
+// 	// 获取充值金额
+// 	rechargeList, err := rechargemod.TotalDistricRechargeAmountV1(ctx, 888, start, end)
+// 	if err != nil {
+// 		log.Error("DirectAmountMethod DirectRechargeAmount ", log.E(err))
+// 		return err
+// 	}
+
+// 	for _, orderinfo := range rechargeList {
+// 		registerTime := utils.BeginWithTime(orderinfo.RegisterTime)
+// 		if registerTime.Before(start.AddDate(0, 0, -20)) {
+// 			continue
+// 		}
+// 		if money, ok := registerMap[registerTime]; ok {
+// 			registerMap[registerTime] = money + orderinfo.Money
+// 		} else {
+// 			registerMap[registerTime] = orderinfo.Money
+// 		}
+// 	}
+
+// 	for key, value := range registerMap {
+// 		fmt.Println(key, "充值金额：", fmt.Sprintf("%02f", float64(value)/100))
+// 	}
+
+// 	return nil
+// }