init

utils/utils.go

@@ -0,0 +1,121 @@
+package utils
+
+import (
+	"bytes"
+	"compress/gzip"
+	"encoding/binary"
+	"io/ioutil"
+	"os"
+	"os/signal"
+	"reflect"
+	"regexp"
+	"strconv"
+	"syscall"
+
+	"github.com/pyihe/go-pkg/rands"
+)
+
+var (
+	mailChecker  = regexp.MustCompile(`^[A-Za-z0-9]+([_\.][A-Za-z0-9]+)*@([A-Za-z0-9\-]+\.)+[A-Za-z]{2,6}$`)
+	phoneChecker = regexp.MustCompile(`^[1](([3][0-9])|([4][5-9])|([5][0-3,5-9])|([6][5,6])|([7][0-8])|([8][0-9])|([9][1,8,9]))[0-9]{8}$`)
+)
+
+//校验邮箱格式
+func CheckMailFormat(mail string) bool {
+	return mailChecker.MatchString(mail)
+}
+
+//校验电话号码格式
+func CheckPhoneFormat(phone string) bool {
+	return phoneChecker.MatchString(phone)
+}
+
+//生成一个1-100的随机数, 用于简单的判断概率
+func LessThanIn100(per int) bool {
+	if per < 1 || per > 100 {
+		panic("input must between 1 and 100")
+	}
+	return per >= rands.Int(1, 100)
+}
+
+//如果监听到系统中断信号，则执行onNotify()
+func Notify(onNotify func()) {
+	//SIGHUP		终端控制进程结束(终端连接断开)
+	//SIGQUIT		用户发送QUIT字符(Ctrl+/)触发
+	//SIGTERM		结束程序(可以被捕获、阻塞或忽略)
+	//SIGINT		用户发送INTR字符(Ctrl+C)触发
+	//SIGSTOP		停止进程(不能被捕获、阻塞或忽略)
+	ch := make(chan os.Signal, 1)
+	signal.Notify(ch, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGSTOP, syscall.SIGINT)
+	for {
+		s := <-ch
+		switch s {
+		case syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGSTOP, syscall.SIGINT, syscall.SIGHUP:
+			if onNotify == nil {
+				return
+			}
+			onNotify()
+		default:
+			return
+		}
+	}
+}
+
+//判断src中是否有元素ele
+func Contain(src interface{}, ele interface{}) bool {
+	switch reflect.TypeOf(src).Kind() {
+	case reflect.Slice:
+		s := reflect.ValueOf(src)
+		for i := 0; i < s.Len(); i++ {
+			if reflect.DeepEqual(ele, s.Index(i).Interface()) {
+				return true
+			}
+		}
+	}
+	return false
+}
+
+//将嵌套的map[string]interface全部转换成一层
+func Interface2Map(data interface{}) map[string]interface{} {
+	result := make(map[string]interface{})
+	for k, v := range data.(map[string]interface{}) {
+		switch v := v.(type) {
+		case map[string]interface{}:
+			for i, u := range v {
+				result[i] = u
+			}
+		default:
+			result[k] = v
+		}
+	}
+	return result
+}
+
+//gzip解压
+func UnGZIP(content []byte) ([]byte, error) {
+	buffer := new(bytes.Buffer)
+	err := binary.Write(buffer, binary.BigEndian, content)
+	if err != nil {
+		return nil, err
+	}
+	zipReader, err := gzip.NewReader(buffer)
+	if err != nil {
+		return nil, err
+	}
+	defer zipReader.Close()
+	result, err := ioutil.ReadAll(zipReader)
+	if err != nil {
+		return nil, err
+	}
+	return result, nil
+}
+
+// ConvertToBinary 十进制转换为二进制字符串
+func ConvertToBinary(data int) string {
+	result := ""
+	for ; data > 0; data = data / 2 {
+		n := data % 2
+		result = strconv.Itoa(n) + result
+	}
+	return result
+}