recover.go

package grpcrecovery

import (
	"github.com/coffeehc/base/errors"
	"github.com/coffeehc/base/log"
	"go.uber.org/zap"
	"google.golang.org/grpc"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/grpclog"
	"google.golang.org/grpc/status"
)

func init() {
	grpclog.SetLoggerV2(NewZapLogger())
	grpc.EnableTracing = false
}

var errCode = codes.Code(18)

func convertRPCError(err interface{}, recover bool, fields ...zap.Field) error {
	if err == nil {
		return nil
	}
	var errs errors.Error = nil
	switch v := err.(type) {
	case errors.Error:
		if errors.IsSystemError(v) {
			if !DisableGrpcLog {
				log.DPanic(v.Error(), v.GetFields()...)
			}
		} else {
			// if !strings.HasPrefix(v.Error(), "context") {
			// 	log.Error(v.Error(), v.GetFields()...)
			// }
		}
		errs = v
	case string:
		if recover && !DisableGrpcLog {
			log.DPanic("不可处理的异常", append(fields, zap.String("error", v))...)
		}
		errs = errors.SystemError(v)
	case error:
		if !DisableGrpcLog {
			if recover {
				log.DPanic("不可处理的异常", append(fields, zap.Error(v))...)
			} else {
				errs = errors.SystemError(v.Error())
			}
		}
	default:
		if !DisableGrpcLog {
			log.DPanic("不可处理的异常", append(fields, zap.Any("err", v))...)
		}
		errs = errors.SystemError("未知异常")
	}
	return status.Errorf(errCode, errs.FormatRPCError())
}

func parseRPCError(err interface{}, recover bool, fields ...zap.Field) error {
	if err == nil {
		return nil
	}
	switch v := err.(type) {
	case errors.Error:
		return v
	case string:
		if !DisableGrpcLog {
			if recover {
				log.DPanic("不可处理的异常", append(fields, zap.String("error", v))...)
			} else {
				log.Warn("rpc错误", append(fields, zap.String("err", v))...)
			}
		}
		return errors.SystemError(v)
	case error:
		s, ok := status.FromError(v)
		if !ok {
			if !DisableGrpcLog {
				log.DPanic("无法识别的RPC异常", append(fields, zap.Error(v))...)
			}
			return errors.SystemError("无法识别的RPC异常")
		}
		if s.Code() == errCode {
			//log.Warn("服务端处理调用异常", append(fields, zap.Error(v))...)
			return errors.ParseError(s.Message())
		}
		if !DisableGrpcLog {
			if recover {
				log.DPanic("不可处理的异常", append(fields, zap.Error(v))...)
			} else {
				log.Error("rpc错误", append(fields, zap.Error(v))...)
			}
		}
		return errors.SystemError("远程服务暂时不可用,请重试")
	}
	if !DisableGrpcLog {
		log.DPanic("未知异常", append(fields, zap.Any("err", err))...)
	}
	return errors.SystemError("未知异常")
}