93 lines
2.5 KiB
Go
93 lines
2.5 KiB
Go
package tokenstore
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"strings"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/shared/respond"
|
||
"google.golang.org/genproto/googleapis/rpc/errdetails"
|
||
"google.golang.org/grpc/codes"
|
||
"google.golang.org/grpc/status"
|
||
)
|
||
|
||
// responseFromRPCError 把 token-store zrpc 错误恢复成 HTTP 层可处理的业务错误。
|
||
//
|
||
// 职责边界：
|
||
// 1. 优先读取 token-store RPC 写入的 ErrorInfo，恢复 respond.Response；
|
||
// 2. 对网络、超时、服务不可用等非业务错误保留为普通 error，让 HTTP 层按 500 处理；
|
||
// 3. 不在这里拼装 HTTP 响应体，handler 仍然统一走 respond.DealWithError。
|
||
func responseFromRPCError(err error) error {
|
||
if err == nil {
|
||
return nil
|
||
}
|
||
|
||
st, ok := status.FromError(err)
|
||
if !ok {
|
||
return wrapRPCError(err)
|
||
}
|
||
if resp, ok := responseFromStatusDetails(st); ok {
|
||
return resp
|
||
}
|
||
|
||
switch st.Code() {
|
||
case codes.Internal, codes.Unknown, codes.Unavailable, codes.DeadlineExceeded, codes.DataLoss, codes.Unimplemented:
|
||
msg := strings.TrimSpace(st.Message())
|
||
if msg == "" {
|
||
msg = "tokenstore zrpc service internal error"
|
||
}
|
||
return wrapRPCError(errors.New(msg))
|
||
case codes.PermissionDenied, codes.Unauthenticated:
|
||
return responseWithFallback(st, respond.ErrUnauthorized)
|
||
case codes.InvalidArgument:
|
||
return responseWithFallback(st, respond.MissingParam)
|
||
}
|
||
|
||
msg := strings.TrimSpace(st.Message())
|
||
if msg == "" {
|
||
msg = "tokenstore zrpc service rejected request"
|
||
}
|
||
return respond.Response{Status: "400", Info: msg}
|
||
}
|
||
|
||
func responseFromStatusDetails(st *status.Status) (respond.Response, bool) {
|
||
if st == nil {
|
||
return respond.Response{}, false
|
||
}
|
||
for _, detail := range st.Details() {
|
||
info, ok := detail.(*errdetails.ErrorInfo)
|
||
if !ok {
|
||
continue
|
||
}
|
||
|
||
statusValue := strings.TrimSpace(info.Reason)
|
||
if statusValue == "" {
|
||
return respond.Response{}, false
|
||
}
|
||
message := strings.TrimSpace(st.Message())
|
||
if message == "" && info.Metadata != nil {
|
||
message = strings.TrimSpace(info.Metadata["info"])
|
||
}
|
||
if message == "" {
|
||
message = statusValue
|
||
}
|
||
return respond.Response{Status: statusValue, Info: message}, true
|
||
}
|
||
return respond.Response{}, false
|
||
}
|
||
|
||
func responseWithFallback(st *status.Status, fallback respond.Response) respond.Response {
|
||
msg := strings.TrimSpace(st.Message())
|
||
if msg == "" {
|
||
msg = fallback.Info
|
||
}
|
||
return respond.Response{Status: fallback.Status, Info: msg}
|
||
}
|
||
|
||
func wrapRPCError(err error) error {
|
||
if err == nil {
|
||
return nil
|
||
}
|
||
return fmt.Errorf("调用 tokenstore zrpc 服务失败: %w", err)
|
||
}
|