python/flask send_from_directory() 的 Golang 替代方案
Golang alternative for python/flask send_from_directory()
我有这个图片网址:
/book/cover/Computer_Science.png
但是图片的位置居然存在
下
/uploads/img/Computer_Science.png
我正在使用 Gin 框架。在 Gin 或内置的 Golang 函数中是否有类似 Flask 的 send_from_directory()
的命令?
如果没有,您能分享一下如何做的片段吗?
谢谢!
使用 gin 的 Context.File
to serve file content. This method internally calls 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
。代码片段中添加了简单的消毒剂。请适应您的需求。
我有这个图片网址:
/book/cover/Computer_Science.png
但是图片的位置居然存在
下/uploads/img/Computer_Science.png
我正在使用 Gin 框架。在 Gin 或内置的 Golang 函数中是否有类似 Flask 的 send_from_directory()
的命令?
如果没有,您能分享一下如何做的片段吗?
谢谢!
使用 gin 的 Context.File
to serve file content. This method internally calls 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
。代码片段中添加了简单的消毒剂。请适应您的需求。