309 lines
7.4 KiB
Go
309 lines
7.4 KiB
Go
package tool
|
||
|
||
import (
|
||
"bytes"
|
||
"crypto/md5"
|
||
"encoding/base64"
|
||
"encoding/hex"
|
||
"fmt"
|
||
"io"
|
||
"io/ioutil"
|
||
"log"
|
||
"math/rand"
|
||
"net/http"
|
||
"os"
|
||
"os/exec"
|
||
"path/filepath"
|
||
"reflect"
|
||
"regexp"
|
||
"runtime"
|
||
"sort"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
/*
|
||
判断文件或文件夹是否存在
|
||
如果返回的错误为nil,说明文件或文件夹存在
|
||
如果返回的错误类型使用os.IsNotExist()判断为true,说明文件或文件夹不存在
|
||
如果返回的错误为其它类型,则不确定是否在存在
|
||
*/
|
||
func PathExists(path string) bool {
|
||
|
||
_, err := os.Stat(path)
|
||
if err == nil {
|
||
return true
|
||
}
|
||
if os.IsNotExist(err) {
|
||
return false
|
||
}
|
||
return false
|
||
}
|
||
|
||
/*
|
||
检查字符串是否在某一数组中
|
||
*/
|
||
func IsContainStr(items []string, item string) bool {
|
||
for _, eachItem := range items {
|
||
if eachItem == item {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// 调用os.MkdirAll递归创建文件夹
|
||
func CreateDir(dirPath string) {
|
||
if !isExist(dirPath) {
|
||
fmt.Println("路径不存在，创建路径", dirPath)
|
||
_ = os.MkdirAll(dirPath, os.ModePerm)
|
||
}
|
||
}
|
||
|
||
// 判断所给路径文件/文件夹是否存在(返回true是存在)
|
||
func isExist(path string) bool {
|
||
_, err := os.Stat(path) //os.Stat获取文件信息
|
||
if err != nil {
|
||
if os.IsExist(err) {
|
||
return true
|
||
}
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
|
||
/*获取uuid*/
|
||
func GetUuid() string {
|
||
str := GetRandstring(32)
|
||
time.Sleep(time.Nanosecond)
|
||
return strings.ToLower(Md5V(str))
|
||
}
|
||
|
||
// #取得随机字符串:使用字符串拼接
|
||
func GetRandstring(lenNum int) string {
|
||
var CHARS = []string{"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",
|
||
"1", "2", "3", "4", "5", "6", "7", "8", "9", "0"}
|
||
str := strings.Builder{}
|
||
length := len(CHARS)
|
||
for i := 0; i < lenNum; i++ {
|
||
l := CHARS[rand.Intn(length)]
|
||
str.WriteString(l)
|
||
}
|
||
return str.String()
|
||
|
||
}
|
||
|
||
// 获取当前执行程序所在的绝对路径
|
||
func GetCurrentAbPathByExecutable() string {
|
||
exePath, err := os.Executable()
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
res, _ := filepath.EvalSymlinks(filepath.Dir(exePath))
|
||
return res
|
||
}
|
||
|
||
/*结构体转map且首字母小写*/
|
||
func Struct2Map(obj interface{}) map[string]interface{} {
|
||
t := reflect.TypeOf(obj)
|
||
v := reflect.ValueOf(obj)
|
||
var data = make(map[string]interface{})
|
||
for i := 0; i < t.NumField(); i++ {
|
||
data[strings.ToLower(string(t.Field(i).Name[0]))+t.Field(i).Name[1:]] = v.Field(i).Interface()
|
||
//data[t.Field(i).Name] = v.Field(i).Interface()
|
||
}
|
||
return data
|
||
}
|
||
func PortInUse(port int) int {
|
||
checkStatement := fmt.Sprintf("lsof -i:%d ", port)
|
||
output, _ := exec.Command("sh", "-c", checkStatement).CombinedOutput()
|
||
fmt.Println(output)
|
||
if len(output) > 0 {
|
||
return 1
|
||
}
|
||
return -1
|
||
}
|
||
|
||
/*检测端口是否在使用中*/
|
||
// 传入查询的端口号
|
||
// 返回端口号对应的进程PID，若没有找到相关进程，返回-1
|
||
func portInUse(portNumber int) int {
|
||
res := -1
|
||
var outBytes bytes.Buffer
|
||
sysType := runtime.GOOS
|
||
fmt.Println(sysType)
|
||
var cmdStr = ""
|
||
if sysType == "linux" {
|
||
cmdStr = fmt.Sprintf("lsof -i:%d ", portNumber)
|
||
}
|
||
//
|
||
if sysType == "windows" {
|
||
cmdStr = fmt.Sprintf("netstat -ano -p tcp | findstr %d", portNumber)
|
||
}
|
||
//checkStatement := fmt.Sprintf("lsof -i:%d ", portNumber)
|
||
fmt.Println(cmdStr)
|
||
cmd := exec.Command("cmd", "/c", cmdStr)
|
||
cmd.Stdout = &outBytes
|
||
cmd.Run()
|
||
resStr := outBytes.String()
|
||
fmt.Println(resStr)
|
||
r := regexp.MustCompile(`\s\d+\s`).FindAllString(resStr, -1)
|
||
if len(r) > 0 {
|
||
pid, err := strconv.Atoi(strings.TrimSpace(r[0]))
|
||
if err != nil {
|
||
res = -1
|
||
} else {
|
||
res = pid
|
||
}
|
||
}
|
||
return res
|
||
}
|
||
|
||
func GetUnUsePort(port int) int {
|
||
isInUse := PortInUse(port)
|
||
if isInUse != -1 {
|
||
fmt.Println("端口：" + strconv.Itoa(port) + " 被占用")
|
||
port++
|
||
port = GetUnUsePort(port)
|
||
return port
|
||
} else {
|
||
return port
|
||
}
|
||
}
|
||
|
||
// WeekIntervalTime 获取某周的开始和结束时间,week为0本周,-1上周，1下周以此类推
|
||
func WeekIntervalTime(week int) (startTime, endTime string) {
|
||
now := time.Now()
|
||
offset := int(time.Monday - now.Weekday())
|
||
//周日做特殊判断 因为time.Monday = 0
|
||
if offset > 0 {
|
||
offset = -6
|
||
}
|
||
|
||
year, month, day := now.Date()
|
||
thisWeek := time.Date(year, month, day, 0, 0, 0, 0, time.Local)
|
||
startTime = thisWeek.AddDate(0, 0, offset+7*week).Format("2006-01-02") + " 00:00:00"
|
||
endTime = thisWeek.AddDate(0, 0, offset+6+7*week).Format("2006-01-02") + " 23:59:59"
|
||
|
||
return startTime, endTime
|
||
}
|
||
|
||
// GetCurrentTime 获取当前系统时间
|
||
func GetCurrentTime() string {
|
||
return time.Now().Format("2006-01-02 15:04:05")
|
||
}
|
||
|
||
/*删除文件或文件夹*/
|
||
func RemoveFile(path string) {
|
||
if isExist(path) {
|
||
err := os.Remove(path)
|
||
if err != nil {
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
// 检查参数
|
||
func Md5V(str string) string {
|
||
h := md5.New()
|
||
h.Write([]byte(str))
|
||
return hex.EncodeToString(h.Sum(nil))
|
||
}
|
||
|
||
// download file会将url下载到本地文件，它会在下载时写入，而不是将整个文件加载到内存中。
|
||
func DownloadFile(filepath string, url string) error {
|
||
|
||
// Get the data
|
||
resp, err := http.Get(url)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer resp.Body.Close()
|
||
// Create the file
|
||
out, err := os.Create(filepath)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer out.Close()
|
||
|
||
// Write the body to file
|
||
_, err = io.Copy(out, resp.Body)
|
||
return err
|
||
}
|
||
|
||
// 通用排序
|
||
// 结构体排序，必须重写数组Len() Swap() Less()函数
|
||
type body_wrapper struct {
|
||
Bodys []interface{}
|
||
by func(p, q *interface{}) bool //内部Less()函数会用到
|
||
}
|
||
type SortBodyBy func(p, q *interface{}) bool //定义一个函数类型
|
||
|
||
// 数组长度Len()
|
||
func (acw body_wrapper) Len() int {
|
||
return len(acw.Bodys)
|
||
}
|
||
|
||
// 元素交换
|
||
func (acw body_wrapper) Swap(i, j int) {
|
||
acw.Bodys[i], acw.Bodys[j] = acw.Bodys[j], acw.Bodys[i]
|
||
}
|
||
|
||
// 比较函数，使用外部传入的by比较函数
|
||
func (acw body_wrapper) Less(i, j int) bool {
|
||
return acw.by(&acw.Bodys[i], &acw.Bodys[j])
|
||
}
|
||
|
||
// 自定义排序字段，参考SortBodyByCreateTime中的传入函数
|
||
func SortBody(bodys []interface{}, by SortBodyBy) {
|
||
sort.Sort(body_wrapper{bodys, by})
|
||
}
|
||
|
||
// 格式化时间
|
||
type LocalTime time.Time
|
||
|
||
func (t *LocalTime) MarshalJSON() ([]byte, error) {
|
||
tTime := time.Time(*t)
|
||
return []byte(fmt.Sprintf("\"%v\"", tTime.Format("2006-01-02 15:04:05"))), nil
|
||
}
|
||
|
||
// 写入文件,保存
|
||
func WriteFile(path string, base64_image_content string) (error, string) {
|
||
|
||
//b, _ := regexp.MatchString(`^data:\s*image\/(\w+);base64,`, base64_image_content)
|
||
//if !b {
|
||
// return errors.New(""), ""
|
||
//}
|
||
base64_image_content = "data:image/png;base64," + base64_image_content
|
||
re, _ := regexp.Compile(`^data:\s*image\/(\w+);base64,`)
|
||
allData := re.FindAllSubmatch([]byte(base64_image_content), 2)
|
||
fileType := string(allData[0][1]) //png ，jpeg 后缀获取
|
||
|
||
base64Str := re.ReplaceAllString(base64_image_content, "")
|
||
|
||
//date := time.Now().Format("2006-01-02")
|
||
//if ok := IsFileExist(path + "/" + date); !ok {
|
||
// os.Mkdir(path+"/"+date, 0666)
|
||
//}
|
||
|
||
curFileStr := strconv.FormatInt(time.Now().UnixNano(), 10)
|
||
|
||
r := rand.New(rand.NewSource(time.Now().UnixNano()))
|
||
n := r.Intn(99999)
|
||
var filename = curFileStr + strconv.Itoa(n) + "." + fileType
|
||
var file = path + "/" + filename
|
||
byte, _ := base64.StdEncoding.DecodeString(base64Str)
|
||
|
||
err := ioutil.WriteFile(file, byte, 0666)
|
||
if err != nil {
|
||
log.Println(err)
|
||
}
|
||
return err, filename
|
||
}
|
||
|
||
func UploadsFile() {
|
||
|
||
}
|