common.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838
  1. package utils
  2. import (
  3. "crypto/md5"
  4. "crypto/sha1"
  5. "encoding/base64"
  6. "encoding/hex"
  7. "encoding/json"
  8. "fmt"
  9. "github.com/PuerkitoBio/goquery"
  10. "html"
  11. "image"
  12. "image/png"
  13. "math"
  14. "math/rand"
  15. "net"
  16. "os"
  17. "os/exec"
  18. "regexp"
  19. "strconv"
  20. "strings"
  21. "time"
  22. )
  23. // 随机数种子
  24. var rnd = rand.New(rand.NewSource(time.Now().UnixNano()))
  25. func GetRandString(size int) string {
  26. 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", "!", "@", "#", "$", "%", "^", "&", "*"}
  27. randomSb := ""
  28. digitSize := len(allLetterDigit)
  29. for i := 0; i < size; i++ {
  30. randomSb += allLetterDigit[rnd.Intn(digitSize)]
  31. }
  32. return randomSb
  33. }
  34. func GetRandStringNoSpecialChar(size int) string {
  35. 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"}
  36. randomSb := ""
  37. digitSize := len(allLetterDigit)
  38. for i := 0; i < size; i++ {
  39. randomSb += allLetterDigit[rnd.Intn(digitSize)]
  40. }
  41. return randomSb
  42. }
  43. func StringsToJSON(str string) string {
  44. rs := []rune(str)
  45. jsons := ""
  46. for _, r := range rs {
  47. rint := int(r)
  48. if rint < 128 {
  49. jsons += string(r)
  50. } else {
  51. jsons += "\\u" + strconv.FormatInt(int64(rint), 16) // json
  52. }
  53. }
  54. return jsons
  55. }
  56. // 序列化
  57. func ToString(v interface{}) string {
  58. data, _ := json.Marshal(v)
  59. return string(data)
  60. }
  61. // md5加密
  62. func MD5(data string) string {
  63. m := md5.Sum([]byte(data))
  64. return hex.EncodeToString(m[:])
  65. }
  66. // 获取数字随机字符
  67. func GetRandDigit(n int) string {
  68. return fmt.Sprintf("%0"+strconv.Itoa(n)+"d", rnd.Intn(int(math.Pow10(n))))
  69. }
  70. // 获取随机数
  71. func GetRandNumber(n int) int {
  72. return rnd.Intn(n)
  73. }
  74. func GetRandInt(min, max int) int {
  75. if min >= max || min == 0 || max == 0 {
  76. return max
  77. }
  78. return rand.Intn(max-min) + min
  79. }
  80. func GetToday(format string) string {
  81. today := time.Now().Format(format)
  82. return today
  83. }
  84. // 获取今天剩余秒数
  85. func GetTodayLastSecond() time.Duration {
  86. today := GetToday(FormatDate) + " 23:59:59"
  87. end, _ := time.ParseInLocation(FormatDateTime, today, time.Local)
  88. return time.Duration(end.Unix()-time.Now().Local().Unix()) * time.Second
  89. }
  90. // 处理出生日期函数
  91. func GetBrithDate(idcard string) string {
  92. l := len(idcard)
  93. var s string
  94. if l == 15 {
  95. s = "19" + idcard[6:8] + "-" + idcard[8:10] + "-" + idcard[10:12]
  96. return s
  97. }
  98. if l == 18 {
  99. s = idcard[6:10] + "-" + idcard[10:12] + "-" + idcard[12:14]
  100. return s
  101. }
  102. return GetToday(FormatDate)
  103. }
  104. // 处理性别
  105. func WhichSexByIdcard(idcard string) string {
  106. var sexs = [2]string{"女", "男"}
  107. length := len(idcard)
  108. if length == 18 {
  109. sex, _ := strconv.Atoi(string(idcard[16]))
  110. return sexs[sex%2]
  111. } else if length == 15 {
  112. sex, _ := strconv.Atoi(string(idcard[14]))
  113. return sexs[sex%2]
  114. }
  115. return "男"
  116. }
  117. // 截取小数点后几位
  118. func SubFloatToString(f float64, m int) string {
  119. n := strconv.FormatFloat(f, 'f', -1, 64)
  120. if n == "" {
  121. return ""
  122. }
  123. if m >= len(n) {
  124. return n
  125. }
  126. newn := strings.Split(n, ".")
  127. if m == 0 {
  128. return newn[0]
  129. }
  130. if len(newn) < 2 || m >= len(newn[1]) {
  131. return n
  132. }
  133. return newn[0] + "." + newn[1][:m]
  134. }
  135. // 截取小数点后几位
  136. func SubFloatToFloat(f float64, m int) float64 {
  137. newn := SubFloatToString(f, m)
  138. newf, _ := strconv.ParseFloat(newn, 64)
  139. return newf
  140. }
  141. // 获取相差时间-年
  142. func GetYearDiffer(start_time, end_time string) int {
  143. t1, _ := time.ParseInLocation("2006-01-02", start_time, time.Local)
  144. t2, _ := time.ParseInLocation("2006-01-02", end_time, time.Local)
  145. age := t2.Year() - t1.Year()
  146. if t2.Month() < t1.Month() || (t2.Month() == t1.Month() && t2.Day() < t1.Day()) {
  147. age--
  148. }
  149. return age
  150. }
  151. // 获取相差时间-秒
  152. func GetSecondDifferByTime(start_time, end_time time.Time) int64 {
  153. diff := end_time.Unix() - start_time.Unix()
  154. return diff
  155. }
  156. func FixFloat(f float64, m int) float64 {
  157. newn := SubFloatToString(f+0.00000001, m)
  158. newf, _ := strconv.ParseFloat(newn, 64)
  159. return newf
  160. }
  161. // 将字符串数组转化为逗号分割的字符串形式 ["str1","str2","str3"] >>> "str1,str2,str3"
  162. func StrListToString(strList []string) (str string) {
  163. if len(strList) > 0 {
  164. for k, v := range strList {
  165. if k == 0 {
  166. str = v
  167. } else {
  168. str = str + "," + v
  169. }
  170. }
  171. return
  172. }
  173. return ""
  174. }
  175. // Token
  176. func GetToken() string {
  177. randStr := GetRandString(64)
  178. token := MD5(randStr + Md5Key)
  179. tokenLen := 64 - len(token)
  180. return strings.ToUpper(token + GetRandString(tokenLen))
  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 ValidateFixedTelephoneFormatat(mobileNum string) bool {
  198. reg := regexp.MustCompile(RegularFixedTelephone)
  199. return reg.MatchString(mobileNum)
  200. }
  201. // 判断文件是否存在
  202. func FileIsExist(filePath string) bool {
  203. _, err := os.Stat(filePath)
  204. return err == nil || os.IsExist(err)
  205. }
  206. // 获取图片扩展名
  207. func GetImgExt(file string) (ext string, err error) {
  208. var headerByte []byte
  209. headerByte = make([]byte, 8)
  210. fd, err := os.Open(file)
  211. if err != nil {
  212. return "", err
  213. }
  214. defer fd.Close()
  215. _, err = fd.Read(headerByte)
  216. if err != nil {
  217. return "", err
  218. }
  219. xStr := fmt.Sprintf("%x", headerByte)
  220. switch {
  221. case xStr == "89504e470d0a1a0a":
  222. ext = ".png"
  223. case xStr == "0000010001002020":
  224. ext = ".ico"
  225. case xStr == "0000020001002020":
  226. ext = ".cur"
  227. case xStr[:12] == "474946383961" || xStr[:12] == "474946383761":
  228. ext = ".gif"
  229. case xStr[:10] == "0000020000" || xStr[:10] == "0000100000":
  230. ext = ".tga"
  231. case xStr[:8] == "464f524d":
  232. ext = ".iff"
  233. case xStr[:8] == "52494646":
  234. ext = ".ani"
  235. case xStr[:4] == "4d4d" || xStr[:4] == "4949":
  236. ext = ".tiff"
  237. case xStr[:4] == "424d":
  238. ext = ".bmp"
  239. case xStr[:4] == "ffd8":
  240. ext = ".jpg"
  241. case xStr[:2] == "0a":
  242. ext = ".pcx"
  243. default:
  244. ext = ""
  245. }
  246. return ext, nil
  247. }
  248. // 保存图片
  249. func SaveImage(path string, img image.Image) (err error) {
  250. //需要保持的文件
  251. imgfile, err := os.Create(path)
  252. defer imgfile.Close()
  253. // 以PNG格式保存文件
  254. err = png.Encode(imgfile, img)
  255. return err
  256. }
  257. // 保存base64数据为文件
  258. func SaveBase64ToFile(content, path string) error {
  259. data, err := base64.StdEncoding.DecodeString(content)
  260. if err != nil {
  261. return err
  262. }
  263. f, err := os.Create(path)
  264. defer f.Close()
  265. if err != nil {
  266. return err
  267. }
  268. f.Write(data)
  269. return nil
  270. }
  271. func SaveBase64ToFileBySeek(content, path string) (err error) {
  272. data, err := base64.StdEncoding.DecodeString(content)
  273. exist, err := PathExists(path)
  274. if err != nil {
  275. return
  276. }
  277. if !exist {
  278. f, err := os.Create(path)
  279. if err != nil {
  280. return err
  281. }
  282. n, _ := f.Seek(0, 2)
  283. // 从末尾的偏移量开始写入内容
  284. _, err = f.WriteAt([]byte(data), n)
  285. defer f.Close()
  286. } else {
  287. f, err := os.OpenFile(path, os.O_WRONLY, 0644)
  288. if err != nil {
  289. return err
  290. }
  291. n, _ := f.Seek(0, 2)
  292. // 从末尾的偏移量开始写入内容
  293. _, err = f.WriteAt([]byte(data), n)
  294. defer f.Close()
  295. }
  296. return nil
  297. }
  298. func PathExists(path string) (bool, error) {
  299. _, err := os.Stat(path)
  300. if err == nil {
  301. return true, nil
  302. }
  303. if os.IsNotExist(err) {
  304. return false, nil
  305. }
  306. return false, err
  307. }
  308. func StartIndex(page, pagesize int) int {
  309. if page > 1 {
  310. return (page - 1) * pagesize
  311. }
  312. return 0
  313. }
  314. func PageCount(count, pagesize int) int {
  315. if count%pagesize > 0 {
  316. return count/pagesize + 1
  317. } else {
  318. return count / pagesize
  319. }
  320. }
  321. func TrimHtml(src string) string {
  322. //将HTML标签全转换成小写
  323. re, _ := regexp.Compile("\\<[\\S\\s]+?\\>")
  324. src = re.ReplaceAllStringFunc(src, strings.ToLower)
  325. re, _ = regexp.Compile("\\<img[\\S\\s]+?\\>")
  326. src = re.ReplaceAllString(src, "[图片]")
  327. re, _ = regexp.Compile("class[\\S\\s]+?>")
  328. src = re.ReplaceAllString(src, "")
  329. re, _ = regexp.Compile("\\<[\\S\\s]+?\\>")
  330. src = re.ReplaceAllString(src, "")
  331. return strings.TrimSpace(src)
  332. }
  333. //1556164246 -> 2019-04-25 03:50:46 +0000
  334. //timestamp
  335. func TimeToTimestamp() {
  336. fmt.Println(time.Unix(1556164246, 0).Format("2006-01-02 15:04:05"))
  337. }
  338. func ToUnicode(text string) string {
  339. textQuoted := strconv.QuoteToASCII(text)
  340. textUnquoted := textQuoted[1 : len(textQuoted)-1]
  341. return textUnquoted
  342. }
  343. func VersionToInt(version string) int {
  344. version = strings.Replace(version, ".", "", -1)
  345. n, _ := strconv.Atoi(version)
  346. return n
  347. }
  348. func IsCheckInList(list []int, s int) bool {
  349. for _, v := range list {
  350. if v == s {
  351. return true
  352. }
  353. }
  354. return false
  355. }
  356. func round(num float64) int {
  357. return int(num + math.Copysign(0.5, num))
  358. }
  359. func toFixed(num float64, precision int) float64 {
  360. output := math.Pow(10, float64(precision))
  361. return float64(round(num*output)) / output
  362. }
  363. // GetWilsonScore returns Wilson Score
  364. func GetWilsonScore(p, n float64) float64 {
  365. if p == 0 && n == 0 {
  366. return 0
  367. }
  368. 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)
  369. }
  370. // 将中文数字转化成数字,比如 第三百四十五章,返回第345章 不支持一亿及以上
  371. func ChangeWordsToNum(str string) (numStr string) {
  372. words := ([]rune)(str)
  373. num := 0
  374. n := 0
  375. for i := 0; i < len(words); i++ {
  376. word := string(words[i : i+1])
  377. switch word {
  378. case "万":
  379. if n == 0 {
  380. n = 1
  381. }
  382. n = n * 10000
  383. num = num*10000 + n
  384. n = 0
  385. case "千":
  386. if n == 0 {
  387. n = 1
  388. }
  389. n = n * 1000
  390. num += n
  391. n = 0
  392. case "百":
  393. if n == 0 {
  394. n = 1
  395. }
  396. n = n * 100
  397. num += n
  398. n = 0
  399. case "十":
  400. if n == 0 {
  401. n = 1
  402. }
  403. n = n * 10
  404. num += n
  405. n = 0
  406. case "一":
  407. n += 1
  408. case "二":
  409. n += 2
  410. case "三":
  411. n += 3
  412. case "四":
  413. n += 4
  414. case "五":
  415. n += 5
  416. case "六":
  417. n += 6
  418. case "七":
  419. n += 7
  420. case "八":
  421. n += 8
  422. case "九":
  423. n += 9
  424. case "零":
  425. default:
  426. if n > 0 {
  427. num += n
  428. n = 0
  429. }
  430. if num == 0 {
  431. numStr += word
  432. } else {
  433. numStr += strconv.Itoa(num) + word
  434. num = 0
  435. }
  436. }
  437. }
  438. if n > 0 {
  439. num += n
  440. n = 0
  441. }
  442. if num != 0 {
  443. numStr += strconv.Itoa(num)
  444. }
  445. return
  446. }
  447. func Sha1(data string) string {
  448. sha1 := sha1.New()
  449. sha1.Write([]byte(data))
  450. return hex.EncodeToString(sha1.Sum([]byte("")))
  451. }
  452. func GetVideoPlaySeconds(videoPath string) (playSeconds float64, err error) {
  453. cmd := `ffmpeg -i ` + videoPath + ` 2>&1 | grep 'Duration' | cut -d ' ' -f 4 | sed s/,//`
  454. out, err := exec.Command("bash", "-c", cmd).Output()
  455. if err != nil {
  456. return
  457. }
  458. outTimes := string(out)
  459. fmt.Println("outTimes:", outTimes)
  460. if outTimes != "" {
  461. timeArr := strings.Split(outTimes, ":")
  462. h := timeArr[0]
  463. m := timeArr[1]
  464. s := timeArr[2]
  465. hInt, err := strconv.Atoi(h)
  466. if err != nil {
  467. return playSeconds, err
  468. }
  469. mInt, err := strconv.Atoi(m)
  470. if err != nil {
  471. return playSeconds, err
  472. }
  473. s = strings.Trim(s, " ")
  474. s = strings.Trim(s, "\n")
  475. sInt, err := strconv.ParseFloat(s, 64)
  476. if err != nil {
  477. return playSeconds, err
  478. }
  479. playSeconds = float64(hInt)*3600 + float64(mInt)*60 + float64(sInt)
  480. }
  481. return
  482. }
  483. func GetMaxTradeCode(tradeCode string) (maxTradeCode string, err error) {
  484. tradeCode = strings.Replace(tradeCode, "W", "", -1)
  485. tradeCode = strings.Trim(tradeCode, " ")
  486. tradeCodeInt, err := strconv.Atoi(tradeCode)
  487. if err != nil {
  488. return
  489. }
  490. tradeCodeInt = tradeCodeInt + 1
  491. maxTradeCode = fmt.Sprintf("W%06d", tradeCodeInt)
  492. return
  493. }
  494. // excel日期字段格式化 yyyy-mm-dd
  495. func ConvertToFormatDay(excelDaysString string) string {
  496. // 2006-01-02 距离 1900-01-01的天数
  497. baseDiffDay := 38719 //在网上工具计算的天数需要加2天,什么原因没弄清楚
  498. curDiffDay := excelDaysString
  499. b, _ := strconv.Atoi(curDiffDay)
  500. // 获取excel的日期距离2006-01-02的天数
  501. realDiffDay := b - baseDiffDay
  502. //fmt.Println("realDiffDay:",realDiffDay)
  503. // 距离2006-01-02 秒数
  504. realDiffSecond := realDiffDay * 24 * 3600
  505. //fmt.Println("realDiffSecond:",realDiffSecond)
  506. // 2006-01-02 15:04:05距离1970-01-01 08:00:00的秒数 网上工具可查出
  507. baseOriginSecond := 1136185445
  508. resultTime := time.Unix(int64(baseOriginSecond+realDiffSecond), 0).Format("2006-01-02")
  509. return resultTime
  510. }
  511. // 字符串转换为time
  512. func StrTimeToTime(strTime string) time.Time {
  513. timeLayout := "2006-01-02 15:04:05" //转化所需模板
  514. loc, _ := time.LoadLocation("Local") //重要:获取时区
  515. resultTime, _ := time.ParseInLocation(timeLayout, strTime, loc)
  516. return resultTime
  517. }
  518. // 时间格式去掉时分秒
  519. func TimeRemoveHms(strTime string) string {
  520. var Ymd string
  521. var resultTime = StrTimeToTime(strTime)
  522. year := resultTime.Year()
  523. month := resultTime.Format("01")
  524. day1 := resultTime.Day()
  525. if day1 < 10 {
  526. Ymd = strconv.Itoa(year) + "." + month + ".0" + strconv.Itoa(day1)
  527. } else {
  528. Ymd = strconv.Itoa(year) + "." + month + "." + strconv.Itoa(day1)
  529. }
  530. return Ymd
  531. }
  532. // 判断时间是当年的第几周
  533. func WeekByDate(t time.Time) string {
  534. var resultSAtr string
  535. t = t.AddDate(0, 0, -8) // 减少八天跟老数据标题统一
  536. yearDay := t.YearDay()
  537. yearFirstDay := t.AddDate(0, 0, -yearDay+1)
  538. firstDayInWeek := int(yearFirstDay.Weekday())
  539. //今年第一周有几天
  540. firstWeekDays := 1
  541. if firstDayInWeek != 0 {
  542. firstWeekDays = 7 - firstDayInWeek + 1
  543. }
  544. var week int
  545. if yearDay <= firstWeekDays {
  546. week = 1
  547. } else {
  548. week = (yearDay-firstWeekDays)/7 + 2
  549. }
  550. resultSAtr = "(" + strconv.Itoa(t.Year()) + "年第" + strconv.Itoa(week) + "周" + ")"
  551. return resultSAtr
  552. }
  553. func Mp3Time(videoPlaySeconds string) string {
  554. var d int
  555. var timeStr string
  556. a, _ := strconv.ParseFloat(videoPlaySeconds, 32)
  557. b := int(a)
  558. c := b % 60
  559. d = b / 60
  560. if b <= 60 {
  561. timeStr = "00:" + strconv.Itoa(b)
  562. } else {
  563. if d < 10 {
  564. timeStr = "0" + strconv.Itoa(d) + ":" + strconv.Itoa(c)
  565. } else {
  566. timeStr = strconv.Itoa(d) + ":" + strconv.Itoa(c)
  567. }
  568. }
  569. return timeStr
  570. }
  571. func GetLocalIP() (ip string, err error) {
  572. addrs, err := net.InterfaceAddrs()
  573. if err != nil {
  574. return
  575. }
  576. for _, addr := range addrs {
  577. ipAddr, ok := addr.(*net.IPNet)
  578. if !ok {
  579. continue
  580. }
  581. if ipAddr.IP.IsLoopback() {
  582. continue
  583. }
  584. if !ipAddr.IP.IsGlobalUnicast() {
  585. continue
  586. }
  587. return ipAddr.IP.String(), nil
  588. }
  589. return
  590. }
  591. // GetOrmInReplace 获取orm的in查询替换?的方法
  592. func GetOrmInReplace(num int) string {
  593. template := make([]string, num)
  594. for i := 0; i < num; i++ {
  595. template[i] = "?"
  596. }
  597. return strings.Join(template, ",")
  598. }
  599. // InArrayByStr php中的in_array(判断String类型的切片中是否存在该string值)
  600. func InArrayByStr(idStrList []string, searchId string) (has bool) {
  601. for _, id := range idStrList {
  602. if id == searchId {
  603. has = true
  604. return
  605. }
  606. }
  607. return
  608. }
  609. // InArrayByInt php中的in_array(判断Int类型的切片中是否存在该int值)
  610. func InArrayByInt(idIntList []int, searchId int) (has bool) {
  611. for _, id := range idIntList {
  612. if id == searchId {
  613. has = true
  614. return
  615. }
  616. }
  617. return
  618. }
  619. // GetNowWeekMonday 获取本周周一的时间
  620. func GetNowWeekMonday() time.Time {
  621. offset := int(time.Monday - time.Now().Weekday())
  622. if offset == 1 { //正好是周日,但是按照中国人的理解,周日是一周最后一天,而不是一周开始的第一天
  623. offset = -6
  624. }
  625. mondayTime := time.Now().AddDate(0, 0, offset)
  626. mondayTime = time.Date(mondayTime.Year(), mondayTime.Month(), mondayTime.Day(), 0, 0, 0, 0, mondayTime.Location())
  627. return mondayTime
  628. }
  629. // GetLastWeekMonday 获取上周周一的时间
  630. func GetLastWeekMonday() time.Time {
  631. offset := int(time.Monday - time.Now().Weekday())
  632. if offset == 1 { //正好是周日,但是按照中国人的理解,周日是一周最后一天,而不是一周开始的第一天
  633. offset = -6
  634. }
  635. mondayTime := time.Now().AddDate(0, 0, offset-7)
  636. mondayTime = time.Date(mondayTime.Year(), mondayTime.Month(), mondayTime.Day(), 0, 0, 0, 0, mondayTime.Location())
  637. return mondayTime
  638. }
  639. // GetNowWeekSunDay 获取本周周日的时间
  640. func GetNowWeekSunday() time.Time {
  641. return GetNowWeekMonday().AddDate(0, 0, 6)
  642. }
  643. // GetLastWeekSunday 获取上周周日的时间
  644. func GetLastWeekSunday() time.Time {
  645. return GetLastWeekMonday().AddDate(0, 0, 6)
  646. }
  647. // GetNowMonthFirstDay 获取本月第一天的时间
  648. func GetNowMonthFirstDay() time.Time {
  649. nowMonthFirstDay := time.Date(time.Now().Year(), time.Now().Month(), 1, 0, 0, 0, 0, time.Now().Location())
  650. return nowMonthFirstDay
  651. }
  652. // GetNowMonthLastDay 获取本月最后一天的时间
  653. func GetNowMonthLastDay() time.Time {
  654. nowMonthLastDay := time.Date(time.Now().Year(), time.Now().Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
  655. nowMonthLastDay = time.Date(nowMonthLastDay.Year(), nowMonthLastDay.Month(), nowMonthLastDay.Day(), 23, 59, 59, 0, nowMonthLastDay.Location())
  656. return nowMonthLastDay
  657. }
  658. // GetNowMonthFirstDay 获取上月第一天的时间
  659. func GetLastMonthFirstDay() time.Time {
  660. nowMonthFirstDay := time.Date(time.Now().Year(), time.Now().AddDate(0, -1, 0).Month(), 1, 0, 0, 0, 0, time.Now().Location())
  661. return nowMonthFirstDay
  662. }
  663. // GetNowMonthLastDay 获取上月最后一天的时间
  664. func GetLastMonthLastDay() time.Time {
  665. nowMonthLastDay := time.Date(time.Now().Year(), time.Now().AddDate(0, -1, 0).Month(), 1, 0, 0, 0, 0, time.Now().Location()).AddDate(0, 1, -1)
  666. nowMonthLastDay = time.Date(nowMonthLastDay.Year(), nowMonthLastDay.Month(), nowMonthLastDay.Day(), 23, 59, 59, 0, nowMonthLastDay.Location())
  667. return nowMonthLastDay
  668. }
  669. // 时间格式去掉时分秒
  670. func TimeRemoveHms2(strTime string) string {
  671. var Ymd string
  672. var resultTime = StrTimeToTime(strTime)
  673. year := resultTime.Year()
  674. month := resultTime.Format("01")
  675. day1 := resultTime.Day()
  676. if day1 < 10 {
  677. Ymd = strconv.Itoa(year) + "-" + month + "-0" + strconv.Itoa(day1)
  678. } else {
  679. Ymd = strconv.Itoa(year) + "-" + month + "-" + strconv.Itoa(day1)
  680. }
  681. return Ymd
  682. }
  683. // 时间格式去掉年
  684. func GetTimeDateRemoveYear(strTime string) (dataStr string) {
  685. slicePublishTime := strings.Split(strTime, "-")
  686. for k, v := range slicePublishTime {
  687. if k == 0 {
  688. continue
  689. }
  690. dataStr += v + "-"
  691. }
  692. dataStr = strings.TrimRight(dataStr, "-")
  693. return dataStr
  694. }
  695. // 时间格式去掉年和秒
  696. func GetTimeDateRemoveYearAndSecond(strTime string) (dataStr string) {
  697. slicePublishTime := strings.Split(strTime, "-")
  698. for k, v := range slicePublishTime {
  699. if k == 0 {
  700. continue
  701. }
  702. dataStr += v + "-"
  703. }
  704. dataStr = strings.TrimRight(dataStr, "-")
  705. dataStr = dataStr[:len(dataStr)-3]
  706. return
  707. }
  708. func ArticleHasImgUrl(body string) (hasImg bool, err error) {
  709. r := strings.NewReader(string(body))
  710. doc, err := goquery.NewDocumentFromReader(r)
  711. if err != nil {
  712. fmt.Println(err)
  713. }
  714. doc.Find("img").Each(func(i int, s *goquery.Selection) {
  715. hasImg = true
  716. })
  717. return
  718. }
  719. func ArticleRemoveImgUrl(body string) (result string) {
  720. // 使用正则表达式去除img标签
  721. re := regexp.MustCompile(`<img[^>]*>`)
  722. result = re.ReplaceAllString(body, "")
  723. return
  724. }
  725. func FindArticleImgUrls(body string) (imgUrls []string, err error) {
  726. r := strings.NewReader(string(body))
  727. doc, err := goquery.NewDocumentFromReader(r)
  728. if err != nil {
  729. fmt.Println(err)
  730. }
  731. doc.Find("img").Each(func(i int, s *goquery.Selection) {
  732. src, _ := s.Attr("src")
  733. imgUrls = append(imgUrls, src)
  734. })
  735. return
  736. }
  737. // 去除部分style
  738. func ExtractText(body string) (result string, err error) {
  739. // 使用正则表达式去除img标签
  740. re := regexp.MustCompile(`<section style[\s\S]*?>`)
  741. result = re.ReplaceAllString(body, "")
  742. re = regexp.MustCompile(`<span style[\s\S]*?>`)
  743. result = re.ReplaceAllString(result, "")
  744. return
  745. }
  746. // 富文本字符串截取指定长度
  747. func InterceptHtmlLength(body string, length int) (newbody string) {
  748. content := html.UnescapeString(body)
  749. doc, err := goquery.NewDocumentFromReader(strings.NewReader(content))
  750. if err != nil {
  751. fmt.Println("create doc err:", err.Error())
  752. return
  753. }
  754. bodyText := doc.Text()
  755. if len(bodyText) < length {
  756. length = len(bodyText)
  757. }
  758. newbody = bodyText[0:length]
  759. return
  760. }