From 563b55b2162e01bde4ba6e0b01e33bdabd9bc509 Mon Sep 17 00:00:00 2001 From: hujiebin Date: Tue, 30 Apr 2024 10:29:46 +0800 Subject: [PATCH] Create fruit_slot_race_charge_history_sum.go --- script/fruit_slot_race_charge_history_sum.go | 90 ++++++++++++++++++++ 1 file changed, 90 insertions(+) create mode 100644 script/fruit_slot_race_charge_history_sum.go diff --git a/script/fruit_slot_race_charge_history_sum.go b/script/fruit_slot_race_charge_history_sum.go new file mode 100644 index 0000000..35847b6 --- /dev/null +++ b/script/fruit_slot_race_charge_history_sum.go @@ -0,0 +1,90 @@ +package main + +import ( + "fmt" + "git.hilo.cn/hilo-common/script/mysql" + "github.com/tealeg/xlsx" +) + +type FruitSlotRaceChargeHistoryData struct { + UserId uint64 + Code string + Area string + Country string + Bet uint64 // 5月份 + Dollar float64 // 累充金额$ +} + +func ats42(a interface{}) string { + return fmt.Sprintf("%v", a) +} + +func main() { + var charges []FruitSlotRaceChargeHistoryData + if err := mysql.ProdReadOnlyDB.Raw("select id as user_id,code,SUM(dollar) as dollar,country FROM " + + "(select u.id,u.`code`,SUM(price) / 100 as dollar,u.country FROM pay_order p, `user` u where u.id = p.user_id AND p.`status` = 2 AND `type` = 0 AND p.created_time >= \"2024-04-29\" AND p.created_time < \"2024-05-06\" group by user_id" + + "UNION ALL " + + "select u.id,u.code,SUM(dollar) / 100 as dollar,u.country FROM dealer_transfer_detail d, `user` u where u.id = d.receiver_id AND d.created_time >= \"2024-04-29\" AND d.created_time < \"2024-05-06\" group by receiver_id " + + "UNION ALL " + + "select u.id,u.code,SUM(dollar) / 100 as dollar,u.country FROM dealer_transfer_detail_pink d, `user` u where u.id = d.receiver_id AND d.created_time >= \"2024-04-29\" AND d.created_time < \"2024-05-06\" group by receiver_id" + + ") t group by code order by dollar DESC").Find(&charges).Error; err != nil { + panic(err) + } + type ResCountry struct { + Name string + Area int + } + var countries []ResCountry + if err := mysql.ProdReadOnlyDB.Model(ResCountry{}).Find(&countries).Error; err != nil { + panic(err) + } + countryMap := make(map[string]string) + for _, c := range countries { + countryMap[c.Name] = "阿语" + if c.Area == 2 { + countryMap[c.Name] = "非阿语" + } + } + var userIds []uint64 + for _, v := range charges { + userIds = append(userIds, v.UserId) + } + bets := GetBets(userIds) + for i, uc := range charges { + charges[i].Area = countryMap[uc.Country] + charges[i].Bet = bets[uc.UserId] + } + + excelFileName := fmt.Sprintf("./5月水果机slot赛车数据.xlsx") + xlFile := xlsx.NewFile() + sheet, err := xlFile.AddSheet("data") + if err != nil { + panic(err) + } + row := sheet.AddRow() + c1, c2, c3, c4, c5 := row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell() + c1.Value, c2.Value, c3.Value, c4.Value, c5.Value = "用户ID", "区域", "国家", "5月水果机+Slot+赛车的总下注钻石额", "上周充值金额" + for _, d := range charges { + row := sheet.AddRow() + c1, c2, c3, c4, c5 := row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell() + c1.Value, c2.Value, c3.Value, c4.Value, c5.Value = d.Code, d.Area, d.Country, ats42(d.Bet), ats42(d.Dollar) + } + _ = xlFile.Save(excelFileName) +} + +func GetBets(userIds []uint64) map[uint64]uint64 { + type Bet struct { + UserId uint64 + Num uint64 + } + var bets []Bet + res := make(map[uint64]uint64) + sql := "select user_id,SUM(num) num FROM diamond_account_detail_202405 where operate_type in (69,39,117) AND created_time >= \"2024-05-01\" AND created_time < \"2024-05-06\" and user_id in ?" + if err := mysql.ProdReadOnlyDB.Raw(sql, userIds).Find(&bets).Error; err != nil { + panic(err) + } + for _, v := range bets { + res[v.UserId] = v.Num + } + return res +} -- 2.22.0