Goで、DBレコードから良い感じにJSON生成する

こんな感じのレスポンスを期待されている時に、JSONをどう生成したかという話。

$ echo '{"books": [{"id":"1","title":"Story about Ruby","amount":20,"created_at":1551354984},{"id":"2","title":"Story about Pearl","amount": -30,"created_at":1551354984}]}' | jq .
{
  "books": [
    {
      "id": "1",
      "title": "Story about Ruby",
      "amount": 20,
      "created_at": 1551354984
    },
    {
      "id": "2",
      "title": "Story about Pearl",
      "amount": -30,
      "created_at": 1551354984
    }
  ]
}

なんか変なデータ構造だけど、実際のプロジェクトコードの構造体名を変えてたりなので、まぁ気にせず😀

// 複数テーブルからのデータを統合した情報
type BookWithAmount struct {
    Book  *models.Book
    Amount int64
}

// 蔵書一覧レスポンス
type booksResponse struct {
    Books []bookRecord `json:"books"`
}

// 蔵書一覧レスポンスに含まれるレコード
type bookRecord struct {
    ID          string `json:"id"`
    Title       string `json:"title"`
    Amount      int64  `json:"amount"`
    CreatedAt   int64  `json:"created_at"`
}

// レスポンス用に蔵書レコードを整形して返す
func getBooksResponse(books []BookWithAmount) booksResponse {
    records := []bookRecord{}
    for _, b := range books {
        r := bookRecord{}
        r.ID = strconv.FormatInt(b.Book.ID, 10)// 文字列希望
        r.Title = b.Book.Title
        r.Amount = map[bool]int64{true: b.Amount, false: -b.Amount}[b.Book.Type == Order]// 場合によってマイナス値に
        r.CreatedAt = b.Book.CreatedAt.Unix()// タイムスタンプ希望
        records = append(records, r)
    }
    return booksResponse{Books: records}
}

func main() {
    // ごにょごにょ...（DBからレコード取得）

    // レスポンス整形
    booksResponse := getBooksResponse(books)
    json, err := json.Marshal(booksResponse)
    if err != nil {
        return
    }

    // ゴニョゴニョ...（レスポンスとして返す）
}

複数テーブルから必要な情報を抜き出して、クライアントが期待する形式にできました👨‍💻🥤