在嵌套函数中设置值

Question

有了这个 golang 代码，随意批评很难，因为我的 golang 已经生锈了。基本刷新令牌流程，但是我们将这些函数包装在 TX 函数中，以便在刷新流程中出现问题时允许回滚。

我们需要返回 newToken 值，但是当它返回到刷新令牌时，它只保存刷新函数期间我们需要的值，具体取决于我是否向其传递对象或指针返回分别是新的空标记或 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
        }

        rec := models.TokenRecord{
            UserID:    userID,
            ClientID:  clientID,
            ProfileID: oldToken.ProfileID,
            Access:    newToken.AccessToken,
            Refresh:   newToken.RefreshToken,
            Expiry:    newToken.Expiry,
        }

        err = repo.UpdateToken(ctx, &rec)
        if err != nil {
            return err
        }

        return nil
    }
}

尝试让 RefreshToken 返回更新后的令牌信息

Answer 1

传递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=...