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  }