[Go] JSON レスポンスから CSV を作成する

やりたいこと

JSON のウェブから取得して、ほしいデータだけを CSV に書き出す。

手順

まずは、JSON の中身を確認する

JSONPlaceholder というサイトから色々な JSON 提供している。
今回は ToDo リストを使う。
それでは、JSON の中身を確認。

[
  {
    userId: 1,
    id: 1,
    title: "delectus aut autem",
    completed: false
  },
  {
    userId: 1,
    id: 2,
    title: "quis ut nam facilis et officia qui",
    completed: false
  },
  {
    userId: 1,
    id: 3,
    title: "fugiat veniam minus",
    completed: false
  },
... と続く

ふむふむ。
userId、id、title、completed がワンセットでリピートしてるのね。

JSON と Go の構造を合わせて struct を作る

ネストがなくシンプルの構造だ。

type ToDo struct {
	UserID int
	ID     int
	Title  string
	Body   string
}

JSON データは、この構造がスライス []ToDo で表現されていることになる

ウェブから JSON を取得する

GET リクエストして、問題なければレスポンス BODY を読む。

	url := "https://jsonplaceholder.typicode.com/posts"
	res, err := http.Get(url)
	if err != nil {
		log.Fatal(err)
	}
	defer res.Body.Close()

	if res.StatusCode != http.StatusOK {
		log.Fatal(err)
	}

	body, err := ioutil.ReadAll(res.Body)
	if err != nil {
		log.Fatal(err)
	}

Unmarshal でJSON → Go 構造に変換

Unmarshal の第2引数にはポインタを渡す
そうしないとエラーになる
Ref: Unmarshal

	var jsonData []ToDo
	err = json.Unmarshal(body, &jsonData)
	if err != nil {
		log.Fatal(err)
	}

CSV への書き込み

UserID はいらないので、CSV には出力しない。
どこでも利用する予定がなければ、Unmarshal で読み込まないこともできる。

	csvFile, err := os.Create("./data.csv")
	if err != nil {
		log.Fatal(err)
	}
	defer csvFile.Close()

	writer := csv.NewWriter(csvFile)

	for _, oneToDo := range jsonData {
		var row []string
		row = append(row, strconv.Itoa(oneToDo.Id))
		row = append(row, oneToDo.Title)
		row = append(row, oneToDo.Body)
		writer.Write(row)
	}

	writer.Flush()

まとめ

読み込む JSON に合った struct ができれば、Unmarshal するだけだから簡単だ。