143 lines
3.5 KiB
Go
143 lines
3.5 KiB
Go
package util
|
||
|
||
import (
|
||
"crypto/md5"
|
||
"encoding/json"
|
||
"fmt"
|
||
"io"
|
||
"math"
|
||
"math/rand"
|
||
"reflect"
|
||
"service/library/logger"
|
||
"strings"
|
||
"time"
|
||
"unsafe"
|
||
|
||
"github.com/qiniu/qmgo"
|
||
)
|
||
|
||
func ToJson(v any) string {
|
||
str, err := json.Marshal(v)
|
||
if err != nil {
|
||
logger.Info("%v marsha failed, err: %v", v, err)
|
||
}
|
||
return string(str)
|
||
}
|
||
|
||
func StringToMd5(s string) string {
|
||
m := md5.New()
|
||
_, _ = io.WriteString(m, s)
|
||
return fmt.Sprintf("%x", m.Sum(nil))
|
||
}
|
||
|
||
// 将实体类结构体指针转为bson.M，采用反射
|
||
func EntityToM(p any) qmgo.M {
|
||
set := qmgo.M{}
|
||
pType := reflect.TypeOf(p).Elem()
|
||
pVal := reflect.ValueOf(p).Elem()
|
||
for i := 0; i < pType.NumField(); i++ {
|
||
name := pType.Field(i).Tag.Get("bson")
|
||
if name != "" {
|
||
field := pVal.Field(i)
|
||
if name != "_id" && !field.IsNil() {
|
||
value := field.Elem().Interface()
|
||
set[name] = value
|
||
}
|
||
}
|
||
}
|
||
return set
|
||
}
|
||
|
||
func StringsContains(elems []string, v string) bool {
|
||
for _, s := range elems {
|
||
if s == v {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// 下划线转大写驼峰
|
||
func UderscoreToUpperCamelCase(s string) string {
|
||
s = strings.Replace(s, "_", " ", -1)
|
||
s = strings.Title(s)
|
||
return strings.Replace(s, " ", "", -1)
|
||
}
|
||
|
||
// 数组转sql数组 []int{1, 2, 3) --> 1,2,3
|
||
func Convert2SqlArr(a ...any) string {
|
||
return strings.Replace(strings.Trim(fmt.Sprint(a), "[]"), " ", ",", -1)
|
||
}
|
||
|
||
// 获取整点时间戳
|
||
func GetHourStartTimeStamp(t time.Time) int64 {
|
||
loc, _ := time.LoadLocation("Asia/Shanghai")
|
||
timeStr := fmt.Sprintf("%02d-%02d-%02d %02d:00:00", t.Year(), t.Month(), t.Day(), t.Hour())
|
||
duetimecst, err := time.ParseInLocation("2006-1-2 15:04:05", timeStr, loc)
|
||
if err != nil {
|
||
logger.Error("parse error : %v", err)
|
||
}
|
||
return duetimecst.Unix()
|
||
}
|
||
|
||
// 获取整分时间戳
|
||
func GetMinuteStartTimeStamp(t time.Time) int64 {
|
||
loc, _ := time.LoadLocation("Asia/Shanghai")
|
||
timeStr := fmt.Sprintf("%02d-%02d-%02d %02d:%02d:00", t.Year(), t.Month(), t.Day(), t.Hour(), t.Minute())
|
||
duetimecst, err := time.ParseInLocation("2006-1-2 15:04:05", timeStr, loc)
|
||
if err != nil {
|
||
logger.Error("parse error : %v", err)
|
||
}
|
||
return duetimecst.Unix()
|
||
}
|
||
|
||
// 获取今天0点
|
||
func GetTodayZeroTime() time.Time {
|
||
currentTime := time.Now()
|
||
zeroTime := time.Date(currentTime.Year(), currentTime.Month(), currentTime.Day(), 0, 0, 0, 0, currentTime.Location())
|
||
return zeroTime
|
||
}
|
||
|
||
// 获取0点时间戳
|
||
func GetDayStartTimeStamp(t time.Time) int64 {
|
||
loc, _ := time.LoadLocation("Asia/Shanghai")
|
||
timeStr := fmt.Sprintf("%02d-%02d-%02d 00:00:00", t.Year(), t.Month(), t.Day())
|
||
duetimecst, err := time.ParseInLocation("2006-1-2 15:04:05", timeStr, loc)
|
||
if err != nil {
|
||
logger.Error("parse error : %v", err)
|
||
}
|
||
return duetimecst.Unix()
|
||
}
|
||
|
||
// 随机生成字符串
|
||
func RandomString(l int) string {
|
||
str := "0123456789AaBbCcDdEeFfGgHhIiJjKkLlMmNnOoPpQqRrSsTtUuVvWwXxYyZz"
|
||
bytes := []byte(str)
|
||
var result []byte = make([]byte, 0, l)
|
||
r := rand.New(rand.NewSource(time.Now().UnixNano()))
|
||
for i := 0; i < l; i++ {
|
||
result = append(result, bytes[r.Intn(len(bytes))])
|
||
}
|
||
return BytesToString(result)
|
||
}
|
||
|
||
// BytesToString 0 拷贝转换 slice byte 为 string
|
||
func BytesToString(b []byte) (s string) {
|
||
_bptr := (*reflect.SliceHeader)(unsafe.Pointer(&b))
|
||
_sptr := (*reflect.StringHeader)(unsafe.Pointer(&s))
|
||
_sptr.Data = _bptr.Data
|
||
_sptr.Len = _bptr.Len
|
||
return s
|
||
}
|
||
|
||
func UnescapeJsonStr(s string) string {
|
||
s = strings.ReplaceAll(s, "\\u003c", "<")
|
||
s = strings.ReplaceAll(s, "\\u003e", ">")
|
||
s = strings.ReplaceAll(s, "\\u0026", "&")
|
||
return s
|
||
}
|
||
|
||
func AbsInt64(x int64) int64 {
|
||
return int64(math.Abs(float64(x)))
|
||
}
|