Go语言用于爬虫的GET和POST请求

最近在做GO服务爬虫发现GO的GET请求和POST请求写起来还挺麻烦,这里放出完整代码,读取数据的时候不使用ioutil.ReadAll(),不符合工程标准。

用法

  • 异步写法

代码采用了通道的写法可以用来做异步爬虫,在处理网络请求的时候响应交给通道,再交给CPU计算(JS加密),等计算完,再从通道取回数据。(通道不用后记得close(通道))

  • 同步写法
    同步的话,就把通道给取消了,return response就可以了。
func getLoginMsg() chan string {
	loginUrl := "login.url"
	data := []byte("param1=value1&param2=value2")
	response := make(chan string)
	go Post(loginUrl, data, response)
	return response
}

func Get(url string, response chan string) {
	// 创建一个请求
	req, err := http.NewRequest("GET", url, nil)
	if err != nil {
	    fmt.Println("GET请求创建失败", err)
		response <- ""
		return
	}

	// 设定Cookie(多个cookie就多次调用AddCookie)
	req.AddCookie(&http.Cookie{
		Name:  "cookie-name",
		Value: "cookie-value",
	})

	// 设定请求头(多个header就多次调用Header)
	req.Header.Add("header-name", "header-value")

	// 创建一个客户端
	client := &http.Client{}

	// 发送请求
	resp, err := client.Do(req)
	if err != nil {
		fmt.Println("GET请求发送失败", err)
		response <- ""
		return
	}
	defer func(Body io.ReadCloser) {
		err := Body.Close()
		if err != nil {
		fmt.Println("POST请求关闭失败", err)
        response <- ""
        return
		}
	}(resp.Body)

	// 读取响应
	buf := new(bytes.Buffer)
	_, err = io.Copy(buf, resp.Body)
	if err != nil {
		fmt.Println("GET读取响应失败", err)
		response <- ""
		return
	}

	response <- buf.String()
}

func Post(url string, data []byte, response chan string) {
	// 创建一个请求
	req, err := http.NewRequest("POST", url, bytes.NewBuffer(data))
	if err != nil {
		fmt.Println("POST请求创建失败", err)
		response <- ""
		return
	}
	// 设定请求头(多个header就多次调用Header)
	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")

	// 创建一个客户端
	client := http.Client{}

	// 发送请求
	resp, err := client.Do(req)
	if err != nil {
		fmt.Println("POST请求发送失败", err)
		response <- ""
		return
	}
	defer func(body io.ReadCloser) {
		err := body.Close()
		if err != nil {
			fmt.Println("POST请求关闭失败", err)
			response <- ""
			return
		}
	}(resp.Body)

	// 读取响应
	buf := new(bytes.Buffer)
	_, err = io.Copy(buf, resp.Body)
	if err != nil {
		fmt.Println("POST读取响应失败", err)
		response <- ""
		return
	}
	response <- buf.String()
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Charles-L

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值