在go语言中验证上传的文件是图片还是允许的文件类型代码示例

2023-06-01 00:00:00 示例 验证 文件类型

在go语言中如何验证上传到我的服务器的文件是否是指定的一个或多个类型,

比如:我只允许图像(JPG、PNG、GIF),验证确实是图片文件。


实现思路:

使用DetectContentType函数检测上传的文件。

仅当上传的文件是图像或允许的文件类型时才继续。

如果不是,则中止并提示错误消息。


示例代码:

 package main
 
 import (
         "fmt"
         "net/http"
         "os"
         "runtime"
 )
 
 func main() {
         //最大限度地提高CPU的使用率,以获得最大的性能
         runtime.GOMAXPROCS(runtime.NumCPU())
         
         //打开上传的文件
         file, err := os.Open("./img.png")
         if err != nil {
                 fmt.Println(err)
                 os.Exit(1)
         }
         
         //为什么是512字节?
         //可以查看手册了解:http://golang.org/pkg/net/http/#DetectContentType
         buff := make([]byte, 512)
         _, err = file.Read(buff)
         if err != nil {
                 fmt.Println(err)
                 os.Exit(1)
         }
         
         filetype := http.DetectContentType(buff)
         fmt.Println(filetype)
         
         switch filetype {
         case "image/jpeg", "image/jpg":
                 fmt.Println(filetype)
                 
         case "image/gif":
                 fmt.Println(filetype)
                 
         case "image/png":
                 fmt.Println(filetype)
                 
         //不是图像,而是应用
             case "application/pdf":
                fmt.Println(filetype)
             default:
                fmt.Println("上传的未知文件类型")
        }

}

相关文章