redshift-data-sql-driver

Redshift-Data API SQL Driver for Go's database/sql package

Usage

for example:

package main

import (
    "context"
    "database/sql"
    "log"

    _ "github.com/mashiike/redshift-data-sql-driver"
)

func main() {
    db, err := sql.Open("redshift-data", "workgroup(default)/dev?timeout=1m")
    if err != nil {
        log.Fatalln(err)
    }
    defer db.Close()
    rows, err := db.QueryContext(
        context.Background(),
        `SELECT table_schema,table_name,table_type FROM svv_tables WHERE table_schema not like :ignore_schema`,
        sql.Named("ignore_schema", "pg_%"),
    )
    if err != nil {
        log.Fatalln(err)
    }
    for rows.Next() {
        var schema, name, tableType string
        err := rows.Scan(&schema, &name, &tableType)
        if err != nil {
            log.Println(err)
            return
        }
        log.Printf("%s.%s\t%s", schema, name, tableType)
    }
}

The pattern for specifying DSNs is as follows

• with redshift serverless: workgroup([name])/[database]
• with provisoned cluster: [dbuser]@cluster([name])/[database]
• with AWS Secrets Manager: arn:aws:secretsmanager:us-east-1:0123456789012:secret:redshift

The DSN parameters include

• timeout: Timeout for query execution. default = 15m0s
• polling: Interval to check for the end of a running query. default = 10ms
• region: Redshift Data API's region. Default is environment setting

Parameter settings are in the format of URL query parameter

workgroup(default)/dev?timeout=1m&polling=1ms

Transaction Notes

The Redshift Data API does not have an interface for pasting transactions and querying sequentially. Therefore, we have implemented an unusual implementation.

package main

import (
    "context"
    "database/sql"
    "log"

    _ "github.com/mashiike/redshift-data-sql-driver"
)

func main() {
    db, err := sql.Open("redshift-data", "workgroup(default)/dev?timeout=1m")
    if err != nil {
        log.Fatalln(err)
    }
    defer db.Close()
    tx, err := db.BeginTx(context.Background(), nil)
    if err != nil {
        log.Fatalln(err)
    }
    tx.ExecContext(context.Background(), "INSERT INTO foo VALUES (1)")
    tx.ExecContext(context.Background(), "INSERT INTO foo VALUES (2)")
    tx.ExecContext(context.Background(), "INSERT INTO foo VALUES (3)")
    tx.Commit() // BatchExecuteStatement API is called here, and the queries called during the transaction are executed together
}

Also, because the interface does not match, Query and QueryContext in the transaction are not supported. Exec and ExecContext in the transaction are not supported.

Unsupported Features

The following functions are not available

Prepare

Prepared Statements were not supported because the Redshift Data API does not have the concept of connecting to a DB.

LICENSE

MIT