92 lines
2.2 KiB
Go
92 lines
2.2 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 (
|
|
FriendDBName = "user"
|
|
FriendCollName = "user_friend"
|
|
FriendColError = errors.New("friend collection open failed")
|
|
)
|
|
|
|
func FriendCollection(plt string) *mongo.Collection {
|
|
s := mongo.MgoSessionMgrSington.GetPltMgoSession(plt, FriendDBName)
|
|
if s != nil {
|
|
c, first := s.DB().C(FriendCollName)
|
|
if first {
|
|
c.EnsureIndex(mgo.Index{Key: []string{"snid"}, Background: true, Sparse: true})
|
|
c.EnsureIndex(mgo.Index{Key: []string{"platform"}, Background: true, Sparse: true})
|
|
}
|
|
return c
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type FriendSvc struct {
|
|
}
|
|
|
|
func (svc *FriendSvc) UpsertFriend(args *model.Friend, ret *model.FriendRet) error {
|
|
fc := FriendCollection(args.Platform)
|
|
if fc == nil {
|
|
return FriendColError
|
|
}
|
|
if args.Id == "" {
|
|
args.Id = bson.NewObjectId()
|
|
}
|
|
_, err := fc.Upsert(bson.M{"platform": args.Platform, "snid": args.SnId}, args)
|
|
if err != nil && err != mgo.ErrNotFound {
|
|
logger.Logger.Error("UpsertFriend is err: ", err)
|
|
return err
|
|
}
|
|
ret.Fri = args
|
|
return nil
|
|
}
|
|
|
|
func (svc *FriendSvc) QueryFriendByKey(args *model.FriendByKey, ret *model.FriendRet) error {
|
|
fc := FriendCollection(args.Platform)
|
|
if fc == nil {
|
|
return FriendColError
|
|
}
|
|
if ret == nil {
|
|
ret = &model.FriendRet{}
|
|
ret.Fri = &model.Friend{}
|
|
}
|
|
err := fc.Find(bson.M{"platform": args.Platform, "snid": args.SnId}).One(&ret.Fri)
|
|
if err != nil && !errors.Is(err, mgo.ErrNotFound) {
|
|
logger.Logger.Error("QueryFriendByKey is err: ", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
func (svc *FriendSvc) QueryFriendsByKey(args *model.FriendsByKey, ret *model.FriendsRet) error {
|
|
fc := FriendCollection(args.Platform)
|
|
if fc == nil {
|
|
return FriendColError
|
|
}
|
|
var sql []bson.M
|
|
if len(args.SnIds) != 0 {
|
|
for _, snid := range args.SnIds {
|
|
sql = append(sql, bson.M{"snid": snid})
|
|
}
|
|
}
|
|
err := fc.Find(bson.M{"$or": sql}).All(&ret.Fris)
|
|
if err != nil && err != mgo.ErrNotFound {
|
|
logger.Logger.Error("QueryFriendsByKey is err: ", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var _FriendSvc = &FriendSvc{}
|
|
|
|
func init() {
|
|
rpc.Register(_FriendSvc)
|
|
}
|