GoのHTTP Clientでリクエストを途中でキャンセルする

Go 1.7からhttp.TransportのCancelRequestがDeprecatedになってcontextという仕組みを使うのが良いとされるようになった。

キャンセルなし

package main

import (
	//"context"
	"fmt"
	"io/ioutil"
	"net/http"
)

func main() {
	client := http.DefaultClient

	//ctx, cancelFunc := context.WithCancel(context.Background())

	req, err := http.NewRequest("GET", "https://google.com/", nil)
	if err != nil {
		panic(err)
	}

	//req = req.WithContext(ctx)

	ch := make(chan struct{})

	go func() {
		resp, err := client.Do(req)
		if err != nil { // リクエストが成功すればここは通らない
			fmt.Println(err)
			ch <- struct{}{}
			return
		}
		defer resp.Body.Close()

		body, err := ioutil.ReadAll(resp.Body)
		if err != nil {
			panic(err)
		}

		fmt.Println(string(body))

		ch <- struct{}{}
	}()

	//cancelFunc()
	<-ch
}

キャンセルあり

package main

import (
	"context"
	"fmt"
	"io/ioutil"
	"net/http"
)

func main() {
	client := http.DefaultClient

	ctx, cancelFunc := context.WithCancel(context.Background())

	req, err := http.NewRequest("GET", "https://google.com/", nil)
	if err != nil {
		panic(err)
	}

	req = req.WithContext(ctx)

	ch := make(chan struct{})

	go func() {
		resp, err := client.Do(req)
		if err != nil { // キャンセルされた場合はエラーが返る
			fmt.Println(err) // Get https://google.com/: net/http: request canceled while waiting for connection
			ch <- struct{}{}
			return
		}
		defer resp.Body.Close()

		body, err := ioutil.ReadAll(resp.Body)
		if err != nil {
			panic(err)
		}

		fmt.Println(string(body))

		ch <- struct{}{}
	}()

	cancelFunc()
	<-ch
}