94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package svc
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/globalsign/mgo"
|
|
"github.com/globalsign/mgo/bson"
|
|
"math"
|
|
"mongo.games.com/game/dbproxy/mongo"
|
|
"mongo.games.com/game/model"
|
|
"mongo.games.com/goserver/core/logger"
|
|
"net/rpc"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
RebateLogDBErr = errors.New("log_rebatetask db open failed.")
|
|
)
|
|
|
|
func RebateCollection(plt string) *mongo.Collection {
|
|
s := mongo.MgoSessionMgrSington.GetPltMgoSession(plt, model.RebateLogDBName)
|
|
if s != nil {
|
|
c, first := s.DB().C(model.RebateLogCollName)
|
|
if first {
|
|
c.EnsureIndex(mgo.Index{Key: []string{"id"}, Unique: true, Background: true, Sparse: true})
|
|
c.EnsureIndex(mgo.Index{Key: []string{"snid"}, Background: true, Sparse: true})
|
|
}
|
|
return c
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func InsertRebateLog(plt string, data *model.Rebate) error {
|
|
rebatec := RebateCollection(plt)
|
|
if rebatec == nil {
|
|
return nil
|
|
}
|
|
data.Id = bson.NewObjectId()
|
|
data.Ts = time.Now().Unix()
|
|
err := rebatec.Insert(data)
|
|
if err != nil {
|
|
logger.Logger.Error("Insert rebate data eror.", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func GetRebateLog(plt string, pageNo, pageSize int, snId int32) (r *model.RebateLog, err error) {
|
|
r = new(model.RebateLog)
|
|
rebatec := RebateCollection(plt)
|
|
if rebatec == nil {
|
|
return
|
|
}
|
|
total, err := rebatec.Find(bson.M{"snid": snId}).Count()
|
|
if err != nil || total == 0 {
|
|
return nil, err
|
|
}
|
|
r.PageSum = int(math.Ceil(float64(total) / float64(pageSize)))
|
|
if pageNo > r.PageSum {
|
|
pageNo = r.PageSum
|
|
}
|
|
if pageNo <= 0 {
|
|
pageNo = 1
|
|
}
|
|
limitNum := (pageNo - 1) * pageSize
|
|
err = rebatec.Find(bson.M{"snid": snId}).Skip(limitNum).Limit(pageSize).Sort("-ts").All(&r.Rebates)
|
|
if err != nil {
|
|
logger.Logger.Error("Find rebate data eror.", err)
|
|
return nil, err
|
|
}
|
|
r.PageNo = pageNo
|
|
r.PageSize = pageSize
|
|
return
|
|
}
|
|
|
|
type RebateLogSvc struct {
|
|
}
|
|
|
|
func (svc *RebateLogSvc) InsertRebateLog(log *model.BankBindLog, ret *bool) error {
|
|
err := InsertBankBindLog(log)
|
|
if err == nil {
|
|
*ret = true
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (svc *RebateLogSvc) GetRebateLog(args *model.GetRebateLogArgs, ret **model.RebateLog) (err error) {
|
|
*ret, err = GetRebateLog(args.Plt, args.PageNo, args.PageSize, args.SnId)
|
|
return err
|
|
}
|
|
|
|
func init() {
|
|
rpc.Register(new(RebateLogSvc))
|
|
}
|