wechat_hitxy/upload.go

package main

import (
	"bytes"
	"fmt"
	"io"
	"io/ioutil"
	"log"
	"mime/multipart"
	"net/http"
	"net/http/httputil"
	"os"
)

//post a file to a url
func postFileForm(filename string, targetURL string, formFieldName string) (strJSON string, err error) {
	bodyBuf := &bytes.Buffer{}
	bodyWriter := multipart.NewWriter(bodyBuf)

	// this step is very important
	fileWriter, err := bodyWriter.CreateFormFile(formFieldName, filename)
	if err != nil {
		fmt.Println(err)
		return "", err
	}

	// open file handle
	fh, err := os.Open(filename)
	if err != nil {
		fmt.Println(err)
		return "", err
	}

	//iocopy
	_, err = io.Copy(fileWriter, fh)
	if err != nil {
		return "", err
	}

	contentType := bodyWriter.FormDataContentType()
	bodyWriter.Close()

	resp, err := http.Post(targetURL, contentType, bodyBuf)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()
	r, err := ioutil.ReadAll(resp.Body)
	strJSON = string(r)
	if err != nil {
		return "", err
	}
	fmt.Println(resp.Status)
	fmt.Println(string(strJSON))
	return strJSON, nil
}

func postJSON(jsonB []byte, targetURL string) (resp string, err error) {
	headers := map[string]string{}
	headers["Content-Type"] = "application/json"
	return postRAW(jsonB, targetURL, headers)
}

func postRAW(data []byte, targetURL string, headers map[string]string) (resp string, err error) {
	return httpRaw("POST", targetURL, data, headers)
}

func putRAW(data []byte, targetURL string, headers map[string]string) (resp string, err error) {
	return httpRaw("PUT", targetURL, data, headers)
}

func patchRAW(data []byte, targetURL string, headers map[string]string) (resp string, err error) {
	return httpRaw("PATCH", targetURL, data, headers)
}
func getRAW(targetURL string, headers map[string]string) (resp string, err error) {
	return httpRaw("GET", targetURL, nil, headers)
}

func deleteRAW(targetURL string, headers map[string]string) (resp string, err error) {
	return httpRaw("DELETE", targetURL, nil, headers)
}

func httpRaw(httpMethod, targetURL string, data []byte, headers map[string]string) (resp string, err error) {
	requestBody := io.Reader(nil)
	if data != nil {
		requestBody = bytes.NewBuffer(data)
	}
	req, err := http.NewRequest(httpMethod, targetURL, requestBody)
	if err != nil {
		return
	}

	if headers != nil {
		for k, v := range headers {
			req.Header.Set(k, v)
		}
	}

	client := &http.Client{}
	r, err := client.Do(req)
	if err != nil {
		return
	}
	defer r.Body.Close()

	b, _ := ioutil.ReadAll(r.Body)
	resp = string(b)

	if r.StatusCode != 200 {
		debugDumpHTTPResponse(r)
		err = errorHTTPResponseNew(r, r.Header.Get("X-Status-Reason"))
	}
	//log.Println(resp)
	return
}

func debugDumpHTTPResponse(r *http.Response) {
	dump, e := httputil.DumpResponse(r, true)
	if e != nil {
		log.Fatal(e)
	}
	fmt.Printf("dump http Response \n : %s", dump)
}

func debugDumpHTTPRequest(r *http.Request) {
	dump, err := httputil.DumpRequest(r, true)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("dump http request : %q", dump)
}