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))) }