d1nfinite / zhttp

zhttp是一个对net/http标准库的封装，参考了python中著名的requests库

zhttp

zhttp 是一个对 net/http 标准库的封装，参考了 python 中著名的 requests 库，支持标准库中的连接池，支持dns缓存，支持流式文件上传，支持多种body格式，很多代码及思路来自grequests

Installation

go get github.com/greyh4t/zhttp

Example

如下为简单示例，更多使用方法请参考godoc

package main

import (
	"log"
	"os"
	"time"

	"github.com/greyh4t/zhttp"
)

func main() {
	z := zhttp.New(&zhttp.HttpOptions{
		UserAgent: "global-useragent",
		Headers: map[string]string{
			"globalheader1": "value1",
			"globalheader2": "value2",
		},
		DNSCacheExpire:      time.Minute * 10,
		DNSServer:           "8.8.8.8:25",
		InsecureSkipVerify:  true,
		DialTimeout:         time.Second * 5,
		TLSHandshakeTimeout: time.Second * 5,
		KeepAlive:           time.Second * 10,
		MaxIdleConns:        10,
	})

	// 请求1
	resp, err := z.Get("http://www.example.com/", nil)
	if err != nil {
		log.Fatal(err)
	}
	log.Println(resp.StatusCode, resp.Status, resp.ContentLength)
	log.Println(resp.RawHeaders())
	log.Println(resp.CookiesMap())
	log.Println(zhttp.RawHTTPRequest(resp.RawResponse.Request))
	log.Println(resp.String())
	resp.Close()

	// 请求2 post表单
	resp, err = z.Post("http://www.example.com/", &zhttp.ReqOptions{
		DisableRedirect: true,
		Timeout:         time.Second * 10,
		Proxies: zhttp.MustProxy(map[string]string{
			"http":  "http://127.0.0.1:8080",
			"https": "http://127.0.0.1:8080",
		}),
		Headers: map[string]string{
			"header1": "value1",
			"header2": "value2",
		},
		Cookie: zhttp.PairsCookie(map[string]string{
			"k1": "v1",
			"k2": "v2",
		}),
		Body: zhttp.Form(map[string]string{
			"key1": "value1",
			"key2": "value2",
		}),
		Query: zhttp.PairsQuery(map[string]string{
			"query1": "value1",
			"query2": "value2",
		}),
	})
	if err != nil {
		log.Fatal(err)
	}
	body := resp.Byte()
	if resp.Error != nil{
		log.Fatal(resp.Error)
	}
	resp.Close()
	log.Println(body)

	// 请求3 post表单
	resp, err = z.Post("http://www.example.com/", &zhttp.ReqOptions{
		Query:     zhttp.RawQuery("query1=value1&query2=value2"),
		Body:      zhttp.RawForm(`fk1=fv1&fk2=fv2`),
		Cookie:    zhttp.RawCookie("k1=v1; k2=v2"),
		UserAgent: "zhttp-ua-test",
	})
	if err != nil {
		log.Fatal(err)
	}
	resp.Close()

	// 请求4 post json
	resp, err = z.Post("http://www.example.com/", &zhttp.ReqOptions{
		Body:      zhttp.RawJSON(`{"jk1":"jv","jk2":2}`),
		Cookie:    zhttp.RawCookie("k1=v1; k2=v2"),
		UserAgent: "zhttp-ua-test",
		IsAjax:    true,
	})
	if err != nil {
		log.Fatal(err)
	}
	resp.Close()

	// 请求5 文件上传
	f, err := os.Open("test.txt")
	if err != nil {
		log.Fatal(err)
	}

	resp, err = z.Post("http://www.example.com/", &zhttp.ReqOptions{
		Body:        zhttp.RawReader(f),
		ContentType: "text/plain",
		Headers: map[string]string{
			"h1": "v1",
			"h2": "v2",
		},
		Auth: zhttp.Auth{
			Username: "username",
			Password: "password",
		},
	})
	if err != nil {
		log.Fatal(err)
	}
	resp.Close()

	// 请求6 文件上传
	file1, err := zhttp.FileFromDisk("file1.txt")
	if err != nil {
		log.Fatal(err)
	}

	file2, err := zhttp.FileFromDisk("file2.txt")
	if err != nil {
		log.Fatal(err)
	}

	resp, err = z.Post("http://www.example.com/", &zhttp.ReqOptions{
		Body: zhttp.Multipart([]*zhttp.File{file1, file2}, map[string]string{
			"field1": "value1",
			"field2": "value2",
		}),
		Host: "file.example.com",
	})
	if err != nil {
		log.Fatal(err)
	}
	resp.Close()

	// 请求7 session的使用
	s := z.NewSession()
	resp, err = s.Post("http://www.example.com/login", &zhttp.ReqOptions{
		Body: zhttp.Form(map[string]string{
			"username": "username",
			"password": "password",
		}),
		Timeout: time.Second * 10,
	})
	if err != nil {
		log.Fatal(err)
	}
	resp.Close()

	resp, err = s.Get("http://www.example.com/userinfo", nil)
	if err != nil {
		log.Fatal(err)
	}
	resp.Close()
}