package utils import ( "bufio" "bytes" "crypto/hmac" "crypto/md5" "crypto/sha1" "crypto/sha256" "encoding/base64" "encoding/hex" "encoding/json" "errors" "fmt" "html" "image" "image/png" "io" "math" "math/rand" "net" "net/http" "net/url" "os" "os/exec" "path" "reflect" "regexp" "sort" "strconv" "strings" "time" "unicode" "gorm.io/gorm" "github.com/PuerkitoBio/goquery" "github.com/microcosm-cc/bluemonday" "github.com/shopspring/decimal" xhtml "golang.org/x/net/html" ) func GetRandString(size int) string { allLetterDigit := []string{"0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z", "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z", "!", "@", "#", "$", "%", "^", "&", "*"} randomSb := "" digitSize := len(allLetterDigit) rnd := rand.New(rand.NewSource(time.Now().UnixNano())) for i := 0; i < size; i++ { randomSb += allLetterDigit[rnd.Intn(digitSize)] } return randomSb } func GetRandStringNoSpecialChar(size int) string { allLetterDigit := []string{"0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z", "a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n", "o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z"} randomSb := "" digitSize := len(allLetterDigit) rnd := rand.New(rand.NewSource(time.Now().UnixNano())) for i := 0; i < size; i++ { randomSb += allLetterDigit[rnd.Intn(digitSize)] } return randomSb } func StringsToJSON(str string) string { rs := []rune(str) jsons := "" for _, r := range rs { rint := int(r) if rint < 128 { jsons += string(r) } else { jsons += "\\u" + strconv.FormatInt(int64(rint), 16) // json } } return jsons } func ToString(v interface{}) string { data, _ := json.Marshal(v) return string(data) } func MD5(data string) string { m := md5.Sum([]byte(data)) return hex.EncodeToString(m[:]) } func HmacMd5(key, data string) string { h := hmac.New(md5.New, []byte(key)) h.Write([]byte(data)) return hex.EncodeToString(h.Sum([]byte(""))) } func GetRandDigit(n int) string { rnd := rand.New(rand.NewSource(time.Now().UnixNano())) return fmt.Sprintf("%0"+strconv.Itoa(n)+"d", rnd.Intn(int(math.Pow10(n)))) } func GetRandNumber(n int) int { rnd := rand.New(rand.NewSource(time.Now().UnixNano())) return rnd.Intn(n) } func GetToday(format string) string { today := time.Now().Format(format) return today } func GetTodayLastSecond() time.Duration { today := GetToday(FormatDate) + " 23:59:59" end, _ := time.ParseInLocation(FormatDateTime, today, time.Local) return time.Duration(end.Unix()-time.Now().Local().Unix()) * time.Second } func GetBrithDate(idcard string) string { l := len(idcard) var s string if l == 15 { s = "19" + idcard[6:8] + "-" + idcard[8:10] + "-" + idcard[10:12] return s } if l == 18 { s = idcard[6:10] + "-" + idcard[10:12] + "-" + idcard[12:14] return s } return GetToday(FormatDate) } func WhichSexByIdcard(idcard string) string { var sexs = [2]string{"女", "男"} length := len(idcard) if length == 18 { sex, _ := strconv.Atoi(string(idcard[16])) return sexs[sex%2] } else if length == 15 { sex, _ := strconv.Atoi(string(idcard[14])) return sexs[sex%2] } return "男" } func SubFloatToString(f float64, m int) string { n := strconv.FormatFloat(f, 'f', -1, 64) if n == "" { return "" } if m >= len(n) { return n } newn := strings.Split(n, ".") if m == 0 { return newn[0] } if len(newn) < 2 || m >= len(newn[1]) { return n } return newn[0] + "." + newn[1][:m] } func SubFloatToFloat(f float64, m int) float64 { newn := SubFloatToString(f, m) newf, _ := strconv.ParseFloat(newn, 64) return newf } func SubFloatToFloatStr(f float64, m int) string { newn := SubFloatToString(f, m) return newn } func GetYearDiffer(start_time, end_time string) int { t1, _ := time.ParseInLocation("2006-01-02", start_time, time.Local) t2, _ := time.ParseInLocation("2006-01-02", end_time, time.Local) age := t2.Year() - t1.Year() if t2.Month() < t1.Month() || (t2.Month() == t1.Month() && t2.Day() < t1.Day()) { age-- } return age } func GetSecondDifferByTime(start_time, end_time time.Time) int64 { diff := end_time.Unix() - start_time.Unix() return diff } func FixFloat(f float64, m int) float64 { newn := SubFloatToString(f+0.00000001, m) newf, _ := strconv.ParseFloat(newn, 64) return newf } func StrListToString(strList []string) (str string) { if len(strList) > 0 { for k, v := range strList { if k == 0 { str = v } else { str = str + "," + v } } return } return "" } func ErrNoRow() string { return " no row found" } func IsErrNoRow(err error) bool { if err == nil { return false } return errors.Is(err, gorm.ErrRecordNotFound) } func ValidateEmailFormatat(email string) bool { reg := regexp.MustCompile(RegularEmail) return reg.MatchString(email) } func ValidateMobileFormatat(mobileNum string) bool { reg := regexp.MustCompile(RegularMobile) return reg.MatchString(mobileNum) } func FileIsExist(filePath string) bool { _, err := os.Stat(filePath) return err == nil || os.IsExist(err) } func GetImgExt(file string) (ext string, err error) { var headerByte []byte headerByte = make([]byte, 8) fd, err := os.Open(file) if err != nil { return "", err } defer fd.Close() _, err = fd.Read(headerByte) if err != nil { return "", err } xStr := fmt.Sprintf("%x", headerByte) switch { case xStr == "89504e470d0a1a0a": ext = ".png" case xStr == "0000010001002020": ext = ".ico" case xStr == "0000020001002020": ext = ".cur" case xStr[:12] == "474946383961" || xStr[:12] == "474946383761": ext = ".gif" case xStr[:10] == "0000020000" || xStr[:10] == "0000100000": ext = ".tga" case xStr[:8] == "464f524d": ext = ".iff" case xStr[:8] == "52494646": ext = ".ani" case xStr[:4] == "4d4d" || xStr[:4] == "4949": ext = ".tiff" case xStr[:4] == "424d": ext = ".bmp" case xStr[:4] == "ffd8": ext = ".jpg" case xStr[:2] == "0a": ext = ".pcx" default: ext = "" } return ext, nil } func SaveImage(path string, img image.Image) (err error) { imgfile, err := os.Create(path) defer imgfile.Close() err = png.Encode(imgfile, img) return err } func DownloadImage(imgUrl string) (filePath string, err error) { imgPath := "./static/imgs/" fileName := path.Base(imgUrl) res, err := http.Get(imgUrl) if err != nil { fmt.Println("A error occurred!") return } defer res.Body.Close() reader := bufio.NewReaderSize(res.Body, 32*1024) filePath = imgPath + fileName file, err := os.Create(filePath) if err != nil { return } defer func() { file.Close() }() writer := bufio.NewWriter(file) written, _ := io.Copy(writer, reader) fmt.Printf("Total length: %d \n", written) return } func DownloadFile(fileUrl, fileDir string) (filePath string, err error) { filePathDir := "./static/imgs/" if fileDir != `` { err = os.MkdirAll(fileDir, DIR_MOD) if err != nil { return } filePathDir = fileDir } fileName := path.Base(fileUrl) res, err := http.Get(fileUrl) if err != nil { fmt.Println("A error occurred!") return } defer res.Body.Close() reader := bufio.NewReaderSize(res.Body, 32*1024) filePath = filePathDir + GetRandString(4) + fileName file, err := os.Create(filePath) if err != nil { return } defer func() { file.Close() }() writer := bufio.NewWriter(file) _, _ = io.Copy(writer, reader) return } func SaveBase64ToFile(content, path string) error { data, err := base64.StdEncoding.DecodeString(content) if err != nil { return err } f, err := os.Create(path) defer f.Close() if err != nil { return err } f.Write(data) return nil } func SaveBase64ToFileBySeek(content, path string) (err error) { data, err := base64.StdEncoding.DecodeString(content) exist, err := PathExists(path) if err != nil { return } if !exist { f, err := os.Create(path) if err != nil { return err } n, _ := f.Seek(0, 2) _, err = f.WriteAt([]byte(data), n) defer f.Close() } else { f, err := os.OpenFile(path, os.O_WRONLY, 0644) if err != nil { return err } n, _ := f.Seek(0, 2) _, err = f.WriteAt([]byte(data), n) defer f.Close() } return nil } func PathExists(path string) (bool, error) { _, err := os.Stat(path) if err == nil { return true, nil } if os.IsNotExist(err) { return false, nil } return false, err } func StartIndex(page, pagesize int) int { if page > 1 { return (page - 1) * pagesize } return 0 } func PageCount(count, pagesize int) int { if count%pagesize > 0 { return count/pagesize + 1 } else { return count / pagesize } } func TrimHtml(src string) string { re, _ := regexp.Compile("\\<[\\S\\s]+?\\>") src = re.ReplaceAllStringFunc(src, strings.ToLower) re, _ = regexp.Compile("\\") src = re.ReplaceAllString(src, "") re, _ = regexp.Compile("class[\\S\\s]+?>") src = re.ReplaceAllString(src, "") re, _ = regexp.Compile("\\<[\\S\\s]+?\\>") src = re.ReplaceAllString(src, "") src = strings.ReplaceAll(src, " ", " ") return strings.TrimSpace(src) } func TimeToTimestamp() { fmt.Println(time.Unix(1556164246, 0).Format("2006-01-02 15:04:05")) } func ToUnicode(text string) string { textQuoted := strconv.QuoteToASCII(text) textUnquoted := textQuoted[1 : len(textQuoted)-1] return textUnquoted } func VersionToInt(version string) int { version = strings.Replace(version, ".", "", -1) n, _ := strconv.Atoi(version) return n } func IsCheckInList(list []int, s int) bool { for _, v := range list { if v == s { return true } } return false } func round(num float64) int { return int(num + math.Copysign(0.5, num)) } func toFixed(num float64, precision int) float64 { output := math.Pow(10, float64(precision)) return float64(round(num*output)) / output } func GetWilsonScore(p, n float64) float64 { if p == 0 && n == 0 { return 0 } return toFixed(((p+1.9208)/(p+n)-1.96*math.Sqrt(p*n/(p+n)+0.9604)/(p+n))/(1+3.8416/(p+n)), 2) } func ChangeWordsToNum(str string) (numStr string) { words := ([]rune)(str) num := 0 n := 0 for i := 0; i < len(words); i++ { word := string(words[i : i+1]) switch word { case "万": if n == 0 { n = 1 } n = n * 10000 num = num*10000 + n n = 0 case "千": if n == 0 { n = 1 } n = n * 1000 num += n n = 0 case "百": if n == 0 { n = 1 } n = n * 100 num += n n = 0 case "十": if n == 0 { n = 1 } n = n * 10 num += n n = 0 case "一": n += 1 case "二": n += 2 case "三": n += 3 case "四": n += 4 case "五": n += 5 case "六": n += 6 case "七": n += 7 case "八": n += 8 case "九": n += 9 case "零": default: if n > 0 { num += n n = 0 } if num == 0 { numStr += word } else { numStr += strconv.Itoa(num) + word num = 0 } } } if n > 0 { num += n n = 0 } if num != 0 { numStr += strconv.Itoa(num) } return } func Sha1(data string) string { sha1 := sha1.New() sha1.Write([]byte(data)) return hex.EncodeToString(sha1.Sum([]byte(""))) } func GetVideoPlaySeconds(videoPath string) (playSeconds float64, err error) { cmd := fmt.Sprintf(`ffmpeg -i %s 2>&1 | grep 'Duration' | cut -d ' ' -f 4 | sed s/,//`, videoPath) out, err := exec.Command("bash", "-c", cmd).Output() if err != nil { return } outTimes := string(out) fmt.Println("outTimes:", outTimes) if outTimes != "" { timeArr := strings.Split(outTimes, ":") h := timeArr[0] m := timeArr[1] s := timeArr[2] hInt, err := strconv.Atoi(h) if err != nil { return playSeconds, err } mInt, err := strconv.Atoi(m) if err != nil { return playSeconds, err } s = strings.Trim(s, " ") s = strings.Trim(s, "\n") sInt, err := strconv.ParseFloat(s, 64) if err != nil { return playSeconds, err } playSeconds = float64(hInt)*3600 + float64(mInt)*60 + float64(sInt) } return } func GetMaxTradeCode(tradeCode string) (maxTradeCode string, err error) { tradeCode = strings.Replace(tradeCode, "W", "", -1) tradeCode = strings.Trim(tradeCode, " ") tradeCodeInt, err := strconv.Atoi(tradeCode) if err != nil { return } tradeCodeInt = tradeCodeInt + 1 maxTradeCode = fmt.Sprintf("W%06d", tradeCodeInt) return } func GetTradeCodeList(tradeCode string, num int) (tradeCodeList []string, err error) { tradeCode = strings.Replace(tradeCode, "W", "", -1) tradeCode = strings.Trim(tradeCode, " ") tradeCodeInt, err := strconv.Atoi(tradeCode) if err != nil { return } for i := 0; i < num; i++ { tradeCodeInt = tradeCodeInt + 1 tradeCodeList = append(tradeCodeList, fmt.Sprintf("W%06d", tradeCodeInt)) } return } func ConvertToFormatDay(excelDaysString string) string { baseDiffDay := 38719 //在网上工具计算的天数需要加2天,什么原因没弄清楚 curDiffDay := excelDaysString b, _ := strconv.Atoi(curDiffDay) realDiffDay := b - baseDiffDay realDiffSecond := realDiffDay * 24 * 3600 baseOriginSecond := 1136185445 resultTime := time.Unix(int64(baseOriginSecond+realDiffSecond), 0).Format("2006-01-02") return resultTime } func GetDaysDiff1900(date string) int { tStart, err := time.ParseInLocation(FormatDate, "1899-12-30", time.Local) if err != nil { return 0 } tEnd, err := time.ParseInLocation(FormatDate, date, time.Local) if err != nil { return 0 } duration := tEnd.Sub(tStart).Hours() / 24 days := int(duration) return days } func CheckPwd(pwd string) bool { compile := `([0-9a-z]+){6,12}|(a-z0-9]+){6,12}` reg := regexp.MustCompile(compile) flag := reg.MatchString(pwd) return flag } func GetMonthStartAndEnd(myYear string, myMonth string) (startDate, endDate string) { if len(myMonth) == 1 { myMonth = "0" + myMonth } yInt, _ := strconv.Atoi(myYear) timeLayout := "2006-01-02 15:04:05" loc, _ := time.LoadLocation("Local") theTime, _ := time.ParseInLocation(timeLayout, myYear+"-"+myMonth+"-01 00:00:00", loc) newMonth := theTime.Month() t1 := time.Date(yInt, newMonth, 1, 0, 0, 0, 0, time.Local).Format("2006-01-02") t2 := time.Date(yInt, newMonth+1, 0, 0, 0, 0, 0, time.Local).Format("2006-01-02") return t1, t2 } func TrimStr(str string) (str2 string) { if str == "" { return str } return strings.Replace(str, " ", "", -1) } func TrimLRStr(str string) (str2 string) { if str == "" { return str } str2 = strings.TrimLeft(str, " ") str2 = strings.TrimRight(str2, " ") return } func StrTimeToTime(strTime string) time.Time { timeLayout := "2006-01-02 15:04:05" //转化所需模板 loc, _ := time.LoadLocation("Local") //重要:获取时区 resultTime, _ := time.ParseInLocation(timeLayout, strTime, loc) return resultTime } func TimeToStr(data time.Time, format string) string { return data.Format(format) } func StrDateTimeToWeek(strTime string) string { var WeekDayMap = map[string]string{ "Monday": "周一", "Tuesday": "周二", "Wednesday": "周三", "Thursday": "周四", "Friday": "周五", "Saturday": "周六", "Sunday": "周日", } var ctime = StrTimeToTime(strTime).Format("2006-01-02") startday, _ := time.Parse("2006-01-02", ctime) staweek_int := startday.Weekday().String() return WeekDayMap[staweek_int] } func EnWeekToCnWeek(enWeekStr string) string { var WeekDayMap = map[string]string{ "Monday": "周一", "Tuesday": "周二", "Wednesday": "周三", "Thursday": "周四", "Friday": "周五", "Saturday": "周六", "Sunday": "周日", } cnWeekStr, ok := WeekDayMap[enWeekStr] if !ok { cnWeekStr = `周一` } return cnWeekStr } func TimeToStrYmd(time2 time.Time) string { var Ymd string year := time2.Year() month := time2.Format("1") day1 := time.Now().Day() Ymd = strconv.Itoa(year) + "年" + month + "月" + strconv.Itoa(day1) + "日" return Ymd } func TimeRemoveHms(strTime string) string { var Ymd string var resultTime = StrTimeToTime(strTime) year := resultTime.Year() month := resultTime.Format("01") day1 := resultTime.Day() if day1 < 10 { Ymd = strconv.Itoa(year) + "." + month + ".0" + strconv.Itoa(day1) } else { Ymd = strconv.Itoa(year) + "." + month + "." + strconv.Itoa(day1) } return Ymd } func TimeRemoveHms2(strTime string) string { var Ymd string var resultTime = StrTimeToTime(strTime) year := resultTime.Year() month := resultTime.Format("01") day1 := resultTime.Day() if day1 < 10 { Ymd = strconv.Itoa(year) + "-" + month + "-0" + strconv.Itoa(day1) } else { Ymd = strconv.Itoa(year) + "-" + month + "-" + strconv.Itoa(day1) } return Ymd } func ArticleLastTime(strTime string) string { var newTime string stamp, _ := time.ParseInLocation("2006-01-02 15:04:05", strTime, time.Local) diffTime := time.Now().Unix() - stamp.Unix() if diffTime <= 60 { newTime = "当前" } else if diffTime < 60*60 { newTime = strconv.FormatInt(diffTime/60, 10) + "分钟前" } else if diffTime < 24*60*60 { newTime = strconv.FormatInt(diffTime/(60*60), 10) + "小时前" } else if diffTime < 30*24*60*60 { newTime = strconv.FormatInt(diffTime/(24*60*60), 10) + "天前" } else if diffTime < 12*30*24*60*60 { newTime = strconv.FormatInt(diffTime/(30*24*60*60), 10) + "月前" } else { newTime = "1年前" } return newTime } func ConvertNumToCny(num float64) (str string, err error) { strNum := strconv.FormatFloat(num*100, 'f', 0, 64) sliceUnit := []string{"仟", "佰", "拾", "亿", "仟", "佰", "拾", "万", "仟", "佰", "拾", "元", "角", "分"} s := sliceUnit[len(sliceUnit)-len(strNum):] upperDigitUnit := map[string]string{"0": "零", "1": "壹", "2": "贰", "3": "叁", "4": "肆", "5": "伍", "6": "陆", "7": "柒", "8": "捌", "9": "玖"} for k, v := range strNum[:] { str = str + upperDigitUnit[string(v)] + s[k] } reg, err := regexp.Compile(`零角零分$`) str = reg.ReplaceAllString(str, "整") reg, err = regexp.Compile(`零角`) str = reg.ReplaceAllString(str, "零") reg, err = regexp.Compile(`零分$`) str = reg.ReplaceAllString(str, "整") reg, err = regexp.Compile(`零[仟佰拾]`) str = reg.ReplaceAllString(str, "零") reg, err = regexp.Compile(`零{2,}`) str = reg.ReplaceAllString(str, "零") reg, err = regexp.Compile(`零亿`) str = reg.ReplaceAllString(str, "亿") reg, err = regexp.Compile(`零万`) str = reg.ReplaceAllString(str, "万") reg, err = regexp.Compile(`零*元`) str = reg.ReplaceAllString(str, "元") reg, err = regexp.Compile(`亿零{0, 3}万`) str = reg.ReplaceAllString(str, "^元") reg, err = regexp.Compile(`零元`) str = reg.ReplaceAllString(str, "零") return } func GetNowWeekMonday() time.Time { offset := int(time.Monday - time.Now().Weekday()) if offset == 1 { //正好是周日,但是按照中国人的理解,周日是一周最后一天,而不是一周开始的第一天 offset = -6 } mondayTime := time.Now().AddDate(0, 0, offset) mondayTime = time.Date(mondayTime.Year(), mondayTime.Month(), mondayTime.Day(), 0, 0, 0, 0, mondayTime.Location()) return mondayTime } func GetNowWeekLastDay() time.Time { offset := int(time.Monday - time.Now().Weekday()) if offset == 1 { //正好是周日,但是按照中国人的理解,周日是一周最后一天,而不是一周开始的第一天 offset = -6 } firstDayTime := time.Now().AddDate(0, 0, offset) firstDayTime = time.Date(firstDayTime.Year(), firstDayTime.Month(), firstDayTime.Day(), 0, 0, 0, 0, firstDayTime.Location()).AddDate(0, 0, 6) lastDayTime := time.Date(firstDayTime.Year(), firstDayTime.Month(), firstDayTime.Day(), 23, 59, 59, 0, firstDayTime.Location()) return lastDayTime } func GetNowMonthFirstDay() time.Time { nowMonthFirstDay := time.Date(time.Now().Year(), time.Now().Month(), 1, 0, 0, 0, 0, time.Now().Location()) return nowMonthFirstDay } func GetNowMonthLastDay() time.Time { nowMonthLastDay := time.Date(time.Now().Year(), time.Now().Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1) nowMonthLastDay = time.Date(nowMonthLastDay.Year(), nowMonthLastDay.Month(), nowMonthLastDay.Day(), 23, 59, 59, 0, nowMonthLastDay.Location()) return nowMonthLastDay } func GetNowQuarterFirstDay() time.Time { month := int(time.Now().Month()) var nowQuarterFirstDay time.Time if month >= 1 && month <= 3 { nowQuarterFirstDay = time.Date(time.Now().Year(), 1, 1, 0, 0, 0, 0, time.Now().Location()) } else if month >= 4 && month <= 6 { nowQuarterFirstDay = time.Date(time.Now().Year(), 4, 1, 0, 0, 0, 0, time.Now().Location()) } else if month >= 7 && month <= 9 { nowQuarterFirstDay = time.Date(time.Now().Year(), 7, 1, 0, 0, 0, 0, time.Now().Location()) } else { nowQuarterFirstDay = time.Date(time.Now().Year(), 10, 1, 0, 0, 0, 0, time.Now().Location()) } return nowQuarterFirstDay } func GetNowQuarterLastDay() time.Time { month := int(time.Now().Month()) var nowQuarterLastDay time.Time if month >= 1 && month <= 3 { nowQuarterLastDay = time.Date(time.Now().Year(), 3, 31, 23, 59, 59, 0, time.Now().Location()) } else if month >= 4 && month <= 6 { nowQuarterLastDay = time.Date(time.Now().Year(), 6, 30, 23, 59, 59, 0, time.Now().Location()) } else if month >= 7 && month <= 9 { nowQuarterLastDay = time.Date(time.Now().Year(), 9, 30, 23, 59, 59, 0, time.Now().Location()) } else { nowQuarterLastDay = time.Date(time.Now().Year(), 12, 31, 23, 59, 59, 0, time.Now().Location()) } return nowQuarterLastDay } func GetNowHalfYearFirstDay() time.Time { month := int(time.Now().Month()) var nowHalfYearLastDay time.Time if month >= 1 && month <= 6 { nowHalfYearLastDay = time.Date(time.Now().Year(), 1, 1, 0, 0, 0, 0, time.Now().Location()) } else { nowHalfYearLastDay = time.Date(time.Now().Year(), 7, 1, 0, 0, 0, 0, time.Now().Location()) } return nowHalfYearLastDay } func GetNowHalfYearLastDay() time.Time { month := int(time.Now().Month()) var nowHalfYearLastDay time.Time if month >= 1 && month <= 6 { nowHalfYearLastDay = time.Date(time.Now().Year(), 6, 30, 23, 59, 59, 0, time.Now().Location()) } else { nowHalfYearLastDay = time.Date(time.Now().Year(), 12, 31, 23, 59, 59, 0, time.Now().Location()) } return nowHalfYearLastDay } func GetNowYearFirstDay() time.Time { nowYearFirstDay := time.Date(time.Now().Year(), 1, 1, 0, 0, 0, 0, time.Now().Location()) return nowYearFirstDay } func GetNowYearLastDay() time.Time { nowYearLastDay := time.Date(time.Now().Year(), 12, 31, 23, 59, 59, 0, time.Now().Location()) return nowYearLastDay } func CalculationDate(startDate, endDate time.Time) (beetweenDay string, err error) { numYear := endDate.Year() - startDate.Year() numMonth := int(endDate.Month()) - int(startDate.Month()) numDay := 0 endDateDays := getMonthDay(endDate.Year(), int(endDate.Month())) endDatePrevMonthDate := endDate.AddDate(0, -1, 0) endDatePrevMonthDays := getMonthDay(endDatePrevMonthDate.Year(), int(endDatePrevMonthDate.Month())) startDateMonthDays := getMonthDay(startDate.Year(), int(startDate.Month())) if endDate.Day() == endDateDays { numDay = startDateMonthDays - startDate.Day() + 1 if numDay == startDateMonthDays { numMonth++ numDay = 0 if numMonth == 12 { numYear++ numMonth = 0 } } } else { numDay = endDate.Day() - startDate.Day() + 1 } if numDay < 0 { numDay += endDatePrevMonthDays numMonth = numMonth - 1 } if numMonth < 0 { numMonth += 12 numYear = numYear - 1 } if numYear < 0 { err = errors.New("日期异常") return } if numYear > 0 { beetweenDay += fmt.Sprint(numYear, "年") } if numMonth > 0 { beetweenDay += fmt.Sprint(numMonth, "个月") } if numDay > 0 { beetweenDay += fmt.Sprint(numDay, "天") } return } func FormatPrice(price float64) (str string) { str = decimal.NewFromFloat(price).String() length := len(str) if length < 4 { return str } arr := strings.Split(str, ".") //用小数点符号分割字符串,为数组接收 length1 := len(arr[0]) if length1 < 4 { return str } count := (length1 - 1) / 3 for i := 0; i < count; i++ { arr[0] = arr[0][:length1-(i+1)*3] + "," + arr[0][length1-(i+1)*3:] } return strings.Join(arr, ".") //将一系列字符串连接为一个字符串,之间用sep来分隔。 } func getMonthDay(year, month int) (days int) { if month != 2 { if month == 4 || month == 6 || month == 9 || month == 11 { days = 30 } else { days = 31 } } else { if ((year%4) == 0 && (year%100) != 0) || (year%400) == 0 { days = 29 } else { days = 28 } } return } func SaveToFile(content, path string) error { f, err := os.Create(path) defer f.Close() if err != nil { return err } f.Write([]byte(content)) return nil } func GetAttendanceDetailSeconds(secondNum int) string { var timeStr string if secondNum <= 60 { if secondNum < 10 { timeStr = "0" + strconv.Itoa(secondNum) + "''" } else { timeStr = strconv.Itoa(secondNum) + "''" } } else { var remainderStr string remainderNum := secondNum % 60 minuteNum := secondNum / 60 if remainderNum < 10 { remainderStr = "0" + strconv.Itoa(remainderNum) + "''" } else { remainderStr = strconv.Itoa(remainderNum) + "''" } if minuteNum < 10 { timeStr = "0" + strconv.Itoa(minuteNum) + "'" + remainderStr } else { timeStr = strconv.Itoa(minuteNum) + "'" + remainderStr } } return timeStr } func SubStr(str string, subLen int) string { strRune := []rune(str) bodyRuneLen := len(strRune) if bodyRuneLen > subLen { bodyRuneLen = subLen } str = string(strRune[:bodyRuneLen]) return str } func HideString(src string, hideLen int) string { if src == "" { return src } str := []rune(src) if hideLen == 0 { hideLen = 4 } hideStr := "" for i := 0; i < hideLen; i++ { hideStr += "*" } strLen := len(str) if strLen == 1 { return string(str[:1]) + hideStr } if strLen <= hideLen+2 { return string(str[:1]) + hideStr + string(str[strLen-1:]) } subLen := strLen - hideLen //剩余需要展示的字符长度 decimal.NewFromFloat(2) frontLenDecimal := decimal.NewFromInt(int64(subLen)).Div(decimal.NewFromInt(2)) //前面需要展示的字符的长度 frontLen := frontLenDecimal.Floor().IntPart() return string(str[:frontLen]) + hideStr + string(str[frontLen+int64(hideLen):]) } func GetLocalIP() (ip string, err error) { addrs, err := net.InterfaceAddrs() if err != nil { return } for _, addr := range addrs { ipAddr, ok := addr.(*net.IPNet) if !ok { continue } if ipAddr.IP.IsLoopback() { continue } if !ipAddr.IP.IsGlobalUnicast() { continue } return ipAddr.IP.String(), nil } return } func GetPreWeekMonday() time.Time { nowMonday := GetNowWeekMonday() return nowMonday.AddDate(0, 0, -7) } func GetPreWeekLastDay() time.Time { nowSunday := GetNowWeekLastDay() return nowSunday.AddDate(0, 0, -7) } func GetNextWeekMonday() time.Time { nowMonday := GetNowWeekMonday() return nowMonday.AddDate(0, 0, 7) } func GetNextWeekLastDay() time.Time { nowSunday := GetNowWeekLastDay() return nowSunday.AddDate(0, 0, 7) } func DealRealName(realName string) string { if realName == "" { return "" } names := bytes.Runes([]byte(realName)) fistName := names[0] length := len(names) if length == 2 { return string(fistName) + "*" } else if length > 2 { lastName := names[length-1] return string(fistName) + "*" + string(lastName) } return realName } func Implode(idIntList []int) (str string) { for _, id := range idIntList { str += fmt.Sprint(id, ",") } str = str[:len(str)-1] return } func InArrayByInt(idIntList []int, searchId int) (has bool) { for _, id := range idIntList { if id == searchId { has = true return } } return } func InArrayByStr(idStrList []string, searchId string) (has bool) { for _, id := range idStrList { if id == searchId { has = true return } } return } func Intersection(a []string, b []string) (inter []string) { low, high := a, b if len(a) > len(b) { low = b high = a } done := false for i, l := range low { for j, h := range high { f1 := i + 1 f2 := j + 1 if l == h { inter = append(inter, h) if f1 < len(low) && f2 < len(high) { if low[f1] != high[f2] { done = true } } high = high[:j+copy(high[j:], high[j+1:])] break } } if done { break } } return } func Minus(a []string, b []string) []string { inter := make([]string, 0) mp := make(map[string]bool) for _, s := range a { if _, ok := mp[s]; !ok { mp[s] = true } } for _, s := range b { if _, ok := mp[s]; ok { delete(mp, s) } } for key := range mp { inter = append(inter, key) } return inter } func IntArr2joinString(arr []int, sep string) string { if len(arr) == 0 { return "" } strArr := make([]string, 0) for _, v := range arr { strArr = append(strArr, strconv.Itoa(v)) } return strings.Join(strArr, sep) } func GetOneMonthFirstDay(time2 time.Time) time.Time { nowMonthFirstDay := time.Date(time2.Year(), time2.Month(), 1, 0, 0, 0, 0, time.Now().Location()) return nowMonthFirstDay } func GetOneMonthLastDay(time2 time.Time) time.Time { nowMonthLastDay := time.Date(time2.Year(), time2.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1) nowMonthLastDay = time.Date(nowMonthLastDay.Year(), nowMonthLastDay.Month(), nowMonthLastDay.Day(), 23, 59, 59, 0, nowMonthLastDay.Location()) return nowMonthLastDay } func GetReportContentTextSub(content string) (contentSub string, err error) { content = html.UnescapeString(content) doc, err := goquery.NewDocumentFromReader(strings.NewReader(content)) docText := doc.Text() bodyRune := []rune(docText) bodyRuneLen := len(bodyRune) if bodyRuneLen > 200 { bodyRuneLen = 200 } body := string(bodyRune[:bodyRuneLen]) contentSub = body contentSub = strings.Replace(body, "Powered by Froala Editor", "", -1) return } func GetOrmInReplace(num int) string { return "?" } func RevSlice(slice []int) []int { for i, j := 0, len(slice)-1; i < j; i, j = i+1, j-1 { slice[i], slice[j] = slice[j], slice[i] } return slice } func ReplaceSpaceAndWrap(str string) string { str = strings.Replace(str, " ", "", -1) str = strings.Replace(str, "\n", "", -1) return str } func JoinStr2IntArr(str, sep string) (arr []int) { arr = make([]int, 0) if str == "" { return } if sep == "" { sep = "," } strArr := strings.Split(str, sep) if len(strArr) == 0 { return } for i := range strArr { v, e := strconv.Atoi(strArr[i]) if e != nil { continue } arr = append(arr, v) } return } func StrDateToDate(strTime string) time.Time { timeLayout := "2006-01-02" //转化所需模板 loc, _ := time.LoadLocation("Local") //重要:获取时区 resultTime, _ := time.ParseInLocation(timeLayout, strTime, loc) return resultTime } func GetTimeSubDay(t1, t2 time.Time) int { var day int swap := false if t1.Unix() > t2.Unix() { t1, t2 = t2, t1 swap = true } t1_ := t1.Add(time.Duration(t2.Sub(t1).Milliseconds()%86400000) * time.Millisecond) day = int(t2.Sub(t1).Hours() / 24) if t1_.Day() != t1.Day() { day += 1 } if swap { day = -day } return day } func StrFilterNonChinese(src *string) { var hzRegexp = regexp.MustCompile("^[\u4e00-\u9fa5]$") strn := "" for _, c := range *src { if hzRegexp.MatchString(string(c)) { strn += string(c) } } *src = strn } func MinusInt(a []int, b []int) []int { var diff []int mpA, mpB := make(map[int]bool), make(map[int]bool) for _, v := range a { mpA[v] = true } for _, v := range b { mpB[v] = true } for _, s := range a { if !mpB[s] { diff = append(diff, s) } } for _, s := range b { if !mpA[s] { diff = append(diff, s) } } return diff } func IntersectInt(nums1 []int, nums2 []int) []int { m := make(map[int]int) var res []int for _, num := range nums1 { m[num]++ } for _, num := range nums2 { if m[num] > 0 { res = append(res, num) m[num]-- } } return res } type MapSorter []Item type Item struct { Key int Val float64 } func NewMapSorter(m map[int]float64) MapSorter { ms := make(MapSorter, 0, len(m)) for k, v := range m { ms = append(ms, Item{k, v}) } return ms } func (ms MapSorter) Len() int { return len(ms) } func (ms MapSorter) Less(i, j int) bool { return ms[i].Val > ms[j].Val // 按值排序 } func (ms MapSorter) Swap(i, j int) { ms[i], ms[j] = ms[j], ms[i] } func FormatFileSize(fileSize int64) (size string) { if fileSize < 1024 { return fmt.Sprintf("%.2fB", float64(fileSize)/float64(1)) } else if fileSize < (1024 * 1024) { return fmt.Sprintf("%.2fKB", float64(fileSize)/float64(1024)) } else if fileSize < (1024 * 1024 * 1024) { return fmt.Sprintf("%.2fMB", float64(fileSize)/float64(1024*1024)) } else if fileSize < (1024 * 1024 * 1024 * 1024) { return fmt.Sprintf("%.2fGB", float64(fileSize)/float64(1024*1024*1024)) } else if fileSize < (1024 * 1024 * 1024 * 1024 * 1024) { return fmt.Sprintf("%.2fTB", float64(fileSize)/float64(1024*1024*1024*1024)) } else { //if fileSize < (1024 * 1024 * 1024 * 1024 * 1024 * 1024) return fmt.Sprintf("%.2fPB", float64(fileSize)/float64(1024*1024*1024*1024*1024)) } } func GetDurationFormatBySecond(sec int) (formatString string) { if sec == 0 { formatString = "0分钟" return } else if sec <= 60 && sec > 0 { formatString = "1分钟" return } duration := time.Duration(int64(sec)) * time.Second h := int(duration.Hours()) m := int(duration.Minutes()) % 60 if h > 0 { formatString = fmt.Sprintf("%d小时", h) } if m > 0 { formatString += fmt.Sprintf("%d分钟", m) } return } func GetDaysBetween2Date(format, date1Str, date2Str string) (int, error) { date1, err := time.ParseInLocation(format, date1Str, time.Local) if err != nil { return 0, err } date2, err := time.ParseInLocation(format, date2Str, time.Local) if err != nil { return 0, err } return int(date1.Sub(date2).Hours() / 24), nil } func GetFrequencyEn(frequency string) (frequencyEn string) { switch frequency { case "日度": frequencyEn = "day" return case "周度": frequencyEn = "week" return case "旬度": frequencyEn = "ten days" return case "月度": frequencyEn = "month" return case "季度": frequencyEn = "quarter" return case "年度": frequencyEn = "year" return } return } func GetLeadUnitEn(unit string) (unitEn string) { switch unit { case "天": unitEn = "day" return case "周": unitEn = "week" return case "月": unitEn = "month" return case "季": unitEn = "quarter" return case "年": unitEn = "year" return } return } func GetDateByDateType(dateType int, tmpStartDate, tmpEndDate string) (startDate, endDate string) { startDate = tmpStartDate endDate = tmpEndDate switch dateType { case 1: startDate = "2000-01-01" endDate = "" case 2: startDate = "2010-01-01" endDate = "" case 3: startDate = "2015-01-01" endDate = "" case 4: startDate = "2021-01-01" endDate = "" case 5: case 6: endDate = "" case 7: startDate = "2018-01-01" endDate = "" case 8: startDate = "2019-01-01" endDate = "" case 9: startDate = "2020-01-01" endDate = "" case 11: startDate = "2022-01-01" endDate = "" } { if strings.Count(startDate, "-") == 1 { startDate = startDate + "-01" } if strings.Count(endDate, "-") == 1 { endDate = endDate + "-01" } } return } func GetDateByDateTypeV2(dateType int, tmpStartDate, tmpEndDate string, startYear, yearMax int) (startDate, endDate string) { startDate = tmpStartDate endDate = tmpEndDate switch dateType { case 1: startDate = "2000-01-01" endDate = "" case 2: startDate = "2010-01-01" endDate = "" case 3: startDate = "2015-01-01" endDate = "" case 4: startDate = "2021-01-01" endDate = "" case 5: case 6: endDate = "" case 7: startDate = "2018-01-01" endDate = "" case 8: startDate = "2019-01-01" endDate = "" case 9: startDate = "2020-01-01" endDate = "" case 11: startDate = "2022-01-01" endDate = "" case DateTypeNYears: if startYear == 0 { //默认取最近5年 startYear = 5 } if yearMax == 0 { return } startYear = startYear - 1 baseDate, _ := time.Parse(FormatDate, fmt.Sprintf("%d-01-01", yearMax)) startDate = baseDate.AddDate(-startYear, 0, 0).Format(FormatDate) endDate = "" } { if strings.Count(startDate, "-") == 1 { startDate = startDate + "-01" } if strings.Count(endDate, "-") == 1 { endDate = endDate + "-01" } } return } func TimeTransferString(format string, t time.Time) string { str := t.Format(format) if t.IsZero() { return "" } return str } func GetDateByDateType2(dateType int, currDate time.Time) (startDate time.Time) { switch dateType { case 1: startDate = currDate.AddDate(0, -3, 0) case 2: startDate = currDate.AddDate(0, -6, 0) case 3: startDate = currDate.AddDate(-1, 0, 0) case 4: startDate = currDate.AddDate(-2, 0, 0) case 5: startDate = currDate.AddDate(-3, 0, 0) case 6: startDate = currDate.AddDate(-5, 0, 0) case 7: startDate = currDate.AddDate(-10, 0, 0) } return } func GetCeilNewNum(num float64, baseLen int) (newNum float64) { if num >= 1 { tmpNum := int(math.Ceil(num)) // 向上取整 str := strconv.Itoa(tmpNum) lenStr := len(str) if lenStr > baseLen { newNumStr := str[0:baseLen] newNumInt, _ := strconv.Atoi(newNumStr) newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen)) if newNum < num { newNumInt += 1 newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen)) } } else { newNum = float64(tmpNum) } return } else if num > 0 { str := strconv.FormatFloat(num, 'f', -1, 64) str = removeDecimalPoint(str) lenStr := len(str) if lenStr > baseLen { newNumStr := str[0:baseLen] newNumInt, _ := strconv.Atoi(newNumStr) newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64() if newNum < num { newNumInt += 1 newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64() } } else { newNum = num } } else if num > -1 { str := strconv.FormatFloat(num, 'f', -1, 64) str = removeDecimalPoint(str) lenStr := len(str) if lenStr > baseLen { newNumStr := str[0:baseLen] newNumInt, _ := strconv.Atoi(newNumStr) newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64() newNum = -newNum if newNum < num { newNumInt -= 1 newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64() newNum = -newNum } } else { newNum = num } if newNum == -0 { newNum = 0 } } else { // 小于等于-1 tmpNumFloat := math.Abs(num) tmpNum := int(math.Floor(tmpNumFloat)) // 向上取整 str := strconv.Itoa(tmpNum) lenStr := len(str) if lenStr > baseLen { newNumStr := str[0:baseLen] newNumInt, _ := strconv.Atoi(newNumStr) newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen)) newNum = -newNum if newNum < num { newNumInt -= 1 newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen)) newNum = -newNum } } else { newNum = float64(-tmpNum) } } return } func GetFloorNewNum(num float64, baseLen int) (newNum float64) { if num >= 1 { tmpNum := int(math.Floor(num)) // 向上取整 str := strconv.Itoa(tmpNum) lenStr := len(str) if lenStr > baseLen { newNumStr := str[0:baseLen] newNumInt, _ := strconv.Atoi(newNumStr) newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen)) if newNum < num { newNumInt -= 1 newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen)) } } else { newNum = float64(tmpNum) } return } else if num > 0 { str := strconv.FormatFloat(num, 'f', -1, 64) str = removeDecimalPoint(str) lenStr := len(str) if lenStr > baseLen { newNumStr := str[0:baseLen] newNumInt, _ := strconv.Atoi(newNumStr) newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64() if newNum > num { newNumInt -= 1 newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64() } } else { newNum = num } } else if num > -1 { str := strconv.FormatFloat(num, 'f', -1, 64) str = removeDecimalPoint(str) lenStr := len(str) if lenStr > baseLen { newNumStr := str[0:baseLen] newNumInt, _ := strconv.Atoi(newNumStr) newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64() newNum = -newNum if newNum > num { newNumInt += 1 newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64() newNum = -newNum } } else { newNum = num } if newNum == -0 { newNum = 0 } } else { // 小于等于-1 tmpNumFloat := math.Abs(num) tmpNum := int(math.Ceil(tmpNumFloat)) // 向上取整 str := strconv.Itoa(tmpNum) lenStr := len(str) if lenStr > baseLen { newNumStr := str[0:baseLen] newNumInt, _ := strconv.Atoi(newNumStr) newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen)) newNum = -newNum if newNum > num { newNumInt += 1 newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen)) newNum = -newNum } } else { newNum = float64(-tmpNum) } } return } func removeDecimalPoint(str string) string { str = str[strings.Index(str, ".")+1:] return str } func GetPredictEdbDayListByEndDate(startDate, endDate time.Time, frequency string) (dayList []time.Time) { switch frequency { case "日度": for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 1) { if currDate.Weekday() == time.Sunday || currDate.Weekday() == time.Saturday { continue } dayList = append(dayList, currDate) } case "周度": for currDate := startDate.AddDate(0, 0, 7); currDate.Before(endDate) || currDate.Equal(endDate); currDate = currDate.AddDate(0, 0, 7) { dayList = append(dayList, currDate) } case "旬度": for currDate := startDate.AddDate(0, 0, 1); currDate.Before(endDate) || currDate.Equal(endDate); { nextDate := currDate.AddDate(0, 0, 1) if nextDate.Day() == 11 || nextDate.Day() == 21 || nextDate.Day() == 1 { dayList = append(dayList, currDate) } currDate = nextDate } case "月度": for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); { currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1) if !currDate.After(endDate) && !currDate.Equal(startDate) { dayList = append(dayList, currDate) } currDate = currDate.AddDate(0, 0, 1) } case "季度": for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); { currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1) if !currDate.After(endDate) && !currDate.Equal(startDate) { if currDate.Month() == 3 || currDate.Month() == 6 || currDate.Month() == 9 || currDate.Month() == 12 { dayList = append(dayList, currDate) } } currDate = currDate.AddDate(0, 0, 1) } case "半年度": for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); { currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1) if !currDate.After(endDate) && !currDate.Equal(startDate) { if currDate.Month() == 6 || currDate.Month() == 12 { dayList = append(dayList, currDate) } } currDate = currDate.AddDate(0, 0, 1) } case "年度": for currDate := startDate; currDate.Before(endDate) || currDate.Equal(endDate); { currDate = time.Date(currDate.Year()+1, 12, 31, 0, 0, 0, 0, time.Now().Location()) if !currDate.After(endDate) && !currDate.Equal(startDate) { dayList = append(dayList, currDate) } } } return } func GetPredictEdbDayListByNum(startDate time.Time, num int, frequency string) (dayList []time.Time) { switch frequency { case "日度": for i := 1; i <= num; { currDate := startDate.AddDate(0, 0, i) if currDate.Weekday() == time.Sunday || currDate.Weekday() == time.Saturday { continue } dayList = append(dayList, currDate) i++ } case "周度": for i := 1; i <= num; { dayList = append(dayList, startDate.AddDate(0, 0, i*7)) i++ } case "旬度": currDate := startDate for i := 1; i <= num; { nextDate := currDate.AddDate(0, 0, 1) if nextDate.Day() == 11 || nextDate.Day() == 21 || nextDate.Day() == 1 { dayList = append(dayList, currDate) i++ } currDate = nextDate } case "月度": currDate := startDate for i := 1; i <= num; { currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1) if !currDate.Equal(startDate) { dayList = append(dayList, currDate) i++ } currDate = currDate.AddDate(0, 0, 1) } case "季度": currDate := startDate for i := 1; i <= num; { currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1) if !currDate.Equal(startDate) { if currDate.Month() == 3 || currDate.Month() == 6 || currDate.Month() == 9 || currDate.Month() == 12 { dayList = append(dayList, currDate) i++ } } currDate = currDate.AddDate(0, 0, 1) } case "半年度": currDate := startDate for i := 1; i <= num; { currDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1) if !currDate.Equal(startDate) { if currDate.Month() == 6 || currDate.Month() == 12 { dayList = append(dayList, currDate) i++ } } currDate = currDate.AddDate(0, 0, 1) } case "年度": currDate := startDate for i := 1; i <= num; { currDate = time.Date(currDate.Year()+1, 12, 31, 0, 0, 0, 0, time.Now().Location()) if !currDate.Equal(startDate) { dayList = append(dayList, currDate) i++ } } } return } func FormatMixTableDataShowValue(x float64) (res string) { res = strconv.FormatFloat(x, 'f', -1, 64) return } func FormatTableDataShowValue(x float64) (res string) { if x > 1 || x < -1 { res = decimal.NewFromFloat(x).Round(2).String() return } xStr := strconv.FormatFloat(x, 'f', -10, 64) xParts := strings.Split(xStr, ".") if len(xParts) < 2 { res = fmt.Sprint(x) return } xDecimals := len(xParts[1]) if xDecimals > 2 { parts := xParts[1] partLen := len(xParts[1]) if partLen <= 2 { res = xParts[0] + "." + parts return } one := 0 for k, p := range parts { if p != 48 { if one == 0 { one = k + 1 } } } n := partLen - one if n >= 1 { n -= 1 } else { one -= 1 } var partFloat float64 partInt, _ := strconv.Atoi(parts) partFloat, _ = decimal.NewFromInt(int64(partInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(n)))).Float64() partFloat = math.Round(partFloat) partFloat, _ = decimal.NewFromFloat(partFloat).Div(decimal.NewFromFloat(math.Pow(10, float64(one+1)))).Float64() resParts := strings.Split(fmt.Sprint(partFloat), ".") resPart := "" if len(resParts) > 1 { resPart = resParts[1] } else { resPart = resParts[0] } res = xParts[0] + "." + resPart } if xDecimals <= 2 { xDecimalsStr := xParts[1] x, _ = strconv.ParseFloat(xParts[0]+"."+xDecimalsStr, 64) res = xParts[0] + "." + xDecimalsStr } return } func DealDateTimeZero(t time.Time, format string) (timeStr string) { if !t.IsZero() { timeStr = t.Format(format) } return } func IsPercentage(input string) (bool, string) { percentagePattern := `^-?(\d+(\.\d+)?)%$` match, _ := regexp.MatchString(percentagePattern, input) if match { re := regexp.MustCompile(percentagePattern) match := re.FindStringSubmatch(input) sign := "-" if input[0] != '-' { sign = "" } return true, sign + match[1] } return false, "" } func GetLikeKeyword(keyword string) string { return `%` + keyword + `%` } func GetLikeKeywordPars(pars []interface{}, keyword string, num int) (newPars []interface{}) { newPars = pars if newPars == nil { newPars = make([]interface{}, 0) } for i := 1; i <= num; i++ { newPars = append(newPars, `%`+keyword+`%`) } return } func GetSign(nonce, timestamp, appId, secret string) (sign string) { signStrMap := map[string]string{ "nonce": nonce, "timestamp": timestamp, "appid": appId, } keys := make([]string, 0, len(signStrMap)) for k := range signStrMap { keys = append(keys, k) } sort.Strings(keys) var signStr string for _, k := range keys { signStr += k + "=" + signStrMap[k] + "&" } signStr = strings.Trim(signStr, "&") fmt.Println("signStr:" + signStr) sign = HmacSha256ToBase64(secret, signStr) return } func HmacSha256ToBase64(key string, data string) string { return base64.URLEncoding.EncodeToString(HmacSha256(key, data)) } func HmacSha256(key string, data string) []byte { mac := hmac.New(sha256.New, []byte(key)) _, _ = mac.Write([]byte(data)) return mac.Sum(nil) } func PascalToSnake(s string) string { var result []rune for i, char := range s { if i > 0 && unicode.IsUpper(char) { result = append(result, '_') } result = append(result, unicode.ToLower(char)) } return string(result) } func ArrContainsStr(arr []string, str string) bool { for _, v := range arr { if v == str { return true } } return false } func GetColorMap() map[int]string { colorMap := make(map[int]string) colors := []string{"#0000FF", "#FF0000", "#999999", "#000000", "#7CB5EC", "#90ED7D", "#F7A35C", "#8085E9", "#F15C80", "#E4D354", "#2B908F", "#F45B5B", "#91E8E1", "#FDA8C7", "#8A4294", "#578B5A", "#0033FF", "#849EC1", "#FFDF0C", "#005496", "#00F0FF", "#4D535B", "#4F4C34", "#804141", "#86BABD", "#8AA3FF", "#960000", "#A173DB", "#A39340", "#CE814A", "#D1D2E6", "#EAB7B7", "#FF2E7A", "#FF4AF8", "#FF785B", "#FF9696", "#FFA800", "#FFBC97", "#FFDFDF"} for k, v := range colors { colorMap[k] = v } return colorMap } func isValidSrc(src string) bool { parsedURL, err := url.Parse(src) if err != nil { validSchemes := regexp.MustCompile(`^data:image\/.*;base64,.*$`) return validSchemes.MatchString(src) } if parsedURL.Host == "" || (parsedURL.Scheme != "http" && parsedURL.Scheme != "https") { validSchemes := regexp.MustCompile(`^data:image\/.*;base64,.*$`) return validSchemes.MatchString(src) } return true } func ContentXssCheck(content string) (err error) { node, err := xhtml.Parse(strings.NewReader(content)) if err != nil { err = fmt.Errorf(" html.Parse Err: %v", err) return } var visit func(n *xhtml.Node) error visit = func(n *xhtml.Node) error { if n.Type == xhtml.ElementNode { lowerData := strings.ToLower(n.Data) switch lowerData { case "script", "javascript": err = fmt.Errorf(" script is forbidden") return err default: for _, attr := range n.Attr { //判断事件 lowerKey := strings.ToLower(attr.Key) lowerVal := strings.ToLower(attr.Val) if lowerKey == "src" || lowerKey == "dynsrc" || lowerKey == "background" || lowerKey == "lowsrc" { if lowerVal != `` && !isValidSrc(lowerVal) { err = fmt.Errorf("invalid src attribute value: %s", attr.Val) return err } } if strings.HasPrefix(lowerKey, "on") { err = fmt.Errorf("the event is forbidden: %s:%s", attr.Key, attr.Val) return err } if lowerKey == "style" { if strings.Contains(lowerVal, "javascript:") || strings.Contains(lowerVal, "script:") { err = fmt.Errorf("invalid style attribute value: %s", attr.Val) return err } } } } } for c := n.FirstChild; c != nil; c = c.NextSibling { if err = visit(c); err != nil { return err } } return nil } if err = visit(node); err != nil { return } return } func ContentXssFilter(content string) (cleanContent string) { p := customXssPolicy() cleanContent = p.Sanitize( content, ) return } func customXssPolicy() (p *bluemonday.Policy) { p = bluemonday.UGCPolicy() p.AllowElements("iframe") p.AllowAttrs("width").OnElements("iframe") p.AllowAttrs("height").OnElements("iframe") p.AllowAttrs("src").OnElements("iframe") p.AllowAttrs("frameborder").Matching(bluemonday.Number).OnElements("iframe") p.AllowAttrs("allow").Matching(regexp.MustCompile(`[a-z; -]*`)).OnElements("iframe") p.AllowAttrs("allowfullscreen").OnElements("iframe") return } func SecondsToHHMMSS(seconds int) string { duration := time.Duration(seconds) * time.Second hours := int(duration.Hours()) minutes := int(duration.Minutes()) % 60 seconds = int(duration.Seconds()) % 60 if hours > 0 { return fmt.Sprintf("%02d:%02d:%02d", hours, minutes, seconds) } else { return fmt.Sprintf("%02d:%02d", minutes, seconds) } } func MillisecondsToHHMMSS(ms int) string { duration := time.Duration(ms) * time.Millisecond hours := int(duration / (time.Hour)) minutes := int((duration % (time.Hour)) / time.Minute) seconds := int((duration % time.Hour % time.Minute) / time.Second) hourStr := strconv.Itoa(hours) minuteStr := strconv.Itoa(minutes) secondStr := strconv.Itoa(seconds) if len(hourStr) == 1 { hourStr = "0" + hourStr } if len(minuteStr) == 1 { minuteStr = "0" + minuteStr } if len(secondStr) == 1 { secondStr = "0" + secondStr } return hourStr + ":" + minuteStr + ":" + secondStr } func ByteToMB(byteCount int) float64 { return float64(byteCount) / (1024 * 1024) } func DateConvMysqlConvMongo(dateCon string) string { cond := "" switch dateCon { case "=": cond = "$eq" case "<": cond = "$lt" case "<=": cond = "$lte" case ">": cond = "$gt" case ">=": cond = "$gte" } return cond } func HandleSystemAppointDateT(currDate time.Time, appointDay, frequency string) (date string, err error, errMsg string) { switch frequency { case "本周": day := int(currDate.Weekday()) if day == 0 { // 周日 day = 7 } num := 0 switch appointDay { case "周一": num = 1 case "周二": num = 2 case "周三": num = 3 case "周四": num = 4 case "周五": num = 5 case "周六": num = 6 case "周日": num = 7 } day = num - day date = currDate.AddDate(0, 0, day).Format(FormatDate) case "本旬": day := currDate.Day() var tmpDate time.Time switch appointDay { case "第一天": if day <= 10 { tmpDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, currDate.Location()) } else if day <= 20 { tmpDate = time.Date(currDate.Year(), currDate.Month(), 11, 0, 0, 0, 0, currDate.Location()) } else { tmpDate = time.Date(currDate.Year(), currDate.Month(), 21, 0, 0, 0, 0, currDate.Location()) } case "最后一天": if day <= 10 { tmpDate = time.Date(currDate.Year(), currDate.Month(), 10, 0, 0, 0, 0, currDate.Location()) } else if day <= 20 { tmpDate = time.Date(currDate.Year(), currDate.Month(), 20, 0, 0, 0, 0, currDate.Location()) } else { tmpDate = time.Date(currDate.Year(), currDate.Month()+1, 1, 0, 0, 0, 0, currDate.Location()).AddDate(0, 0, -1) } } date = tmpDate.Format(FormatDate) case "本月": var tmpDate time.Time switch appointDay { case "第一天": tmpDate = time.Date(currDate.Year(), currDate.Month(), 1, 0, 0, 0, 0, currDate.Location()) case "最后一天": tmpDate = time.Date(currDate.Year(), currDate.Month()+1, 1, 0, 0, 0, 0, currDate.Location()).AddDate(0, 0, -1) } date = tmpDate.Format(FormatDate) case "本季": month := currDate.Month() var tmpDate time.Time switch appointDay { case "第一天": if month <= 3 { tmpDate = time.Date(currDate.Year(), 1, 1, 0, 0, 0, 0, currDate.Location()) } else if month <= 6 { tmpDate = time.Date(currDate.Year(), 4, 1, 0, 0, 0, 0, currDate.Location()) } else if month <= 9 { tmpDate = time.Date(currDate.Year(), 7, 1, 0, 0, 0, 0, currDate.Location()) } else { tmpDate = time.Date(currDate.Year(), 10, 1, 0, 0, 0, 0, currDate.Location()) } case "最后一天": if month <= 3 { tmpDate = time.Date(currDate.Year(), 3, 31, 0, 0, 0, 0, currDate.Location()) } else if month <= 6 { tmpDate = time.Date(currDate.Year(), 6, 30, 0, 0, 0, 0, currDate.Location()) } else if month <= 9 { tmpDate = time.Date(currDate.Year(), 9, 30, 0, 0, 0, 0, currDate.Location()) } else { tmpDate = time.Date(currDate.Year(), 12, 31, 0, 0, 0, 0, currDate.Location()) } } date = tmpDate.Format(FormatDate) case "本半年": month := currDate.Month() var tmpDate time.Time switch appointDay { case "第一天": if month <= 6 { tmpDate = time.Date(currDate.Year(), 1, 1, 0, 0, 0, 0, currDate.Location()) } else { tmpDate = time.Date(currDate.Year(), 7, 1, 0, 0, 0, 0, currDate.Location()) } case "最后一天": if month <= 6 { tmpDate = time.Date(currDate.Year(), 6, 30, 0, 0, 0, 0, currDate.Location()) } else { tmpDate = time.Date(currDate.Year(), 12, 31, 0, 0, 0, 0, currDate.Location()) } } date = tmpDate.Format(FormatDate) case "本年": var tmpDate time.Time switch appointDay { case "第一天": tmpDate = time.Date(currDate.Year(), 1, 1, 0, 0, 0, 0, currDate.Location()) case "最后一天": tmpDate = time.Date(currDate.Year(), 12, 31, 0, 0, 0, 0, currDate.Location()) } date = tmpDate.Format(FormatDate) default: errMsg = "错误的日期频度:" + frequency err = errors.New(errMsg) return } return } func CompareFloatByOpStrings(op string, a, b float64) bool { switch op { case "=": return a == b case ">": return a > b case ">=": return a >= b case "<=": return a <= b case "<": return a < b } return false } func GetDuration(filePath string) (duration string, err error) { cmd := exec.Command("ffprobe", "-i", filePath, "-show_entries", "format=duration", "-v", "quiet", "-of", "csv=p=0") var out bytes.Buffer cmd.Stdout = &out err = cmd.Run() if err != nil { return "", err } re := regexp.MustCompile(`\d+\.\d+`) duration = re.FindString(out.String()) return duration, nil } func ForwardPars(pars []interface{}, values ...interface{}) []interface{} { if len(values) <= 0 { return pars } return append(values, pars...) } func AppendPars(pars []interface{}, values ...interface{}) []interface{} { if len(values) <= 0 { return pars } return append(pars, values...) } func ExcelDateToDate(excelDate string) time.Time { excelTime := time.Date(1899, time.December, 30, 0, 0, 0, 0, time.UTC) var days, _ = strconv.Atoi(excelDate) return excelTime.Add(time.Second * time.Duration(days*86400)) } func ExcelTimeToTime(excelTime float64) time.Time { baseTime := time.Date(1899, time.December, 30, 0, 0, 0, 0, time.UTC) // 注意:这里使用 1899-12-30 是因为 Excel 的日期系统从这一天开始(考虑到 1900 年错误) hours := int(excelTime * 24) minutes := int(math.Mod(excelTime*24*60, 60)) seconds := int(math.Mod(excelTime*24*60*60, 60)) return time.Date(baseTime.Year(), baseTime.Month(), baseTime.Day(), hours, minutes, seconds, 0, time.UTC) } func TimeToFormatDate(t time.Time) string { if t.IsZero() { return "0000-00-00" } return t.Format(FormatDate) } func TimeToFormatDateTime(t time.Time) string { if t.IsZero() { return "0000-00-00 00:00:00" } return t.Format(FormatDateTime) } func FormatDateStrToTime(timeStr string) (currTime time.Time, err error) { if timeStr != `` { return } if strings.Contains(timeStr, "T") { return time.ParseInLocation(FormatDateWallWithLoc, timeStr, time.Local) } if strings.Contains(timeStr, ":") { return time.ParseInLocation(FormatDateTime, timeStr, time.Local) } return time.ParseInLocation(FormatDate, timeStr, time.Local) } func GormDateStrToDateTimeStr(originalString string) (formatStr string) { formatStr = originalString if !strings.Contains(originalString, "T") { return } t, err := time.Parse(FormatDateWallWithLoc, originalString) if err != nil { fmt.Println("Error parsing time:", err) return } formatStr = t.Format(FormatDateTime) return } func GormDateStrToDateStr(originalString string) (formatStr string) { formatStr = originalString if !strings.Contains(originalString, "T") { return } t, err := time.Parse(FormatDateWallWithLoc, originalString) if err != nil { fmt.Println("Error parsing time:", err) return } formatStr = t.Format(FormatDate) return } func ReflectSetField(original, result interface{}, fieldsToProcess []string, formatDate string) error { fieldsToProcess = []string{"CreateTime", "ModifyTime", "BaseModifyTime", "StartDate", "EndDate"} for _, fieldName := range fieldsToProcess { value := reflect.ValueOf(original).FieldByName(fieldName) if !value.IsValid() { continue } if reflect.TypeOf(time.Time{}) == value.Type() { t := value.Interface().(time.Time) tmpVal := `` if t.IsZero() { tmpVal = `` } else { tmpVal = t.Format(formatDate) } v := reflect.ValueOf(result).Elem() f := v.FieldByName(fieldName) if !f.IsValid() { fmt.Printf("Field '%s' not found in the struct.\n", fieldName) continue } f.Set(reflect.ValueOf(tmpVal)) } } return nil } func RoundNumber(num string, decimalPlaces int, hasPercent bool) string { numDecimal, _ := decimal.NewFromString(num) if hasPercent { numDecimal = numDecimal.Mul(decimal.NewFromInt(100)) } numFloat, _ := numDecimal.Round(int32(decimalPlaces)).Float64() numStr := strconv.FormatFloat(numFloat, 'f', decimalPlaces, 64) if hasPercent { numStr += "%" } return numStr }