common_utils.go

package main

import (
	"crypto/md5"
	"encoding/hex"
	"github.com/go-resty/resty"
	"github.com/tidwall/gjson"
	"strings"
	"time"
)

const (
	cityNameCodeVersion string = "267040"
	cityNameCodeURL            = "https://tce.alicdn.com/api/data.htm?ids=" + cityNameCodeVersion
	UserAgent           string = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/78.0.3904.70 Safari/537.36"
)

// 初始化一个城市名和城市代码的映射
var cityNameCode = make(map[string]string)

// 初始化城市名和城市代码的数据
func initCityNameCodeData() {
	dataResp, err := resty.New().R().
		SetHeader("user-agent", UserAgent).
		Get(cityNameCodeURL)
	if err != nil {
		logger.Fatalf("[Flight-Go]初始化数据接口异常, 错误原因: %v", err)
	}
	if dataResp.String() != "" {
		jsonData := gjson.Parse(dataResp.String())
		cityArray := jsonData.Get(cityNameCodeVersion).Get("value").Get("cityArray").Array()[1:]
		for _, cities := range cityArray {
			cityDD := cities.Get("tabdata").Array()
			for _, cityData := range cityDD {
				for _, city := range cityData.Get("dd").Array() {
					cityNameCode[city.Get("cityName").String()] = city.Get("cityCode").String()
				}
			}
		}
		logger.Info("[Flight-Go]初始化数据成功!")
	}
}

// 时间戳转时间
func timestampToTime(timestamp int64) string {
	if timestamp != 0 {
		return time.Unix(timestamp, 0).Format("2006-01-02 15:04:05")
	} else {
		return "--:--"
	}
}

// 字符串 format
func stringFormat(format string, args ...string) (formatString string) {
	return strings.NewReplacer(args...).Replace(format)
}

// 通过自定义字符串获取随机 MD5 字符串
func getRandomMD5ByCustomStr(str string) string {
	md5Context := md5.New()
	md5Context.Write([]byte(str))
	return hex.EncodeToString(md5Context.Sum(nil))
}

// 分钟转 小时分钟 （例如: 75分钟 转换为 1小时15分钟）
const (
	Minute = 60
	Hour   = Minute * 60
)

func minutesToHour(minutes int64) (int64, int64) {
	return minutes / Minute, minutes % 60
}