```
All checks were successful
ci/woodpecker/push/my-first-workflow Pipeline was successful

feat(dict): 添加字典数据压缩加密功能

- 在字典数据接口中集成压缩和加密机制
- 实现Gzip压缩、XOR加密和Base64编码流程
- 新增CompressAndEncrypt和DecryptAndDecompress工具函数
- 在middleware中启用压缩中间件支持
```
This commit is contained in:
昔念
2026-02-28 00:21:33 +08:00
parent 9485c510c0
commit 8bc3fd3cb7
3 changed files with 154 additions and 3 deletions

View File

@@ -0,0 +1,70 @@
package middleware
import (
"compress/flate"
"compress/gzip"
"strings"
"github.com/andybalholm/brotli"
"github.com/gogf/gf/v2/net/ghttp"
)
func CompressMiddleware(r *ghttp.Request) {
ae := strings.ToLower(r.Header.Get("Accept-Encoding"))
// 先执行后面的逻辑
r.Middleware.Next()
// 已经压缩过就跳过
if r.Response.Header().Get("Content-Encoding") != "" {
return
}
// 只压缩文本类型
contentType := r.Response.Header().Get("Content-Type")
compressTypes := []string{
"text/",
"application/json",
"application/javascript",
"application/xml",
}
need := false
for _, t := range compressTypes {
if strings.HasPrefix(contentType, t) {
need = true
break
}
}
if !need {
return
}
buf := r.Response.Buffer()
if len(buf) == 0 {
return
}
// 优先 Brotli
if strings.Contains(ae, "br") {
r.Response.Header().Set("Content-Encoding", "br")
r.Response.Header().Del("Content-Length")
r.Response.ClearBuffer()
wr := brotli.NewWriterLevel(r.Response.Writer, brotli.BestCompression)
defer wr.Close()
wr.Write(buf)
return
}
// 降级 Gzip
if strings.Contains(ae, "gzip") {
r.Response.Header().Set("Content-Encoding", "gzip")
r.Response.Header().Del("Content-Length")
r.Response.ClearBuffer()
wr, _ := gzip.NewWriterLevel(r.Response.Writer, flate.BestCompression)
defer wr.Close()
wr.Write(buf)
return
}
}

View File

@@ -156,7 +156,7 @@ func init() {
tt.ServeHTTP(r.Response.Writer, r.Request)
})
g.Server().Use(CompressMiddleware)
g.Server().BindHandler("/server/*", func(r *ghttp.Request) {
servert := new(ServerHandler)
id := gconv.Uint16(r.URL.Query().Get("id"))

View File

@@ -1,12 +1,17 @@
package admin
import (
"bytes"
"compress/gzip"
"context"
"encoding/base64"
"fmt"
"blazing/cool"
"blazing/modules/dict/service"
"github.com/gogf/gf/v2/encoding/gjson"
"github.com/gogf/gf/v2/frame/g"
)
@@ -35,7 +40,83 @@ type DictInfoDataReq struct {
// Data 方法 获得字典数据
func (c *DictInfoController) Data(ctx context.Context, req *DictInfoDataReq) (res *cool.BaseRes, err error) {
service := service.NewDictInfoService()
data, err := service.Data(ctx, req.Types)
res = cool.Ok(data)
data, _ := service.Data(ctx, req.Types)
r, _ := gjson.EncodeString(data)
encryptedStr, _ := CompressAndEncrypt(r)
res = cool.Ok(encryptedStr)
return
}
// 异或密钥(两端必须一致)
const xorKey = "simple_key_123"
// Xor 异或加解密(复用同一个函数)
func Xor(data []byte) []byte {
keyLen := len(xorKey)
result := make([]byte, len(data))
for i := 0; i < len(data); i++ {
result[i] = data[i] ^ xorKey[i%keyLen]
}
return result
}
// GzipCompress Gzip压缩
func GzipCompress(data []byte) ([]byte, error) {
var buf bytes.Buffer
gz := gzip.NewWriter(&buf)
_, err := gz.Write(data)
if err != nil {
return nil, err
}
if err := gz.Close(); err != nil {
return nil, err
}
return buf.Bytes(), nil
}
// GzipDecompress Gzip解压
func GzipDecompress(data []byte) ([]byte, error) {
buf := bytes.NewBuffer(data)
gz, err := gzip.NewReader(buf)
if err != nil {
return nil, err
}
defer gz.Close()
var res bytes.Buffer
_, err = res.ReadFrom(gz)
return res.Bytes(), err
}
// CompressAndEncrypt 压缩+异或+Base64
func CompressAndEncrypt(plainText string) (string, error) {
// 1. 字符串转字节数组
rawData := []byte(plainText)
// 2. Gzip压缩
compressed, err := GzipCompress(rawData)
if err != nil {
return "", fmt.Errorf("压缩失败:%v", err)
}
// 3. 异或加密
encrypted := Xor(compressed)
// 4. Base64编码
return base64.StdEncoding.EncodeToString(encrypted), nil
}
// DecryptAndDecompress Base64+异或+解压
func DecryptAndDecompress(encryptedStr string) (string, error) {
// 1. Base64解码
encryptedData, err := base64.StdEncoding.DecodeString(encryptedStr)
if err != nil {
return "", fmt.Errorf("Base64解码失败%v", err)
}
// 2. 异或解密
decrypted := Xor(encryptedData)
// 3. Gzip解压
decompressed, err := GzipDecompress(decrypted)
if err != nil {
return "", fmt.Errorf("解压失败:%v", err)
}
// 4. 字节数组转字符串
return string(decompressed), nil
}