package util import ( "bytes" "context" "crypto/tls" "encoding/json" "encoding/pem" "encoding/xml" "fmt" "io" "io/ioutil" "log" "mime/multipart" "net/http" "os" "golang.org/x/crypto/pkcs12" ) // HTTPGet get 请求 func HTTPGet(uri string) ([]byte, error) { return HTTPGetContext(context.Background(), uri) } // HTTPGetContext get 请求 func HTTPGetContext(ctx context.Context, uri string) ([]byte, error) { request, err := http.NewRequestWithContext(ctx, http.MethodGet, uri, nil) if err != nil { return nil, err } response, err := http.DefaultClient.Do(request) if err != nil { return nil, err } defer response.Body.Close() if response.StatusCode != http.StatusOK { return nil, fmt.Errorf("http get error : uri=%v , statusCode=%v", uri, response.StatusCode) } return ioutil.ReadAll(response.Body) } // HTTPPost post 请求 func HTTPPost(uri string, data string) ([]byte, error) { return HTTPPostContext(context.Background(), uri, data) } // HTTPPostContext post 请求 func HTTPPostContext(ctx context.Context, uri string, data string) ([]byte, error) { body := bytes.NewBuffer([]byte(data)) request, err := http.NewRequestWithContext(ctx, http.MethodPost, uri, body) if err != nil { return nil, err } response, err := http.DefaultClient.Do(request) if err != nil { return nil, err } defer response.Body.Close() if response.StatusCode != http.StatusOK { return nil, fmt.Errorf("http get error : uri=%v , statusCode=%v", uri, response.StatusCode) } return ioutil.ReadAll(response.Body) } // PostJSON post json 数据请求 func PostJSON(uri string, obj interface{}) ([]byte, error) { jsonBuf := new(bytes.Buffer) enc := json.NewEncoder(jsonBuf) enc.SetEscapeHTML(false) err := enc.Encode(obj) if err != nil { return nil, err } response, err := http.Post(uri, "application/json;charset=utf-8", jsonBuf) if err != nil { return nil, err } defer response.Body.Close() if response.StatusCode != http.StatusOK { return nil, fmt.Errorf("http get error : uri=%v , statusCode=%v", uri, response.StatusCode) } return ioutil.ReadAll(response.Body) } // PostJSONWithRespContentType post json数据请求,且返回数据类型 func PostJSONWithRespContentType(uri string, obj interface{}) ([]byte, string, error) { jsonBuf := new(bytes.Buffer) enc := json.NewEncoder(jsonBuf) enc.SetEscapeHTML(false) err := enc.Encode(obj) if err != nil { return nil, "", err } response, err := http.Post(uri, "application/json;charset=utf-8", jsonBuf) if err != nil { return nil, "", err } defer response.Body.Close() if response.StatusCode != http.StatusOK { return nil, "", fmt.Errorf("http get error : uri=%v , statusCode=%v", uri, response.StatusCode) } responseData, err := ioutil.ReadAll(response.Body) contentType := response.Header.Get("Content-Type") return responseData, contentType, err } // PostFile 上传文件 func PostFile(fieldname, filename, uri string) ([]byte, error) { fields := []MultipartFormField{ { IsFile: true, Fieldname: fieldname, Filename: filename, }, } return PostMultipartForm(fields, uri) } // MultipartFormField 保存文件或其他字段信息 type MultipartFormField struct { IsFile bool Fieldname string Value []byte Filename string } // PostMultipartForm 上传文件或其他多个字段 func PostMultipartForm(fields []MultipartFormField, uri string) (respBody []byte, err error) { bodyBuf := &bytes.Buffer{} bodyWriter := multipart.NewWriter(bodyBuf) for _, field := range fields { if field.IsFile { fileWriter, e := bodyWriter.CreateFormFile(field.Fieldname, field.Filename) if e != nil { err = fmt.Errorf("error writing to buffer , err=%v", e) return } fh, e := os.Open(field.Filename) if e != nil { err = fmt.Errorf("error opening file , err=%v", e) return } defer fh.Close() if _, err = io.Copy(fileWriter, fh); err != nil { return } } else { partWriter, e := bodyWriter.CreateFormField(field.Fieldname) if e != nil { err = e return } valueReader := bytes.NewReader(field.Value) if _, err = io.Copy(partWriter, valueReader); err != nil { return } } } contentType := bodyWriter.FormDataContentType() bodyWriter.Close() resp, e := http.Post(uri, contentType, bodyBuf) if e != nil { err = e return } defer resp.Body.Close() if resp.StatusCode != http.StatusOK { return nil, err } respBody, err = ioutil.ReadAll(resp.Body) return } // PostXML perform a HTTP/POST request with XML body func PostXML(uri string, obj interface{}) ([]byte, error) { xmlData, err := xml.Marshal(obj) if err != nil { return nil, err } body := bytes.NewBuffer(xmlData) response, err := http.Post(uri, "application/xml;charset=utf-8", body) if err != nil { return nil, err } defer response.Body.Close() if response.StatusCode != http.StatusOK { return nil, fmt.Errorf("http code error : uri=%v , statusCode=%v", uri, response.StatusCode) } return ioutil.ReadAll(response.Body) } // httpWithTLS CA证书 func httpWithTLS(rootCa, key string) (*http.Client, error) { var client *http.Client certData, err := ioutil.ReadFile(rootCa) if err != nil { return nil, fmt.Errorf("unable to find cert path=%s, error=%v", rootCa, err) } cert := pkcs12ToPem(certData, key) config := &tls.Config{ Certificates: []tls.Certificate{cert}, } tr := &http.Transport{ TLSClientConfig: config, DisableCompression: true, } client = &http.Client{Transport: tr} return client, nil } // pkcs12ToPem 将Pkcs12转成Pem func pkcs12ToPem(p12 []byte, password string) tls.Certificate { blocks, err := pkcs12.ToPEM(p12, password) defer func() { if x := recover(); x != nil { log.Print(x) } }() if err != nil { panic(err) } var pemData []byte for _, b := range blocks { pemData = append(pemData, pem.EncodeToMemory(b)...) } cert, err := tls.X509KeyPair(pemData, pemData) if err != nil { panic(err) } return cert } // PostXMLWithTLS perform a HTTP/POST request with XML body and TLS func PostXMLWithTLS(uri string, obj interface{}, ca, key string) ([]byte, error) { xmlData, err := xml.Marshal(obj) if err != nil { return nil, err } body := bytes.NewBuffer(xmlData) client, err := httpWithTLS(ca, key) if err != nil { return nil, err } response, err := client.Post(uri, "application/xml;charset=utf-8", body) if err != nil { return nil, err } defer response.Body.Close() if response.StatusCode != http.StatusOK { return nil, fmt.Errorf("http code error : uri=%v , statusCode=%v", uri, response.StatusCode) } return ioutil.ReadAll(response.Body) }