为了账号安全,请及时绑定邮箱和手机立即绑定

在嵌套函数中设置值

在嵌套函数中设置值

Go
阿晨1998 2022-11-23 10:22:25

给出了这个 golang 代码,因为我的 golang 生锈了,所以很难自由批评。基本的刷新令牌流,但是我们将函数包装在 TX 函数中,以便在刷新流出现问题时允许回滚。


我们需要返回该newToken值,但是refresh当它返回 refreshToken 时它只保存函数期间我们需要的值,具体取决于我传递给它的对象或指针是否分别返回新的空标记或 nil。


package service


import (

    "context"

    "database/sql"

    "errors"

    "strings"


    "github.com/api-oauth/pkg/models"

    "github.com/api-oauth/pkg/repository"

    "golang.org/x/oauth2"

)


const minParts = 2


type 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贡献1512条经验 获得超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函数更改为:


func (s *Service) refresh(ctx context.Context, clientID string, userID int64, newToken **oauth2.Token) func(repo repository.Auth) error {

  ...

  *newToken, err=...


查看完整回答
反对 回复 2022-11-23

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信