hvchien216 · November 17, 2022 15:18 · Dec 29, 2017 · Dec 29, 2017
diff --git a/server.go b/server.go
@@ -12,6 +12,7 @@ func main() {
 	router := gin.Default()
 	// Set a lower memory limit for multipart forms (default is 32 MiB)
 	// router.MaxMultipartMemory = 8 << 20  // 8 MiB
+	router.Static("/file", "saved")
 	router.POST("/many", func(c *gin.Context) {
 		// Multipart form
 		form, _ := c.MultipartForm()
@@ -24,6 +25,8 @@ func main() {
 				log.Fatal(err)
 			}
 		}
+
+		fmt.Println(c.PostForm("key"))
 		c.String(http.StatusOK, fmt.Sprintf("%d files uploaded!", len(files)))
 	})
 	router.POST("/one", func(c *gin.Context) {

diff --git a/upload.go b/upload.go
@@ -7,17 +7,48 @@ import (
 	"mime/multipart"
 	"net/http"
 	"os"
+	"strings"
 )
 
 func main() {
 	// err := Upload("http://localhost:8002/one", "main.go")
 	// if err != nil {
 	//  panic(err)
 	// }
-	err := UploadTwo()
+	// err := UploadTwo()
+	// if err != nil {
+	//  panic(err)
+	// }
+	downloadFromUrl("http://localhost:8062/file/file1.bin")
+}
+
+func downloadFromUrl(url string) {
+	tokens := strings.Split(url, "/")
+	fileName := tokens[len(tokens)-1]
+	fmt.Println("Downloading", url, "to", fileName)
+
+	// TODO: check file existence first with io.IsExist
+	output, err := os.Create(fileName)
+	if err != nil {
+		fmt.Println("Error while creating", fileName, "-", err)
+		return
+	}
+	defer output.Close()
+
+	response, err := http.Get(url)
+	if err != nil {
+		fmt.Println("Error while downloading", url, "-", err)
+		return
+	}
+	defer response.Body.Close()
+
+	n, err := io.Copy(output, response.Body)
 	if err != nil {
-		panic(err)
+		fmt.Println("Error while downloading", url, "-", err)
+		return
 	}
+
+	fmt.Println(n, "bytes downloaded.")
 }
 
 func Upload(url, file string) (err error) {
@@ -72,7 +103,7 @@ func Upload(url, file string) (err error) {
 }
 
 func UploadTwo() (err error) {
-	url := "https://music.schollz.com/many"
+	url := "http://localhost:8062/many"
 	// Prepare a form that you will submit to that URL.
 	var b bytes.Buffer
 	w := multipart.NewWriter(&b)
@@ -97,6 +128,17 @@ func UploadTwo() (err error) {
 		f.Close()
 	}
 
+	// Add the other fields
+	fw, err2 := w.CreateFormField("key")
+	if err2 != nil {
+		err = err2
+		return
+	}
+	if _, err2 = fw.Write([]byte("KEY")); err2 != nil {
+		err = err2
+		return
+	}
+
 	// Don't forget to close the multipart writer.
 	// If you don't close it, your request will be missing the terminating boundary.
 	w.Close()

diff --git a/files.sh b/files.sh
@@ -0,0 +1,4 @@
+#! /bin/bash
+for n in {1..100}; do
+    dd if=/dev/urandom of=file$( printf %d "$n" ).bin bs=1 count=$(( RANDOM + 1024 ))
+done
diff --git a/server.go b/server.go
@@ -0,0 +1,44 @@
+package main
+
+import (
+	"fmt"
+	"log"
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+)
+
+func main() {
+	router := gin.Default()
+	// Set a lower memory limit for multipart forms (default is 32 MiB)
+	// router.MaxMultipartMemory = 8 << 20  // 8 MiB
+	router.POST("/many", func(c *gin.Context) {
+		// Multipart form
+		form, _ := c.MultipartForm()
+		files := form.File["file"]
+
+		for _, file := range files {
+			log.Println(file.Filename)
+			err := c.SaveUploadedFile(file, "saved/"+file.Filename)
+			if err != nil {
+				log.Fatal(err)
+			}
+		}
+		c.String(http.StatusOK, fmt.Sprintf("%d files uploaded!", len(files)))
+	})
+	router.POST("/one", func(c *gin.Context) {
+		// single file
+		file, err := c.FormFile("file")
+		if err != nil {
+			log.Fatal(err)
+		}
+		log.Println(file.Filename)
+
+		err = c.SaveUploadedFile(file, "saved/"+file.Filename)
+		if err != nil {
+			log.Fatal(err)
+		}
+		c.String(http.StatusOK, fmt.Sprintf("'%s' uploaded!", file.Filename))
+	})
+	router.Run(":8062")
+}
diff --git a/upload.go b/upload.go
@@ -0,0 +1,124 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"mime/multipart"
+	"net/http"
+	"os"
+)
+
+func main() {
+	// err := Upload("http://localhost:8002/one", "main.go")
+	// if err != nil {
+	//  panic(err)
+	// }
+	err := UploadTwo()
+	if err != nil {
+		panic(err)
+	}
+}
+
+func Upload(url, file string) (err error) {
+	// Prepare a form that you will submit to that URL.
+	var b bytes.Buffer
+	w := multipart.NewWriter(&b)
+	// Add your image file
+	f, err := os.Open(file)
+	if err != nil {
+		return
+	}
+	defer f.Close()
+	fw, err := w.CreateFormFile("file", file)
+	if err != nil {
+		return
+	}
+	if _, err = io.Copy(fw, f); err != nil {
+		return
+	}
+
+	// Add the other fields
+	if fw, err = w.CreateFormField("key"); err != nil {
+		return
+	}
+	if _, err = fw.Write([]byte("KEY")); err != nil {
+		return
+	}
+	// 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
+	client := &http.Client{}
+	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 UploadTwo() (err error) {
+	url := "https://music.schollz.com/many"
+	// Prepare a form that you will submit to that URL.
+	var b bytes.Buffer
+	w := multipart.NewWriter(&b)
+
+	for i := 1; i <= 100; i++ {
+		fname := fmt.Sprintf("file%d.bin", i)
+		fw, err2 := w.CreateFormFile("file", fname)
+		if err2 != nil {
+			err = err2
+			return
+		}
+
+		f, err2 := os.Open(fname)
+		if err2 != nil {
+			err = err2
+			return
+		}
+		if _, err2 = io.Copy(fw, f); err2 != nil {
+			err = err2
+			return
+		}
+		f.Close()
+	}
+
+	// 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
+	client := &http.Client{}
+	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
+}
No results found