context.go

package mycontext

import (
	"context"
	"git.hilo.cn/hilo-common/mylogrus"
	"github.com/satori/go.uuid"
	"github.com/sirupsen/logrus"
	"strconv"
	"strings"
)

const (
	TRACEID       = "traceId"
	USERID        = "userId"
	EXTERNAL_ID   = "externalId"
	CODE          = "code"
	NICK          = "nick"
	AVATAR        = "avatar"
	COUNTRY       = "country"
	EXTERNALID1   = "externalId1"
	EXTERNALID2   = "externalId2"
	MGRID         = "mgrId"
	DEVICETYPE    = "deviceType"
	DEVICEVERSION = "deviceVersion"
	APP_VERSION   = "appVersion"
	ACTION_RESULt = "actionResult"
	URL           = "url"
	METHOD        = "method"
	IMEI          = "imei"
	LANGUAGE      = "language"
)

/**
 * 主要是完成日志打印
 * @param
 * @return
 **/

type MyContext struct {
	context.Context
	Log *logrus.Entry
	Cxt map[string]interface{}
}

func CreateMyContextWith(traceId interface{}) *MyContext {
	cxt := map[string]interface{}{}
	cxt[TRACEID] = traceId
	return CreateMyContext(cxt)
}

func CreateMyContext(ctx map[string]interface{}) *MyContext {
	var traceId string
	if traceIdTemp, ok := ctx[TRACEID]; ok {
		traceId, ok = traceIdTemp.(string)
	} else {
		traceId = strings.Replace(uuid.NewV4().String(), "-", "", -1)
	}

	var userId string
	if userIdTemp, ok := ctx[USERID]; ok {
		userId = strconv.FormatUint(userIdTemp.(uint64), 10)
	}

	var deviceTypeStr string
	if deviceTypeTemp, ok := ctx[DEVICETYPE]; ok {
		deviceTypeStr, ok = deviceTypeTemp.(string)
	}

	var sAppVersion string
	if appVersionTmp, ok := ctx[APP_VERSION]; ok {
		sAppVersion, ok = appVersionTmp.(string)
	}

	var url string
	if urlTmp, ok := ctx[URL]; ok {
		url, ok = urlTmp.(string)
	}

	var method string
	if methodTmp, ok := ctx[METHOD]; ok {
		method, ok = methodTmp.(string)
	}
	_ctx := context.WithValue(context.Background(), "traceId", traceId)
	_ctx = context.WithValue(_ctx, "userId", userId)
	return &MyContext{
		Context: _ctx,
		Log:     CreateContextLog(userId, traceId, deviceTypeStr, sAppVersion, url, method),
		Cxt:     ctx,
	}
}

/**
 * 创建上下文的日志
 **/
func CreateContextLog(userId string, traceId string, deviceType string, deviceVersion string, url string, method string) *logrus.Entry {
	return mylogrus.MyLog.WithFields(logrus.Fields{
		USERID:      userId,
		TRACEID:     traceId,
		DEVICETYPE:  deviceType,
		APP_VERSION: deviceVersion,
		URL:         url,
		METHOD:      method,
	})
}