三分钟用Golang搭建一个HTTP文件上传下载服务器

已于 2022-09-30 11:03:53 修改
阅读量4.3k 收藏 9
点赞数
分类专栏： 工具类 文章标签： 服务器 golang http 文件服务器
于 2022-09-30 09:31:50 首次发布
著作权归作者所有,商业转载请联系作者获得授权，非商业转载请注明出处！更多精彩内容，欢迎访问一只海星的主页:https://www.hnoyi.com/
本文链接：https://blog.csdn.net/haysonzeng/article/details/127117806
版权

背景

简单需要一个文件服务器来传递数据，只要两个功能，一个上传接口，一个下载接口。

选用go http模块实现，比nginx、ftp等更方便快捷。

需求整理

  • 上传接口"/v1/file_upload/"
  • 上传接口增加简单BasicAuth鉴权
  • 上传成功返回下载URL
    json格式返回

电梯直达

只想文件上传服务器测试接口，以下电梯直达即可
CSDN 5积分下载
白嫖

测试效果

在这里插入图片描述在这里插入图片描述

GO代码实现

package main

import (
	"bytes"
	"encoding/base64"
	"encoding/json"
	"flag"
	"fmt"
	"io"
	"log"
	"net/http"
	"os"
	"path/filepath"
	"strings"
)

type ViewFunc func(http.ResponseWriter, *http.Request)

func http_resp(code int, msg string, w http.ResponseWriter) {
	var Result map[string]interface{}
	Result = make(map[string]interface{})

	Result["code"] = code
	Result["msg"] = msg

	data, err := json.Marshal(Result)
	if err != nil {
		log.Printf("%v\n", err)
	}

	w.Write([]byte(string(data)))
}

func upload(w http.ResponseWriter, r *http.Request) {
	if r.Method == "GET" {
		t, _ := template.ParseFiles("upload.html")
		t.Execute(w, nil)
		return
	}
	
	contentType := r.Header.Get("content-type")
	contentLen := r.ContentLength

	if !strings.Contains(contentType, "multipart/form-data") {
		http_resp(-1001, "The content-type must be multipart/form-data", w)
		return
	}
	//限制最大文件大小
	if contentLen >= 50*1024*1024 {
		http_resp(-1002, "Failed to large,limit 50MB", w)
		return
	}

	err := r.ParseMultipartForm(50 * 1024 * 1024)
	if err != nil {
		http_resp(-1003, "Failed to ParseMultipartForm", w)
		return
	}

	if len(r.MultipartForm.File) == 0 {
		http_resp(-1004, "File is NULL", w)
		return
	}

	var Result map[string]interface{}
	Result = make(map[string]interface{})

	Result["code"] = 0
	DownLoadUrl := "http://192.168.10.9:8080/"

	FileCount := 0
	for _, headers := range r.MultipartForm.File {
		for _, header := range headers {
			log.Printf("recv file: %s\n", header.Filename)

			filePath := filepath.Join("./upload", header.Filename)
			dstFile, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE, 0666)
			if err != nil {
				log.Printf("Create file %s error: %s\n", filePath, err)
				return
			}
			srcFile, err := header.Open()
			if err != nil {
				log.Printf("Open header failed: %s\n", err)
			}
			_, err = io.Copy(dstFile, srcFile)
			if err != nil {
				log.Printf("Write file %s error: %s\n", filePath, err)
			}
			_, _ = srcFile.Close(), dstFile.Close()
			FileCount++
			name := fmt.Sprintf("file%d_url", FileCount)
			Result[name] = (DownLoadUrl + header.Filename)
		}
	}
	data, err := json.Marshal(Result)
	if err != nil {
		log.Printf("%v \n", err)
	}
	w.Write([]byte(string(data)))
}

func BasicAuth(f ViewFunc, user, passwd []byte) ViewFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		basicAuthPrefix := "Basic "

		// 获取 request header
		auth := r.Header.Get("Authorization")
		// 如果是 http basic auth
		if strings.HasPrefix(auth, basicAuthPrefix) {
			// 解码认证信息
			payload, err := base64.StdEncoding.DecodeString(
				auth[len(basicAuthPrefix):],
			)
			if err == nil {
				pair := bytes.SplitN(payload, []byte(":"), 2)
				if len(pair) == 2 && bytes.Equal(pair[0], user) &&
					bytes.Equal(pair[1], passwd) {
					// 执行被装饰的函数
					f(w, r)
					return
				}
			}
		}

		// 认证失败，提示 401 Unauthorized
		// Restricted 可以改成其他的值
		w.Header().Set("WWW-Authenticate", `Basic realm="Restricted"`)
		// 401 状态码
		w.WriteHeader(http.StatusUnauthorized)
	}
}

func main() {
	addr := flag.String("addr", "0.0.0.0:8080", "服务绑定地址端口")
	user := flag.String("user", "", "http认证用户名(如果用户名为空,则不需要认证)")
	psw := flag.String("psw", "", "http认证密码")
	root := flag.String("root", "./upload/", "文件服务根目录")
	flag.Parse()

	os.Mkdir(*root, 0777)

	http_user := []byte(*user)
	http_psw := []byte(*psw)

	log.Printf("http server listen on %v \n", *addr)

	if *user != "" {
		http.HandleFunc("/v1/file_upload", BasicAuth(upload, http_user, http_psw))
	} else {
		http.HandleFunc("/v1/file_upload", upload)
	}

	http.Handle("/", http.StripPrefix("/", http.FileServer(http.Dir(*root))))
	err := http.ListenAndServe(*addr, nil)
	if err != nil {
		log.Printf("%v", err)
	}
}

网页上传

新建文件upload.html放到http服务器root目录。例如upload文件夹

<!DOCTYPE html>
 <html>
      <head>
      <meta charset="utf-8">
      <title>go 上传test</title>
     </head>
<body>
<form method="POST" action="/v1/file_upload" enctype="multipart/form-data" >
<input type="file" name="uploadfile" />
<input type="submit" value="上传">
</form>
</body>
</html>

在这里插入图片描述
在这里插入图片描述

参考

《如何为 HTTP Server 增加 HTTP Basic Auth 》

一只海星
关注
  • 0
    点赞
  • 踩
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
从Java到Go：使用Go语言构建文件上传和下载服务器
sybh的博客
06-07 281
在本篇博客中，我们学习了如何使用Go语言构建一个文件上传和下载服务器，并探讨了从Java到Go的过渡。我们了解了Go和Java的基本语法差异，并通过实际示例展示了如何在Go中实现HTTP服务器以及文件上传和下载功能。最后，我们讨论了如何优化和扩展该服务器以满足实际需求。希望本文能帮助你更好地理解Go语言，并为你在实际项目中应用Go提供一些启示。祝你学习Go愉快！
golang语言实现的文件上传与文件下载功能示例
01-20
本文实例讲述了golang实现的文件上传与文件下载功能。分享给大家供大家参考，具体如下： upload.go 复制代码 代码如下:package common import (  “io/ioutil”  “os”  “path”  “github.com/gin-gonic/gin”  “googo.co/goo”  “googo.co/utils” ) const (  UPLOAD_DIR = “static/” ) type Upload struct { } func (this Upload) DoHandle(c *gin.Context) *goo.Result {  fh,
golang搭建文件管理服务器,搭建本地 golang 文档服务器
weixin_34928052的博客
08-03 296
平时我们看 golang 文档的方式通常有：1. `go doc `2. golang.org/pkg 或者国内的 godoc.golangtc.com/pkg/3. 看源码，当然这是最好的方式，只是需要更多的时间和耐心。如果没有网络，我一般会选择 方式1 + 方式3，但缺点是可读性差。后来了解到，`godoc -html ` 可以生成 HTML 文档，所以就思考如何搭建本地的 localhost...
golang实现文件上传api接口（基于net/http），不使用三方框架
最新发布
m0_61174587的博客
03-12 412
文件上传
golang简单获取上传文件大小的实现代码
09-21
主要介绍了golang简单获取上传文件大小的方法,涉及Go语言文件传输及文件属性操作的相关技巧,需要的朋友可以参考下
golang并发下载多个文件的方法
09-19
今天小编就为大家分享一篇golang并发下载多个文件的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
go之搭建一个文件服务器
萧海的博客
10-02 215
【代码】go之搭建一个文件服务器。
文件下载和上传功能服务器搭建配置
https://wangpengcheng0616.github.io/
06-30 6444
三种实现文件下载和上传功能服务器搭建配置 1.nginx 2.miniserve 3.nodejs
[Python http.server] 搭建http服务器用于下载/上传文件
热门推荐
Norths_的博客
05-17 1万+
动机：笔者需测试bs架构下的文件上传与下载性能，故想通过Python搭建http服务器并实现客户端与服务器之间的文件上传和下载需求 难点：这应该是很基础的东西，不过笔者之前未接触过http编程，谨在此记录下学习的过程，可能不是最优解 方法：在服务器端部署html页面，并执行监听的Python代码，在客户端操作实现文件的上传与下载
golang文件服务器的两种方式(可以访问任何目录)
n_fly的博客
04-07 2344
一、方法1： 主要用到的方法是http包的FileServer,参数很简单,就是要路由的文件夹的路径。 1 2 3 4 5 6 7 8 9 10 11 12 13 package main import ( "fmt" "net/http" ) func main() { http.Handle(...
golang大文件上传
12-20
go大文件上传的example
Golang实现异步上传文件支持进度条查询的方法
01-03
业务背景 业务需求要求开发一个异步上传文件的接口，并支持上传进度的查询。 需求分析 ZIP压缩包中，包含一个csv文件和一个图片文件夹，要求：解析csv数据存入mongo，将图片文件夹中的图片信息对应上csv中的人员信息。 ZIP压缩包解压 使用golang自带的 “archive/zip” 包解压。 func decompressZip(filePath, dest string) (string, string, error) { var csvName string imageFolder := path.Base(filePath) ext := path.Ext(
golang 文件服务器-命令行工具
01-08
背景 经常进入容器操作，有时候会有在容器之间或在容器和本机之间传输文件的需求。于是用golang写了一个命令行工具，可以把当前目录启动为文件服务器的根目录。需要下载时就可以使用curl命令下载相应文件。 fileserver 项目地址: https://github.com/FengGeSe/fileserver 使用 帮助 $ fileserver -h 文件服务器 Usage: fileserver [flags] Flags: -d, --dir string 指定文件服务器的根目录 -h, --help help for fileserver
golang-file-sever：用golang编写的性能文件服务器
02-12
golang文件服务器 用golang编写的性能文件服务器
golang搭建静态web服务器的实现方法
09-20
主要介绍了golang搭建静态web服务器的实现方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
golang实现http server提供文件下载功能
09-17
主要介绍了golang实现http server提供文件下载功能，本文给大家简单介绍了Golang的相关知识，非常不错，具有一定的参考借鉴价值,需要的朋友可以参考下
vuejs学习之路 vue上传文件控件往golang服务器上传文件
01-20
golang代码： func UploadHandler(w http.ResponseWriter, r *http.Request) GetFile(w, r, ../test.dat) } func GetFile(w http.ResponseWriter, r *http.Request, filepath string) { fmt....
终于找到透明背景BMP格式Logo制作方法，超简单一步到位
一只海星的主页
08-05 4694
背景 海思平台显示osd使用BMP格式图片，为了图片漂亮我们不需把LOGO图片周围抠成镂空，不然显示白色背景，很难看！ 所以问题来了，美工做不出来背景透明在海思OSD模块可以显示的BMP图片!!! 原理分析 BMP是位图，由于我们需要背景透明，所以除了RGB三色通道我们还需要透明通道Alpha Bitmap.Config ARGB_8888：每个像素占四位，即A=8，R=8，G=8，B=8，那么一个像素点占8+8+8+8=32位 所以，我们的图片格式位深度必须是32位，海思OSD模块格式请选择ARGB_
用golang写一个具有下载功能的http server
02-18
很高兴为您解答。Golang可以很容易地编写一个具有下载功能的HTTP服务器，只需要几行代码即可实现。可以使用net/http包来实现，下面是一个示例：package mainimport ( "fmt" "io" "net/http" )func downloadFile(w http.ResponseWriter, r *http.Request) { // Set the file name and content type fileName := "name-of-file.pdf" w.Header().Set("Content-Type", "application/octet-stream") w.Header().Set("Content-Disposition", "attachment; filename="+fileName) // Create a buffer to store our output buffer := make([]byte, 4096) // Open the file file, err := os.Open(fileName) if err != nil { fmt.Println(err) } defer file.Close() // Read the file into the buffer for { n, err := file.Read(buffer) if err != nil && err != io.EOF { fmt.Println(err) return } if n == 0 { break } // Write the buffer to the client if _, err := w.Write(buffer[:n]); err != nil { fmt.Println(err) return } } }func main() { http.HandleFunc("/", downloadFile) http.ListenAndServe(":8080", nil) }

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值