common.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983
  1. package utils
  2. import (
  3. "bufio"
  4. "crypto/md5"
  5. "crypto/sha1"
  6. "encoding/base64"
  7. "encoding/hex"
  8. "encoding/json"
  9. "fmt"
  10. "github.com/shopspring/decimal"
  11. "image"
  12. "image/png"
  13. "io"
  14. "math"
  15. "math/rand"
  16. "net/http"
  17. "os"
  18. "os/exec"
  19. "path"
  20. "regexp"
  21. "strconv"
  22. "strings"
  23. "time"
  24. )
  25. // 随机数种子
  26. var rnd = rand.New(rand.NewSource(time.Now().UnixNano()))
  27. func GetRandString(size int) string {
  28. 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", "!", "@", "#", "$", "%", "^", "&", "*"}
  29. randomSb := ""
  30. digitSize := len(allLetterDigit)
  31. for i := 0; i < size; i++ {
  32. randomSb += allLetterDigit[rnd.Intn(digitSize)]
  33. }
  34. return randomSb
  35. }
  36. func GetRandStringNoSpecialChar(size int) string {
  37. 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"}
  38. randomSb := ""
  39. digitSize := len(allLetterDigit)
  40. for i := 0; i < size; i++ {
  41. randomSb += allLetterDigit[rnd.Intn(digitSize)]
  42. }
  43. return randomSb
  44. }
  45. func StringsToJSON(str string) string {
  46. rs := []rune(str)
  47. jsons := ""
  48. for _, r := range rs {
  49. rint := int(r)
  50. if rint < 128 {
  51. jsons += string(r)
  52. } else {
  53. jsons += "\\u" + strconv.FormatInt(int64(rint), 16) // json
  54. }
  55. }
  56. return jsons
  57. }
  58. // 序列化
  59. func ToString(v interface{}) string {
  60. data, _ := json.Marshal(v)
  61. return string(data)
  62. }
  63. // md5加密
  64. func MD5(data string) string {
  65. m := md5.Sum([]byte(data))
  66. return hex.EncodeToString(m[:])
  67. }
  68. // 获取数字随机字符
  69. func GetRandDigit(n int) string {
  70. return fmt.Sprintf("%0"+strconv.Itoa(n)+"d", rnd.Intn(int(math.Pow10(n))))
  71. }
  72. // 获取随机数
  73. func GetRandNumber(n int) int {
  74. return rnd.Intn(n)
  75. }
  76. func GetRandInt(min, max int) int {
  77. if min >= max || min == 0 || max == 0 {
  78. return max
  79. }
  80. return rand.Intn(max-min) + min
  81. }
  82. func GetToday(format string) string {
  83. today := time.Now().Format(format)
  84. return today
  85. }
  86. // 获取今天剩余秒数
  87. func GetTodayLastSecond() time.Duration {
  88. today := GetToday(FormatDate) + " 23:59:59"
  89. end, _ := time.ParseInLocation(FormatDateTime, today, time.Local)
  90. return time.Duration(end.Unix()-time.Now().Local().Unix()) * time.Second
  91. }
  92. // 处理出生日期函数
  93. func GetBrithDate(idcard string) string {
  94. l := len(idcard)
  95. var s string
  96. if l == 15 {
  97. s = "19" + idcard[6:8] + "-" + idcard[8:10] + "-" + idcard[10:12]
  98. return s
  99. }
  100. if l == 18 {
  101. s = idcard[6:10] + "-" + idcard[10:12] + "-" + idcard[12:14]
  102. return s
  103. }
  104. return GetToday(FormatDate)
  105. }
  106. // 处理性别
  107. func WhichSexByIdcard(idcard string) string {
  108. var sexs = [2]string{"女", "男"}
  109. length := len(idcard)
  110. if length == 18 {
  111. sex, _ := strconv.Atoi(string(idcard[16]))
  112. return sexs[sex%2]
  113. } else if length == 15 {
  114. sex, _ := strconv.Atoi(string(idcard[14]))
  115. return sexs[sex%2]
  116. }
  117. return "男"
  118. }
  119. // 截取小数点后几位
  120. func SubFloatToString(f float64, m int) string {
  121. n := strconv.FormatFloat(f, 'f', -1, 64)
  122. if n == "" {
  123. return ""
  124. }
  125. if m >= len(n) {
  126. return n
  127. }
  128. newn := strings.Split(n, ".")
  129. if m == 0 {
  130. return newn[0]
  131. }
  132. if len(newn) < 2 || m >= len(newn[1]) {
  133. return n
  134. }
  135. return newn[0] + "." + newn[1][:m]
  136. }
  137. // 截取小数点后几位
  138. func SubFloatToFloat(f float64, m int) float64 {
  139. newn := SubFloatToString(f, m)
  140. newf, _ := strconv.ParseFloat(newn, 64)
  141. return newf
  142. }
  143. // 截取小数点后几位
  144. func SubFloatToFloatStr(f float64, m int) string {
  145. newn := SubFloatToString(f, m)
  146. return newn
  147. }
  148. // 获取相差时间-年
  149. func GetYearDiffer(start_time, end_time string) int {
  150. t1, _ := time.ParseInLocation("2006-01-02", start_time, time.Local)
  151. t2, _ := time.ParseInLocation("2006-01-02", end_time, time.Local)
  152. age := t2.Year() - t1.Year()
  153. if t2.Month() < t1.Month() || (t2.Month() == t1.Month() && t2.Day() < t1.Day()) {
  154. age--
  155. }
  156. return age
  157. }
  158. // 获取相差时间-秒
  159. func GetSecondDifferByTime(start_time, end_time time.Time) int64 {
  160. diff := end_time.Unix() - start_time.Unix()
  161. return diff
  162. }
  163. func FixFloat(f float64, m int) float64 {
  164. newn := SubFloatToString(f+0.00000001, m)
  165. newf, _ := strconv.ParseFloat(newn, 64)
  166. return newf
  167. }
  168. // 将字符串数组转化为逗号分割的字符串形式 ["str1","str2","str3"] >>> "str1,str2,str3"
  169. func StrListToString(strList []string) (str string) {
  170. if len(strList) > 0 {
  171. for k, v := range strList {
  172. if k == 0 {
  173. str = v
  174. } else {
  175. str = str + "," + v
  176. }
  177. }
  178. return
  179. }
  180. return ""
  181. }
  182. // 数据没有记录
  183. func ErrNoRow() string {
  184. return "<QuerySeter> no row found"
  185. }
  186. // 校验邮箱格式
  187. func ValidateEmailFormatat(email string) bool {
  188. reg := regexp.MustCompile(RegularEmail)
  189. return reg.MatchString(email)
  190. }
  191. // 验证是否是手机号
  192. func ValidateMobileFormatat(mobileNum string) bool {
  193. reg := regexp.MustCompile(RegularMobile)
  194. return reg.MatchString(mobileNum)
  195. }
  196. // 判断文件是否存在
  197. func FileIsExist(filePath string) bool {
  198. _, err := os.Stat(filePath)
  199. return err == nil || os.IsExist(err)
  200. }
  201. // 获取图片扩展名
  202. func GetImgExt(file string) (ext string, err error) {
  203. var headerByte []byte
  204. headerByte = make([]byte, 8)
  205. fd, err := os.Open(file)
  206. if err != nil {
  207. return "", err
  208. }
  209. defer fd.Close()
  210. _, err = fd.Read(headerByte)
  211. if err != nil {
  212. return "", err
  213. }
  214. xStr := fmt.Sprintf("%x", headerByte)
  215. switch {
  216. case xStr == "89504e470d0a1a0a":
  217. ext = ".png"
  218. case xStr == "0000010001002020":
  219. ext = ".ico"
  220. case xStr == "0000020001002020":
  221. ext = ".cur"
  222. case xStr[:12] == "474946383961" || xStr[:12] == "474946383761":
  223. ext = ".gif"
  224. case xStr[:10] == "0000020000" || xStr[:10] == "0000100000":
  225. ext = ".tga"
  226. case xStr[:8] == "464f524d":
  227. ext = ".iff"
  228. case xStr[:8] == "52494646":
  229. ext = ".ani"
  230. case xStr[:4] == "4d4d" || xStr[:4] == "4949":
  231. ext = ".tiff"
  232. case xStr[:4] == "424d":
  233. ext = ".bmp"
  234. case xStr[:4] == "ffd8":
  235. ext = ".jpg"
  236. case xStr[:2] == "0a":
  237. ext = ".pcx"
  238. default:
  239. ext = ""
  240. }
  241. return ext, nil
  242. }
  243. // 保存图片
  244. func SaveImage(path string, img image.Image) (err error) {
  245. //需要保持的文件
  246. imgfile, err := os.Create(path)
  247. defer imgfile.Close()
  248. // 以PNG格式保存文件
  249. err = png.Encode(imgfile, img)
  250. return err
  251. }
  252. // 下载图片
  253. func DownloadImage(imgUrl string) (filePath string, err error) {
  254. imgPath := "./static/imgs/"
  255. fileName := path.Base(imgUrl)
  256. res, err := http.Get(imgUrl)
  257. if err != nil {
  258. fmt.Println("A error occurred!")
  259. return
  260. }
  261. defer res.Body.Close()
  262. // 获得get请求响应的reader对象
  263. reader := bufio.NewReaderSize(res.Body, 32*1024)
  264. filePath = imgPath + fileName
  265. file, err := os.Create(filePath)
  266. if err != nil {
  267. return
  268. }
  269. // 获得文件的writer对象
  270. writer := bufio.NewWriter(file)
  271. written, _ := io.Copy(writer, reader)
  272. fmt.Printf("Total length: %d \n", written)
  273. return
  274. }
  275. // 保存base64数据为文件
  276. func SaveBase64ToFile(content, path string) error {
  277. data, err := base64.StdEncoding.DecodeString(content)
  278. if err != nil {
  279. return err
  280. }
  281. f, err := os.Create(path)
  282. defer f.Close()
  283. if err != nil {
  284. return err
  285. }
  286. f.Write(data)
  287. return nil
  288. }
  289. func SaveBase64ToFileBySeek(content, path string) (err error) {
  290. data, err := base64.StdEncoding.DecodeString(content)
  291. exist, err := PathExists(path)
  292. if err != nil {
  293. return
  294. }
  295. if !exist {
  296. f, err := os.Create(path)
  297. if err != nil {
  298. return err
  299. }
  300. n, _ := f.Seek(0, 2)
  301. // 从末尾的偏移量开始写入内容
  302. _, err = f.WriteAt([]byte(data), n)
  303. defer f.Close()
  304. } else {
  305. f, err := os.OpenFile(path, os.O_WRONLY, 0644)
  306. if err != nil {
  307. return err
  308. }
  309. n, _ := f.Seek(0, 2)
  310. // 从末尾的偏移量开始写入内容
  311. _, err = f.WriteAt([]byte(data), n)
  312. defer f.Close()
  313. }
  314. return nil
  315. }
  316. func PathExists(path string) (bool, error) {
  317. _, err := os.Stat(path)
  318. if err == nil {
  319. return true, nil
  320. }
  321. if os.IsNotExist(err) {
  322. return false, nil
  323. }
  324. return false, err
  325. }
  326. func StartIndex(page, pagesize int) int {
  327. if page > 1 {
  328. return (page - 1) * pagesize
  329. }
  330. return 0
  331. }
  332. func PageCount(count, pagesize int) int {
  333. if count%pagesize > 0 {
  334. return count/pagesize + 1
  335. } else {
  336. return count / pagesize
  337. }
  338. }
  339. func TrimHtml(src string) string {
  340. //将HTML标签全转换成小写
  341. re, _ := regexp.Compile("\\<[\\S\\s]+?\\>")
  342. src = re.ReplaceAllStringFunc(src, strings.ToLower)
  343. re, _ = regexp.Compile("\\<img[\\S\\s]+?\\>")
  344. src = re.ReplaceAllString(src, "[图片]")
  345. re, _ = regexp.Compile("class[\\S\\s]+?>")
  346. src = re.ReplaceAllString(src, "")
  347. re, _ = regexp.Compile("\\<[\\S\\s]+?\\>")
  348. src = re.ReplaceAllString(src, "")
  349. return strings.TrimSpace(src)
  350. }
  351. //1556164246 -> 2019-04-25 03:50:46 +0000
  352. //timestamp
  353. func TimeToTimestamp() {
  354. fmt.Println(time.Unix(1556164246, 0).Format("2006-01-02 15:04:05"))
  355. }
  356. func ToUnicode(text string) string {
  357. textQuoted := strconv.QuoteToASCII(text)
  358. textUnquoted := textQuoted[1 : len(textQuoted)-1]
  359. return textUnquoted
  360. }
  361. func VersionToInt(version string) int {
  362. version = strings.Replace(version, ".", "", -1)
  363. n, _ := strconv.Atoi(version)
  364. return n
  365. }
  366. func IsCheckInList(list []int, s int) bool {
  367. for _, v := range list {
  368. if v == s {
  369. return true
  370. }
  371. }
  372. return false
  373. }
  374. func round(num float64) int {
  375. return int(num + math.Copysign(0.5, num))
  376. }
  377. func toFixed(num float64, precision int) float64 {
  378. output := math.Pow(10, float64(precision))
  379. return float64(round(num*output)) / output
  380. }
  381. // GetWilsonScore returns Wilson Score
  382. func GetWilsonScore(p, n float64) float64 {
  383. if p == 0 && n == 0 {
  384. return 0
  385. }
  386. 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)
  387. }
  388. // 将中文数字转化成数字,比如 第三百四十五章,返回第345章 不支持一亿及以上
  389. func ChangeWordsToNum(str string) (numStr string) {
  390. words := ([]rune)(str)
  391. num := 0
  392. n := 0
  393. for i := 0; i < len(words); i++ {
  394. word := string(words[i : i+1])
  395. switch word {
  396. case "万":
  397. if n == 0 {
  398. n = 1
  399. }
  400. n = n * 10000
  401. num = num*10000 + n
  402. n = 0
  403. case "千":
  404. if n == 0 {
  405. n = 1
  406. }
  407. n = n * 1000
  408. num += n
  409. n = 0
  410. case "百":
  411. if n == 0 {
  412. n = 1
  413. }
  414. n = n * 100
  415. num += n
  416. n = 0
  417. case "十":
  418. if n == 0 {
  419. n = 1
  420. }
  421. n = n * 10
  422. num += n
  423. n = 0
  424. case "一":
  425. n += 1
  426. case "二":
  427. n += 2
  428. case "三":
  429. n += 3
  430. case "四":
  431. n += 4
  432. case "五":
  433. n += 5
  434. case "六":
  435. n += 6
  436. case "七":
  437. n += 7
  438. case "八":
  439. n += 8
  440. case "九":
  441. n += 9
  442. case "零":
  443. default:
  444. if n > 0 {
  445. num += n
  446. n = 0
  447. }
  448. if num == 0 {
  449. numStr += word
  450. } else {
  451. numStr += strconv.Itoa(num) + word
  452. num = 0
  453. }
  454. }
  455. }
  456. if n > 0 {
  457. num += n
  458. n = 0
  459. }
  460. if num != 0 {
  461. numStr += strconv.Itoa(num)
  462. }
  463. return
  464. }
  465. func Sha1(data string) string {
  466. sha1 := sha1.New()
  467. sha1.Write([]byte(data))
  468. return hex.EncodeToString(sha1.Sum([]byte("")))
  469. }
  470. func GetVideoPlaySeconds(videoPath string) (playSeconds float64, err error) {
  471. cmd := `ffmpeg -i ` + videoPath + ` 2>&1 | grep 'Duration' | cut -d ' ' -f 4 | sed s/,//`
  472. out, err := exec.Command("bash", "-c", cmd).Output()
  473. if err != nil {
  474. return
  475. }
  476. outTimes := string(out)
  477. fmt.Println("outTimes:", outTimes)
  478. if outTimes != "" {
  479. timeArr := strings.Split(outTimes, ":")
  480. h := timeArr[0]
  481. m := timeArr[1]
  482. s := timeArr[2]
  483. hInt, err := strconv.Atoi(h)
  484. if err != nil {
  485. return playSeconds, err
  486. }
  487. mInt, err := strconv.Atoi(m)
  488. if err != nil {
  489. return playSeconds, err
  490. }
  491. s = strings.Trim(s, " ")
  492. s = strings.Trim(s, "\n")
  493. sInt, err := strconv.ParseFloat(s, 64)
  494. if err != nil {
  495. return playSeconds, err
  496. }
  497. playSeconds = float64(hInt)*3600 + float64(mInt)*60 + float64(sInt)
  498. }
  499. return
  500. }
  501. func GetMaxTradeCode(tradeCode string) (maxTradeCode string, err error) {
  502. tradeCode = strings.Replace(tradeCode, "W", "", -1)
  503. tradeCode = strings.Trim(tradeCode, " ")
  504. tradeCodeInt, err := strconv.Atoi(tradeCode)
  505. if err != nil {
  506. return
  507. }
  508. tradeCodeInt = tradeCodeInt + 1
  509. maxTradeCode = fmt.Sprintf("W%06d", tradeCodeInt)
  510. return
  511. }
  512. // excel日期字段格式化 yyyy-mm-dd
  513. func ConvertToFormatDay(excelDaysString string) string {
  514. // 2006-01-02 距离 1900-01-01的天数
  515. baseDiffDay := 38719 //在网上工具计算的天数需要加2天,什么原因没弄清楚
  516. curDiffDay := excelDaysString
  517. b, _ := strconv.Atoi(curDiffDay)
  518. // 获取excel的日期距离2006-01-02的天数
  519. realDiffDay := b - baseDiffDay
  520. //fmt.Println("realDiffDay:",realDiffDay)
  521. // 距离2006-01-02 秒数
  522. realDiffSecond := realDiffDay * 24 * 3600
  523. //fmt.Println("realDiffSecond:",realDiffSecond)
  524. // 2006-01-02 15:04:05距离1970-01-01 08:00:00的秒数 网上工具可查出
  525. baseOriginSecond := 1136185445
  526. resultTime := time.Unix(int64(baseOriginSecond+realDiffSecond), 0).Format("2006-01-02")
  527. return resultTime
  528. }
  529. func CheckPwd(pwd string) bool {
  530. compile := `([0-9a-z]+){6,12}|(a-z0-9]+){6,12}`
  531. reg := regexp.MustCompile(compile)
  532. flag := reg.MatchString(pwd)
  533. return flag
  534. }
  535. func GetMonthStartAndEnd(myYear string, myMonth string) (startDate, endDate string) {
  536. // 数字月份必须前置补零
  537. if len(myMonth) == 1 {
  538. myMonth = "0" + myMonth
  539. }
  540. yInt, _ := strconv.Atoi(myYear)
  541. timeLayout := "2006-01-02 15:04:05"
  542. loc, _ := time.LoadLocation("Local")
  543. theTime, _ := time.ParseInLocation(timeLayout, myYear+"-"+myMonth+"-01 00:00:00", loc)
  544. newMonth := theTime.Month()
  545. t1 := time.Date(yInt, newMonth, 1, 0, 0, 0, 0, time.Local).Format("2006-01-02")
  546. t2 := time.Date(yInt, newMonth+1, 0, 0, 0, 0, 0, time.Local).Format("2006-01-02")
  547. return t1, t2
  548. }
  549. // 移除字符串中的空格
  550. func TrimStr(str string) (str2 string) {
  551. return strings.Replace(str, " ", "", -1)
  552. }
  553. // 字符串转换为time
  554. func StrTimeToTime(strTime string) time.Time {
  555. timeLayout := "2006-01-02 15:04:05" //转化所需模板
  556. loc, _ := time.LoadLocation("Local") //重要:获取时区
  557. resultTime, _ := time.ParseInLocation(timeLayout, strTime, loc)
  558. return resultTime
  559. }
  560. // GetOrmInReplace 获取orm的in查询替换?的方法
  561. func GetOrmInReplace(num int) string {
  562. template := make([]string, num)
  563. for i := 0; i < num; i++ {
  564. template[i] = "?"
  565. }
  566. return strings.Join(template, ",")
  567. }
  568. // RevSlice 反转切片
  569. func RevSlice(slice []int) []int {
  570. for i, j := 0, len(slice)-1; i < j; i, j = i+1, j-1 {
  571. slice[i], slice[j] = slice[j], slice[i]
  572. }
  573. return slice
  574. }
  575. // GetTimeSubDay 计算两个时间的自然日期差
  576. func GetTimeSubDay(t1, t2 time.Time) int {
  577. var day int
  578. swap := false
  579. if t1.Unix() > t2.Unix() {
  580. t1, t2 = t2, t1
  581. swap = true
  582. }
  583. t1_ := t1.Add(time.Duration(t2.Sub(t1).Milliseconds()%86400000) * time.Millisecond)
  584. day = int(t2.Sub(t1).Hours() / 24)
  585. // 计算在t1+两个时间的余数之后天数是否有变化
  586. if t1_.Day() != t1.Day() {
  587. day += 1
  588. }
  589. if swap {
  590. day = -day
  591. }
  592. return day
  593. }
  594. // GetDaysBetween2Date 计算两个日期之间相差几天
  595. func GetDaysBetween2Date(format, date1Str, date2Str string) (int, error) {
  596. // 将字符串转化为Time格式
  597. date1, err := time.ParseInLocation(format, date1Str, time.Local)
  598. if err != nil {
  599. return 0, err
  600. }
  601. // 将字符串转化为Time格式
  602. date2, err := time.ParseInLocation(format, date2Str, time.Local)
  603. if err != nil {
  604. return 0, err
  605. }
  606. //计算相差天数
  607. return int(date1.Sub(date2).Hours() / 24), nil
  608. }
  609. // MapSorter 对于map 排序
  610. type MapSorter []Item
  611. type Item struct {
  612. Key int
  613. Val float64
  614. }
  615. func NewMapSorter(m map[int]float64) MapSorter {
  616. ms := make(MapSorter, 0, len(m))
  617. for k, v := range m {
  618. ms = append(ms, Item{k, v})
  619. }
  620. return ms
  621. }
  622. func (ms MapSorter) Len() int {
  623. return len(ms)
  624. }
  625. func (ms MapSorter) Less(i, j int) bool {
  626. return ms[i].Val > ms[j].Val // 按值排序
  627. //return ms[i].Key < ms[j].Key // 按键排序
  628. }
  629. func (ms MapSorter) Swap(i, j int) {
  630. ms[i], ms[j] = ms[j], ms[i]
  631. }
  632. // InArrayByInt php中的in_array(判断Int类型的切片中是否存在该int值)
  633. func InArrayByInt(idIntList []int, searchId int) (has bool) {
  634. for _, id := range idIntList {
  635. if id == searchId {
  636. has = true
  637. return
  638. }
  639. }
  640. return
  641. }
  642. // InArrayByStr php中的in_array(判断String类型的切片中是否存在该string值)
  643. func InArrayByStr(idStrList []string, searchId string) (has bool) {
  644. for _, id := range idStrList {
  645. if id == searchId {
  646. has = true
  647. return
  648. }
  649. }
  650. return
  651. }
  652. // GetDateByDateType 通过dateType获取需要的开始/结束日期
  653. func GetDateByDateType(dateType int, tmpStartDate, tmpEndDate string) (startDate, endDate string) {
  654. startDate = tmpStartDate
  655. endDate = tmpEndDate
  656. switch dateType {
  657. case 1:
  658. startDate = "2000-01-01"
  659. endDate = ""
  660. case 2:
  661. startDate = "2010-01-01"
  662. endDate = ""
  663. case 3:
  664. startDate = "2015-01-01"
  665. endDate = ""
  666. case 4:
  667. //startDate = strconv.Itoa(time.Now().Year()) + "-01-01"
  668. startDate = "2021-01-01"
  669. endDate = ""
  670. case 5:
  671. //startDate = startDate + "-01"
  672. //endDate = endDate + "-01"
  673. case 6:
  674. //startDate = startDate + "-01"
  675. endDate = ""
  676. case 7:
  677. startDate = "2018-01-01"
  678. endDate = ""
  679. case 8:
  680. startDate = "2019-01-01"
  681. endDate = ""
  682. case 9:
  683. startDate = "2020-01-01"
  684. endDate = ""
  685. case 11:
  686. startDate = "2022-01-01"
  687. endDate = ""
  688. }
  689. // 兼容日期错误
  690. {
  691. if strings.Count(startDate, "-") == 1 {
  692. startDate = startDate + "-01"
  693. }
  694. if strings.Count(endDate, "-") == 1 {
  695. endDate = endDate + "-01"
  696. }
  697. }
  698. return
  699. }
  700. // GetDateByDateType2 通过dateType获取需要的开始/结束日期(日期类型:1:最近3月;2:最近6月;3:最近1年;4:最近2年;5:最近3年;6:最近5年;7:最近10年,8:自定义时间)
  701. func GetDateByDateType2(dateType int, currDate time.Time) (startDate time.Time) {
  702. switch dateType {
  703. case 1:
  704. startDate = currDate.AddDate(0, -3, 0)
  705. case 2:
  706. startDate = currDate.AddDate(0, -6, 0)
  707. case 3:
  708. startDate = currDate.AddDate(-1, 0, 0)
  709. case 4:
  710. startDate = currDate.AddDate(-2, 0, 0)
  711. case 5:
  712. startDate = currDate.AddDate(-3, 0, 0)
  713. case 6:
  714. startDate = currDate.AddDate(-5, 0, 0)
  715. case 7:
  716. startDate = currDate.AddDate(-10, 0, 0)
  717. }
  718. return
  719. }
  720. // GetCeilNewNum 保留n位有效数字的向上取整
  721. // @params num 实际数据
  722. // @params baseLen 需要保留的有效位数
  723. func GetCeilNewNum(num float64, baseLen int) (newNum float64) {
  724. if num >= 1 {
  725. tmpNum := int(math.Ceil(num)) // 向上取整
  726. str := strconv.Itoa(tmpNum)
  727. lenStr := len(str)
  728. if lenStr > baseLen {
  729. newNumStr := str[0:baseLen]
  730. newNumInt, _ := strconv.Atoi(newNumStr)
  731. newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen))
  732. if newNum < num {
  733. newNumInt += 1
  734. newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen))
  735. }
  736. } else {
  737. newNum = float64(tmpNum)
  738. }
  739. return
  740. } else if num > 0 {
  741. // 这是小数
  742. str := strconv.FormatFloat(num, 'f', -1, 64)
  743. // 去除小数点和负号
  744. str = removeDecimalPoint(str)
  745. // 计算字符串长度
  746. lenStr := len(str)
  747. if lenStr > baseLen {
  748. newNumStr := str[0:baseLen]
  749. newNumInt, _ := strconv.Atoi(newNumStr)
  750. newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64()
  751. if newNum < num {
  752. newNumInt += 1
  753. newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64()
  754. }
  755. } else {
  756. newNum = num
  757. }
  758. } else if num > -1 {
  759. // 这是小数
  760. str := strconv.FormatFloat(num, 'f', -1, 64)
  761. // 去除小数点和负号
  762. str = removeDecimalPoint(str)
  763. // 计算字符串长度
  764. lenStr := len(str)
  765. if lenStr > baseLen {
  766. newNumStr := str[0:baseLen]
  767. newNumInt, _ := strconv.Atoi(newNumStr)
  768. newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64()
  769. newNum = -newNum
  770. if newNum < num {
  771. newNumInt -= 1
  772. newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64()
  773. newNum = -newNum
  774. }
  775. } else {
  776. newNum = num
  777. }
  778. if newNum == -0 {
  779. newNum = 0
  780. }
  781. } else { // 小于等于-1
  782. tmpNumFloat := math.Abs(num)
  783. tmpNum := int(math.Floor(tmpNumFloat)) // 向上取整
  784. str := strconv.Itoa(tmpNum)
  785. lenStr := len(str)
  786. if lenStr > baseLen {
  787. newNumStr := str[0:baseLen]
  788. //fmt.Println("newNumStr:", newNumStr)
  789. newNumInt, _ := strconv.Atoi(newNumStr)
  790. newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen))
  791. newNum = -newNum
  792. if newNum < num {
  793. newNumInt -= 1
  794. newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen))
  795. newNum = -newNum
  796. }
  797. } else {
  798. newNum = float64(-tmpNum)
  799. }
  800. }
  801. return
  802. }
  803. // GetFloorNewNum 保留n位有效数字的向下取整
  804. // @params num 实际数据
  805. // @params baseLen 需要保留的有效位数
  806. func GetFloorNewNum(num float64, baseLen int) (newNum float64) {
  807. if num >= 1 {
  808. tmpNum := int(math.Floor(num)) // 向上取整
  809. str := strconv.Itoa(tmpNum)
  810. lenStr := len(str)
  811. if lenStr > baseLen {
  812. newNumStr := str[0:baseLen]
  813. newNumInt, _ := strconv.Atoi(newNumStr)
  814. newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen))
  815. if newNum < num {
  816. newNumInt -= 1
  817. newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen))
  818. }
  819. } else {
  820. newNum = float64(tmpNum)
  821. }
  822. return
  823. } else if num > 0 {
  824. // 这是小数
  825. str := strconv.FormatFloat(num, 'f', -1, 64)
  826. // 去除小数点和负号
  827. str = removeDecimalPoint(str)
  828. // 计算字符串长度
  829. lenStr := len(str)
  830. if lenStr > baseLen {
  831. newNumStr := str[0:baseLen]
  832. newNumInt, _ := strconv.Atoi(newNumStr)
  833. newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64()
  834. if newNum > num {
  835. newNumInt -= 1
  836. newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64()
  837. }
  838. } else {
  839. newNum = num
  840. }
  841. } else if num > -1 {
  842. // 这是小数
  843. str := strconv.FormatFloat(num, 'f', -1, 64)
  844. // 去除小数点和负号
  845. str = removeDecimalPoint(str)
  846. // 计算字符串长度
  847. lenStr := len(str)
  848. if lenStr > baseLen {
  849. newNumStr := str[0:baseLen]
  850. newNumInt, _ := strconv.Atoi(newNumStr)
  851. newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64()
  852. newNum = -newNum
  853. if newNum > num {
  854. newNumInt += 1
  855. newNum, _ = decimal.NewFromInt(int64(newNumInt)).Div(decimal.NewFromFloat(math.Pow(10, float64(baseLen)))).Float64()
  856. newNum = -newNum
  857. }
  858. } else {
  859. newNum = num
  860. }
  861. if newNum == -0 {
  862. newNum = 0
  863. }
  864. } else { // 小于等于-1
  865. tmpNumFloat := math.Abs(num)
  866. tmpNum := int(math.Ceil(tmpNumFloat)) // 向上取整
  867. str := strconv.Itoa(tmpNum)
  868. lenStr := len(str)
  869. if lenStr > baseLen {
  870. newNumStr := str[0:baseLen]
  871. //fmt.Println("newNumStr:", newNumStr)
  872. newNumInt, _ := strconv.Atoi(newNumStr)
  873. newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen))
  874. newNum = -newNum
  875. if newNum > num {
  876. newNumInt += 1
  877. newNum = float64(newNumInt) * math.Pow(10, float64(lenStr-baseLen))
  878. newNum = -newNum
  879. }
  880. } else {
  881. newNum = float64(-tmpNum)
  882. }
  883. }
  884. return
  885. }
  886. // 去除小数点和负号
  887. func removeDecimalPoint(str string) string {
  888. // 去除小数点
  889. str = str[strings.Index(str, ".")+1:]
  890. return str
  891. }