feat(compressor): deflate compress

pkg/compressor/compressor.go

@@ -17,22 +17,8 @@
 
 package compressor
 
-type CompressorType int8
-
-const (
-	CompressorNone CompressorType = iota
-	CompressorGzip
-	CompressorZip
-	CompressorSevenz
-	CompressorBzip2
-	CompressorLz4
-	CompressorDefault
-	CompressorZstd
-	CompressorMax
-)
-
 type Compressor interface {
 	Compress([]byte) ([]byte, error)
 	Decompress([]byte) ([]byte, error)
-	GetCompressorType() CompressorType
+	GetCompressorType() CompressType
 }

pkg/compressor/defalte_compress_test.go

@@ -0,0 +1,40 @@
+package compressor
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestDeflateCompress(t *testing.T) {
+	ts := []struct {
+		text string
+	}{
+		{
+			text: "Don't communicate by sharing memory, share memory by communicating.",
+		},
+		{
+			text: "Concurrency is not parallelism.",
+		},
+		{
+			text: "The bigger the interface, the weaker the abstraction.",
+		},
+		{
+			text: "Documentation is for users.",
+		},
+	}
+
+	dc := &DeflateCompress{}
+	assert.EqualValues(t, CompressDeflate, dc.GetCompressorType())
+
+	for _, s := range ts {
+		var data []byte = []byte(s.text)
+		fmt.Println(len(data))
+		dataCompressed, _ := dc.Compress(data)
+		fmt.Println(len(dataCompressed))
+		ret, _ := dc.Decompress(dataCompressed)
+		fmt.Println(len(ret))
+		assert.EqualValues(t, s.text, string(ret))
+	}
+}

pkg/compressor/deflate_compress.go

@@ -0,0 +1,51 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package compressor
+
+import (
+	"bytes"
+	"compress/flate"
+	"io"
+
+	"github.com/seata/seata-go/pkg/util/log"
+)
+
+type DeflateCompress struct{}
+
+func (*DeflateCompress) Compress(data []byte) ([]byte, error) {
+	var buf bytes.Buffer
+	fw, err := flate.NewWriter(&buf, flate.BestCompression)
+	if err != nil {
+		log.Error(err)
+		return nil, err
+	}
+	defer fw.Close()
+	fw.Write(data)
+	fw.Flush()
+	return buf.Bytes(), nil
+}
+
+func (*DeflateCompress) Decompress(data []byte) ([]byte, error) {
+	fr := flate.NewReader(bytes.NewBuffer(data))
+	defer fr.Close()
+	return io.ReadAll(fr)
+}
+
+func (*DeflateCompress) GetCompressorType() CompressType {
+	return CompressDeflate
+}

pkg/compressor/deflater_compress.go → pkg/compressor/types.go

@@ -16,3 +16,18 @@
  */
 
 package compressor
+
+//go:generate stringer -type=CompressType
+type CompressType int8
+
+const (
+	_ CompressType = iota
+	CompressGzip
+	CompressZip
+	CompressSevenz
+	CompressBzip2
+	CompressLz4
+	CompressDeflate
+	CompressZstd
+	CompressMax
+)