Initial commit

2026-05-20 16:13:58 +08:00
commit ceffbe382e
500 changed files with 157469 additions and 0 deletions
@@ -0,0 +1,51 @@
+package agentSer
+
+import (
+	"context"
+	"go.mongodb.org/mongo-driver/bson"
+	"stat/common/log"
+	"stat/common/utils"
+	"stat/models/channelmod"
+)
+
+func RepairData() {
+	wg := utils.NewWaitGroup()
+	wg.Add(func() {
+		ctx := context.Background()
+		if err := FixChannelSecondParentAgent(ctx); err != nil {
+			log.Error("fix data err", log.E(err))
+		}
+	})
+	go wg.RunWaits()
+}
+
+func FixChannelSecondParentAgent(ctx context.Context) error {
+	filter := bson.M{
+		channelmod.FN.ChannelInfo.Type: channelmod.ChannelTypeSecond,
+		"$or": bson.A{
+			bson.M{channelmod.FN.ChannelInfo.ParentAgent: bson.M{"$eq": nil}},
+			bson.M{channelmod.FN.ChannelInfo.ParentAgent: bson.M{"$exists": false}},
+			bson.M{channelmod.FN.ChannelInfo.ParentAgent: bson.M{"$eq": ""}},
+		},
+		channelmod.FN.ChannelInfo.ParentChannel: bson.M{"$gt": ""},
+	}
+	count, err := channelmod.CountChannelInfos(ctx, filter)
+	if err != nil {
+		return err
+	}
+	if count == 0 {
+		return nil
+	}
+	// 修复
+	channels, err := channelmod.FindManyChannelInfo(ctx, filter)
+	for _, channel := range channels {
+		filter := bson.M{channelmod.FN.ChannelInfo.Channel: channel.Channel}
+		update := bson.M{"$set": bson.M{
+			channelmod.FN.ChannelInfo.ParentAgent: channel.ParentChannel,
+		}}
+		if err = channelmod.UpdateOneChannelInfo(ctx, filter, update); err != nil {
+			return err
+		}
+	}
+	return nil
+}