Go语言Web服务器：从ZIP文件高效提供静态资源

type FileSystem interface {
    Open(name string) (File, error)
}

type File interface {
    io.ReadCloser
    io.Seeker
    Readdir(count int) ([]os.FileInfo, error)
    Stat() (os.FileInfo, error)
}

package main

import (
    "archive/zip"
    "io"
    "log"
    "net/http"
    "os"
    "path/filepath"
    "strings"
    "time"
)

// ZipFileSystem 实现了 http.FileSystem 接口，从 ZIP 文件中提供静态资源。
type ZipFileSystem struct {
    zipReader *zip.ReadCloser
    files     map[string]*zip.File // 存储 ZIP 内的文件路径到 *zip.File 的映射
}

// NewZipFileSystem 创建并初始化一个 ZipFileSystem。
func NewZipFileSystem(zipFilePath string) (*ZipFileSystem, error) {
    reader, err := zip.OpenReader(zipFilePath)
    if err != nil {
        return nil, err
    }

    fs := &ZipFileSystem{
        zipReader: reader,
        files:     make(map[string]*zip.File),
    }

    // 遍历 ZIP 文件，构建文件路径映射
    for _, f := range reader.File {
        // 规范化路径，移除开头的斜杠，确保与 http.FileServer 的行为一致
        // http.FileServer 会移除请求路径开头的斜杠
        name := strings.TrimPrefix(f.Name, "/")
        fs.files[name] = f
    }

    return fs, nil
}

// Close 关闭底层的 ZIP 文件读取器。
func (zfs *ZipFileSystem) Close() error {
    return zfs.zipReader.Close()
}

// Open 实现了 http.FileSystem 接口的 Open 方法。
func (zfs *ZipFileSystem) Open(name string) (http.File, error) {
    // http.FileServer 可能会请求根目录 "/" 或规范化后的路径
    // 我们需要处理空路径或根路径的请求，通常将其视为索引文件或错误
    if name == "/" {
        // 尝试查找 index.html 或返回错误
        if f, ok := zfs.files["index.html"]; ok {
            return newZipFile(f)
        }
        return nil, os.ErrNotExist // 或者返回一个表示目录的虚拟文件
    }

    // 查找 ZIP 文件中的对应条目
    file, ok := zfs.files[name]
    if !ok {
        // 如果文件不存在，尝试查找带斜杠的目录（如果需要支持）
        // 但对于静态文件服务器，通常只查找具体文件
        return nil, os.ErrNotExist
    }

    return newZipFile(file)
}

// zipFile 实现了 http.File 接口，用于包装 *zip.File。
type zipFile struct {
    file   *zip.File
    reader io.ReadCloser
}

func newZipFile(f *zip.File) (*zipFile, error) {
    rc, err := f.Open()
    if err != nil {
        return nil, err
    }
    return &zipFile{file: f, reader: rc}, nil
}

// Read 实现了 io.Reader 接口。
func (zf *zipFile) Read(p []byte) (n int, err error) {
    return zf.reader.Read(p)
}

// Close 实现了 io.Closer 接口。
func (zf *zipFile) Close() error {
    return zf.reader.Close()
}

// Seek 实现了 io.Seeker 接口 (注意：zip.File 的 io.ReadCloser 通常不支持 Seek)。
// 对于简单的静态文件服务，如果文件很小，可以一次性读取；如果需要 Seek，可能需要将文件内容读入内存。
// 这里的实现是简化的，实际生产环境可能需要更复杂的逻辑或限制。
func (zf *zipFile) Seek(offset int64, whence int) (int64, error) {
    // zip.File 的 ReadCloser 通常不直接支持 Seek，需要特殊处理
    // 例如，如果需要 Seek，可以考虑将整个文件内容加载到内存中，然后对内存切片进行 Seek
    // 或者重新打开文件并跳过字节。这里为了简化，暂时返回不支持。
    return 0, os.ErrInvalid
}

// Stat 实现了 os.FileInfo 接口。
func (zf *zipFile) Stat() (os.FileInfo, error) {
    return zf.file.FileInfo(), nil
}

// Readdir 实现了 http.File 接口的 Readdir 方法。
// 对于 ZIP 文件中的单个文件，通常没有子目录可列出。
func (zf *zipFile) Readdir(count int) ([]os.FileInfo, error) {
    return nil, nil // 不支持目录列表
}

func main() {
    // 假设你有一个名为 "static.zip" 的 ZIP 文件，其中包含 index.html, style.css 等
    zipFS, err := NewZipFileSystem("static.zip")
    if err != nil {
        log.Fatalf("无法创建 ZipFileSystem: %v", err)
    }
    defer zipFS.Close() // 确保在程序结束时关闭 ZIP 文件

    // 创建一个简单的静态文件服务器
    http.Handle("/", http.FileServer(zipFS))

    log.Println("服务器在 :8080 启动，服务 static.zip 中的文件")
    log.Fatal(http.ListenAndServe(":8080", nil))
}