ginrpc/middleware.go

package ginrpc

import (
	"context"
	"fmt"
	"time"

	"github.com/xxjwxc/public/message"
	"google.golang.org/grpc/status"

	"github.com/xxjwxc/public/mylog"

	"github.com/gin-gonic/gin"
)

// GinBeforeAfterInfo 对象调用前后执行中间件参数
type GinBeforeAfterInfo struct {
	C        *gin.Context
	FuncName string      // 函数名
	Req      interface{} // 调用前的请求参数
	Resp     interface{} // 调用后的返回参数
	Error    error       // 错误信息
	// Other options for implementations of the interface
	// can be stored in a context
	Context context.Context // 占位参数，可用于存储其他参数，前后连接可用

}

// GinBeforeAfter 对象调用前后执行中间件(支持总的跟对象单独添加)
type GinBeforeAfter interface {
	GinBefore(req *GinBeforeAfterInfo) bool
	GinAfter(req *GinBeforeAfterInfo) bool
}

// DefaultGinBeforeAfter 创建一个默认 BeforeAfter Middleware
type DefaultGinBeforeAfter struct {
}

type timeTrace struct{}

// GinBefore call之前调用
func (d *DefaultGinBeforeAfter) GinBefore(req *GinBeforeAfterInfo) bool {
	req.Context = context.WithValue(req.Context, timeTrace{}, time.Now())
	return true
}

// GinAfter call之后调用
func (d *DefaultGinBeforeAfter) GinAfter(req *GinBeforeAfterInfo) bool {
	begin := (req.Context.Value(timeTrace{})).(time.Time)
	now := time.Now()
	mylog.Info(fmt.Sprintf("[middleware] call[%v] [%v]", req.FuncName, now.Sub(begin)))

	msg := message.GetSuccessMsg()
	if req.Error != nil {
		msg = message.GetErrorMsg(message.InValidOp)
		gerr := status.Convert(req.Error)
		if gerr != nil {
			msg.Code = int(gerr.Code())
			msg.Error = gerr.Message()
		} else {
			msg.Error = req.Error.Error()
		}
	} else {
		msg.Data = req.Resp
	}

	req.Resp = msg // 设置resp 结果

	return true
}

// ----------------end