smartmate/backend/gateway/client/notification/errors.go

package notification

import (
	"errors"
	"fmt"
	"strings"

	"github.com/LoveLosita/smartflow/backend/respond"
	"google.golang.org/genproto/googleapis/rpc/errdetails"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

// responseFromRPCError 负责把 notification 的 gRPC 错误反解回项目内的 respond.Response。
//
// 职责边界：
// 1. 只在 gateway 边缘层使用，不下沉到服务实现里；
// 2. 业务错误尽量恢复成 respond.Response，方便 API 层继续复用现有 DealWithError；
// 3. 只要拿不到业务语义，就退化成普通 error，让上层按 500 处理。
func responseFromRPCError(err error) error {
	if err == nil {
		return nil
	}

	st, ok := status.FromError(err)
	if !ok {
		return wrapRPCError(err)
	}

	if resp, ok := responseFromStatus(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 = "notification zrpc service internal error"
		}
		return wrapRPCError(errors.New(msg))
	}

	msg := strings.TrimSpace(st.Message())
	if msg == "" {
		msg = "notification zrpc service rejected request"
	}
	return respond.Response{
		Status: grpcCodeToRespondStatus(st.Code()),
		Info:   msg,
	}
}

func responseFromStatus(st *status.Status) (respond.Response, bool) {
	if st == nil {
		return respond.Response{}, false
	}

	if resp, ok := responseFromStatusDetails(st); ok {
		return resp, true
	}
	if resp, ok := responseFromLegacyStatus(st.Code(), st.Message()); ok {
		return resp, true
	}
	return respond.Response{}, false
}

func responseFromStatusDetails(st *status.Status) (respond.Response, bool) {
	for _, detail := range st.Details() {
		info, ok := detail.(*errdetails.ErrorInfo)
		if !ok {
			continue
		}

		statusValue := strings.TrimSpace(info.Reason)
		if statusValue == "" {
			statusValue = grpcCodeToRespondStatus(st.Code())
		}
		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 responseFromLegacyStatus(code codes.Code, message string) (respond.Response, bool) {
	trimmed := strings.TrimSpace(message)
	if resp, ok := respondResponseByMessage(trimmed); ok {
		return resp, true
	}

	switch code {
	case codes.Unauthenticated:
		if trimmed == "" {
			trimmed = "unauthorized"
		}
		return respond.Response{Status: respond.ErrUnauthorized.Status, Info: trimmed}, true
	case codes.InvalidArgument:
		if trimmed == "" {
			trimmed = "invalid argument"
		}
		return respond.Response{Status: respond.MissingParam.Status, Info: trimmed}, true
	case codes.Internal, codes.Unknown, codes.DataLoss:
		if trimmed == "" {
			trimmed = "notification service internal error"
		}
		return respond.InternalError(errors.New(trimmed)), true
	}

	return respond.Response{}, false
}

func respondResponseByMessage(message string) (respond.Response, bool) {
	switch strings.TrimSpace(message) {
	case respond.MissingParam.Info:
		return respond.MissingParam, true
	case respond.WrongParamType.Info:
		return respond.WrongParamType, true
	case respond.ErrUnauthorized.Info:
		return respond.ErrUnauthorized, true
	}
	return respond.Response{}, false
}

func grpcCodeToRespondStatus(code codes.Code) string {
	switch code {
	case codes.Unauthenticated:
		return respond.ErrUnauthorized.Status
	case codes.InvalidArgument:
		return respond.MissingParam.Status
	case codes.Internal, codes.Unknown, codes.DataLoss:
		return "500"
	default:
		return "400"
	}
}

func wrapRPCError(err error) error {
	if err == nil {
		return nil
	}
	return fmt.Errorf("调用 notification zrpc 服务失败: %w", err)
}