apinto/utils/function.go

package utils

import (
	"bytes"
	"compress/gzip"
	"crypto/md5"
	"encoding/hex"
	"fmt"
	"math/rand"
	"net"
	"os"
	"os/exec"
	"regexp"
	"strconv"
	"strings"
	"time"
)

// ConvertIntArrayToString 转换整型数组
func ConvertIntArrayToString(ids []int) string {
	idLen := len(ids)
	if idLen < 1 {
		return ""
	}
	idStr := ""
	for i, id := range ids {
		idStr += strconv.Itoa(id)
		if i < idLen-1 {
			idStr += ","
		}
	}
	return idStr
}

// ConvertArray 将[]string转为[]int
func ConvertArray(arr []string) (bool, []int) {
	result := make([]int, 0)
	for _, i := range arr {
		res, err := strconv.Atoi(i)
		if err != nil {
			return false, result
		}
		result = append(result, res)
	}
	return true, result
}

// ValidateRemoteAddr 判断ip端口是否合法
func ValidateRemoteAddr(ip string) bool {
	match, err := regexp.MatchString(`^(?:(?:1[0-9][0-9]\.)|(?:2[0-4][0-9]\.)|(?:25[0-5]\.)|(?:[1-9][0-9]\.)|(?:[0-9]\.)){3}(?:(?:1[0-9][0-9])|(?:2[0-4][0-9])|(?:25[0-5])|(?:[1-9][0-9])|(?:[0-9]))\:(([0-9])|([1-9][0-9]{1,3})|([1-6][0-9]{0,4}))$`, ip)
	if err != nil {
		return false
	}
	return match
}

// ValidateURL 判断ip端口是否合法
func ValidateURL(url string) bool {
	match, err := regexp.MatchString(`^/(([a-zA-Z][0-9a-zA-Z+\-\.]*:)?/{0,2}[0-9a-zA-Z;/?:@&=+$\.\-_!~*'()%]+)?(#[0-9a-zA-Z;/?:@&=+$\.\-_!~*'()%]+)?$`, url)
	if err != nil {
		return false
	}
	return match
}

// Intercept 获取IP
func Intercept(str, substr string) (string, string) {
	result := strings.Index(str, substr)
	var rs string
	var bs string
	if result != -1 {
		rs = str[:result]
		bs = str[result+1:]
	} else {
		rs = str
		bs = str
	}
	return rs, bs
}

// Md5 md5加密
func Md5(encodeString string) string {
	h := md5.New()
	h.Write([]byte(encodeString))
	return hex.EncodeToString(h.Sum(nil)) // 输出加密结果
}

// GetRandomStringBack 生成随机字符串
func GetRandomStringBack(num int) string {
	str := "123456789abcdefghijklmnpqrstuvwxyzABCDEFGHIJKLMNPQRSTUVWXYZ"
	bytes := []byte(str)
	result := []byte{}
	r := rand.New(rand.NewSource(time.Now().UnixNano()))
	for i := 0; i < num; i++ {
		result = append(result, bytes[r.Intn(len(bytes))])
	}
	return string(result)
}

var (
	randManager = rand.New(rand.NewSource(time.Now().UnixNano()))
	randBytes   = []byte("123456789abcdefghijklmnpqrstuvwxyzABCDEFGHIJKLMNPQRSTUVWXYZ")
	randSize    = len(randBytes)
	reg, _      = regexp.Compile(`[0-9a-f][0-9a-f][:-][0-9a-f][0-9a-f][:-][0-9a-f][0-9a-f][:-][0-9a-f][0-9a-f][:-][0-9a-f][0-9a-f][:-][0-9a-f][0-9a-f]`)
)

// GetRandomString 生成随机字符串
func GetRandomString(num int) string {
	result := make([]byte, num)
	for i := 0; i < num; i++ {
		result[i] = randBytes[int(randManager.Int31())%randSize]
	}
	return string(result)
}

// CheckFileIsExist 判断文件是否存在  存在返回 true 不存在返回false
func CheckFileIsExist(filename string) bool {
	if _, err := os.Stat(filename); os.IsNotExist(err) {
		return false
	}
	return true
}

// Stop 关闭网关服务，重启读取配置文件
func Stop() bool {
	id := os.Getpid()
	cmd := exec.Command("/bin/bash", "-c", "kill -HUP "+strconv.Itoa(id))
	if _, err := cmd.Output(); err != nil {
		return false
	}
	return true
}

// GetMac 获取MAC地址
func GetMac() (bool, string) {
	interfaces, err := net.Interfaces()
	if err != nil {
		return false, "Poor soul, here is what you got: " + err.Error()
	}

	for _, inter := range interfaces {
		mac := inter.HardwareAddr //获取本机MAC地址
		m := fmt.Sprintf("%s", mac)
		match := reg.MatchString(m)

		if match {
			return true, string(m)
		}
	}
	return false, ""
}

// GzipCompress 转成Gzip
func GzipCompress(origin []byte) ([]byte, error) {
	var b bytes.Buffer
	gz := gzip.NewWriter(&b)
	defer gz.Close()
	if _, err := gz.Write(origin); err != nil {
		return nil, err
	}
	if err := gz.Flush(); err != nil {
		return nil, err
	}
	if err := gz.Close(); err != nil {
		return nil, err
	}
	return b.Bytes(), nil
}