package pagoda

import (
	"context"
	"go_dreamfactory/comm"
	"go_dreamfactory/lego/core"
	"go_dreamfactory/lego/sys/redis"
	"go_dreamfactory/modules"
	"go_dreamfactory/pb"

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

type ModelRank struct {
	modules.MCompModel
	moduleUser *Pagoda
}

func (this *ModelRank) Init(service core.IService, module core.IModule, comp core.IModuleComp, options core.IModuleOptions) (err error) {
	this.TableName = comm.TablePagodaRank
	err = this.MCompModel.Init(service, module, comp, options)
	this.moduleUser = module.(*Pagoda)
	return
}

//获取用户
func (this *ModelRank) getUserSession(uid string) (cuser *pb.CacheUser) {
	cuser = &pb.CacheUser{}
	if err := this.Get(uid, cuser); err != nil {
		this.moduleUser.Errorf("GetUserSession err:%v", err)
		return
	}
	return
}

func (this *ModelRank) AddRank(uId string, data *pb.DBPagodaRank) (err error) {
	if err = this.Add(uId, data); err != nil {
		//this.Errorf("err:%v", err)
		return
	}
	return nil
}

//获取用户通过扩展表
func (this *ModelRank) GetUserRandData(uid string) (result *pb.DBPagodaRank, err error) {
	result = &pb.DBPagodaRank{}
	if err = this.Get(uid, result); err != nil && redis.RedisNil != err {
		return
	}
	err = nil
	return result, err
}

func (this *ModelRank) ChangeUserRank(uid string, value map[string]interface{}) (err error) {
	if len(value) == 0 {
		return nil
	}
	return this.Change(uid, value)
}

// 临时测试
func (this *ModelRank) GetRankData(floorId int32, pagodaType int32) (data []*pb.DBPagodaRank, err error) {
	if floorId == 0 {
		if _data, err := this.DB.Find(comm.TablePagodaRank, bson.M{"type": pagodaType}, options.Find().SetSort(bson.M{"pagodaId": -1})); err == nil {
			for _data.Next(context.TODO()) {
				temp := &pb.DBPagodaRank{}
				if err = _data.Decode(temp); err == nil {
					data = append(data, temp)
				}
			}
		}
	} else {
		if _data, err := this.DB.Find(comm.TablePagodaRank, bson.M{"pagodaId": bson.M{"$gte": floorId}, "type": pagodaType}, options.Find().SetSort(bson.M{"pagodaId": -1})); err == nil {
			for _data.Next(context.TODO()) {
				temp := &pb.DBPagodaRank{}
				if err = _data.Decode(temp); err == nil {
					data = append(data, temp)
				}
			}
		}
	}

	return
}