286 lines
9.2 KiB
Go
286 lines
9.2 KiB
Go
|
package dao
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/jinzhu/gorm"
|
|||
|
|
"go-common/app/admin/main/growup/dao/shell"
|
|||
|
|
"go-common/app/admin/main/growup/model/offlineactivity"
|
|||
|
|
"go-common/app/admin/main/up/util"
|
|||
|
|
"go-common/library/log"
|
|||
|
|
"strconv"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func generateDelimiter(delimiter string) func(r rune) bool {
|
|||
|
|
return func(r rune) bool {
|
|||
|
|
for _, v := range delimiter {
|
|||
|
|
if v == r {
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
trimSet = "\r\n "
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func strlistToInt64List(list []string, trimSet string) (result []int64) {
|
|||
|
|
for _, v := range list {
|
|||
|
|
mid, e := strconv.ParseInt(strings.Trim(v, trimSet), 10, 64)
|
|||
|
|
if e != nil {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
result = append(result, mid)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func trimString(strlist []string, trimset string) (result []string) {
|
|||
|
|
for _, v := range strlist {
|
|||
|
|
var trimed = strings.Trim(v, trimSet)
|
|||
|
|
if trimed == "" {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
result = append(result, trimed)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ParseMidsFromString parse []int64 from string
|
|||
|
|
func ParseMidsFromString(str string) (result []int64) {
|
|||
|
|
var midstrlist = trimString(strings.FieldsFunc(str, generateDelimiter(",\r\n")), trimSet)
|
|||
|
|
result = util.Unique(strlistToInt64List(midstrlist, trimSet))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//OfflineActivityAddActivity add acitvity
|
|||
|
|
func (d *Dao) OfflineActivityAddActivity(ctx context.Context, arg *offlineactivity.AddActivityArg) (err error) {
|
|||
|
|
var activityInfo = offlineactivity.OfflineActivityInfo{
|
|||
|
|
Title: arg.Title,
|
|||
|
|
Link: arg.Link,
|
|||
|
|
BonusType: arg.BonusType,
|
|||
|
|
Memo: arg.Memo,
|
|||
|
|
State: int8(offlineactivity.ActivityStateInit),
|
|||
|
|
Creator: arg.Creator,
|
|||
|
|
}
|
|||
|
|
var bonusList = arg.BonusList
|
|||
|
|
var hasBonus = false
|
|||
|
|
for _, bonus := range bonusList {
|
|||
|
|
if bonus.TotalMoney <= 0 {
|
|||
|
|
err = fmt.Errorf("bonus money < 0, money=%f", bonus.TotalMoney)
|
|||
|
|
log.Error(err.Error())
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
bonus.MidList = ParseMidsFromString(bonus.Mids)
|
|||
|
|
bonus.MemberCount = int64(len(bonus.MidList))
|
|||
|
|
if bonus.MemberCount == 0 {
|
|||
|
|
log.Warn("no mid for this bonus, bonus：money=%.2f", bonus.TotalMoney)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
hasBonus = true
|
|||
|
|
log.Info("bonus info:money=%.2f, membercount=%d", bonus.TotalMoney/1000, bonus.MemberCount)
|
|||
|
|
}
|
|||
|
|
// 如果一个bonus都没有，则什么也不做
|
|||
|
|
if !hasBonus {
|
|||
|
|
log.Error("no bonus")
|
|||
|
|
err = fmt.Errorf("no bonus info")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if err = d.db.Create(&activityInfo).Error; err != nil {
|
|||
|
|
log.Error("err insert offline activity, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var tx = d.db.Begin()
|
|||
|
|
defer func() {
|
|||
|
|
if r := recover(); r != nil || err != nil {
|
|||
|
|
tx.Rollback()
|
|||
|
|
activityInfo.State = int8(offlineactivity.ActivityStateCreateFail)
|
|||
|
|
if err = d.db.Select("state").Save(&activityInfo).Error; err != nil {
|
|||
|
|
log.Error("err update offline activity, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
// create
|
|||
|
|
|
|||
|
|
for _, bonus := range bonusList {
|
|||
|
|
if bonus.MemberCount <= 0 {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
// 插入OfflineActivityBonus
|
|||
|
|
//if activityInfo.BonusType == offlineactivity.BonusTypeMoney {
|
|||
|
|
bonusMoney := offlineactivity.GetMoneyForDb(bonus.TotalMoney)
|
|||
|
|
//}
|
|||
|
|
var bonusInfo = offlineactivity.OfflineActivityBonus{
|
|||
|
|
TotalMoney: bonusMoney,
|
|||
|
|
MemberCount: uint32(bonus.MemberCount),
|
|||
|
|
State: int8(offlineactivity.BonusStateInit),
|
|||
|
|
ActivityID: activityInfo.ID,
|
|||
|
|
}
|
|||
|
|
if err = tx.Create(&bonusInfo).Error; err != nil {
|
|||
|
|
log.Error("err insert offline activity bonus, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 插入OfflineActivityResult
|
|||
|
|
var insertSchema []string
|
|||
|
|
var vals []interface{}
|
|||
|
|
var dbmoney = bonusInfo.TotalMoney / int64(bonusInfo.MemberCount)
|
|||
|
|
for _, mid := range bonus.MidList {
|
|||
|
|
insertSchema = append(insertSchema, "(?,?,?,?,?)")
|
|||
|
|
vals = append(vals, activityInfo.ID, bonusInfo.ID, activityInfo.BonusType, mid, dbmoney)
|
|||
|
|
}
|
|||
|
|
var insertPre = "insert into offline_activity_result (activity_id, bonus_id, bonus_type, mid, bonus_money) values "
|
|||
|
|
var insertSQL = insertPre + strings.Join(insertSchema, ",")
|
|||
|
|
if err = tx.Table(offlineactivity.TableOfflineActivityResult).Exec(insertSQL, vals...).Error; err != nil {
|
|||
|
|
log.Error("err insert offline activity result, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
activityInfo.State = int8(offlineactivity.ActivityStateSending)
|
|||
|
|
if arg.BonusType == int8(offlineactivity.BonusTypeThing) {
|
|||
|
|
activityInfo.State = int8(offlineactivity.ActivityStateSucess)
|
|||
|
|
}
|
|||
|
|
if err = tx.Select("state").Save(&activityInfo).Error; err != nil {
|
|||
|
|
log.Error("err update offline activity, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
return tx.Commit().Error
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ShellCallbackUpdate shell callback
|
|||
|
|
func (d *Dao) ShellCallbackUpdate(ctx context.Context, result *shell.OrderCallbackJSON, msgid string) (orderInfo offlineactivity.OfflineActivityShellOrder, err error) {
|
|||
|
|
|
|||
|
|
// 找到order info, 写入结果
|
|||
|
|
var stateResult = offlineactivity.ActivityStateWaitResult
|
|||
|
|
if result.IsSuccess() {
|
|||
|
|
stateResult = offlineactivity.ActivityStateSucess
|
|||
|
|
} else if result.IsFail() {
|
|||
|
|
stateResult = offlineactivity.ActivityStateFail
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err = d.db.Where("order_id=?", result.ThirdOrderNo).Find(&orderInfo).Error; err != nil {
|
|||
|
|
log.Error("order id=%s, find order id fail, err=%s", result.ThirdOrderNo, err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var tx = d.db.Begin()
|
|||
|
|
defer func() {
|
|||
|
|
if r := recover(); r != nil || err != nil {
|
|||
|
|
tx.Rollback()
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
if err = tx.Table(offlineactivity.TableOfflineActivityShellOrder).
|
|||
|
|
Where("id=?", orderInfo.ID).
|
|||
|
|
Update("order_status", result.Status).
|
|||
|
|
Error; err != nil {
|
|||
|
|
log.Error("order id=%s, update order fail, err=%s", result.ThirdOrderNo, err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err = tx.Table(offlineactivity.TableOfflineActivityResult).
|
|||
|
|
Where("id=?", orderInfo.ResultID).
|
|||
|
|
Update("state", stateResult).
|
|||
|
|
Error; err != nil {
|
|||
|
|
log.Error("order id=%s, update result fail, err=%s", result.ThirdOrderNo, err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
log.Info("order id=%s, shell callback, success, mid=%s, status=%s", result.ThirdOrderNo, result.Mid, result.Status)
|
|||
|
|
|
|||
|
|
err = tx.Commit().Error
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//OfflineActivityGetUpBonusResult Up主所有Activity一起统计
|
|||
|
|
func (d *Dao) OfflineActivityGetUpBonusResult(ctx context.Context, needCount bool, limit, offset int, query string, args ...interface{}) (upResult []*offlineactivity.OfflineActivityResult, totalCount int, err error) {
|
|||
|
|
// 查询所有的 result for mid
|
|||
|
|
// 区分已结算、未结算
|
|||
|
|
// 已结算= 成功， 未结算= 初始、发送、等待
|
|||
|
|
if needCount {
|
|||
|
|
if err = d.db.Table(offlineactivity.TableOfflineActivityResult).Where(query, args...).Count(&totalCount).Error; err != nil {
|
|||
|
|
log.Error("fail to get from db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err = d.db.Where(query, args...).Limit(limit).Offset(offset).Find(&upResult).Error; err != nil {
|
|||
|
|
log.Error("fail to get from db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//OfflineActivityGetUpBonusResultSelect select by selection
|
|||
|
|
func (d *Dao) OfflineActivityGetUpBonusResultSelect(ctx context.Context, selectQuery string, query string, args ...interface{}) (upResult []*offlineactivity.OfflineActivityResult, err error) {
|
|||
|
|
if err = d.db.Where(query, args...).Select(selectQuery).Find(&upResult).Error; err != nil {
|
|||
|
|
log.Error("fail to get from db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//OfflineActivityGetUpBonusByActivityResult 根据Activity来分别统计
|
|||
|
|
func (d *Dao) OfflineActivityGetUpBonusByActivityResult(ctx context.Context, limit, offset int, mid int64) (upResult []*offlineactivity.OfflineActivityResult, totalCount int, err error) {
|
|||
|
|
// 查询所有的 result for mid
|
|||
|
|
// 区分已结算、未结算
|
|||
|
|
// 已结算= 成功， 未结算= 初始、发送、等待
|
|||
|
|
var actividyIds []struct {
|
|||
|
|
ActivityID int64 `gorm:"column:activity_id"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err = d.db.Table(offlineactivity.TableOfflineActivityResult).Select("count(distinct(activity_id))").Where("mid=?", mid).Count(&totalCount).Error; err != nil {
|
|||
|
|
log.Error("fail to get from db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if err = d.db.Table(offlineactivity.TableOfflineActivityResult).Select("distinct(activity_id)").Where("mid=?", mid).Offset(offset).Limit(limit).Find(&actividyIds).Error; err != nil {
|
|||
|
|
log.Error("fail to get from db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
//totalCount = len(actividyIds)
|
|||
|
|
var ids []int64
|
|||
|
|
for i := 0; i < len(actividyIds); i++ {
|
|||
|
|
ids = append(ids, actividyIds[i].ActivityID)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err = d.db.Where("mid=? and activity_id in(?)", mid, ids).Find(&upResult).Error; err != nil {
|
|||
|
|
log.Error("fail to get from db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//OfflineActivityGetDB get gorm db
|
|||
|
|
func (d *Dao) OfflineActivityGetDB() *gorm.DB {
|
|||
|
|
return d.db
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//UpdateActivityState 更新Activity的状态，根据每一个付款记录的状态
|
|||
|
|
func (d *Dao) UpdateActivityState(ctx context.Context, activityID int64) (affectedRow int64, err error) {
|
|||
|
|
// 找到所有result中，state最小的一个
|
|||
|
|
// 更新到activity的状态
|
|||
|
|
var record offlineactivity.OfflineActivityResult
|
|||
|
|
if err = d.db.Model(&record).Select("min(state) as state").Where("activity_id=?", activityID).Find(&record).Error; err != nil {
|
|||
|
|
log.Error("fail to get from db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
var db = d.db.Table(offlineactivity.TableOfflineActivityInfo).
|
|||
|
|
Where("id=? and state<10 and state<?", activityID, record.State).
|
|||
|
|
Update("state", record.State)
|
|||
|
|
if err = db.Error; err != nil {
|
|||
|
|
log.Error("fail to get from db, err=%s", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
affectedRow = db.RowsAffected
|
|||
|
|
if affectedRow > 0 {
|
|||
|
|
log.Info("update activity state, id=%d, newstate=%d", activityID, record.State)
|
|||
|
|
} else {
|
|||
|
|
log.Info("no need to update activity, id=%d, newstate=%d", activityID, record.State)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|