SwordFish_Interface_Service/sword_base/utils/stringutil.go

package utils

import (
	"crypto/md5"
	cryptoRand "crypto/rand"
	"encoding/hex"
	"fmt"
	"io"
	"math/big"
	"math/rand"
	"net/url"
	"strconv"
	"strings"
	"time"

	"github.com/dchest/captcha"
)

func FilterXSS(str string) string {
	str = strings.Replace(str, "<", "&#60;", -1)
	str = strings.Replace(str, ">", "&#62;", -1)
	str = strings.Replace(str, "%3C", "&#60;", -1)
	str = strings.Replace(str, "%3E", "&#62;", -1)
	str = strings.Replace(str, "expression", "ｅｘｐｒｅｓｓｉｏｎ", -1)
	str = strings.Replace(str, "javascript", "ｊａｖａｓｃｒｉｐｔ", -1)
	return str
}

func MakeSimpleCaptcha(n int) string {
	var idChars = []byte("0123456789")
	b := captcha.RandomDigits(n)
	for i, c := range b {
		b[i] = idChars[c]
	}
	return string(b)
}

/**
获取当前时间的毫秒值
*/
func Now4Millis() int64 {
	return time.Now().UnixNano() / 1e6
}

/*产生n以内的随机数*/
func MakeIntRand(n int) int {
	rand.Seed(time.Now().UnixNano())
	return rand.Intn(n)
}

//生成32位md5字串
func GetMd5String(s string) string {
	h := md5.New()
	h.Write([]byte(s))
	return hex.EncodeToString(h.Sum(nil))
}

func GenerateSimpleToken() string {
	rand.Seed(time.Now().UnixNano())
	str := fmt.Sprintf("%d%d", time.Now().UnixNano(), rand.Intn(999999))
	h := md5.New()
	h.Write([]byte(str))
	return hex.EncodeToString(h.Sum(nil))
}

//var pool = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz@#$"

//获取复杂的随机数
func GetLetterRandom(length int, flag ...bool) string {
	var idChars = []byte("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz")
	var mod byte = 52
	if len(flag) > 0 && flag[0] {
		idChars = []byte("ABCDEFGHIJKLMNOPQRSTUVWXYZ")
		mod = 26
	}
	b := make([]byte, length)
	maxrb := byte(256 - (256 % int(mod)))
	i := 0
EXIT:
	for {
		r := make([]byte, length+(length/4))
		if _, err := io.ReadFull(cryptoRand.Reader, r); err != nil {
			panic("GetLetterRandom: error reading random source: " + err.Error())
		}
		for _, c := range r {
			if c > maxrb {
				continue
			}
			b[i] = c % mod
			i++
			if i == length {
				break EXIT
			}
		}
	}
	for i, c := range b {
		b[i] = idChars[c]
	}
	return string(b)
}

/*获取复杂的随机数,数字和字母的组合
 * c > 2 数字的个数和字母的个数随机分配
 * n 数字的个数
 * l 字母的个数
 */
func GetComplexRandom(c, n, l int) string {
	if c < 2 && (n < 1 || l < 1) {
		return "--"
	}
	r := rand.New(rand.NewSource(time.Now().UnixNano()))
	myCommonMethod := func(flag bool) int {
		if flag {
			return r.Intn(c-1) + 1
		} else {
			return r.Intn(c)
		}
	}
	if c >= 2 {
		n = myCommonMethod(true)
		l = c - n
	} else {
		c = l + n
	}
	value := MakeSimpleCaptcha(n) + GetLetterRandom(l)
	var array = strings.Split(value, "")
	for i := 0; i < c/2; i++ {
		r1 := myCommonMethod(false)
		r2 := myCommonMethod(false)
		o := array[r1]
		array[r1] = array[r2]
		array[r2] = o
	}
	return strings.Join(array, "")
}

func GetRandom(n int) string {
	var idChars = []byte("0123456789")
	b := captcha.RandomDigits(n)
	for i, c := range b {
		b[i] = idChars[c]
	}
	return string(b)
}

func ObjToString(old interface{}) string {
	if nil == old {
		return ""
	} else {
		r, _ := old.(string)
		return r
	}
}

func IntAll(num interface{}) int {
	return IntAllDef(num, 0)
}

func IntAllDef(num interface{}, defaultNum int) int {
	if i, ok := num.(int); ok {
		return int(i)
	} else if i0, ok0 := num.(int32); ok0 {
		return int(i0)
	} else if i1, ok1 := num.(float64); ok1 {
		return int(i1)
	} else if i2, ok2 := num.(int64); ok2 {
		return int(i2)
	} else if i3, ok3 := num.(float32); ok3 {
		return int(i3)
	} else if i4, ok4 := num.(string); ok4 {
		in, _ := strconv.Atoi(i4)
		return int(in)
	} else if i5, ok5 := num.(int16); ok5 {
		return int(i5)
	} else if i6, ok6 := num.(int8); ok6 {
		return int(i6)
	} else if i7, ok7 := num.(*big.Int); ok7 {
		in, _ := strconv.Atoi(fmt.Sprint(i7))
		return int(in)
	} else if i8, ok8 := num.(*big.Float); ok8 {
		in, _ := strconv.Atoi(fmt.Sprint(i8))
		return int(in)
	} else {
		return defaultNum
	}
}

func ObjArrToMapArr(old []interface{}) []map[string]interface{} {
	if old != nil {
		new := make([]map[string]interface{}, len(old))
		for i, v := range old {
			new[i], _ = v.(map[string]interface{})
		}
		return new
	} else {
		return nil
	}
}

func Int64All(num interface{}) int64 {
	if i, ok := num.(int64); ok {
		return int64(i)
	} else if i0, ok0 := num.(int32); ok0 {
		return int64(i0)
	} else if i1, ok1 := num.(float64); ok1 {
		return int64(i1)
	} else if i2, ok2 := num.(int); ok2 {
		return int64(i2)
	} else if i3, ok3 := num.(float32); ok3 {
		return int64(i3)
	} else if i4, ok4 := num.(string); ok4 {
		i64, _ := strconv.ParseInt(i4, 10, 64)
		//in, _ := strconv.Atoi(i4)
		return i64
	} else if i5, ok5 := num.(int16); ok5 {
		return int64(i5)
	} else if i6, ok6 := num.(int8); ok6 {
		return int64(i6)
	} else if i7, ok7 := num.(*big.Int); ok7 {
		in, _ := strconv.ParseInt(fmt.Sprint(i7), 10, 64)
		return int64(in)
	} else if i8, ok8 := num.(*big.Float); ok8 {
		in, _ := strconv.ParseInt(fmt.Sprint(i8), 10, 64)
		return int64(in)
	} else {
		return 0
	}
}

func Float64All(num interface{}) float64 {
	if i, ok := num.(float64); ok {
		return float64(i)
	} else if i0, ok0 := num.(int32); ok0 {
		return float64(i0)
	} else if i1, ok1 := num.(int64); ok1 {
		return float64(i1)
	} else if i2, ok2 := num.(int); ok2 {
		return float64(i2)
	} else if i3, ok3 := num.(float32); ok3 {
		return float64(i3)
	} else if i4, ok4 := num.(string); ok4 {
		in, _ := strconv.ParseFloat(i4, 64)
		return in
	} else if i5, ok5 := num.(int16); ok5 {
		return float64(i5)
	} else if i6, ok6 := num.(int8); ok6 {
		return float64(i6)
	} else if i6, ok6 := num.(uint); ok6 {
		return float64(i6)
	} else if i6, ok6 := num.(uint8); ok6 {
		return float64(i6)
	} else if i6, ok6 := num.(uint16); ok6 {
		return float64(i6)
	} else if i6, ok6 := num.(uint32); ok6 {
		return float64(i6)
	} else if i6, ok6 := num.(uint64); ok6 {
		return float64(i6)
	} else if i7, ok7 := num.(*big.Float); ok7 {
		in, _ := strconv.ParseFloat(fmt.Sprint(i7), 64)
		return float64(in)
	} else if i8, ok8 := num.(*big.Int); ok8 {
		in, _ := strconv.ParseFloat(fmt.Sprint(i8), 64)
		return float64(in)
	} else {
		return 0
	}
}

//通用加密
func CommonEncodeArticle(stype string, keys ...string) (id string) {
	//正文
	var SE = &SimpleEncrypt{Key: "topnet2015topnet2015"}
	var SE2 = &SimpleEncrypt{Key: "2017jianyu"}
	if stype == "content" {
		id = BEncodeArticleId2ByCheck("A", SE, SE2, keys...)
	}
	return
}

//短地址加密,二次加密带校验和
func BEncodeArticleId2ByCheck(h string, s1, s2 *SimpleEncrypt, keys ...string) string {
	kstr := strings.Join(keys, ",")
	kstr = s1.EncodeStringByCheck(kstr)
	return url.QueryEscape(h + GetLetterRandom(2) + s2.EncodeStringByCheck(kstr))
}

/**
产生一般订单编号方法
*/
func CreateOrderCode() string {
	return fmt.Sprint(time.Now().Unix()) + fmt.Sprint(GetRandom(6))
}

func GetDayMinMax(t time.Time) (int64, int64) {
	min := time.Date(t.Year(), t.Month(), t.Day(), 0, 0, 0, 0, time.Local).Unix()
	return min, min + 86400
}