溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

怎么使用GO實現(xiàn)文件上傳

發(fā)布時間:2022-07-27 10:07:40 來源:億速云 閱讀:208 作者:iii 欄目:開發(fā)技術(shù)

本篇內(nèi)容介紹了“怎么使用GO實現(xiàn)文件上傳”的有關(guān)知識,在實際案例的操作過程中,不少人都會遇到這樣的困境,接下來就讓小編帶領(lǐng)大家學習一下如何處理這些情況吧!希望大家仔細閱讀,能夠?qū)W有所成!

1、表單操作

要使表單能夠上傳文件,首先第一步就要添加form的enctype屬性進去,enctype屬性有如下三種情況:

application/x-www-form-urlencoded   表示在發(fā)送前編碼所有字符(默認)
multipart/form-data      不對字符編碼。在使用包含文件上傳控件的表單時,必須使用該值。
text/plain      空格轉(zhuǎn)換為 "+" 加號,但不對特殊字符編碼。

所以可以創(chuàng)建如下上傳表單:

<html>
<head>
    <title>上傳文件</title>
</head>
<body>
<form enctype="multipart/form-data" action="/upload" method="post">
  <input type="file" name="uploadfile" />
  <input type="hidden" name="token" value="{{.}}"/>
  <input type="submit" value="upload" />
</form>
</body>
</html>

2、服務(wù)端操作

在服務(wù)端只需要添加一個handlerFunc并完善相關(guān)功能即可:

// 處理/upload 邏輯
func upload(w http.ResponseWriter, r *http.Request) {
    //獲取請求的方法
    fmt.Println("method:", r.Method)
    //GET的處理操作
    if r.Method == "GET" {
        crutime := time.Now().Unix()
        h := md5.New()
        io.WriteString(h, strconv.FormatInt(crutime, 10))
        token := fmt.Sprintf("%x", h.Sum(nil))

        t, _ := template.ParseFiles("upload.gtpl")
        t.Execute(w, token)
    } else {
        //設(shè)置內(nèi)存大小
        r.ParseMultipartForm(32 << 20)
        //獲取上傳文件
        file, handler, err := r.FormFile("uploadfile")
        if err != nil {
            fmt.Println(err)
            return
        }
        defer file.Close()
        fmt.Fprintf(w, "%v", handler.Header)
        //創(chuàng)建上傳目錄
        os.Mkdir("./test", os.ModePerm)
        //創(chuàng)建上傳文件
        f, err := os.Create("./test/" + handler.Filename)
        //f, err := os.OpenFile("./test/"+handler.Filename, os.O_WRONLY|os.O_CREATE, 0666) // 此處假設(shè)當前目錄下已存在test目錄
        if err != nil {
            fmt.Println(err)
            return
        }
        defer f.Close()
        io.Copy(f, file)
    }
}

main()函數(shù)中記得添加http.HandleFunc("/upload", upload)即可。

通過http://127.0.0.1:9999/upload來測試文件上傳。 截圖

怎么使用GO實現(xiàn)文件上傳

選擇文件之后就會在當前目錄下的test文件夾中成功上傳文件。

3、流程解析

通過上面的代碼可以看到,處理文件上傳我們需要調(diào)用r.ParseMultipartForm,里面的參數(shù)表示maxMemory,調(diào)用ParseMultipartForm之后,上傳的文件存儲在maxMemory大小的內(nèi)存里面,如果文件大小超過了maxMemory,那么剩下的部分將存儲在系統(tǒng)的臨時文件中。我們可以通過r.FormFile獲取上面的文件句柄,然后實例中使用了io.Copy來存儲文件。我們可以嘗試使用看一下使用的相關(guān)原函數(shù):

ParseMultipartForm函數(shù)如下:

func (r *Request) ParseMultipartForm(maxMemory int64) error {
    if r.MultipartForm == multipartByReader {
        return errors.New("http: multipart handled by MultipartReader")
    }
    if r.Form == nil {
        err := r.ParseForm()
        if err != nil {
            return err
        }
    }
    if r.MultipartForm != nil {
        return nil
    }

    mr, err := r.multipartReader(false)
    if err != nil {
        return err
    }

    f, err := mr.ReadForm(maxMemory)
    if err != nil {
        return err
    }

    if r.PostForm == nil {
        r.PostForm = make(url.Values)
    }
    for k, v := range f.Value {
        r.Form[k] = append(r.Form[k], v...)
        // r.PostForm should also be populated. See Issue 9305.
        r.PostForm[k] = append(r.PostForm[k], v...)
    }

    r.MultipartForm = f

    return nil
}

FormFile函數(shù)如下:

func (r *Request) FormFile(key string) (multipart.File, *multipart.FileHeader, error) {
    if r.MultipartForm == multipartByReader {
        return nil, nil, errors.New("http: multipart handled by MultipartReader")
    }
    if r.MultipartForm == nil {
        err := r.ParseMultipartForm(defaultMaxMemory)
        if err != nil {
            return nil, nil, err
        }
    }
    if r.MultipartForm != nil && r.MultipartForm.File != nil {
        if fhs := r.MultipartForm.File[key]; len(fhs) > 0 {
            f, err := fhs[0].Open()
            return f, fhs[0], err
        }
    }
    return nil, nil, ErrMissingFile
}

文件handlermultipart.FileHeader里面的結(jié)構(gòu)體如下

// A FileHeader describes a file part of a multipart request.
type FileHeader struct {
    Filename string
    Header   textproto.MIMEHeader
    Size     int64

    content []byte
    tmpfile string
}

4、成功結(jié)果

瀏覽器端顯示如下消息。

怎么使用GO實現(xiàn)文件上傳

“怎么使用GO實現(xiàn)文件上傳”的內(nèi)容就介紹到這里了,感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識可以關(guān)注億速云網(wǎng)站,小編將為大家輸出更多高質(zhì)量的實用文章!

向AI問一下細節(jié)

免責聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。

go
AI