使用golang怎么实现一个文件上传服务

使用golang怎么实现一个文件上传服务,很多新手对此不是很清楚,为了帮助大家解决这个难题,下面小编将为大家详细讲解,有这方面需求的人可以来学习下,希望你能有所收获。

成都网站建设、做网站的关注点不是能为您做些什么网站,而是怎么做网站,有没有做好网站,给成都创新互联一个展示的机会来证明自己,这并不会花费您太多时间,或许会给您带来新的灵感和惊喜。面向用户友好,注重用户体验,一切以用户为中心。

package main

import (
	"encoding/json"
	"io"
	"math/rand"
	"net/http"
	"os"
	"strconv"
	"strings"
	"time"
)

const POS = "."

const DIR = "/"

const PATH = "image"

const HTTP = "https://www.image.com"

func getFileName(ext string) string {
	random := rand.New(rand.NewSource(time.Now().UnixNano()))
	datetime := time.Now().Format("200612")

	dstPath := PATH + DIR + datetime
	dstFile := dstPath + DIR + datetime + strconv.Itoa(random.Int()) + POS + strings.Replace(ext, DIR, POS, 1)

	_, err := os.Stat(dstPath)
	res := os.IsNotExist(err)
	if res == true {
		os.MkdirAll(dstPath, os.ModePerm)
	}

	return dstFile
}

func uploadFile(w http.ResponseWriter, r *http.Request) {
	r.ParseMultipartForm(10 << 20)
	file, handler, err := r.FormFile("myFile")
	if err != nil {
		w.WriteHeader(500)
		return
	}
	defer file.Close()

	fileName := getFileName(handler.Header.Get("Content-Type"))

	fp, err := os.OpenFile(fileName, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666)
	if err != nil {
		w.WriteHeader(500)
		return
	}
	defer fp.Close()

	size, err := io.Copy(fp, file)
	if err != nil {
		w.WriteHeader(500)
		return
	}

	go callUploadResult(w, fileName, size)
}

func callUploadResult(w http.ResponseWriter, fileName string, size int64) {
	var list = make(map[string]string)
	list["image"] = HTTP + DIR + fileName
	list["size"] = strconv.FormatInt(size/1024, 10) + "KB"
	list["action"] = "call-upload-result"
	jsonStr, _ := json.Marshal(list)
	w.Header().Set("Content-Type", "application/json")
	w.Write([]byte(jsonStr))
}

func main() {
	http.HandleFunc("/upload", uploadFile)
	http.ListenAndServe("127.0.0.1:8090", nil)
}

使用golang怎么实现一个文件上传服务

看完上述内容是否对您有帮助呢?如果还想对相关知识有进一步的了解或阅读更多相关文章,请关注创新互联行业资讯频道,感谢您对创新互联的支持。


新闻名称:使用golang怎么实现一个文件上传服务
文章转载:http://myzitong.com/article/gipipg.html