mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-09 03:50:26 +08:00
[API节点]支持HTTP API
This commit is contained in:
40
internal/rpc/services/service_api_access_token.go
Normal file
40
internal/rpc/services/service_api_access_token.go
Normal file
@@ -0,0 +1,40 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/TeaOSLab/EdgeAPI/internal/db/models"
|
||||
"github.com/TeaOSLab/EdgeAPI/internal/errors"
|
||||
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
||||
)
|
||||
|
||||
// AccessToken相关服务
|
||||
type APIAccessTokenService struct {
|
||||
}
|
||||
|
||||
// 获取AccessToken
|
||||
func (this *APIAccessTokenService) GetAPIAccessToken(ctx context.Context, req *pb.GetAPIAccessTokenRequest) (*pb.GetAPIAccessTokenResponse, error) {
|
||||
if req.Type == "user" { // 用户
|
||||
accessKey, err := models.SharedUserAccessKeyDAO.FindAccessKeyWithUniqueId(req.AccessKeyId)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if accessKey == nil {
|
||||
return nil, errors.New("access key not found")
|
||||
}
|
||||
if accessKey.Secret != req.AccessKey {
|
||||
return nil, errors.New("access key not found")
|
||||
}
|
||||
|
||||
// 创建AccessToken
|
||||
token, expiresAt, err := models.SharedAPIAccessTokenDAO.GenerateAccessToken(int64(accessKey.UserId))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &pb.GetAPIAccessTokenResponse{
|
||||
Token: token,
|
||||
ExpiresAt: expiresAt,
|
||||
}, nil
|
||||
} else {
|
||||
return nil, errors.New("unsupported type '" + req.Type + "'")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user