golang压缩方法

2023-05-22 09:05:43 方法 压缩 Golang

golang 是一种非常流行的高性能编程语言，具有比较强大的代码执行能力，并且它的标准库中也拥有许多对各种文件压缩格式进行处理的方法。本文将介绍 Golang 压缩方法的使用。

首先需要引入 "compress" 和 "arcHive" 包，它们都是 golang 标准库中的压缩处理相关包。但具体选用哪一个，需要根据压缩格式来进行选择。

  1. gzip 压缩

在 golang 中进行 gzip 压缩的方法，同样利用标准库中的 "compress/gzip" 包。

这里我们以一个字符串为例进行压缩：

package main

import (
    "bytes"
    "compress/gzip"
    "fmt"
)

func main() {
    str := "golang gzip test"
    var buf bytes.Buffer
    z := gzip.NewWriter(&buf)
    _, err := z.Write([]byte(str))
    if err != nil {
        panic(err)
    }
    err = z.Close()
    if err != nil {
        panic(err)
    }
    fmt.Println("gzip:", buf.String())
}

这段代码中首先创建了一个 buffer 缓存，然后再创建了一个 gzip.Write 对象，并且将 buffer 传递给了这个对象。接着，将需要进行压缩的字符串写入这个对象中，最后关闭这个 writer 对象。

打印结果为：gzip: �▒H-IM0189WVnV-I�HI�J-.�V�R,Q�P.�-NMV-.WVN��O�,�R��S�Q�L��KՅ_(�з)_/�

  1. zlib 压缩

zlib 是一种无损数据压缩格式，采用 Lempel-Ziv 算法和 Huffam 编码。它的压缩率比较高，且压缩速度也比较快，可用于某些数据传输和存储场景。

在 golang 中进行 zlib 压缩的方法，同样利用标准库中的 "compress/zlib" 包。

这里我们以一个字符串为例进行压缩：

package main

import (
    "bytes"
    "compress/zlib"
    "fmt"
)

func main() {
    str := "golang zlib test"
    var buf bytes.Buffer
    w := zlib.NewWriter(&buf)
    _, err := w.Write([]byte(str))
    if err != nil {
        panic(err)
    }
    err = w.Close()
    if err != nil {
        panic(err)
    }
    fmt.Println("zlib:", buf.String())
}

这段代码中首先创建了一个 buffer 缓存，然后再创建了一个 zlib.Write 对象，并且将 buffer 传递给了这个对象。接着，将需要进行压缩的字符串写入这个对象中，最后关闭这个 writer 对象。

打印结果为：zlib: x��+��J-.N�(,�QP.I,�M-V-.Q�

  1. tar 压缩

tar 是一种归档文件格式，经常用于将多个文件或目录打包成一个文件。在 golang 中进行 tar 压缩的方法，可以使用标准库中的 "archive/tar" 包。

这里我们以将一个目录进行 tar 压缩为例：

package main

import (
    "archive/tar"
    "fmt"
    "io"
    "os"
)

func tarPath(dst, src string) error {
    info, err := os.Stat(src)
    if err != nil {
        return err 
    }

    // 如果源目录是一个文件，直接对这个文件进行压缩
    if !info.IsDir() {
        srcFile, err := os.Open(src)
        if err != nil {
            return err
        }
        defer srcFile.Close()

        dstFile, err := os.Create(dst + ".tar")
        if err != nil {
            return err
        }
        defer dstFile.Close()

        tarWriter := tar.NewWriter(dstFile)
        defer tarWriter.Close()

        hdr := &tar.Header {
            Name: src,
            Mode: int64(info.Mode()),
            Size: info.Size(),
        }

        if err := tarWriter.WriteHeader(hdr); err != nil {
            return err 
        }

        if _, err := io.Copy(tarWriter, srcFile); err != nil {
            return err
        }

        fmt.Println("tar file created:", dst+".tar")
        return nil 
    }

    // 如果源目录是一个文件夹，先遍历源目录
    files, err := os.ReadDir(src)
    if err != nil {
        return err 
    }

    for _, file := range files {
        fileName := file.Name()

        // 这里需要再次判断是否是一个目录
        if file.IsDir() {
            fmt.Println("skipping directory:", fileName)
            continue
        }

        srcFile, err := os.Open(filepath.Join(src, fileName))
        if err != nil {
            return err
        }
        defer srcFile.Close()

        dstFile, err := os.Create(filepath.Join(dst, fileName) + ".tar")
        if err != nil {
            return err
        }
        defer dstFile.Close()

        tarWriter := tar.NewWriter(dstFile)
        defer tarWriter.Close()

        hdr := &tar.Header {
            Name: fileName,
            Mode: int64(file.Mode()),
            Size: file.Size(),
        }

        if err := tarWriter.WriteHeader(hdr); err != nil {
            return err 
        }

        if _, err := io.Copy(tarWriter, srcFile); err != nil {
            return err
        }

        fmt.Println("tar file created:", filepath.Join(dst, fileName)+".tar")
    }

    return nil 
}

func main() {
    srcPath := "./testdir"
    dstPath := "./"

    err := tarPath(dstPath, srcPath)
    if err != nil {
        fmt.Println(err)
    }
}

这段代码中，首先对源路径进行了一个判断，如果是一个文件，则直接进行压缩保存；如果是一个文件夹，则遍历其中的所有文件并对每个文件进行压缩保存，并将所有文件进行打包压缩。

  1. zip 压缩

zip 是一种归档文件格式，它通常用于将一组文件或者目录打包成一个文件，同时也会对这些文件进行压缩。在 golang 中进行 zip 压缩的方法，可以使用标准库中的 "archive/zip" 包。

这里我们以将两个文件进行 zip 压缩为例：

package main

import (
    "archive/zip"
    "fmt"
    "io"
    "os"
)

func zipFiles(dst string, files []string) error {
    newZipFile, err := os.Create(dst + ".zip")
    if err != nil {
        return err
    }
    defer newZipFile.Close()
    zipWriter := zip.NewWriter(newZipFile)
    defer zipWriter.Close()

    for _, file := range files {
        srcFile, err := os.Open(file)
        if err != nil {
            return err
        }
        defer srcFile.Close()

        info, _ := srcFile.Stat()

        header, err := zip.FileInfoHeader(info)
        if err != nil {
            return err
        }
        header.Name = file
        header.Method = zip.Deflate

        writer, err := zipWriter.CreateHeader(header)
        if err != nil {
            return err
        }
        if _, err := io.Copy(writer, srcFile); err != nil {
            return err
        }
    }

    fmt.Println("zip file created:", dst+".zip")
    return nil 
}

func main() {
    files := []string{"test.txt", "test1.txt"}
    dstPath := "./"

    err := zipFiles(dstPath, files)
    if err != nil {
        fmt.Println(err)
    }
}

这段代码中，主要是通过遍历需要进行压缩的文件列表，并将它们逐个添加到压缩包中。

总结

golang 对于各种压缩格式都有相应的处理类库，极大的方便了对文件压缩的处理流程，这里仅展示了几种流行的文件压缩方式的处理方法。

以上就是golang压缩方法的详细内容，更多请关注其它相关文章！

相关文章