73 lines
1.9 KiB
Go
73 lines
1.9 KiB
Go
package svc
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/globalsign/mgo"
|
|
"github.com/globalsign/mgo/bson"
|
|
"mongo.games.com/game/dbproxy/mongo"
|
|
"mongo.games.com/game/model"
|
|
"mongo.games.com/goserver/core/logger"
|
|
"net/rpc"
|
|
)
|
|
|
|
var (
|
|
RankPlayerPermitDBName = "log"
|
|
RankPlayerPermitCollName = "log_rankplayerpermit"
|
|
RankPlayerPermitColError = errors.New("RankPlayerPermit collection open failed")
|
|
)
|
|
|
|
func RankPlayerPermitCollection(plt string) *mongo.Collection {
|
|
s := mongo.MgoSessionMgrSington.GetPltMgoSession(plt, RankPlayerPermitDBName)
|
|
if s != nil {
|
|
c, first := s.DB().C(RankPlayerPermitCollName)
|
|
if first {
|
|
c.EnsureIndex(mgo.Index{Key: []string{"snid"}, Background: true, Sparse: true})
|
|
c.EnsureIndex(mgo.Index{Key: []string{"startts", "-exp", "-ts"}, Background: true, Sparse: true})
|
|
}
|
|
return c
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func RankPlayerPermitUpsert(args *model.PermitScore) error {
|
|
cc := RankPlayerPermitCollection(args.Platform)
|
|
if cc == nil {
|
|
return RankPlayerPermitColError
|
|
}
|
|
_, err := cc.Upsert(bson.M{"startts": args.StartTs, "snid": args.SnId}, args)
|
|
if err != nil && !errors.Is(err, mgo.ErrNotFound) {
|
|
logger.Logger.Error("RankPlayerPermitSvc.Upsert is err: ", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type RankPlayerPermitSvc struct {
|
|
}
|
|
|
|
func (svc *RankPlayerPermitSvc) Upsert(args *model.PermitScore, ret *bool) error {
|
|
err := RankPlayerPermitUpsert(args)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*ret = true
|
|
return nil
|
|
}
|
|
|
|
func (svc *RankPlayerPermitSvc) Find(args *model.FindPlayerPermitListArgs, ret *model.FindPlayerPermitListReply) error {
|
|
fc := RankPlayerPermitCollection(args.Platform)
|
|
if fc == nil {
|
|
return RankPlayerPermitColError
|
|
}
|
|
|
|
err := fc.Find(bson.M{"startts": args.StartTs}).Sort("-exp", "-ts").Limit(int(model.GameParamData.RankPlayerPermitMaxNum)).All(&ret.List)
|
|
if err != nil && !errors.Is(err, mgo.ErrNotFound) {
|
|
logger.Logger.Error("QueryPermit is err: ", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
func init() {
|
|
rpc.Register(new(RankPlayerPermitSvc))
|
|
}
|