給出了這個 golang 代碼,因為我的 golang 生銹了,所以很難自由批評?;镜乃⑿铝钆屏?,但是我們將函數(shù)包裝在 TX 函數(shù)中,以便在刷新流出現(xiàn)問題時允許回滾。我們需要返回該newToken值,但是refresh當(dāng)它返回 refreshToken 時它只保存函數(shù)期間我們需要的值,具體取決于我傳遞給它的對象或指針是否分別返回新的空標(biāo)記或 nil。package serviceimport ( "context" "database/sql" "errors" "strings" "github.com/api-oauth/pkg/models" "github.com/api-oauth/pkg/repository" "golang.org/x/oauth2")const minParts = 2type OAuthAPI interface { Exchange(context.Context, string) (*oauth2.Token, error) Refresh(context.Context, string) (*oauth2.Token, error)}type Service struct { repo repository.RepoInterface oauthAPI OAuthAPI}func New(api OAuthAPI, repo repository.RepoInterface) *Service { return &Service{repo: repo, oauthAPI: api}}var ( ErrNoRefreshTokenFound = errors.New("no refresh token found") errNoScopesReturn = errors.New("no scopes returned") errNoProfileFound = errors.New("no profile found"))func (s *Service) RefreshToken(ctx context.Context, clientID string, userID int64) (*oauth2.Token, error) { var newToken *oauth2.Token err := s.repo.InTx(s.refresh(ctx, clientID, userID, newToken)) if err != nil { return nil, err } return newToken, nil}func (s *Service) refresh(ctx context.Context, clientID string, userID int64, newToken *oauth2.Token) func(repo repository.Auth) error { return func(repo repository.Auth) error { oldToken, err := repo.GetToken(ctx, userID, clientID) if err != nil && err != sql.ErrNoRows { return err } if oldToken == nil { return ErrNoRefreshTokenFound } newToken, err = s.oauthAPI.Refresh(ctx, oldToken.Refresh) if err != nil { return err } }}試圖RefreshToken返回更新的令牌信息
1 回答

慕婉清6462132
TA貢獻(xiàn)1804條經(jīng)驗 獲得超2個贊
通過地址newToken:
func (s *Service) RefreshToken(ctx context.Context, clientID string, userID int64) (*oauth2.Token, error) {
var newToken *oauth2.Token
err := s.repo.InTx(s.refresh(ctx, clientID, userID, &newToken))
...
并且,將refresh函數(shù)更改為:
func (s *Service) refresh(ctx context.Context, clientID string, userID int64, newToken **oauth2.Token) func(repo repository.Auth) error {
...
*newToken, err=...
- 1 回答
- 0 關(guān)注
- 100 瀏覽
添加回答
舉報
0/150
提交
取消