Go圖片上傳

這是一個建立於 的文章，其中的資訊可能已經有所發展或是發生改變。

使用Go語言實現圖片上傳功能，伺服器及用戶端的原始碼。

伺服器端代碼：

// uploadfile_server.gopackage mainimport (//"crypto/md5""fmt"//"html/template""io""log""net/http""os"//"strconv"//"time")//檢查目錄是否存在func checkFileIsExist(filename string) bool {var exist = trueif _, err := os.Stat(filename); os.IsNotExist(err) {fmt.Print(filename + " not exist")exist = false}return exist}func main() {http.HandleFunc("/upload", upload)err := http.ListenAndServe(":9090", nil) //設定監聽的連接埠if err != nil {log.Fatal("ListenAndServe: ", err)}log.Println("success!")}// 處理/upload 邏輯func upload(w http.ResponseWriter, r *http.Request) {fmt.Println("method:", r.Method) //擷取請求的方法/*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 */fmt.Println(r)if r.Method == "POST" {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)//bool_fileexist := checkFileIsExist("./test")fmt.Println("check file 1-----------------")fmt.Println("-------------------------bool_fileexist:", bool_fileexist)if bool_fileexist { //如果檔案夾存在//f, err1 = os.OpenFile(filename, os.O_APPEND, 0666) //開啟檔案f, err := os.OpenFile("./test/"+handler.Filename, os.O_WRONLY|os.O_CREATE, 0666)if err != nil {fmt.Println(err)return}defer f.Close()io.Copy(f, file)fmt.Println("檔案夾存在")} else { //不存在檔案夾時 先建立檔案夾再上傳err1 := os.Mkdir("./test", os.ModePerm) //建立檔案夾if err1 != nil {fmt.Println(err)return}fmt.Println("檔案夾不存在")fmt.Println("檔案夾建立成功！")f, err := os.OpenFile("./test/"+handler.Filename, os.O_WRONLY|os.O_CREATE, 0666)if err != nil {fmt.Println(err)return}defer f.Close()io.Copy(f, file)}}}

用戶端代碼：

// uploadfile_client.gopackage mainimport ("bytes""fmt""io""io/ioutil""mime/multipart""nettp""os")func postFile(filename string, targetUrl string) error {bodyBuf := &bytes.Buffer{}bodyWriter := multipart.NewWriter(bodyBuf)//關鍵的一步操作fileWriter, err := bodyWriter.CreateFormFile("uploadfile", filename)if err != nil {fmt.Println("error writing to buffer")return err}//開啟檔案控制代碼操作fh, err := os.Open(filename)if err != nil {fmt.Println("error opening file")return err}defer fh.Close()//iocopy_, err = io.Copy(fileWriter, fh)if err != nil {return err}contentType := bodyWriter.FormDataContentType()bodyWriter.Close()resp, err := http.Post(targetUrl, contentType, bodyBuf)fmt.Println(bodyBuf)if err != nil {return err}defer resp.Body.Close()resp_body, err := ioutil.ReadAll(resp.Body)if err != nil {return err}fmt.Println(resp.Status)fmt.Println(string(resp_body))return nil}// sample usagefunc main() {target_url := "http://xxx.xxx.xxx.xxx:9090/upload"//此處為自己的ip地址filename := "image2.jpg"postFile(filename, target_url)}

著作權聲明：本文為博主原創文章，未經博主允許不得轉載。