70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package svc
|
|
|
|
import (
|
|
"errors"
|
|
"net/rpc"
|
|
|
|
"github.com/globalsign/mgo"
|
|
"github.com/globalsign/mgo/bson"
|
|
"mongo.games.com/goserver/core/logger"
|
|
|
|
"mongo.games.com/game/common"
|
|
"mongo.games.com/game/dbproxy/mongo"
|
|
"mongo.games.com/game/model"
|
|
)
|
|
|
|
var (
|
|
RankInviteDBName = "log"
|
|
RankInviteCollName = "log_rankinvite"
|
|
RankInviteColError = errors.New("RankInvite collection open failed")
|
|
)
|
|
|
|
func RankInviteCollection(plt string) *mongo.Collection {
|
|
s := mongo.MgoSessionMgrSington.GetPltMgoSession(plt, RankInviteDBName)
|
|
if s != nil {
|
|
c, first := s.DB().C(RankInviteCollName)
|
|
if first {
|
|
c.EnsureIndex(mgo.Index{Key: []string{"snid"}, Background: true, Sparse: true})
|
|
c.EnsureIndex(mgo.Index{Key: []string{"num"}, Background: true, Sparse: true})
|
|
c.EnsureIndex(mgo.Index{Key: []string{"-score"}, Background: true, Sparse: true})
|
|
c.EnsureIndex(mgo.Index{Key: []string{"score"}, Background: true, Sparse: true})
|
|
c.EnsureIndex(mgo.Index{Key: []string{"ts"}, Background: true, Sparse: true})
|
|
c.EnsureIndex(mgo.Index{Key: []string{"week", "-score"}, Background: true, Sparse: true})
|
|
}
|
|
return c
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func SaveRankInvite(data *model.RankInvite) error {
|
|
c := RankInviteCollection(data.Platform)
|
|
if c == nil {
|
|
return RankInviteColError
|
|
}
|
|
|
|
data.Week = common.GetWeekStartTs(data.Ts)
|
|
|
|
_, err := c.Upsert(bson.M{"snid": data.SnId, "week": data.Week}, data)
|
|
if err != nil {
|
|
logger.Logger.Tracef("SaveRankInvite error:%v", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type RankInviteSvc struct {
|
|
}
|
|
|
|
func (svc *RankInviteSvc) SaveRankInvite(data *model.RankInvite, ret *bool) (err error) {
|
|
err = SaveRankInvite(data)
|
|
if err == nil {
|
|
*ret = true
|
|
}
|
|
return
|
|
}
|
|
|
|
func init() {
|
|
rpc.Register(new(RankInviteSvc))
|
|
}
|