在Go Martini框架中高效服务动态生成图像的实践指南

go get github.com/codegangsta/martini
go get github.com/nfnt/resize

package main

import (
    "image"
    "image/jpeg"
    "log"
    "net/http"
    "os"

    "github.com/codegangsta/martini"
    "github.com/nfnt/resize"
)

// thumb 函数负责读取、解码、缩放图像并返回 image.Image 对象
func thumb() image.Image {
    // 打开本地图片文件
    file, err := os.Open("test.jpg")
    if err != nil {
        log.Fatal(err) // 实际应用中应进行更优雅的错误处理
    }
    defer file.Close() // 确保文件关闭

    // 解码JPEG图片
    img, err := jpeg.Decode(file)
    if err != nil {
        log.Fatal(err) // 实际应用中应进行更优雅的错误处理
    }

    // 使用 nfnt/resize 库缩放图片
    // 0 表示根据宽度自动计算高度，200 是指定的高度
    m := resize.Resize(0, 200, img, resize.MitchellNetravali)

    return m
}

func main() {
    m := martini.Classic()

    // 定义 GET / 路由处理器
    m.Get("/", func(res http.ResponseWriter, req *http.Request) {
        // 1. 设置 Content-Type 头部为 image/jpeg
        res.Header().Set("Content-Type", "image/jpeg")

        // 2. 调用 thumb() 获取处理后的图像
        img := thumb()

        // 3. 将 image.Image 对象编码为JPEG格式并直接写入 ResponseWriter
        // jpeg.Options{100} 设置JPEG质量为100
        err := jpeg.Encode(res, img, &jpeg.Options{Quality: 100})

        // 4. 根据编码结果设置HTTP状态码
        if err != nil {
            log.Printf("Error encoding image: %v", err)
            res.WriteHeader(http.StatusInternalServerError) // 编码失败返回500
        } else {
            res.WriteHeader(http.StatusOK) // 成功返回200
        }
    })

    // 启动Martini服务
    m.Run()
}

package main

import (
    "image"
    "image/jpeg"
    "log"
    "net/http"
    "os"

    "github.com/codegangsta/martini"
    "github.com/nfnt/resize"
)

// thumb 函数负责读取、解码、缩放图像并返回 image.Image 对象
func thumb() image.Image {
    // 打开本地图片文件
    file, err := os.Open("test.jpg")
    if err != nil {
        log.Fatal(err) // 在生产环境中，应返回错误并让上层处理，而不是直接退出
    }
    defer file.Close() // 确保文件关闭

    // 解码JPEG图片
    img, err := jpeg.Decode(file)
    if err != nil {
        log.Fatal(err) // 同上
    }

    // 使用 nfnt/resize 库缩放图片
    // 0 表示根据宽度自动计算高度，200 是指定的高度
    m := resize.Resize(0, 200, img, resize.MitchellNetravali)

    return m
}

func main() {
    m := martini.Classic()

    // 定义 GET / 路由处理器
    m.Get("/", func(res http.ResponseWriter, req *http.Request) {
        // 1. 设置 Content-Type 头部为 image/jpeg
        res.Header().Set("Content-Type", "image/jpeg")

        // 2. 调用 thumb() 获取处理后的图像
        img := thumb()

        // 3. 将 image.Image 对象编码为JPEG格式并直接写入 ResponseWriter
        // jpeg.Options{Quality: 100} 设置JPEG质量为100 (0-100)
        err := jpeg.Encode(res, img, &jpeg.Options{Quality: 100})

        // 4. 根据编码结果设置HTTP状态码
        if err != nil {
            log.Printf("Error encoding image: %v", err)
            // 编码失败时，返回500 Internal Server Error
            // 注意：如果头部已经写入，WriteHeader可能无效，但这里是在写入体之前
            res.WriteHeader(http.StatusInternalServerError)
            // 可以选择写入一个错误消息到响应体
            // fmt.Fprintf(res, "Failed to encode image")
        } else {
            // 成功时返回200 OK
            res.WriteHeader(http.StatusOK)
        }
    })

    // 启动Martini服务，默认监听 :3000
    m.Run()
}