package util
|
|
import (
|
"bytes"
|
"encoding/json"
|
"fmt"
|
"io"
|
"io/ioutil"
|
"net/http"
|
"os"
|
"strings"
|
"time"
|
|
"github.com/ajg/form"
|
"github.com/dustin/go-humanize"
|
)
|
|
const (
|
DATA_URL_PREFIX = "/data/api-v"
|
CONTENT_TYPE_FORM = "application/x-www-form-urlencoded"
|
CONTENT_TYPE_MULFORM = "multipart/form-data"
|
CONTENT_TYPE_JSON = "application/json"
|
)
|
|
func DoGetRequest(url string, params map[string]string, headers map[string]string) ([]byte, error) {
|
var body []byte
|
request, err := http.NewRequest("GET", url, nil)
|
if err != nil {
|
return body, err
|
}
|
// add params
|
query := request.URL.Query()
|
if params != nil {
|
for key, val := range params {
|
query.Add(key, val)
|
}
|
request.URL.RawQuery = query.Encode()
|
}
|
// add headers
|
if headers != nil {
|
for key, val := range headers {
|
request.Header.Add(key, val)
|
}
|
}
|
client := &http.Client{}
|
resp, err := client.Do(request)
|
if err != nil {
|
return body, err
|
}
|
defer resp.Body.Close()
|
body, err = ioutil.ReadAll(resp.Body)
|
if err != nil {
|
return body, err
|
}
|
return body, nil
|
}
|
|
func DoPostRequest(url string, contentType string, body map[string]interface{}, params map[string]string, headers map[string]string, timeOut time.Duration) ([]byte, error) {
|
var resultBytes []byte
|
var request *http.Request
|
var err error
|
if contentType == CONTENT_TYPE_FORM || contentType == CONTENT_TYPE_MULFORM {
|
formData, err := form.EncodeToString(body)
|
if err != nil {
|
return resultBytes, err
|
}
|
request, err = http.NewRequest("POST", url, strings.NewReader(formData))
|
if err != nil {
|
return resultBytes, err
|
}
|
} else {
|
var bodyJson []byte
|
if body != nil {
|
var err error
|
bodyJson, err = json.Marshal(body)
|
if err != nil {
|
return resultBytes, err
|
}
|
}
|
request, err = http.NewRequest("POST", url, bytes.NewBuffer(bodyJson))
|
if err != nil {
|
return resultBytes, err
|
}
|
}
|
|
request.Header.Set("Content-type", contentType)
|
//add params
|
q := request.URL.Query()
|
if params != nil {
|
for key, val := range params {
|
q.Add(key, val)
|
}
|
request.URL.RawQuery = q.Encode()
|
}
|
// add headers
|
if headers != nil {
|
for key, val := range headers {
|
request.Header.Add(key, val)
|
}
|
}
|
client := &http.Client{
|
Timeout: timeOut,
|
}
|
resp, err := client.Do(request)
|
if err != nil {
|
return resultBytes, err
|
}
|
defer resp.Body.Close()
|
resultBytes, err = ioutil.ReadAll(resp.Body)
|
if err != nil {
|
return resultBytes, err
|
}
|
return resultBytes, nil
|
}
|
|
type WriteCounter struct {
|
Total uint64
|
}
|
|
func (wc WriteCounter) PrintProgress() {
|
fmt.Printf("\r%s", strings.Repeat(" ", 35))
|
fmt.Printf("/rDownloading... %s complete", humanize.Bytes(wc.Total))
|
}
|
|
func (wc *WriteCounter) Write(p []byte) (int, error) {
|
n := len(p)
|
wc.Total = uint64(n)
|
wc.PrintProgress()
|
return n, nil
|
}
|
|
//下载文件
|
func DownloadFile(fPath string, url string) error {
|
f, err := os.Create(fPath)
|
if err != nil {
|
return err
|
}
|
resp, err := http.Get(url)
|
if err != nil {
|
f.Close()
|
return err
|
}
|
defer resp.Body.Close()
|
counter := &WriteCounter{}
|
if _, err = io.Copy(f, io.TeeReader(resp.Body, counter)); err != nil {
|
f.Close()
|
return err
|
}
|
fmt.Printf("\n")
|
f.Close()
|
if err = os.Rename(fPath, fPath); err != nil {
|
return err
|
}
|
return nil
|
}
|
