360 lines
8.9 KiB
Go
360 lines
8.9 KiB
Go
|
package offlineactivity
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"encoding/csv"
|
|||
|
|
"fmt"
|
|||
|
|
"go-common/library/time"
|
|||
|
|
"math"
|
|||
|
|
"strconv"
|
|||
|
|
time2 "time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
dateFmt = "20060102"
|
|||
|
|
dateTimeFmt = "20060102_150405"
|
|||
|
|
filenamePre = "活动"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func round(num float64) int {
|
|||
|
|
return int(num + math.Copysign(0.5, num))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ToFixed fix float precision
|
|||
|
|
func ToFixed(num float64, precision int) float64 {
|
|||
|
|
output := math.Pow(10, float64(precision))
|
|||
|
|
return float64(round(num*output)) / output // since go 1.9 doesn't have a math.Round function...
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ExportArg export arg
|
|||
|
|
type ExportArg struct {
|
|||
|
|
Export string `form:"export"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ExportFormat export format
|
|||
|
|
func (e *ExportArg) ExportFormat() string {
|
|||
|
|
return e.Export
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//BonusInfo bonus struct
|
|||
|
|
type BonusInfo struct {
|
|||
|
|
TotalMoney float64 `json:"total_money"`
|
|||
|
|
MemberCount int64 `json:"member_count"`
|
|||
|
|
Mids string `json:"mids"`
|
|||
|
|
MidList []int64 `json:"-"` // 存储转换后的mids
|
|||
|
|
Filename string `json:"filename"` // 如果有文件名，优先使用文件
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//AddActivityArg add activity arg
|
|||
|
|
type AddActivityArg struct {
|
|||
|
|
Title string `json:"title"`
|
|||
|
|
Link string `json:"link"`
|
|||
|
|
BonusType int8 `json:"bonus_type"`
|
|||
|
|
Memo string `json:"memo"`
|
|||
|
|
BonusList []*BonusInfo `json:"bonus_list"`
|
|||
|
|
Creator string `json:"-"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//PreAddActivityResult result, nothing to return
|
|||
|
|
type PreAddActivityResult struct {
|
|||
|
|
BonusType int8 `json:"bonus_type"`
|
|||
|
|
MemberCount int64 `json:"member_count"`
|
|||
|
|
TotalMoney float64 `json:"total_money"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//AddActivityResult result, nothing to return
|
|||
|
|
type AddActivityResult struct {
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//PageArg page arg
|
|||
|
|
type PageArg struct {
|
|||
|
|
Page int `form:"page"`
|
|||
|
|
Size int `form:"size"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//CheckPageValidation check the page validte, return limit offset
|
|||
|
|
func (arg *PageArg) CheckPageValidation() (limit, offset int) {
|
|||
|
|
if arg.Page < 1 {
|
|||
|
|
arg.Page = 1
|
|||
|
|
}
|
|||
|
|
if arg.Size > 100 || arg.Size <= 0 {
|
|||
|
|
arg.Size = 10
|
|||
|
|
}
|
|||
|
|
limit = arg.Size
|
|||
|
|
offset = (arg.Page - 1) * limit
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ToPageResult cast to page result
|
|||
|
|
func (arg *PageArg) ToPageResult(total int) (res PageResult) {
|
|||
|
|
res.TotalCount = total
|
|||
|
|
res.Page = arg.Page
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//PageResult page result
|
|||
|
|
type PageResult struct {
|
|||
|
|
Page int `json:"page"`
|
|||
|
|
TotalCount int `json:"total_count"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//QueryActivityByIDArg arg
|
|||
|
|
type QueryActivityByIDArg struct {
|
|||
|
|
FromDate string `form:"from_date"` // 20180101
|
|||
|
|
ToDate string `form:"to_date"` // 20180102 closed interval [20180101, 20180102]
|
|||
|
|
ID int64 `form:"id"` // activity id, if not 0, FromDate and toDate not used
|
|||
|
|
PageArg
|
|||
|
|
ExportArg
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//QueryActivityInfo activity info
|
|||
|
|
type QueryActivityInfo struct {
|
|||
|
|
OfflineActivityInfo
|
|||
|
|
CreateTime string `json:"create_time"`
|
|||
|
|
TotalMoney float64 `json:"total_money"`
|
|||
|
|
MemberCount uint32 `json:"member_count"`
|
|||
|
|
PayDate string `json:"pay_date"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//QueryActivityResult result
|
|||
|
|
type QueryActivityResult struct {
|
|||
|
|
Result []*QueryActivityInfo `json:"result"`
|
|||
|
|
PageResult
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//GetFileName get file name
|
|||
|
|
func (q *QueryActivityResult) GetFileName() string {
|
|||
|
|
return fmt.Sprintf("%s-%s_%s.csv", filenamePre, "按活动", time2.Now().Format(dateTimeFmt))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ToCsv to buffer
|
|||
|
|
func (q *QueryActivityResult) ToCsv(writer *csv.Writer) {
|
|||
|
|
var title = []string{
|
|||
|
|
"添加日期",
|
|||
|
|
"活动id",
|
|||
|
|
"活动标题",
|
|||
|
|
"活动链接",
|
|||
|
|
"支出金额",
|
|||
|
|
"中奖人数",
|
|||
|
|
"付款日期",
|
|||
|
|
"发奖状态"}
|
|||
|
|
writer.Write(title)
|
|||
|
|
if q == nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
for _, v := range q.Result {
|
|||
|
|
var record []string
|
|||
|
|
record = append(record,
|
|||
|
|
v.CreateTime,
|
|||
|
|
strconv.Itoa(int(v.ID)),
|
|||
|
|
v.Title,
|
|||
|
|
v.Link,
|
|||
|
|
fmt.Sprintf("%0.2f", v.TotalMoney),
|
|||
|
|
strconv.Itoa(int(v.MemberCount)),
|
|||
|
|
v.PayDate,
|
|||
|
|
StateToString(int(v.State)),
|
|||
|
|
)
|
|||
|
|
writer.Write(record)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//CopyFromActivityDB copy from db
|
|||
|
|
func (q *QueryActivityInfo) CopyFromActivityDB(v *OfflineActivityInfo) {
|
|||
|
|
q.OfflineActivityInfo = *v
|
|||
|
|
q.CreateTime = v.CTime.Time().Format(dateFmt)
|
|||
|
|
q.PayDate = q.CreateTime
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//QueryUpBonusByMidArg arg
|
|||
|
|
type QueryUpBonusByMidArg struct {
|
|||
|
|
Mid int64 `form:"mid"` // activity id, if not 0, FromDate and toDate not used
|
|||
|
|
PageArg
|
|||
|
|
ExportArg
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//UpSummaryBonusInfo bonus for one up info
|
|||
|
|
type UpSummaryBonusInfo struct {
|
|||
|
|
Mid int64 `json:"mid"`
|
|||
|
|
BilledMoney float64 `json:"billed_money"`
|
|||
|
|
UnbilledMoney float64 `json:"unbilled_money"`
|
|||
|
|
LastBillTime string `json:"last_bill_time"` // 20180101， 最近结算时间
|
|||
|
|
TmpBillTime time.Time `json:"-"` // 用来计算LastBillTime
|
|||
|
|
TotalBonusMoney float64 `json:"total_bonus_money"` // 所有的中奖金额
|
|||
|
|
ActivityID int64 `json:"activity_id"` // 中奖活动ID
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//Finish finish it before send back
|
|||
|
|
func (s *UpSummaryBonusInfo) Finish() {
|
|||
|
|
if s.TmpBillTime > 0 {
|
|||
|
|
s.LastBillTime = s.TmpBillTime.Time().Format(dateFmt)
|
|||
|
|
}
|
|||
|
|
s.BilledMoney = ToFixed(s.BilledMoney, 2)
|
|||
|
|
s.UnbilledMoney = ToFixed(s.UnbilledMoney, 2)
|
|||
|
|
s.TotalBonusMoney = ToFixed(s.TotalBonusMoney, 2)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//QueryUpBonusByMidResult query result
|
|||
|
|
type QueryUpBonusByMidResult struct {
|
|||
|
|
Result []*UpSummaryBonusInfo `json:"result"`
|
|||
|
|
PageResult
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//GetFileName get file name
|
|||
|
|
func (q *QueryUpBonusByMidResult) GetFileName() string {
|
|||
|
|
return fmt.Sprintf("%s-%s_%s.csv", filenamePre, "结算记录", time2.Now().Format(dateTimeFmt))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ToCsv to buffer
|
|||
|
|
func (q *QueryUpBonusByMidResult) ToCsv(writer *csv.Writer) {
|
|||
|
|
var title = []string{
|
|||
|
|
"UID",
|
|||
|
|
"已结算",
|
|||
|
|
"未结算",
|
|||
|
|
"最近结算时间"}
|
|||
|
|
writer.Write(title)
|
|||
|
|
if q == nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
for _, v := range q.Result {
|
|||
|
|
var record []string
|
|||
|
|
record = append(record,
|
|||
|
|
intFormat(v.Mid),
|
|||
|
|
floatFormat(v.BilledMoney),
|
|||
|
|
floatFormat(v.UnbilledMoney),
|
|||
|
|
v.LastBillTime,
|
|||
|
|
)
|
|||
|
|
writer.Write(record)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//QueryUpBonusByActivityResult use same result, we define another type, because we need to implement interface again
|
|||
|
|
type QueryUpBonusByActivityResult QueryUpBonusByMidResult
|
|||
|
|
|
|||
|
|
//GetFileName get file name
|
|||
|
|
func (q *QueryUpBonusByActivityResult) GetFileName() string {
|
|||
|
|
return fmt.Sprintf("%s-%s_%s.csv", filenamePre, "详细信息", time2.Now().Format(dateTimeFmt))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ToCsv to buffer
|
|||
|
|
func (q *QueryUpBonusByActivityResult) ToCsv(writer *csv.Writer) {
|
|||
|
|
var title = []string{
|
|||
|
|
"UID",
|
|||
|
|
"中奖活动id",
|
|||
|
|
"中奖金额",
|
|||
|
|
"结算日期",
|
|||
|
|
"结算收入",
|
|||
|
|
"未结算收入"}
|
|||
|
|
writer.Write(title)
|
|||
|
|
if q == nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
for _, v := range q.Result {
|
|||
|
|
var record []string
|
|||
|
|
record = append(record,
|
|||
|
|
intFormat(v.Mid),
|
|||
|
|
intFormat(v.ActivityID),
|
|||
|
|
floatFormat(v.TotalBonusMoney),
|
|||
|
|
v.LastBillTime,
|
|||
|
|
floatFormat(v.BilledMoney),
|
|||
|
|
floatFormat(v.UnbilledMoney),
|
|||
|
|
)
|
|||
|
|
writer.Write(record)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//QueryActvityMonthArg arg
|
|||
|
|
type QueryActvityMonthArg struct {
|
|||
|
|
ExportArg
|
|||
|
|
PageArg
|
|||
|
|
FromDate string `form:"from_date"`
|
|||
|
|
ToDate string `form:"to_date"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ActivityMonthInfo info
|
|||
|
|
type ActivityMonthInfo struct {
|
|||
|
|
CreateTime string `json:"create_time"`
|
|||
|
|
ActivityCount int64 `json:"activity_count"`
|
|||
|
|
MemberCount uint32 `json:"member_count"`
|
|||
|
|
AverageMoneyMember float64 `json:"average_money_member"`
|
|||
|
|
AverageMoneyActivity float64 `json:"average_money_activity"`
|
|||
|
|
TotalBonusMoneyMonth float64 `json:"total_bonus_money"` // 所有的中奖金额
|
|||
|
|
TotalMoneyAccumulate float64 `json:"total_money_accumulate"`
|
|||
|
|
GenerateDay string `json:"generate_day"` // 生成日期
|
|||
|
|
|
|||
|
|
activityMap map[int64]struct{} // 用来保存活动数
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//AddBonus 增加bonus信息
|
|||
|
|
func (s *ActivityMonthInfo) AddBonus(v *OfflineActivityBonus) {
|
|||
|
|
s.MemberCount += v.MemberCount
|
|||
|
|
s.TotalBonusMoneyMonth += GetMoneyFromDb(v.TotalMoney)
|
|||
|
|
|
|||
|
|
if s.activityMap == nil {
|
|||
|
|
s.activityMap = make(map[int64]struct{})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
s.activityMap[v.ActivityID] = struct{}{}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//Finish finish
|
|||
|
|
func (s *ActivityMonthInfo) Finish() {
|
|||
|
|
if s.MemberCount > 0 {
|
|||
|
|
s.AverageMoneyMember = ToFixed(s.TotalBonusMoneyMonth/float64(s.MemberCount), 2)
|
|||
|
|
}
|
|||
|
|
s.ActivityCount = int64(len(s.activityMap))
|
|||
|
|
if s.ActivityCount > 0 {
|
|||
|
|
s.AverageMoneyActivity = ToFixed(s.TotalBonusMoneyMonth/float64(s.ActivityCount), 2)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//QueryActivityMonthResult result
|
|||
|
|
type QueryActivityMonthResult struct {
|
|||
|
|
PageResult
|
|||
|
|
Result []*ActivityMonthInfo `json:"result"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//GetFileName get file name
|
|||
|
|
func (q *QueryActivityMonthResult) GetFileName() string {
|
|||
|
|
return fmt.Sprintf("%s-%s_%s.csv", filenamePre, "按月", time2.Now().Format(dateTimeFmt))
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func floatFormat(f float64) string {
|
|||
|
|
return strconv.FormatFloat(f, 'f', 2, 64)
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
func intFormat(i int64) string {
|
|||
|
|
return strconv.Itoa(int(i))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ToCsv to csv
|
|||
|
|
func (q *QueryActivityMonthResult) ToCsv(writer *csv.Writer) {
|
|||
|
|
var title = []string{
|
|||
|
|
"月份",
|
|||
|
|
"最新数据截止日期",
|
|||
|
|
"月支出",
|
|||
|
|
"活动数",
|
|||
|
|
"中奖人数",
|
|||
|
|
"人均支出",
|
|||
|
|
"活动均支出",
|
|||
|
|
"累计支出"}
|
|||
|
|
writer.Write(title)
|
|||
|
|
if q == nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
for _, v := range q.Result {
|
|||
|
|
var record []string
|
|||
|
|
record = append(record,
|
|||
|
|
v.CreateTime,
|
|||
|
|
v.GenerateDay,
|
|||
|
|
floatFormat(v.TotalBonusMoneyMonth),
|
|||
|
|
intFormat(v.ActivityCount),
|
|||
|
|
intFormat(int64(v.MemberCount)),
|
|||
|
|
floatFormat(v.AverageMoneyMember),
|
|||
|
|
floatFormat(v.AverageMoneyActivity),
|
|||
|
|
floatFormat(v.TotalMoneyAccumulate),
|
|||
|
|
)
|
|||
|
|
writer.Write(record)
|
|||
|
|
}
|
|||
|
|
}
|