"일꾼이 일을 잘하려면 먼저 도구를 갈고 닦아야 한다." - 공자, 『논어』.
첫 장 > 프로그램 작성 > `zip` 패키지를 사용하여 Go에서 파일의 압축을 쉽게 풀 수 있는 방법은 무엇입니까?

`zip` 패키지를 사용하여 Go에서 파일의 압축을 쉽게 풀 수 있는 방법은 무엇입니까?

2024년 11월 17일에 게시됨
검색:736

How to Easily Unzip Files in Go with the `zip` Package?

Go로 쉽게 압축 풀기

zip 패키지를 사용하면 Go에서 파일 압축을 쉽게 풀 수 있습니다. 단순함 덕분에 다음과 같은 유틸리티 함수를 만들 수 있습니다:

func Unzip(src, dest string) error {
    r, err := zip.OpenReader(src)
    if err != nil {
        return err
    }
    defer func() {
        if err := r.Close(); err != nil {
            panic(err)
        }
    }()

    os.MkdirAll(dest, 0755)

    // Closure isolates file descriptor .Close() calls
    extractAndWriteFile := func(f *zip.File) error {
        rc, err := f.Open()
        if err != nil {
            return err
        }
        defer func() {
            if err := rc.Close(); err != nil {
                panic(err)
            }
        }()

        path := filepath.Join(dest, f.Name)

        // Prevent ZipSlip vulnerability
        if !strings.HasPrefix(path, filepath.Clean(dest) string(os.PathSeparator)) {
            return fmt.Errorf("illegal file path: %s", path)
        }

        if f.FileInfo().IsDir() {
            os.MkdirAll(path, f.Mode())
        } else {
            os.MkdirAll(filepath.Dir(path), f.Mode())
            f, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
            if err != nil {
                return err
            }
            defer func() {
                if err := f.Close(); err != nil {
                    panic(err)
                }
            }()

            _, err = io.Copy(f, rc)
            if err != nil {
                return err
            }
        }
        return nil
    }

    for _, f := range r.File {
        err := extractAndWriteFile(f)
        if err != nil {
            return err
        }
    }

    return nil
}

이 코드는 zip 아카이브의 각 파일을 꼼꼼하게 반복하여 파일 경로 유효성을 보장하는 동시에 디렉터리와 일반 파일을 적절하게 처리합니다. 클로저 접근 방식은 불필요한 지연 스택을 제거하고, 캡슐화를 강화하며, 오류 처리를 간소화합니다.

최신 튜토리얼 더>

부인 성명: 제공된 모든 리소스는 부분적으로 인터넷에서 가져온 것입니다. 귀하의 저작권이나 기타 권리 및 이익이 침해된 경우 자세한 이유를 설명하고 저작권 또는 권리 및 이익에 대한 증거를 제공한 후 이메일([email protected])로 보내주십시오. 최대한 빨리 처리해 드리겠습니다.

Copyright© 2022 湘ICP备2022001581号-3