使用Martini框架在Go中直接提供动态图像流

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

package main

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

    "github.com/nfnt/resize"
)

// thumb 函数用于加载图片，并将其调整大小
func thumb() image.Image {
    // 假设存在一个名为 "test.jpg" 的图片文件
    file, err := os.Open("test.jpg")
    if err != nil {
        log.Fatalf("无法打开图片文件: %v", err) // 使用 Fatalf 确保错误时程序退出
    }
    defer file.Close() // 确保文件句柄被关闭

    // 解码 JPEG 图片
    img, err := jpeg.Decode(file)
    if err != nil {
        log.Fatalf("无法解码图片: %v", err)
    }

    // 调整图片大小，宽度为0表示按比例缩放，高度为200像素
    m := resize.Resize(0, 200, img, resize.MitchellNetravali)

    return m
}

package main

import (
    "image/jpeg"
    "log"
    "net/http"
    "os" // 导入 os 包以创建 test.jpg 文件

    "github.com/go-martini/martini" // 更新为 go-martini
    "github.com/nfnt/resize"
)

// thumb 函数与上文相同
func thumb() image.Image {
    file, err := os.Open("test.jpg")
    if err != nil {
        log.Fatalf("无法打开图片文件: %v", err)
    }
    defer file.Close()

    img, err := jpeg.Decode(file)
    if err != nil {
        log.Fatalf("无法解码图片: %v", err)
    }

    m := resize.Resize(0, 200, img, resize.MitchellNetravali)
    return m
}

func main() {
    // 为了运行示例，先创建一个 test.jpg 文件
    // 实际应用中，你可能从其他地方获取图片
    createTestImage()

    m := martini.Classic()

    // 定义 GET / 路由处理器
    m.Get("/", func(res http.ResponseWriter, req *http.Request) {
        // 1. 设置 Content-Type 头部为 image/jpeg
        // 这告诉浏览器响应内容是一个JPEG图像
        res.Header().Set("Content-Type", "image/jpeg")

        // 2. 调用 thumb() 获取处理后的 image.Image 对象
        imgToServe := thumb()

        // 3. 将 image.Image 对象编码为 JPEG 格式并直接写入 ResponseWriter
        // jpeg.Encode 的第一个参数是 io.Writer，http.ResponseWriter 实现了此接口
        // jpeg.Options{100} 设置 JPEG 质量为 100（最高质量）
        err := jpeg.Encode(res, imgToServe, &jpeg.Options{Quality: 90}) // 通常90-95质量足够

        // 4. 根据编码结果设置 HTTP 状态码
        if err != nil {
            log.Printf("编码图片失败: %v", err)
            res.WriteHeader(http.StatusInternalServerError) // 500 Internal Server Error
        } else {
            res.WriteHeader(http.StatusOK) // 200 OK
        }
    })

    log.Println("Martini 服务器在 :3000 端口启动...")
    m.Run() // 默认在 :3000 端口运行
}

// createTestImage 用于生成一个简单的 test.jpg 文件以供示例使用
func createTestImage() {
    // 检查文件是否已存在，如果不存在则创建
    if _, err := os.Stat("test.jpg"); os.IsNotExist(err) {
        log.Println("创建 test.jpg 示例文件...")
        // 这里只是一个简单的占位符，实际应用中你需要一个真实的图片文件
        // 为了快速测试，你可以下载一个小的JPEG图片并命名为 test.jpg
        // 例如，从网络下载一个小的 logo.jpg 并重命名为 test.jpg
        // 或者使用 Go 的 image/draw 包生成一个简单的图片
        // 这里我们简化处理，假设用户会提供一个 test.jpg 文件
        log.Println("请确保当前目录下存在一个名为 'test.jpg' 的图片文件。")
        // 创建一个空的占位文件，以便程序能够编译运行，但实际图片内容需要手动提供
        file, err := os.Create("test.jpg")
        if err != nil {
            log.Fatalf("无法创建 test.jpg 文件: %v", err)
        }
        file.Close()
    }
}