fruit_fix.go

package main

import (
	"fmt"
	"git.hilo.cn/hilo-common/script/mysql"
	"github.com/tealeg/xlsx"
	"gorm.io/gorm"
)

var userIds = []uint64{3117191}

type FruitMachine struct {
	Date    string
	Round   int64
	FruitId int
}

type FruitMachineStake2 struct {
	Stake int64
}

func main() {
	var Plus = make(map[int]int64) // fruitId->plush
	Plus[1] = 5
	Plus[2] = 5
	Plus[3] = 5
	Plus[4] = 5
	Plus[5] = 10
	Plus[6] = 15
	Plus[7] = 25
	Plus[8] = 45
	var data = make(map[uint64]int64)
	for round := 194; round <= 216; round++ {
		var fm = new(FruitMachine)
		if err := mysql.ProdReadOnlyDB.Model(FruitMachine{}).Where("`date` = '2023-08-09'").Where("round = ?", round).First(fm).Error; err != nil {
			panic(err)
		}
		for _, userId := range userIds {
			var fs = new(FruitMachineStake2)
			if err := mysql.ProdReadOnlyDB.Table("fruit_machine_stake").
				Where("`date` = '2023-08-09'").
				Where("round = ?", round).
				Where("fruit_id = ?", fm.FruitId).
				Where("user_id = ?", userId).First(&fs).Error; err != nil {
				if err != gorm.ErrRecordNotFound {
					panic(err)
				}
				continue
			}
			data[userId] += fs.Stake * Plus[fm.FruitId]
		}
	}
	excelFileName := fmt.Sprintf("./水果机.xlsx")
	xlFile := xlsx.NewFile()
	sheet, _ := xlFile.AddSheet("charge")
	row := sheet.AddRow()
	c1, c2 := row.AddCell(), row.AddCell()
	c1.Value, c2.Value = "userId", "should"
	for userId, award := range data {
		row := sheet.AddRow()
		c1, c2 := row.AddCell(), row.AddCell()
		c1.Value, c2.Value = fmt.Sprintf("%d", userId), fmt.Sprintf("%d", award)
	}
	_ = xlFile.Save(excelFileName)

}