修改代码

Author: jaryf

utils/base64_test.go

@@ -1,4 +1,4 @@
-package utils
+package gencode
 
 import (
 	"encoding/base64"

utils/base64.go utils/gencode/base64.goutils/base64.go renamed to utils/gencode/base64.go

@@ -0,0 +1,14 @@
+package gencode
+
+import "testing"
+
+func TestB64EncodeString(t *testing.T) {
+	s := B64EncodeString("123456")
+	t.Log(s)
+}
+
+func BenchmarkB64EncodeString(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		B64EncodeString("123456")
+	}
+}

utils/gencode/base64_test.go

@@ -1,4 +1,4 @@
-package utils
+package gencrypt
 
 import (
 	"crypto/hmac"

utils/hmac.go utils/gencrypt/hmac.goutils/hmac.go renamed to utils/gencrypt/hmac.go

@@ -0,0 +1,28 @@
+package gencrypt
+
+import (
+	"crypto/md5"
+	"encoding/hex"
+	"io/ioutil"
+)
+
+func Md5Bytes(data []byte) (res string, err error) {
+	h := md5.New()
+	if _, err = h.Write(data); err != nil {
+		return "", err
+	}
+	return hex.EncodeToString(h.Sum(nil)), nil
+}
+
+func Md5String(data string) (res string, err error) {
+	return Md5Bytes([]byte(data))
+}
+
+func Md5File(path string) (res string, err error) {
+	var fileBytes []byte
+	fileBytes, err = ioutil.ReadFile(path)
+	if err != nil {
+		return "", err
+	}
+	return Md5Bytes(fileBytes)
+}

utils/gencrypt/md5.go

@@ -1,4 +1,4 @@
-package utils
+package gencrypt
 
 import (
 	"crypto/rand"

utils/rsa.go utils/gencrypt/rsa.goutils/rsa.go renamed to utils/gencrypt/rsa.go

@@ -0,0 +1,59 @@
+package gencrypt
+
+import (
+	"crypto/sha1"
+	"crypto/sha256"
+	"crypto/sha512"
+	"encoding/hex"
+	"io/ioutil"
+)
+
+func Sha1File(path string) (encrypt string, err error) {
+	var fileBytes []byte
+	fileBytes, err = ioutil.ReadFile(path)
+	if err != nil {
+		return "", err
+	}
+	h := sha1.New()
+	h.Write(fileBytes)
+	encrypt = hex.EncodeToString(h.Sum(nil))
+	return
+}
+
+func Sha1Bytes(data []byte) string {
+	h := sha1.New()
+	h.Write(data)
+	return hex.EncodeToString(h.Sum(nil))
+}
+
+func Sha1String(data string) (encrypt string) {
+	return Sha1Bytes([]byte(data))
+}
+
+func Sha256String(data string) string {
+	return Sha256Bytes([]byte(data))
+}
+
+func Sha256Bytes(data []byte) string {
+	return hex.EncodeToString(Sha256(data))
+}
+
+func Sha256(data []byte) []byte {
+	digest := sha256.New()
+	digest.Write(data)
+	return digest.Sum(nil)
+}
+
+func Sha512String(data string) string {
+	return Sha512Bytes([]byte(data))
+}
+
+func Sha512Bytes(data []byte) string {
+	return hex.EncodeToString(Sha512(data))
+}
+
+func Sha512(data []byte) []byte {
+	digest := sha512.New()
+	digest.Write(data)
+	return digest.Sum(nil)
+}

utils/gencrypt/sha.go

@@ -1,4 +1,4 @@
-package utils
+package grand
 
 import (
 	"crypto/rand"

utils/rand.go utils/grand/rand.goutils/rand.go renamed to utils/grand/rand.go

@@ -1,4 +1,4 @@
-package utils
+package grand
 
 import (
 	"testing"
@@ -9,11 +9,23 @@ func TestIntn(t *testing.T) {
 	t.Log(s)
 }
 
+func BenchmarkIntn(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		Intn(500)
+	}
+}
+
 func TestRandBytes(t *testing.T) {
 	b := RandBytes(10)
 	t.Log(b)
 }
 
+func BenchmarkRandBytes(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		RandBytes(10)
+	}
+}
+
 func TestRandRangeIntN(t *testing.T) {
 	n := RandRangeIntN(1, 4)
 	t.Log(n)
@@ -29,6 +41,12 @@ func TestRandStr(t *testing.T) {
 	t.Log(randStr)
 }
 
+func BenchmarkRandStr(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		RandStr("01234你耗阿萨德绿卡结束勒肯定就看", 5)
+	}
+}
+
 func TestRandDigits(t *testing.T) {
 	s := RandDigits(5)
 	t.Log(s)

utils/rand_test.go utils/grand/rand_test.goutils/rand_test.go renamed to utils/grand/rand_test.go

@@ -0,0 +1,20 @@
+package gstr
+
+const (
+	NotFoundIndex = -1
+)
+
+// SearchArray 查询子串在字符串列表里的索引,不存在则返回-1
+func SearchArray(strList []string, s string) int {
+	for i, v := range strList {
+		if s == v {
+			return i
+		}
+	}
+	return NotFoundIndex
+}
+
+// InArray 查询子串是否在字符串列表里
+func InArray(a []string, s string) bool {
+	return SearchArray(a, s) != NotFoundIndex
+}