jyOrderManager/internal/logic/middleware/middleware.go

package middleware

import (
	elastic "app.yhyue.com/moapp/jybase/es"
	"strings"

	//. "app.yhyue.com/moapp/jybase/api"
	"context"
	"fmt"
	"jyOrderManager/internal/consts"
	"jyOrderManager/internal/jyutil"
	"jyOrderManager/internal/model"
	"jyOrderManager/internal/service"
	"net/http"
	"sync"
	"time"

	util "app.yhyue.com/moapp/jybase/common"
	"github.com/gogf/gf/v2/errors/gcode"
	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/ghttp"
)

type sMiddleware struct {
}

var (
	logArr       = []map[string]interface{}{}
	logLock      = &sync.Mutex{}
	logSaveCount int
)

func init() {
	logSaveCount = g.Cfg().MustGet(context.Background(), "logger.logSaveCount", 500).Int()
	elastic.NewEs("v7", g.Cfg().MustGet(context.Background(), "esAddress").String(), g.Cfg().MustGet(context.Background(), "esSize").Int(), g.Cfg().MustGet(context.Background(), "esUserName").String(), g.Cfg().MustGet(context.Background(), "esPassword").String())
	service.RegisterMiddleware(&sMiddleware{})
}

// 定时保存日志
func SaveLogTask() {
	logLock.Lock()
	if len(logArr) >= 1 {
		tmpArr := logArr
		logArr = make([]map[string]interface{}, 0)
		go func() {
			g.Log().Info(context.TODO(), "timer..save..visit..log", len(tmpArr))
			jyutil.MG.DB("log").SaveBulk("jy_order_manager", tmpArr...)
		}()
	}
	logLock.Unlock()
	time.AfterFunc(5*time.Minute, SaveLogTask)
}

// 访问日志
func (s *sMiddleware) Log(r *ghttp.Request) {
	timeNow := time.Now()
	data := map[string]interface{}{
		"date":        timeNow.Unix(),
		"ip":          r.GetClientIp(),
		"refer":       r.Referer(),
		"year":        timeNow.Year(),
		"month":       timeNow.Month(),
		"day":         timeNow.Day(),
		"hour":        timeNow.Hour(),
		"minutes":     timeNow.Minute(),
		"mdescribe":   r.GetBodyString(),
		"client":      r.UserAgent(),
		"os":          util.GetOS(r.UserAgent()),
		"browse":      util.GetBrowse(r.UserAgent()),
		"method":      r.Method,
		"url":         r.RequestURI,
		"entUserId":   r.Session.MustGet("entUserId", 0).Int64(),
		"entUserName": r.Session.MustGet("entUserName", "").String(),
	}

	logLock.Lock()
	logArr = append(logArr, data)
	if len(logArr) >= logSaveCount {
		tmpArr := logArr
		logArr = make([]map[string]interface{}, 0)
		go func() {
			g.Log().Info(r.Context(), "save..visit..log", len(tmpArr))
			jyutil.MG.DB("log").SaveBulk("jy_order_manager", tmpArr...)
		}()
	}
	logLock.Unlock()
	r.Middleware.Next()
}

// LoginFilter 登录过滤拦截
func (s *sMiddleware) LoginFilter(r *ghttp.Request) {
	var (
		uMsg *model.User
		err  error
	)
	if !strings.HasPrefix(r.RequestURI, "/debug") {
		if token := r.Header.Get("Token"); token != "" {
			uMsg, err = jyutil.GetUserMsgFromToken(r.Context(), token)
			if err != nil {
				r.SetError(fmt.Errorf("无效token"))
				return
			}
		} else {
			uMsg = &model.User{
				PositionId: r.Session.MustGet("positionId", 0).Int64(),
				EntUserId:  r.Session.MustGet("entUserId", 0).Int64(),
				EntId:      r.Session.MustGet("entId", 0).Int64(),
				EntDeptId:  r.Session.MustGet("entDeptId", 0).Int64(),
				EntRole:    r.Session.MustGet("entRole", 0).Int64(),
				AccountId:  r.Session.MustGet("accountId", 0).Int64(),
				//MgoUserId:   r.Session.MustGet("mgoUserId", 0).String(),
				EntUserName: r.Session.MustGet("entUserName", "").String(),
			}
		}
		if uMsg == nil || uMsg.EntId == 0 || uMsg.EntUserId == 0 {
			r.SetError(fmt.Errorf("身份异常"))
			return
		}
		if uMsg.EntId != g.Cfg("global").MustGet(r.Context(), "powerEntId", 25917).Int64() {
			r.SetError(fmt.Errorf("非法请求"))
			return
		}
		r.SetCtxVar(consts.ContextKey, uMsg)
	}
	r.Middleware.Next()
}

// CORS 允许跨域请求
func (s *sMiddleware) CORS(r *ghttp.Request) {
	r.Response.CORSDefault()
	r.Middleware.Next()
}

// MiddlewareHandlerResponse is the default middleware handling handler response object and its error.
func (s *sMiddleware) MiddlewareHandlerResponse(r *ghttp.Request) {
	r.Middleware.Next()

	// There's custom buffer content, it then exits current handler.
	if r.Response.BufferLength() > 0 {
		return
	}

	var (
		msg  string
		err  = r.GetError()
		res  = r.GetHandlerResponse()
		code = gerror.Code(err)
	)
	if err != nil {
		if code == gcode.CodeNil {
			code = gcode.CodeInternalError
		}
		msg = err.Error()
	} else {
		if r.Response.Status > 0 && r.Response.Status != http.StatusOK {
			msg = http.StatusText(r.Response.Status)
			switch r.Response.Status {
			case http.StatusNotFound:
				code = gcode.CodeNotFound
			case http.StatusForbidden:
				code = gcode.CodeNotAuthorized
			default:
				code = gcode.CodeUnknown
			}
			// It creates error as it can be retrieved by otherProduct middlewares.
			err = gerror.NewCode(code, msg)
			r.SetError(err)
		} else {
			code = gcode.CodeOK
		}
	}
	r.Response.WriteJson(model.DefaultResponse{
		Code:    code.Code(),
		Message: msg,
		Data:    res,
	})
}