118 lines
2.9 KiB
Go
118 lines
2.9 KiB
Go
package script
|
|
|
|
import (
|
|
"bytes"
|
|
"golang.org/x/text/encoding/simplifiedchinese"
|
|
"golang.org/x/text/transform"
|
|
"io/ioutil"
|
|
"math/rand"
|
|
"net"
|
|
"os/exec"
|
|
"time"
|
|
)
|
|
|
|
func checkCommandExists(command string) bool {
|
|
cmd := exec.Command(command, "--version")
|
|
_, err := cmd.Output()
|
|
return err == nil
|
|
}
|
|
|
|
func GetIpAddress() string {
|
|
addrs, _ := net.InterfaceAddrs()
|
|
for _, value := range addrs {
|
|
if ipnet, ok := value.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
|
|
if ipnet.IP.To4() != nil {
|
|
return ipnet.IP.String()
|
|
}
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
const char = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
|
|
|
func RandChar(size int) string {
|
|
source := rand.NewSource(time.Now().UnixNano()) // 产生随机种子
|
|
var s bytes.Buffer
|
|
for i := 0; i < size; i++ {
|
|
s.WriteByte(char[source.Int63()%int64(len(char))])
|
|
}
|
|
return s.String()
|
|
}
|
|
|
|
type Timestamp int64
|
|
|
|
const time_layout string = "2006-01-02 15:04:05"
|
|
|
|
// 日期字符串转为时间戳 工具函数
|
|
func getTimeStamp(str_date string) Timestamp {
|
|
if len(str_date) == 10 {
|
|
str_date = str_date + " 00:00:00"
|
|
}
|
|
// 获取时区
|
|
loc, err := time.LoadLocation("Local")
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
// 转换为时间戳
|
|
the_time, err := time.ParseInLocation(time_layout, str_date, loc)
|
|
if err != nil {
|
|
return 0
|
|
}
|
|
// 返回时间戳
|
|
return Timestamp(the_time.Unix())
|
|
}
|
|
|
|
// 获取1到2个日期字符串中更大的日期
|
|
func getMaxDate(str_date1 string, str_date2 string) string {
|
|
var max_date string
|
|
if str_date1 != "" && str_date2 == "" {
|
|
// 只定义了第一个账户，取第一个账户的日期为准
|
|
max_date = str_date1
|
|
} else if str_date1 == "" && str_date2 != "" {
|
|
// 只定义了第二个账户，取第二个账户的日期为准
|
|
max_date = str_date2
|
|
} else if str_date1 != "" && str_date2 != "" {
|
|
// 重复定义的账户，取最晚的时间为准
|
|
t1 := getTimeStamp(str_date1)
|
|
t2 := getTimeStamp(str_date2)
|
|
if t1 > t2 {
|
|
max_date = str_date1
|
|
} else {
|
|
max_date = str_date2
|
|
}
|
|
} else if str_date1 == "" && str_date2 == "" {
|
|
// 没有定义账户，取当前日期为准
|
|
max_date = time.Now().Format(time_layout)
|
|
}
|
|
return max_date
|
|
}
|
|
|
|
// ConvertGBKToUTF8 将 GBK 编码的字符串转换为 UTF-8 编码
|
|
func ConvertGBKToUTF8(gbkStr string) (string, error) {
|
|
if !isWindows() {
|
|
return gbkStr, nil
|
|
}
|
|
// 创建一个 GBK 到 UTF-8 的转换器
|
|
reader := transform.NewReader(bytes.NewReader([]byte(gbkStr)), simplifiedchinese.GBK.NewDecoder())
|
|
|
|
// 将转换后的内容读出为 UTF-8 字符串
|
|
utf8Bytes, err := ioutil.ReadAll(reader)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return string(utf8Bytes), nil
|
|
}
|
|
|
|
func GetMonth(date string) (string, error) {
|
|
// 解析日期字符串
|
|
parsedDate, err := time.Parse("2006-01-02", date)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
// 格式化日期为 "YYYY-MM" 格式
|
|
formattedDate := parsedDate.Format("2006-01")
|
|
return formattedDate, nil
|
|
}
|