Cold_Api/models/tools.go

package models

import (
	"math/rand"
	"strconv"
	"strings"
	"time"
)

// #取得随机字符串：通过打乱slice来操作
func GetRandstring(length int, char string, rand_x int64) string {
	if length < 1 {
		return ""
	}

	if len(char) <= 6 || len(char) <= length {
		char = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
	}

	charArr := strings.Split(char, "")
	ran := rand.New(rand.NewSource(time.Now().Unix() + rand_x))

	l := len(charArr)
	for i := l - 1; i > 0; i-- {
		r := ran.Intn(i)
		charArr[r], charArr[i] = charArr[i], charArr[r]
	}
	rchar := charArr[:length]
	return strings.Join(rchar, "")
}

func SplitStringIds(str string, prefix string) (r []string) {
	Ids_str := strings.TrimRight(str, "|")
	Ids := strings.Split(Ids_str, "|")
	for _, v := range Ids {
		r = append(r, strings.TrimLeft(v, prefix))
	}
	return r
}

func SplitStringToDotStr(str string, prefix string) (r string) {
	Ids_str := strings.TrimRight(str, "|")
	Ids := strings.Split(Ids_str, "|")
	for _, v := range Ids {
		r += strings.TrimLeft(v, prefix) + ","
	}
	r = strings.TrimRight(r, ",")
	return r
}

func SplitStringToIntIds(str string, prefix string) (r []int) {
	Ids_str := strings.TrimRight(str, "|")
	Ids := strings.Split(Ids_str, "|")
	for _, v := range Ids {
		id, _ := strconv.Atoi(strings.TrimLeft(v, prefix))
		r = append(r, id)
	}
	return r
}

func IntIdsDistinct(Ids []int) (result []int) {
	distinctMap := make(map[int]int, len(Ids))
	for _, Id := range Ids {
		if _, ok := distinctMap[Id]; !ok {
			distinctMap[Id] = 1
			result = append(result, Id)
		}
	}
	return result
}