125 lines
3.5 KiB
Go
125 lines
3.5 KiB
Go
package statctrl
|
|
|
|
import (
|
|
"91porn-server/common"
|
|
"91porn-server/common/stderr"
|
|
"91porn-server/web/service/statser"
|
|
"bytes"
|
|
"encoding/csv"
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func DataStat(ctx *gin.Context) {
|
|
var args struct {
|
|
Start time.Time `form:"startTime" json:"startTime" binding:"required"`
|
|
End time.Time `form:"endTime" json:"endTime" binding:"required"`
|
|
}
|
|
if err := ctx.ShouldBind(&args); err != nil {
|
|
common.ServeJSON(ctx, stderr.ErrParamError, "stat RchSucRate arg error "+err.Error())
|
|
return
|
|
}
|
|
res, err, titles := statser.DataStat(args.Start, args.End)
|
|
if err != nil {
|
|
common.ServeJSON(ctx, stderr.Failure, "DataStat DataStat error: "+err.Error())
|
|
return
|
|
}
|
|
var (
|
|
headList []string
|
|
dataBytes = new(bytes.Buffer)
|
|
)
|
|
headList = []string{"时间", "楼凤", "vip充值", "金币充值", "新增用户"}
|
|
for i := range titles {
|
|
headList = append(headList, titles[i]+"金额", titles[i]+"-数量")
|
|
}
|
|
headList = append(headList, titles...)
|
|
//设置编码格式
|
|
dataBytes.WriteString("\xEF\xBB\xBF")
|
|
wr := csv.NewWriter(dataBytes)
|
|
wr.Write(headList)
|
|
for _, v := range res {
|
|
bodyList := []string{
|
|
v.SumData,
|
|
strconv.FormatInt(v.LoufengTrade, 10),
|
|
strconv.FormatInt(v.VIPRecharge, 10),
|
|
strconv.FormatInt(v.CoinRecharge, 10),
|
|
strconv.FormatInt(v.NewUser, 10),
|
|
}
|
|
vipTrans := make([]string, 0)
|
|
for i := range titles {
|
|
if len(v.VipTrande) < 1 {
|
|
vipTrans = append(vipTrans, "0", "0")
|
|
} else if vipTrande, ok := v.VipTrande[titles[i]]; ok {
|
|
vipTrans = append(vipTrans, fmt.Sprintf("%d", vipTrande.Amount), fmt.Sprintf("%d", vipTrande.Count))
|
|
} else {
|
|
vipTrans = append(vipTrans, "0", "0")
|
|
}
|
|
}
|
|
bodyList = append(bodyList, vipTrans...)
|
|
wr.Write(bodyList)
|
|
}
|
|
//清空
|
|
wr.Flush()
|
|
ctx.Writer.Header().Set("Content-type", "application/octet-stream")
|
|
//c.Writer.Header().Set("Content-Type", "text/csv")
|
|
ctx.Writer.Header().Set("Content-Disposition", fmt.Sprintf("attachment;filename=%s", "companys.csv"))
|
|
ctx.String(200, dataBytes.String())
|
|
return
|
|
}
|
|
func VidStat(ctx *gin.Context) {
|
|
var args struct {
|
|
Start time.Time `form:"startTime" json:"startTime" binding:"required"`
|
|
End time.Time `form:"endTime" json:"endTime" binding:"required"`
|
|
}
|
|
if err := ctx.ShouldBind(&args); err != nil {
|
|
common.ServeJSON(ctx, stderr.ErrParamError, "stat RchSucRate arg error "+err.Error())
|
|
return
|
|
}
|
|
var (
|
|
headList []string
|
|
dataBytes = new(bytes.Buffer)
|
|
)
|
|
headList = []string{"时间", "短视频销售额", "短视频数量", "长视频销售额", "长视频数量"}
|
|
//设置编码格式
|
|
dataBytes.WriteString("\xEF\xBB\xBF")
|
|
wr := csv.NewWriter(dataBytes)
|
|
wr.Write(headList)
|
|
start := args.Start
|
|
end := args.End
|
|
|
|
for start.Before(end) {
|
|
head := start
|
|
tail := head.Add(time.Hour * 24 * 15)
|
|
if tail.After(end) {
|
|
tail = end
|
|
}
|
|
res, err := statser.PayVidStat(head, tail)
|
|
if err != nil {
|
|
common.ServeJSON(ctx, stderr.Failure, "statser PayVidStat error: "+err.Error())
|
|
return
|
|
}
|
|
|
|
for _, v := range res {
|
|
bodyList := []string{
|
|
v.SumData,
|
|
strconv.FormatInt(v.ShortVidAmount, 10),
|
|
strconv.FormatInt(v.ShortVidCount, 10),
|
|
strconv.FormatInt(v.LongVidAmount, 10),
|
|
strconv.FormatInt(v.LongVidCount, 10),
|
|
}
|
|
wr.Write(bodyList)
|
|
}
|
|
start = tail
|
|
}
|
|
|
|
//清空
|
|
wr.Flush()
|
|
ctx.Writer.Header().Set("Content-type", "application/octet-stream")
|
|
ctx.Writer.Header().Set("Content-Disposition", fmt.Sprintf("attachment;filename=%s", "videoSales.csv"))
|
|
ctx.String(200, dataBytes.String())
|
|
return
|
|
}
|