Go言語でRedshiftとつなぐ（というかただのPostgreSQL）

ゴール

• Go言語からRedshiftにつないでクエリとかする

Redshiftとは

• Amazon Redshift および PostgreSQL

Redshiftへの接続およびクエリはPostgresだと思ってだいたい問題ないっぽいです。（一度、ISNULL使っててRedshiftにはISNULL無いのでCOALESCE使う、みたいなつまずきがありましたが...）

Go言語からPostgresにつなぐ

どこにでもあるGoでsql使うサンプルです

package main

import (
    "database/sql"
    "log"

    _ "github.com/lib/pq"
)

// User 適当に
type User struct {
    Name string `sql:"username"`
}

func main() {
    db, err := sql.Open("postgres", "postgres://user:pass@host:port/dbname")
    if err != nil {
        log.Fatalln("つなぐの", err)
    }
    defer db.Close()

    row, err := db.Query("SELECT username FROM users WHERE username LIKE $1 LIMIT $2", "otiai%", 10)
    if err != nil {
        log.Fatalln("クエリするの", err)
    }

    for row.Next() { // 行があるまでtrue
        user := new(User)
        err := row.Scan(&user.Name)
        if err != nil {
            log.Fatalln("結果をスキャンするの", err)
        }

        log.Printf("%+v\n", user)
    }
    if err := row.Err(); err != nil {
        log.Fatalln("イテレーション中の", err)
    }
}

上記の、postgres://user:pass@host:port/dbnameに該当するところを、お使いの（RDSなり）Redshiftのインスタンスであるところのfoo.bar.region.redshift.amazonaws.com:5439とかにすれば、普通に動くはずです。

めんどいのでORMつかおう

もはやRedshift関係無いんですが、Goのdatabase/sql十分可愛いけど、row.Nextとかrow.Scanとかめんどいので、そのへんもいい感じにマッピングしてくれるやつ使う。

今回はgormで

 package main

 import (
-   "database/sql"
    "log"

+   "github.com/jinzhu/gorm"
    _ "github.com/lib/pq"
 )

 // User 適当に
 type User struct {
-   Name string `sql:"username"`
+   Name string `gorm:"column:username"`
 }

 func main() {
-   db, err := sql.Open("postgres", "postgres://user:pass@host:port/dbname")
+   db, err := gorm.Open("postgres", "postgres://user:pass@host:port/dbname")
    if err != nil {
        log.Fatalln("つなぐの", err)
    }
    defer db.Close()

-   row, err := db.Query("SELECT username FROM users WHERE username LIKE $1", "otiai%")
+   users := []User{}
+   err = db.Where("username LIKE ?", "otiai%").Limit(10).Find(&users).Error
    if err != nil {
-       log.Fatalln("クエリするの", err)
-   }
-
-   for row.Next() { // 行があるまでtrue
-       user := new(User)
-       err := row.Scan(&user.Name)
-       if err != nil {
-           log.Fatalln("結果をスキャンするの", err)
-       }
-
-       log.Printf("%+v\n", user)
-   }
-   if err := row.Err(); err != nil {
-       log.Fatalln("イテレーション中の", err)
+       log.Fatalln("クエリしてスキャンするの", err)
    }
+   log.Printf("%+v\n", users)
 }

すげー楽。gormが、Findとかで渡すstructのtype名見て、この場合勝手にusersというテーブル名を入れます。発行されるSQLを生で見たい場合は

db.LogMode(true)

と、どっかにはさめばstdoutに出ます。

• https://github.com/jinzhu/gorm

というか

ぜんぜんRedshiftの記事じゃなかった...