jyMicroservices/jyBXCore/api/internal/util/util.go

package util

import (
	MC "app.yhyue.com/moapp/jybase/common"
	"fmt"
	"net/http"
	"regexp"
	"strings"
	"time"
)

var (
	mobileReg = regexp.MustCompile("(?i)(Android|Mobile|Phone)")
)

func CheckPlatform(r *http.Request) (p string) {
	p = "PC"
	if CheckIsMobile(r) {
		if CheckWxBrowser(r) {
			p = "WX"
		} else {
			p = "APP"
		}
	}
	return
}

// 判断是否是微信访问
func CheckWxBrowser(Request *http.Request) bool {
	if strings.Index(Request.UserAgent(), "MicroMessenger") > -1 || strings.Index(Request.UserAgent(), "Wechat") > -1 {
		return true
	} else {
		return false
	}
}

// 是否是移动端
func CheckIsMobile(r *http.Request) bool {
	client := r.UserAgent()
	if mobileReg.MatchString(client) {
		return true
	}
	return false
}

func GetFlag(r *http.Request, w http.ResponseWriter, limitFlag string) (string, bool) {
	if limitFlag == "" {
		c, _ := r.Cookie("limitSearchTextFlag")
		if c != nil {
			limitFlag = c.Value
			if limitFlag == "" {
				sid, _ := r.Cookie("SESSIONID")
				limitFlag = sid.Value
				//limitFlag, _ = s.Get("limitSearchTextFlag").(string)
			}
		}
	}
	if limitFlag != "" {
		return limitFlag, true
	}
	limitFlag = MC.GetLetterRandom(5) + fmt.Sprint(time.Now().UnixNano())
	//s.Set("limitSearchTextFlag", limitFlag)
	c := &http.Cookie{
		Name:     "limitSearchTextFlag",
		Value:    limitFlag,
		Path:     "/",
		HttpOnly: false,
		MaxAge:   2592000, //一个月
	}
	http.SetCookie(w, c)
	return limitFlag, false
}

// aData 词库 rData过滤词 count数量
func FilterWord(aData, fData []string, count int) []string {
	var rsData []string
	for _, datum := range aData {
		if len(rsData) >= count {
			break
		}
		var isExist bool
		for _, datum1 := range fData {
			if datum == datum1 {
				isExist = true
				break
			}
		}
		if !isExist {
			rsData = append(rsData, datum)
		}
	}

	if ct := count - len(rsData); ct > 0 && len(aData) >= ct {
		rsData = append(rsData, aData[:ct]...) //追加
	}
	return rsData
}