package compression

import (
?? ?"bytes"
?? ?"compress/gzip"
?? ?"encoding/json"
)

func GzipEncode(input []byte) ([]byte, error) {
?? ?// 創(chuàng)建一個新的 byte 輸出流
?? ?var buf bytes.Buffer
?? ?// 創(chuàng)建一個新的 gzip 輸出流

?? ?//NoCompression ? ? ?= flate.NoCompression ? ? ?// 不壓縮
?? ?//BestSpeed ? ? ? ? ?= flate.BestSpeed ? ? ? ? ?// 最快速度
?? ?//BestCompression ? ?= flate.BestCompression ? ?// 最佳壓縮比
?? ?//DefaultCompression = flate.DefaultCompression // 默認(rèn)壓縮比
?? ?//gzip.NewWriterLevel()
?? ?gzipWriter := gzip.NewWriter(&buf)
?? ?// 將 input byte 數(shù)組寫入到此輸出流中
?? ?_, err := gzipWriter.Write(input)
?? ?if err != nil {
?? ??? ?_ = gzipWriter.Close()
?? ??? ?return nil, err
?? ?}
?? ?if err := gzipWriter.Close(); err != nil {
?? ??? ?return nil, err
?? ?}
?? ?// 返回壓縮后的 bytes 數(shù)組
?? ?return buf.Bytes(), nil
}

func GzipDecode(input []byte) ([]byte, error) {
?? ?// 創(chuàng)建一個新的 gzip.Reader
?? ?bytesReader := bytes.NewReader(input)
?? ?gzipReader, err := gzip.NewReader(bytesReader)
?? ?if err != nil {
?? ??? ?return nil, err
?? ?}
?? ?defer func() {
?? ??? ?// defer 中關(guān)閉 gzipReader
?? ??? ?_ = gzipReader.Close()
?? ?}()
?? ?buf := new(bytes.Buffer)
?? ?// 從 Reader 中讀取出數(shù)據(jù)
?? ?if _, err := buf.ReadFrom(gzipReader); err != nil {
?? ??? ?return nil, err
?? ?}
?? ?return buf.Bytes(), nil
}

// 壓縮
func MarshalJsonAndGzip(data interface{}) ([]byte, error) {
?? ?marshalData, err := json.Marshal(data)
?? ?if err != nil {
?? ??? ?return nil, err
?? ?}
?? ?gzipData, err := GzipEncode(marshalData)
?? ?if err != nil {
?? ??? ?return nil, err
?? ?}
?? ?return gzipData, err
}

// 解壓
func UnmarshalDataFromJsonWithGzip(input []byte, output interface{}) error {
?? ?decodeData, err := GzipDecode(input)
?? ?if err != nil {
?? ??? ?return err
?? ?}

?? ?err = json.Unmarshal(decodeData, output)
?? ?if err != nil {
?? ??? ?return err
?? ?}
?? ?return nil
}