package db

import (
	"go_dreamfactory/lego/core"
	"go_dreamfactory/pb"

	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

const (
	DB_FriendTable core.SqlTable = "friend"
)

type IFriend interface {
	Friend_Apply(data *pb.Cache_FriendData) (err error)
}

//好友申请
func (this *DB) Friend_Apply(data *pb.Cache_FriendData) (err error) {
	err = this.mgo.FindOneAndUpdate(DB_FriendTable,
		bson.M{"_id": data.UserId},
		bson.M{"$set": bson.M{
			"friendids": data.FriendIds,
			"applyids":  data.ApplyIds}},
		options.FindOneAndUpdate().SetUpsert(true)).Err()
	if err != nil {
		if err == mongo.ErrNoDocuments {
			_, err = this.mgo.InsertOne(DB_FriendTable, data)
		}
	}
	return
}