github.com/infraboard/keyauth@v0.8.1/apps/token/impl/dao.go (about) 1 package impl 2 3 import ( 4 "context" 5 6 "github.com/infraboard/mcube/exception" 7 "go.mongodb.org/mongo-driver/bson" 8 9 "github.com/infraboard/keyauth/apps/token" 10 ) 11 12 func (s *service) saveToken(tk *token.Token) error { 13 if _, err := s.col.InsertOne(context.TODO(), tk); err != nil { 14 return exception.NewInternalServerError("inserted token(%s) document error, %s", 15 tk.AccessToken, err) 16 } 17 18 return nil 19 } 20 21 func (s *service) updateToken(tk *token.Token) error { 22 _, err := s.col.UpdateOne(context.TODO(), bson.M{"_id": tk.AccessToken}, bson.M{"$set": tk}) 23 if err != nil { 24 return exception.NewInternalServerError("update token(%s) error, %s", tk.AccessToken, err) 25 } 26 27 return nil 28 }