server/game/game_cluster/nodes/logstash/fakedata/main.go

package main

import (
	"fmt"
	"log"
	"math/rand"
	"os"
	"time"

	"github.com/bxcodec/faker/v3"
	"github.com/mhaya/game/game_cluster/internal/mdb/eventmodels"
	"gorm.io/driver/clickhouse"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

// `insert into assets_event_contents (user_id, user_name, tg_id, x_id, parent_user_id, is_robot, points_rank_seq,
//                                    throw_dice_rank_seq, invite_user_rank_seq, is_success, create_at, operation_type,
//                                    currency, reason, before_balance, amount, after_balance, server_id)
// values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`

type NameData struct {
	User_id   string
	User_name string
	Tg_id     string
	X_id      string
}

func main() {
	dsn := "clickhouse://default:123456@192.168.0.193:9000/default?dial_timeout=10s&read_timeout=20s"

	newLogger := logger.New(
		log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer（日志输出的地方，这里是标准输出）
		logger.Config{
			SlowThreshold: time.Second, // 慢 SQL 阈值
			LogLevel:      logger.Info, // 日志级别
			Colorful:      true,        // 彩色打印
		},
	)

	db, err := gorm.Open(clickhouse.Open(dsn), &gorm.Config{
		Logger: newLogger,
	})
	if err != nil {
		panic("failed to connect database")
	}

	datas := make([]*eventmodels.AssetsEventContent, 0, 8)
	count := 100

	nameDatas := make([]*NameData, 0, 8)
	for i := 0; i < 10; i++ {
		user_id := faker.Name()
		name := faker.Name()
		tg_id := faker.Name()
		x_id := faker.Name()
		nameObj := &NameData{
			User_id:   user_id,
			User_name: name,
			Tg_id:     tg_id,
			X_id:      x_id,
		}
		nameDatas = append(nameDatas, nameObj)
	}

	server_id := "m-web-admin-1"
	for i := 0; i < count; i++ {
		index := rand.Intn(len(nameDatas))
		user_id := nameDatas[index].User_id
		user_name := nameDatas[index].User_name
		tg_id := nameDatas[index].Tg_id
		x_id := nameDatas[index].X_id
		parent_user_id := nameDatas[index].User_name
		is_robot := false
		points_rank_seq := faker.RandomUnixTime()
		throw_dice_rank_seq := faker.RandomUnixTime()
		invite_user_rank_seq := faker.RandomUnixTime()
		is_success := func() bool {
			return rand.Intn(2) == 1
		}
		create_at := faker.RandomUnixTime()
		operation_type := func() string {
			if rand.Intn(2) == 1 {
				return "increase"
			}
			return "decrease"
		}
		currency := func() string {
			index := rand.Intn(3)
			if index == 1 {
				return "POINTS"
			}
			if index == 2 {
				return "TON"
			}
			return "USDT"
		}
		reason := func() string {
			if rand.Intn(2) == 1 {
				return "reason"
			}
			return ""
		}
		var before_balance int64
		var amount int64
		var after_balance int64
		before_balance = rand.Int63n(100000)
		amount = rand.Int63n(1000)
		after_balance = before_balance + amount

		// email := faker.Email()
		// address := faker.Address()
		// phone := faker.Phone()

		datas = append(datas, &eventmodels.AssetsEventContent{
			UserBasic: eventmodels.UserBasic{
				UserId:            user_id,
				UserName:          user_name,
				TgId:              tg_id,
				XId:               x_id,
				ParentUserId:      parent_user_id,
				IsRobot:           is_robot,
				PointsRankSeq:     points_rank_seq,
				ThrowDiceRankSeq:  throw_dice_rank_seq,
				InviteUserRankSeq: invite_user_rank_seq,
			},
			EventBasic: eventmodels.EventBasic{
				ServerId:  server_id,
				IsSuccess: is_success(),
				CreateAt:  create_at,
			},
			OperationType: operation_type(),
			Currency:      currency(),
			Reason:        reason(),
			BeforeBalance: before_balance,
			Amount:        amount,
			AfterBalance:  after_balance,
		})
	}

	err = db.CreateInBatches(datas, len(datas)).Error
	if err != nil {
		fmt.Printf("CreateInBatches error:%v", err)
		return
	}

	// sqlStr := `select SUM(amount) as total_amount from assets_event_contents where user_id = ? and currency = ?`
	var total_amount int64
	for _, nameObj := range nameDatas {
		// err := db.Raw(sqlStr, name, "USDT").Pluck("SUM(amount) as total_amount", &total_amount).Error
		// if err != nil {
		// 	fmt.Printf("Pluck error:%v", err)
		// 	return
		// }

		where := &eventmodels.AssetsEventContent{
			UserBasic: eventmodels.UserBasic{
				UserId: nameObj.User_id,
				// UserId: "1", // 不存在的用户
			},
			EventBasic: eventmodels.EventBasic{
				ServerId: server_id,
			},
			Currency: "USDT",
		}
		err := db.Model(&eventmodels.AssetsEventContent{}).Where(where).Pluck("SUM(amount) as total_amount", &total_amount).Error
		if err != nil {
			fmt.Printf("Pluck error:%v", err)
			return
		}

		fmt.Printf("User_id:%s total_amount:%v", nameObj.User_id, total_amount)
	}
}