common.go 24 KB

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