feat(message): add message encryption with batch decryption optimization
- Add encryption configuration with environment variable support - Implement batch parallel decryption for improved query performance - Integrate message encryptor via wire dependency injection - Add explicit Decrypt() method for single message decryption - Optimize AfterFind hook to defer decryption for batch processing
This commit is contained in:
137
internal/pkg/crypto/batch_decrypt_test.go
Normal file
137
internal/pkg/crypto/batch_decrypt_test.go
Normal file
@@ -0,0 +1,137 @@
|
||||
package crypto
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// BenchmarkEncryptDecryptSingle 单条加密解密性能基准
|
||||
func BenchmarkEncryptDecryptSingle(b *testing.B) {
|
||||
key := "12345678901234567890123456789012"
|
||||
_ = InitMessageEncryptor(key, 1)
|
||||
encryptor := GetMessageEncryptor()
|
||||
|
||||
plaintext := []byte(`{"type":"text","data":{"content":"这是一条测试消息"}}`)
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
ciphertext, _ := encryptor.Encrypt(plaintext)
|
||||
_, _ = encryptor.Decrypt(ciphertext)
|
||||
}
|
||||
}
|
||||
|
||||
// BenchmarkBatchDecrypt 批量并行解密性能基准
|
||||
func BenchmarkBatchDecrypt(b *testing.B) {
|
||||
key := "12345678901234567890123456789012"
|
||||
_ = InitMessageEncryptor(key, 1)
|
||||
encryptor := GetMessageEncryptor()
|
||||
|
||||
// 准备测试数据
|
||||
sizes := []int{10, 50, 100, 500}
|
||||
|
||||
for _, size := range sizes {
|
||||
b.Run(fmt.Sprintf("size_%d", size), func(b *testing.B) {
|
||||
// 生成加密数据
|
||||
ciphertexts := make([]string, size)
|
||||
for i := 0; i < size; i++ {
|
||||
msg := map[string]interface{}{
|
||||
"type": "text",
|
||||
"data": map[string]string{"content": fmt.Sprintf("消息内容 %d", i)},
|
||||
}
|
||||
plaintext, _ := json.Marshal(msg)
|
||||
ciphertexts[i], _ = encryptor.Encrypt(plaintext)
|
||||
}
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
_ = encryptor.BatchDecrypt(ciphertexts, 4)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// BenchmarkSerialDecrypt 串行解密性能基准(对比用)
|
||||
func BenchmarkSerialDecrypt(b *testing.B) {
|
||||
key := "12345678901234567890123456789012"
|
||||
_ = InitMessageEncryptor(key, 1)
|
||||
encryptor := GetMessageEncryptor()
|
||||
|
||||
sizes := []int{10, 50, 100, 500}
|
||||
|
||||
for _, size := range sizes {
|
||||
b.Run(fmt.Sprintf("size_%d", size), func(b *testing.B) {
|
||||
ciphertexts := make([]string, size)
|
||||
for i := 0; i < size; i++ {
|
||||
msg := map[string]interface{}{
|
||||
"type": "text",
|
||||
"data": map[string]string{"content": fmt.Sprintf("消息内容 %d", i)},
|
||||
}
|
||||
plaintext, _ := json.Marshal(msg)
|
||||
ciphertexts[i], _ = encryptor.Encrypt(plaintext)
|
||||
}
|
||||
|
||||
b.ResetTimer()
|
||||
for i := 0; i < b.N; i++ {
|
||||
for _, ct := range ciphertexts {
|
||||
_, _ = encryptor.Decrypt(ct)
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestBatchDecrypt(t *testing.T) {
|
||||
key := "12345678901234567890123456789012"
|
||||
_ = InitMessageEncryptor(key, 1)
|
||||
encryptor := GetMessageEncryptor()
|
||||
|
||||
// 准备测试数据
|
||||
count := 100
|
||||
ciphertexts := make([]string, count)
|
||||
expectedContents := make([]string, count)
|
||||
|
||||
for i := 0; i < count; i++ {
|
||||
content := fmt.Sprintf("测试消息 %d", i)
|
||||
expectedContents[i] = content
|
||||
msg := fmt.Sprintf(`{"type":"text","data":{"content":"%s"}}`, content)
|
||||
ciphertext, err := encryptor.Encrypt([]byte(msg))
|
||||
if err != nil {
|
||||
t.Fatalf("Encrypt failed: %v", err)
|
||||
}
|
||||
ciphertexts[i] = ciphertext
|
||||
}
|
||||
|
||||
// 批量解密
|
||||
results := encryptor.BatchDecrypt(ciphertexts, 4)
|
||||
|
||||
// 验证结果
|
||||
for i, result := range results {
|
||||
if result == nil {
|
||||
t.Errorf("Decrypt result[%d] is nil", i)
|
||||
continue
|
||||
}
|
||||
|
||||
var msg map[string]interface{}
|
||||
if err := json.Unmarshal(result, &msg); err != nil {
|
||||
t.Errorf("Unmarshal result[%d] failed: %v", i, err)
|
||||
continue
|
||||
}
|
||||
|
||||
data, ok := msg["data"].(map[string]interface{})
|
||||
if !ok {
|
||||
t.Errorf("result[%d] data is not map", i)
|
||||
continue
|
||||
}
|
||||
|
||||
content, ok := data["content"].(string)
|
||||
if !ok {
|
||||
t.Errorf("result[%d] content is not string", i)
|
||||
continue
|
||||
}
|
||||
|
||||
if content != expectedContents[i] {
|
||||
t.Errorf("result[%d] content = %s, want %s", i, content, expectedContents[i])
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user