Golang中处理文件服务时遇到的错误解决方案

在处理Golang文件服务时，常见的错误包括文件路径问题、权限错误和并发访问冲突。以下是针对这些问题的解决方案和示例代码：

1. 文件路径问题

最常见的错误是相对路径导致的文件查找失败。使用绝对路径或工作目录检查：

func dogPic(w http.ResponseWriter, req *http.Request) {
    // 方法1：使用绝对路径
    wd, _ := os.Getwd()
    filePath := filepath.Join(wd, "toby.jpg")
    
    // 方法2：检查文件是否存在
    if _, err := os.Stat("toby.jpg"); os.IsNotExist(err) {
        http.Error(w, "file not found", http.StatusNotFound)
        return
    }
    
    f, err := os.Open("toby.jpg")
    if err != nil {
        http.Error(w, "file open error", http.StatusInternalServerError)
        return
    }
    defer f.Close()
    
    // 设置正确的Content-Type
    w.Header().Set("Content-Type", "image/jpeg")
    io.Copy(w, f)
}

2. 并发访问问题

当多个请求同时访问同一文件时，可能会遇到并发问题：

var (
    fileCache     []byte
    fileCacheOnce sync.Once
    fileCacheErr  error
)

func dogPic(w http.ResponseWriter, req *http.Request) {
    fileCacheOnce.Do(func() {
        data, err := os.ReadFile("toby.jpg")
        if err != nil {
            fileCacheErr = err
            return
        }
        fileCache = data
    })
    
    if fileCacheErr != nil {
        http.Error(w, fileCacheErr.Error(), http.StatusInternalServerError)
        return
    }
    
    w.Header().Set("Content-Type", "image/jpeg")
    w.Header().Set("Content-Length", strconv.Itoa(len(fileCache)))
    w.Write(fileCache)
}

3. 大文件处理优化

对于大文件，使用缓冲区和正确的错误处理：

func dogPic(w http.ResponseWriter, req *http.Request) {
    f, err := os.Open("toby.jpg")
    if err != nil {
        if os.IsNotExist(err) {
            http.Error(w, "file not found", http.StatusNotFound)
        } else if os.IsPermission(err) {
            http.Error(w, "permission denied", http.StatusForbidden)
        } else {
            http.Error(w, "internal server error", http.StatusInternalServerError)
        }
        return
    }
    defer f.Close()
    
    // 获取文件信息
    fi, err := f.Stat()
    if err != nil {
        http.Error(w, "file stat error", http.StatusInternalServerError)
        return
    }
    
    // 设置正确的headers
    w.Header().Set("Content-Type", "image/jpeg")
    w.Header().Set("Content-Length", strconv.FormatInt(fi.Size(), 10))
    
    // 使用带缓冲的复制
    buf := make([]byte, 4096)
    _, err = io.CopyBuffer(w, f, buf)
    if err != nil && err != io.EOF {
        // 处理复制过程中的错误
        log.Printf("Error copying file: %v", err)
    }
}

4. 完整的错误处理示例

结合所有错误处理的最佳实践：

func dogPic(w http.ResponseWriter, req *http.Request) {
    // 验证请求方法
    if req.Method != http.MethodGet {
        http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
        return
    }
    
    // 打开文件
    f, err := os.Open("toby.jpg")
    if err != nil {
        handleFileError(w, err)
        return
    }
    defer f.Close()
    
    // 获取文件状态
    fi, err := f.Stat()
    if err != nil {
        http.Error(w, "Could not get file info", http.StatusInternalServerError)
        return
    }
    
    // 设置headers
    w.Header().Set("Content-Type", getContentType("toby.jpg"))
    w.Header().Set("Content-Length", strconv.FormatInt(fi.Size(), 10))
    
    // 支持范围请求（部分内容）
    http.ServeContent(w, req, "toby.jpg", fi.ModTime(), f)
}

func handleFileError(w http.ResponseWriter, err error) {
    switch {
    case os.IsNotExist(err):
        http.Error(w, "File not found", http.StatusNotFound)
    case os.IsPermission(err):
        http.Error(w, "Permission denied", http.StatusForbidden)
    default:
        http.Error(w, "Internal server error", http.StatusInternalServerError)
    }
}

func getContentType(filename string) string {
    ext := filepath.Ext(filename)
    switch ext {
    case ".jpg", ".jpeg":
        return "image/jpeg"
    case ".png":
        return "image/png"
    case ".gif":
        return "image/gif"
    default:
        return "application/octet-stream"
    }
}

5. 使用http.FileServer替代方案

对于静态文件服务，考虑使用内置的FileServer：

func main() {
    // 单个文件服务
    http.HandleFunc("/", dog)
    http.Handle("/toby.jpg", http.StripPrefix("/toby.jpg", 
        http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            http.ServeFile(w, r, "toby.jpg")
        })))
    
    // 或者整个目录服务
    fs := http.FileServer(http.Dir("./static"))
    http.Handle("/static/", http.StripPrefix("/static/", fs))
    
    http.ListenAndServe(":8080", nil)
}

这些解决方案覆盖了文件服务中常见的错误场景，包括路径解析、权限验证、并发处理和性能优化。根据具体需求选择合适的方法。