grpcmid

Documentation

Index

• Variables
• func WhiteIpMiddleware(whiteList map[string][]string) server.Option
• type GrpcToken
  • func (gt GrpcToken) GetGrpcToken() string
• type GrpcTokenGet
• type MetaTTHeaderHandler
  • func (ch *MetaTTHeaderHandler) ReadMeta(ctx context.Context, msg remote.Message) (context.Context, error)
  • func (ch *MetaTTHeaderHandler) WriteMeta(ctx context.Context, msg remote.Message) (context.Context, error)

Variables

var ClientMetaHandler = client.WithMetaHandler(&MetaTTHeaderHandler{})

var ErrClientHandlerOption = client.WithErrorHandler(func(ctx context.Context, err error) error {
	transErr, ok := err.(*remote.TransError)
	if ok && transErr.TypeID() == 6 {
		errStr := strings.ReplaceAll(transErr.Error(), "biz error: ", "")
		var errInfo e.Err
		err = json.Unmarshal([]byte(errStr), &errInfo)
		if err == nil {
			return errInfo
		}
		err = e.RPCClientErrorCodeError.SetErr(err, strings.ReplaceAll(transErr.Error(), "biz error: ", ""))
	} else {
		err = e.RPCClientErrorCodeError.SetErr(err, "数据访问失败")
	}
	return err
})

ErrClientHandlerOption 业务错误再封装

var ErrHandlerOption = client.WithErrorHandler(func(ctx context.Context, err error) error {
	transErr, ok := err.(*remote.TransError)
	if ok && transErr.TypeID() == 6 {
		err = e.RPCClientErrorCodeError.SetErr(err, strings.ReplaceAll(transErr.Error(), "biz error: ", ""))
	} else {
		statusErr, ok := err.(*status.Error)
		if ok && uint32(statusErr.GRPCStatus().Code()) == 13 {
			err = e.RPCClientErrorCodeError.SetErr(err, strings.TrimSpace(strings.ReplaceAll(statusErr.GRPCStatus().Message(), "[biz error]", "")))
		} else {
			err = e.RPCClientErrorCodeError.SetErr(err, "数据访问失败")
		}
	}
	return err
})

ErrHandlerOption 业务错误再封装

var ErrServerHandlerOption = server.WithErrorHandler(func(ctx context.Context, err error) error {
	transErr, ok := err.(*kerrors.DetailedError)
	if ok {
		var errInfo e.Err
		if transErr.As(&errInfo) {
			var b []byte
			b, err = json.Marshal(errInfo)
			if err != nil {
				err = e.RPCServerErrorCodeError.SetErr(err, "错误转换失败")
			} else {
				err = e.RPCServerErrorCodeError.SetErr(err, string(b))
			}
		}
	} else {
		err = e.RPCServerErrorCodeError.SetErr(err, "数据访问失败")
	}
	return err
})

var GrpcTokenKey struct{}

var GrpcTokenStringKey = "grpcTokenString"

var ServerMetaHandler = server.WithMetaHandler(&MetaTTHeaderHandler{})

Functions

func WhiteIpMiddleware

func WhiteIpMiddleware(whiteList map[string][]string) server.Option

Types

type GrpcToken

type GrpcToken struct {
	GrpcToken string `json:"grpcToken"`
}

func (GrpcToken) GetGrpcToken

func (gt GrpcToken) GetGrpcToken() string

type GrpcTokenGet

type GrpcTokenGet interface {
	GetGrpcToken() string
}

type MetaTTHeaderHandler

type MetaTTHeaderHandler struct {
	WhiteApi   []string //接口白名单
	CheckWhite bool
}

func (*MetaTTHeaderHandler) ReadMeta

func (ch *MetaTTHeaderHandler) ReadMeta(ctx context.Context, msg remote.Message) (context.Context, error)

ReadMeta of MetaTTHeaderHandler reads headers of TTHeader protocol from transport

func (*MetaTTHeaderHandler) WriteMeta

func (ch *MetaTTHeaderHandler) WriteMeta(ctx context.Context, msg remote.Message) (context.Context, error)