소스 검색

修改AES加密

yjp 6 달 전
부모
커밋
b5a3638b05
1개의 변경된 파일17개의 추가작업 그리고 0개의 파일을 삭제
  1. 17 0
      encoding/encoding.go

+ 17 - 0
encoding/encoding.go

@@ -10,9 +10,18 @@ import (
 	"encoding/base64"
 	"encoding/hex"
 	"fmt"
+	"git.sxidc.com/go-tools/utils/strutils"
 )
 
 func AESEncrypt(originData string, key string) (string, error) {
+	if strutils.IsStringEmpty(originData) {
+		return "", nil
+	}
+
+	if strutils.IsStringEmpty(key) {
+		return originData, nil
+	}
+
 	keyBytes := []byte(key)
 
 	block, err := aes.NewCipher(keyBytes)
@@ -31,6 +40,14 @@ func AESEncrypt(originData string, key string) (string, error) {
 }
 
 func AESDecrypt(encrypted string, key string) (string, error) {
+	if strutils.IsStringEmpty(encrypted) {
+		return "", nil
+	}
+
+	if strutils.IsStringEmpty(key) {
+		return encrypted, nil
+	}
+
 	keyBytes := []byte(key)
 
 	block, err := aes.NewCipher(keyBytes)