package db import ( "errors" cherryApp "f1-game/internal/cherry_app" dbFacade "f1-game/internal/component/db/facade" nameDB "f1-game/internal/name/db" dbQuest "f1-game/nodes/game/internal/db/data/quest" clog "github.com/cherry-game/cherry/logger" "go.mongodb.org/mongo-driver/v2/bson" "go.mongodb.org/mongo-driver/v2/mongo" "go.mongodb.org/mongo-driver/v2/mongo/options" ) // QuestTable 任务表 type QuestTable struct { PlayerID int64 `bson:"PlayerID,omitempty"` NodeID string `bson:"NodeID,omitempty"` dbQuest.Data `bson:"Data,omitempty"` dbFacade.TableBase `bson:",inline"` } func (*QuestTable) TableName() string { return nameDB.Table_Quest } func (p *QuestTable) PrimaryValue() any { return p.PlayerID } func (p *QuestTable) CheckField() { } func (p *QuestTable) Indexes() []mongo.IndexModel { var list []mongo.IndexModel list = append(list, mongo.IndexModel{ Keys: bson.D{ { Key: nameDB.Field_PlayerID, Value: 1, // 升序 }, { Key: nameDB.Field_NodeID, Value: 1, // 升序 }, }, Options: options.Index().SetUnique(true), }) return list } func (p *QuestTable) Save2Queue() { save2Queue(p) } func GetQuestTable(playerID int64) *QuestTable { if val, ok := questTableCache.GetIfPresent(playerID); ok { return val.(*QuestTable) } questTable := GetQuestTableFromDB(playerID) questTableCache.Put(playerID, questTable) return questTable } func GetQuestTableFromDB(playerID int64) *QuestTable { table := newQuestTable(playerID) filter := bson.M{ nameDB.Field_NodeID: table.NodeID, nameDB.Field_PlayerID: table.PlayerID, } err := dbComponent.FindOne(table, filter) if err != nil { if errors.Is(err, mongo.ErrNoDocuments) { clog.Warnf("[playerID = %d] not found, create new one", playerID) return table } clog.Errorf("[playerID = %d] db error = %v", playerID, err) return table } table.CheckField() return table } func newQuestTable(playerID int64) *QuestTable { return &QuestTable{ PlayerID: playerID, NodeID: cherryApp.NodeID(), Data: dbQuest.NewData(), } }