diff --git a/model/model_test.go b/model/model_test.go index 2e2e7ba..c7c790b 100644 --- a/model/model_test.go +++ b/model/model_test.go @@ -2580,6 +2580,71 @@ func GetMonth(date string) string { return date[len(date)-2:] + "月份" } +// 拉新锁卡数据 +func TestLachineCardInfo(t *testing.T) { + InitDBProd() + DB = DBProd + LachineCardInfo() +} + +// 拉新锁卡数据 +func LachineCardInfo() { + DB = DBProd + var cards []OrderCard + err := DBProd.Raw("SELECT * FROM order_card WHERE uid IN (41062552,61322858,79502493,5151314,75683046,92301496,84098843) AND card_status IN (1,2,3);").Scan(&cards).Error + if err != nil { + logger.Error("err:", err) + } + storeIds := make([]uint32, 0, len(cards)) + gameIds := make([]uint32, 0, len(cards)) + + for i, _ := range cards { + storeIds = append(storeIds, cards[i].StoreId) + gameIds = append(gameIds, cards[i].GameCardId) + } + storeMap, err := GetStoreMap(storeIds) + if err != nil { + logger.Error("err:", err) + } + gameMap := GetGameCardMap(gameIds) + file := excelize.NewFile() + streamWriter, err := file.NewStreamWriter("Sheet1") + if err != nil { + fmt.Println(err) + } + + fileName := time.Now().Format(TimeFormat) + "锁卡数据" + ".xlsx" + + title := []interface{}{"订单id", "门店id", "门店名称", "游戏id", "游戏名称", "游戏卡编码", "锁卡时间", "订单状态"} + cell, _ := excelize.CoordinatesToCellName(1, 1) + if err = streamWriter.SetRow(cell, title); err != nil { + fmt.Println(err) + } + list := cards + var row []interface{} + for rowId := 0; rowId < len(list); rowId++ { + store, _ := storeMap[list[rowId].StoreId] + gameCard := gameMap[list[rowId].GameCardId] + + row = []interface{}{list[rowId].ID, store.ID, store.Name, list[rowId].GameCardId, gameCard.Name, list[rowId].SerialNumber, + list[rowId].CreatedAt.Format(TimeFormat), GetOrderStatus(list[rowId].CardStatus), + } + cell, _ := excelize.CoordinatesToCellName(1, rowId+2) + if err := streamWriter.SetRow(cell, row); err != nil { + fmt.Println(err) + } + } + if err := streamWriter.Flush(); err != nil { + fmt.Println(err) + } + //if err := file.SaveAs("/www/server/images/export/" + fileName); err != nil { + if err := file.SaveAs("./" + fileName); err != nil { + fmt.Println(err) + } + + return +} + func TestNameSunHash(t *testing.T) { }