GolangでHTTP/1.1のチャンク受信

fullChankClient.go

package main

import (
	"net"
	"net/http"
	"time"
	"bufio"
	"io"
	"strconv"
	"log"
)

func main() {
	// TCPソケットオープン
	dialer := &net.Dialer{
		Timeout:   30 * time.Second,
		KeepAlive: 30 * time.Second,
	}
	conn, err := dialer.Dial("tcp", "localhost:8080")
	if err != nil {
		panic(err)
	}
	defer conn.Close()

	// コネクションを通してリクエストの送信
	request, err := http.NewRequest("GET", "http://localhost:8080/chunked", nil)
	err = request.Write(conn)
	if err != nil {
		panic(err)
	}

	//　コネクションからリーダーを作成
	reader := bufio.NewReader(conn)

	// ヘッダの読み込み
	resp, err := http.ReadResponse(reader, request)
	if err != nil {
		panic(err)
	}

	if resp.TransferEncoding[0] != "chunked" {
		panic("wront transfer encoding")
	}

	for {
		// サイズを取得
		sizeStr, err := reader.ReadBytes('\n')
		if err == io.EOF {
			break
		}

		// 16進数のサイズをパース。サイズが０ならクローズ
		// HTTPではボディ要素以外の行末印をCRLFとして扱っているため、その分は読み込まない。
		size, err := strconv.ParseInt(string(sizeStr[:len(sizeStr)-2]), 16, 64)
		if size == 0 {
			break
		}
		if err != nil {
			panic(err)
		}
		// サイズ数分バッファを確保して読み込み
		line := make([]byte, int(size))
		reader.Read(line)
		reader.Discard(2)
		log.Println(" ", string(line))
	}

}