common.go 20 KB

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