🎉 Initial commit

cache/compress.go

@@ -0,0 +1,142 @@
+package cache
+
+import (
+	"bytes"
+	"compress/gzip"
+	"io"
+	"strings"
+)
+
+// CompressOptions 压缩选项
+type CompressOptions struct {
+	Enabled      bool    // 是否启用压缩
+	Level        int     // 压缩级别 (1-9)，1为最快压缩，9为最佳压缩
+	MinSizeBytes int     // 最小压缩大小，小于此大小的数据不进行压缩
+	Ratio        float64 // 压缩比阈值，压缩后大小/原始大小，小于此值才保存压缩结果
+}
+
+// DefaultCompressOptions 默认压缩选项
+var DefaultCompressOptions = CompressOptions{
+	Enabled:      true,
+	Level:        6,   // 默认压缩级别为6，平衡压缩率和性能
+	MinSizeBytes: 100, // 默认最小压缩大小为100字节
+	Ratio:        0.9, // 默认压缩比阈值为0.9，即至少要压缩到原始大小的90%以下才保存压缩结果
+}
+
+// CompressSVG 压缩SVG数据
+// 返回压缩后的数据和是否进行了压缩
+func CompressSVG(svg string, options CompressOptions) ([]byte, bool) {
+	if !options.Enabled || len(svg) < options.MinSizeBytes {
+		return []byte(svg), false
+	}
+
+	// 创建一个bytes.Buffer来存储压缩数据
+	var buf bytes.Buffer
+
+	// 创建一个gzip.Writer，设置压缩级别
+	writer, err := gzip.NewWriterLevel(&buf, options.Level)
+	if err != nil {
+		return []byte(svg), false
+	}
+
+	// 写入SVG数据
+	_, err = writer.Write([]byte(svg))
+	if err != nil {
+		return []byte(svg), false
+	}
+
+	// 关闭writer，确保所有数据都被写入
+	err = writer.Close()
+	if err != nil {
+		return []byte(svg), false
+	}
+
+	// 获取压缩后的数据
+	compressed := buf.Bytes()
+
+	// 计算压缩比
+	ratio := float64(len(compressed)) / float64(len(svg))
+
+	// 如果压缩比不理想，返回原始数据
+	if ratio >= options.Ratio {
+		return []byte(svg), false
+	}
+
+	return compressed, true
+}
+
+// DecompressSVG 解压缩SVG数据
+func DecompressSVG(data []byte, isCompressed bool) (string, error) {
+	// 如果数据未压缩，直接返回字符串
+	if !isCompressed {
+		return string(data), nil
+	}
+
+	// 检查数据是否为gzip格式
+	if !isGzipped(data) {
+		return string(data), nil
+	}
+
+	// 创建一个gzip.Reader
+	reader, err := gzip.NewReader(bytes.NewReader(data))
+	if err != nil {
+		return string(data), err
+	}
+	defer reader.Close()
+
+	// 读取解压缩后的数据
+	decompressed, err := io.ReadAll(reader)
+	if err != nil {
+		return string(data), err
+	}
+
+	return string(decompressed), nil
+}
+
+// isGzipped 检查数据是否为gzip格式
+func isGzipped(data []byte) bool {
+	// gzip文件的魔数是0x1f 0x8b
+	return len(data) > 2 && data[0] == 0x1f && data[1] == 0x8b
+}
+
+// OptimizeSVG 优化SVG字符串，移除不必要的空白和注释
+func OptimizeSVG(svg string) string {
+	// 移除XML注释
+	svg = removeXMLComments(svg)
+
+	// 移除多余的空白
+	svg = removeExtraWhitespace(svg)
+
+	return svg
+}
+
+// removeXMLComments 移除XML注释
+func removeXMLComments(svg string) string {
+	for {
+		start := strings.Index(svg, "<!--")
+		if start == -1 {
+			break
+		}
+		end := strings.Index(svg[start:], "-->") + start
+		if end > start {
+			svg = svg[:start] + svg[end+3:]
+		} else {
+			break
+		}
+	}
+	return svg
+}
+
+// removeExtraWhitespace 移除多余的空白
+func removeExtraWhitespace(svg string) string {
+	// 替换多个空白字符为单个空格
+	svg = strings.Join(strings.Fields(svg), " ")
+
+	// 优化常见的SVG标签周围的空白
+	svg = strings.ReplaceAll(svg, "> <", "><")
+	svg = strings.ReplaceAll(svg, " />", "/>")
+	svg = strings.ReplaceAll(svg, " =", "=")
+	svg = strings.ReplaceAll(svg, "= ", "=")
+
+	return svg
+}