Last active
March 28, 2020 07:24
-
-
Save sanatgersappa/5150439 to your computer and use it in GitHub Desktop.
Alternative handler for multiple file uploads in Go.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package main | |
import ( | |
"html/template" | |
"io" | |
"net/http" | |
"os" | |
) | |
//Compile templates on start | |
var templates = template.Must(template.ParseFiles("tmpl/upload.html")) | |
//Display the named template | |
func display(w http.ResponseWriter, tmpl string, data interface{}) { | |
templates.ExecuteTemplate(w, tmpl+".html", data) | |
} | |
//This is where the action happens. | |
func uploadHandler(w http.ResponseWriter, r *http.Request) { | |
switch r.Method { | |
//GET displays the upload form. | |
case "GET": | |
display(w, "upload", nil) | |
//POST takes the uploaded file(s) and saves it to disk. | |
case "POST": | |
//get the multipart reader for the request. | |
reader, err := r.MultipartReader() | |
if err != nil { | |
http.Error(w, err.Error(), http.StatusInternalServerError) | |
return | |
} | |
//copy each part to destination. | |
for { | |
part, err := reader.NextPart() | |
if err == io.EOF { | |
break | |
} | |
//if part.FileName() is empty, skip this iteration. | |
if part.FileName() == "" { | |
continue | |
} | |
dst, err := os.Create("/home/sanat/" + part.FileName()) | |
defer dst.Close() | |
if err != nil { | |
http.Error(w, err.Error(), http.StatusInternalServerError) | |
return | |
} | |
if _, err := io.Copy(dst, part); err != nil { | |
http.Error(w, err.Error(), http.StatusInternalServerError) | |
return | |
} | |
} | |
//display success message. | |
display(w, "upload", "Upload successful.") | |
default: | |
w.WriteHeader(http.StatusMethodNotAllowed) | |
} | |
} | |
func main() { | |
http.HandleFunc("/upload", uploadHandler) | |
//static file handler. | |
http.Handle("/assets/", http.StripPrefix("/assets/", http.FileServer(http.Dir("assets")))) | |
//Listen on port 8080 | |
http.ListenAndServe(":8080", nil) | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment