Golang发送GET、POST、PUT、DELETE请求

golang发送GET、POST、PUT、DELETE请求

package requests

import (
	"compress/gzip"
	"fmt"
	"io"
	"io/ioutil"
	"log"
	"net/http"
)

type Headers map[string]string

type Cookies []*http.Cookie

type Data map[string]interface{}

// params
type Params struct {
	Data Data
	Headers Headers
	Cookies Cookies
}

// debugger
var Debugger bool = false

// response
type responses struct {
	Response *http.Response
	Body string
}

func request(method string, url string, params Params) (r responses, e error) {
	// 打印debugger发起请求
	printSendDebugger(method, url, params)
	r, e = do(method, url, params)
	if e != nil {
		return
	}
	// 打印debugger返回请求
	printResultDebugger(r)
	return
}

func do(method string, url string, params Params) (r responses, e error) {
	request, e := http.NewRequest(method, url, nil)
	// header
	addHeaders(request, params.Headers)
	// cookie
	addCookies(request, params.Cookies)
	// 参数
	addData(request, params.Data)
	r.Response, e = http.DefaultClient.Do(request)
	defer r.Response.Body.Close()
	if e != nil {
		return
	}
	unCoding(&r)
	return
}

func addHeaders(request *http.Request, headers Headers) {
	for k, v := range headers{
		request.Header.Add(k, v)
	}
}

// 添加cookie
func addCookies(request *http.Request, cookies Cookies) {
	for _, v := range cookies{
		request.AddCookie(v)
	}
}

// 添加参数
func addData(request *http.Request, data Data) {
	query := request.URL.Query()
	for k, v := range data{
		query.Add(k, fmt.Sprint(v))
	}
	request.URL.RawQuery = query.Encode()
}

func unCoding(r *responses) {
	if r.Response.StatusCode == 200 {
		switch r.Response.Header.Get("Content-Encoding") {
		case "gzip":
			reader, _ := gzip.NewReader(r.Response.Body)
			for {
				buf := make([]byte, 1024)
				n, err := reader.Read(buf)
				if err != nil && err != io.EOF {
					panic(err)
				}
				if n == 0 {
					break
				}
				r.Body += string(buf)
			}
		default:
			bodyByte, _ := ioutil.ReadAll(r.Response.Body)
			r.Body = string(bodyByte)
		}
	} else {
		bodyByte, _ := ioutil.ReadAll(r.Response.Body)
		r.Body = string(bodyByte)
	}
}

// debugger发起请求
func printSendDebugger(method string, url string, params Params) {
	if Debugger {
		log.Println("debug log start ----------")
		fmt.Println("Method", method)
		fmt.Println("Host", ":", url)
		for k, v := range params.Headers{
			fmt.Println(k, ":", v)
		}
		fmt.Println("----------------------------------------------------")
	}
}

func printResultDebugger(r responses) {
	if Debugger {
		fmt.Println("Status", ":", r.Response.Status)
		for key, val := range r.Response.Header {
			fmt.Println(key, ":", val[0])
		}
		log.Println("debug log end ----------")
	}
}

// request
func Request(method string, url string, params Params) (responses, error) {
	return request(method, url , params)
}

// get
func Get(url string, params Params) (responses, error) {
	return request("GET", url , params)
}

// post
func Post(url string, params Params) (responses, error) {
	return request("POST", url , params)
}

// put
func Put(url string, params Params) (responses, error) {
	return request("PUT", url , params)
}

// delete
func Delete(url string, params Params) (responses, error) {
	return request("DELETE", url , params)
}

测试请求

func main() {
	requests.Debugger = true
	var params requests.Params
	params.Data = requests.Data{
		"id": 100000,
		"name": "赵云",
		"sid": 10,
	}
	params.Cookies = requests.Cookies{
		{Name: "id", Value: "1"},
		{Name: "name", Value: "zhaoyun"},
		{Name: "sid", Value: "10"},
	}
	params.Headers = requests.Headers{
		"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 11_2_2) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/88.0.4324.96 Safari/537.36",
		"Content-Type": "text/plain; charset=UTF-8",
		"Transfer-Encoding": "chunked",
		"Accept-Encoding": "gzip, deflate",
		"Authorization": "",
	}
	r, e := requests.Request("GET", "https://www.demo.com/demo.php", params)
	if e != nil {
		fmt.Println(e)
		os.Exit(0)
	}
	fmt.Println(r.Body)
}