diff --git a/tokenMongoStore.go b/tokenMongoStore.go index 0edaa73..eeeefe7 100644 --- a/tokenMongoStore.go +++ b/tokenMongoStore.go @@ -22,7 +22,7 @@ func NewTokenMongoStore(mongoConfig *MongoConfig, cName string) (TokenStore, err if cName == "" { cName = DefaultTokenCollectionName } - err = mHandler.C(cName).EnsureIndexKey("ID", "AccessToken", "RefreshToken") + err = mHandler.C(cName).EnsureIndexKey("AccessToken", "RefreshToken") if err != nil { return nil, err } @@ -58,7 +58,7 @@ func (tm *TokenMongoStore) Create(item *Token) (id int64, err error) { // Update Modify item func (tm *TokenMongoStore) Update(id int64, info map[string]interface{}) (err error) { tm.mHandler.CHandle(tm.cName, func(c *mgo.Collection) { - err = c.Update(bson.M{"ID": id}, bson.M{"$set": info}) + err = c.UpdateId(id, bson.M{"$set": info}) if err != nil { return } @@ -69,7 +69,7 @@ func (tm *TokenMongoStore) Update(id int64, info map[string]interface{}) (err er func (tm *TokenMongoStore) findOne(query interface{}) (token *Token, err error) { tm.mHandler.CHandle(tm.cName, func(c *mgo.Collection) { var result []Token - err = c.Find(query).Sort("-ID").Limit(1).All(&result) + err = c.Find(query).Sort("-_id").Limit(1).All(&result) if err != nil { return } diff --git a/tokenStore.go b/tokenStore.go index cee1458..dbbf3f7 100644 --- a/tokenStore.go +++ b/tokenStore.go @@ -6,7 +6,7 @@ import ( // Token 令牌信息 type Token struct { - ID int64 `bson:"ID"` // 唯一标识(自增ID) + ID int64 `bson:"_id"` // 唯一标识(自增ID) ClientID string `bson:"ClientID"` // 客户端标识 UserID string `bson:"UserID"` // 用户标识 AccessToken string `bson:"AccessToken"` // 访问令牌