POST-данные, используя multipart/form-data Content-Type
Я пытаюсь загрузить изображения со своего компьютера на веб-сайт с помощью go. Обычно я использую скрипт bash, который отправляет файл и ключ на сервер:
curl -F "image"[email protected]"IMAGEFILE" -F "key"="KEY" URL
он работает нормально, но я пытаюсь преобразовать этот запрос в мою программу golang.
http://matt.aimonetti.net/posts/2013/07/01/golang-multipart-file-upload-example/
Я пробовал эту ссылку и многие другие, но для каждого кода, который я пробую, ответ от сервера - "изображение не отправлено", и я не знаю почему. Если кто-то знает, что происходит с примером выше.
Ответы
Ответ 1
Вот пример кода.
Короче говоря, вам нужно использовать пакет mime/multipart
для построения формы.
package main
import (
"bytes"
"fmt"
"io"
"mime/multipart"
"net/http"
"net/http/httptest"
"net/http/httputil"
"os"
"strings"
)
func main() {
var client *http.Client
var remoteURL string
{
//setup a mocked http client.
ts := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
b, err := httputil.DumpRequest(r, true)
if err != nil {
panic(err)
}
fmt.Printf("%s", b)
}))
defer ts.Close()
client = ts.Client()
remoteURL = ts.URL
}
//prepare the reader instances to encode
values := map[string]io.Reader{
"file": mustOpen("main.go"), // lets assume its this file
"other": strings.NewReader("hello world!"),
}
err := Upload(client, remoteURL, values)
if err != nil {
panic(err)
}
}
func Upload(client *http.Client, url string, values map[string]io.Reader) (err error) {
// Prepare a form that you will submit to that URL.
var b bytes.Buffer
w := multipart.NewWriter(&b)
for key, r := range values {
var fw io.Writer
if x, ok := r.(io.Closer); ok {
defer x.Close()
}
// Add an image file
if x, ok := r.(*os.File); ok {
if fw, err = w.CreateFormFile(key, x.Name()); err != nil {
return
}
} else {
// Add other fields
if fw, err = w.CreateFormField(key); err != nil {
return
}
}
if _, err = io.Copy(fw, r); err != nil {
return err
}
}
// Don't forget to close the multipart writer.
// If you don't close it, your request will be missing the terminating boundary.
w.Close()
// Now that you have a form, you can submit it to your handler.
req, err := http.NewRequest("POST", url, &b)
if err != nil {
return
}
// Don't forget to set the content type, this will contain the boundary.
req.Header.Set("Content-Type", w.FormDataContentType())
// Submit the request
res, err := client.Do(req)
if err != nil {
return
}
// Check the response
if res.StatusCode != http.StatusOK {
err = fmt.Errorf("bad status: %s", res.Status)
}
return
}
func mustOpen(f string) *os.File {
r, err := os.Open(f)
if err != nil {
panic(err)
}
return r
}
Ответ 2
Что касается сообщения attila-o, заголовок запроса не имеет границы, так как Writer уже закрыт.
// after the close, the bounday will be nil.
w.Close()
...
req.Header.Set("Content-Type", w.FormDataContentType())
Значит, он должен закрыться после набора, я думаю.
req.Header.Set("Content-Type", w.FormDataContentType())
w.Close()
Ответ 3
Возможно, это может быть полезно https://github.com/030/go-curl
./go-curl -url \
http://localhost:9999/service/rest/v1/components?repository=maven-releases \
-user admin -pass admin123 -F \
"[email protected]/test-files-multipart/file1.pom,\
maven2.asset1.extension=pom,\
[email protected]/test-files-multipart/file1.jar,\
maven2.asset2.extension=jar,\
[email protected]/test-files-multipart/file1-sources.jar,\
maven2.asset3.extension=sources.jar"
Ответ 4
После декодирования принятого ответа на этот вопрос для использования в моем модульном тестировании я, наконец, получил следующий переработанный код:
func createMultipartFormData(t *testing.T, fieldName, fileName string) (bytes.Buffer, *multipart.Writer) {
var b bytes.Buffer
var err error
w := multipart.NewWriter(&b)
var fw io.Writer
file := mustOpen(fileName)
if fw, err = w.CreateFormFile(fieldName, file.Name()); err != nil {
t.Errorf("Error creating writer: %v", err)
}
if _, err = io.Copy(fw, file); err != nil {
t.Errorf("Error with io.Copy: %v", err)
}
w.Close()
return b, w
}
func mustOpen(f string) *os.File {
r, err := os.Open(f)
if err != nil {
pwd, _ := os.Getwd()
fmt.Println("PWD: ", pwd)
panic(err)
}
return r
}
Теперь это должно быть довольно легко использовать:
b, w := createMultipartFormData(t, "image","../luke.png")
req, err := http.NewRequest("POST", url, &b)
if err != nil {
return
}
// Don't forget to set the content type, this will contain the boundary.
req.Header.Set("Content-Type", w.FormDataContentType())
Ответ 5
Здесь я использовал функцию, которая использует io.Pipe()
, чтобы избежать чтения всего файла в память или необходимости управлять какими-либо буферами. Он обрабатывает только один файл, но может быть легко расширен, чтобы обрабатывать больше, добавляя больше частей в программу. Счастливый путь работает хорошо. Пути ошибок не прошли много испытаний.
import (
"fmt"
"io"
"mime/multipart"
"net/http"
"os"
)
func UploadMultipartFile(client *http.Client, uri, key, path string) (*http.Response, error) {
body, writer := io.Pipe()
req, err := http.NewRequest(http.MethodPost, uri, body)
if err != nil {
return nil, err
}
mwriter := multipart.NewWriter(writer)
req.Header.Add("Content-Type", mwriter.FormDataContentType())
errchan := make(chan error)
go func() {
defer close(errchan)
defer writer.Close()
defer mwriter.Close()
w, err := mwriter.CreateFormFile(key, path)
if err != nil {
errchan <- err
return
}
in, err := os.Open(path)
if err != nil {
errchan <- err
return
}
defer in.Close()
if written, err := io.Copy(w, in); err != nil {
errchan <- fmt.Errorf("error copying %s (%d bytes written): %v", path, written, err)
return
}
if err := mwriter.Close(); err != nil {
errchan <- err
return
}
}()
resp, err := client.Do(req)
merr := <-errchan
if err != nil || merr != nil {
return resp, fmt.Errorf("http error: %v, multipart error: %v", err, merr)
}
return resp, nil
}
Ответ 6
Я нашел этот учебник, очень полезный, чтобы прояснить мои недоумения относительно загрузки файлов в Go.
В основном вы загружаете файл через ajax с помощью form-data
на клиенте и используете следующий небольшой фрагмент кода Go на сервере:
file, handler, err := r.FormFile("img") // img is the key of the form-data
if err != nil {
fmt.Println(err)
return
}
defer file.Close()
fmt.Println("File is good")
fmt.Println(handler.Filename)
fmt.Println()
fmt.Println(handler.Header)
f, err := os.OpenFile(handler.Filename, os.O_WRONLY|os.O_CREATE, 0666)
if err != nil {
fmt.Println(err)
return
}
defer f.Close()
io.Copy(f, file)
Здесь r
есть *http.Request
. P.S., это просто сохраняет файл в той же папке и не выполняет проверки безопасности.