Create & Init Project...
This commit is contained in:
285
app/admin/main/growup/dao/offlineactivity.go
Normal file
285
app/admin/main/growup/dao/offlineactivity.go
Normal file
@ -0,0 +1,285 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user