Golang中如何创建包含目录的ZIP文件

问题在于你的代码没有正确处理目录条目。当创建ZIP文件时，目录需要特殊的处理方式。以下是修复后的代码：

package main

import (
	"archive/zip"
	"fmt"
	"io"
	"io/fs"
	"log"
	"os"
	"path/filepath"
	"strings"
	"time"
)

func main() {
	fmt.Println("Hello from Dir Walker")

	args := os.Args[1:]
	if len(args) < 2 {
		log.Fatal("Usage: program <source_dir> <output_zip>")
	}

	srcDir := args[0]
	outputFile := args[1]

	f, err := os.Create(outputFile)
	if err != nil {
		log.Fatal("Error creating zip file:", err)
	}
	defer f.Close()

	writer := zip.NewWriter(f)
	defer writer.Close()

	start := time.Now()
	
	// 使用正确的相对路径处理
	err = filepath.WalkDir(srcDir, func(path string, d fs.DirEntry, err error) error {
		if err != nil {
			return err
		}

		// 获取相对于源目录的路径
		relPath, err := filepath.Rel(srcDir, path)
		if err != nil {
			return err
		}

		// 跳过源目录本身
		if relPath == "." {
			return nil
		}

		// 确保目录路径以斜杠结尾
		if d.IsDir() {
			relPath += "/"
		}

		info, err := d.Info()
		if err != nil {
			return err
		}

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

		// 设置正确的文件名（使用相对路径）
		header.Name = filepath.ToSlash(relPath)
		
		// 对于目录，设置正确的标志
		if d.IsDir() {
			header.Method = zip.Store
		} else {
			header.Method = zip.Deflate
		}

		hw, err := writer.CreateHeader(header)
		if err != nil {
			return err
		}

		// 如果是目录或符号链接，不需要写入内容
		if d.IsDir() || (d.Type()&fs.ModeSymlink != 0) {
			return nil
		}

		// 处理普通文件
		f, err := os.Open(path)
		if err != nil {
			return err
		}
		defer f.Close()

		_, err = io.Copy(hw, f)
		return err
	})

	if err != nil {
		log.Printf("Error walking directory: %s\n", err)
	}

	t := time.Now()
	elapsed := t.Sub(start)
	fmt.Printf("Elapsed time: %v\n", elapsed)
}

关键修改点：

1. 目录路径处理：为目录路径添加斜杠后缀（relPath += "/"），这是ZIP格式中标识目录的标准方式。

2. 相对路径：使用filepath.Rel()获取相对于源目录的路径，避免在ZIP文件中包含绝对路径。

3. 目录压缩方法：为目录设置zip.Store方法（不压缩），因为目录条目没有实际内容需要压缩。

4. 路径分隔符：使用filepath.ToSlash()确保使用正斜杠作为路径分隔符，这是ZIP文件的标准。

5. 参数检查：添加了更清晰的参数检查和使用说明。

6. 跳过源目录：添加逻辑跳过源目录本身（当relPath == "."时）。

这里还有一个更简洁的版本，使用archive/zip包的CreateRaw方法：

func walker(path string, d fs.DirEntry, err error) error {
	if err != nil {
		return err
	}

	// 获取相对路径
	relPath, err := filepath.Rel(srcDir, path)
	if err != nil {
		return err
	}
	
	if relPath == "." {
		return nil
	}

	// 转换为ZIP格式的路径（使用正斜杠）
	zipPath := filepath.ToSlash(relPath)
	
	// 如果是目录，确保以斜杠结尾
	if d.IsDir() {
		zipPath += "/"
	}

	info, err := d.Info()
	if err != nil {
		return err
	}

	// 使用Create而不是CreateHeader来简化处理
	if d.IsDir() {
		_, err := writer.Create(zipPath)
		return err
	}

	hw, err := writer.Create(zipPath)
	if err != nil {
		return err
	}

	f, err := os.Open(path)
	if err != nil {
		return err
	}
	defer f.Close()

	_, err = io.Copy(hw, f)
	return err
}

使用writer.Create()方法会自动处理文件头信息，包括正确设置目录标志。对于目录，只需要创建一个空的条目即可。