golang替代python / flask send_from_directory()

时间:2017-06-03 18:09:16

标签: go go-gin

我有这个图片网址:

/book/cover/Computer_Science.png

但图像的位置实际上存在于

/uploads/img/Computer_Science.png

我正在使用 Gin框架。在Gin或内置的Golang函数中是否有像Flask的send_from_directory()这样的命令?

如果没有,你能分享一下如何做到吗?

谢谢!

1 个答案:

答案 0 :(得分:2)

使用gin的Context.File来提供文件内容。此方法在内部调用http.ServeFile内置函数。代码段将是:

import "path/filepath"


// ...
router := gin.Default()
// ... 

router.GET("/book/cover/:filename", func(c *gin.Context) {
    rootDir := "/uploads/img/"
    name := c.Param("filename")
    filePath, err :=  filepath.Abs(rootDir + name)
    if err != nil {
        c.AbortWithStatus(404)
    }

    //Only allow access to file/directory under rootDir
    //The following code is for ilustration since HasPrefix is deprecated.
    //Replace with correct one when https://github.com/golang/dep/issues/296 fixed
    if !filepath.HasPrefix(filePath, rootDir) {
        c.AbortWithStatus(404)
    }

    c.File(filePath)
})

<强>更新

正如zerkms所指出的,在传递路径名称之前必须对其进行清理 Context.File。片段中添加了简单的消毒剂。请适应您的需求。