package util
|
|
import (
|
"encoding/json"
|
"fmt"
|
"github.com/gin-gonic/gin"
|
"math/rand"
|
"strconv"
|
"strings"
|
"time"
|
"video-analysis-shop/api/code"
|
)
|
|
func ResponseFormat(c *gin.Context, respStatus *code.Code, data interface{}) {
|
if respStatus == nil {
|
respStatus = code.RequestParamError
|
}
|
c.JSON(respStatus.Status, gin.H{
|
"code": respStatus.Status,
|
"success": respStatus.Success,
|
"msg": respStatus.Message,
|
"data": data,
|
})
|
return
|
}
|
|
func GenValidateCode(width int) string {
|
numeric := [10]byte{0,1,2,3,4,5,6,7,8,9}
|
r := len(numeric)
|
rand.Seed(time.Now().UnixNano())
|
|
var sb strings.Builder
|
for i := 0; i < width; i++ {
|
fmt.Fprintf(&sb, "%d", numeric[ rand.Intn(r) ])
|
}
|
str := sb.String()
|
if strings.HasPrefix(str, "0") {
|
pI := RandInt(1,9)
|
str = strconv.Itoa(pI) + str[1:]
|
}
|
return str
|
}
|
|
const (
|
KC_RAND_KIND_NUM = 0 // 纯数字
|
KC_RAND_KIND_LOWER = 1 // 小写字母
|
KC_RAND_KIND_UPPER = 2 // 大写字母
|
KC_RAND_KIND_ALL = 3 // 数字、大小写字母
|
)
|
|
// 随机字符串
|
func Krand(size int, kind int) []byte {
|
ikind, kinds, result := kind, [][]int{[]int{10, 48}, []int{26, 97}, []int{26, 65}}, make([]byte, size)
|
is_all := kind > 2 || kind < 0
|
rand.Seed(time.Now().UnixNano())
|
for i :=0; i < size; i++ {
|
if is_all { // random ikind
|
ikind = rand.Intn(3)
|
}
|
scope, base := kinds[ikind][0], kinds[ikind][1]
|
result[i] = uint8(base+rand.Intn(scope))
|
}
|
return result
|
}
|
|
//生成区间随机数
|
func RandInt(min, max int) int {
|
if min >= max || min ==0 ||max == 0 {
|
return max
|
}
|
return rand.Intn(max-min) + min
|
}
|
|
func Struct2Map(obj interface{}) map[string]interface{} {
|
resultMap := make(map[string]interface{}, 0)
|
bytesData, err := json.Marshal(obj)
|
if err != nil {
|
return resultMap
|
}
|
json.Unmarshal(bytesData, &resultMap)
|
return resultMap
|
}
|
