Merge pull request #32 from mashiike/feature/tx

implement Transaction With BatchExecuteStatement

Author: mashiike

README.md

@@ -16,36 +16,36 @@ for example:
 package main
 
 import (
-	"context"
-	"database/sql"
-	"log"
+    "context"
+    "database/sql"
+    "log"
 
-	_ "github.com/mashiike/redshift-data-sql-driver"
+    _ "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)
-	}
+    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)
+    }
 }
 ```
 
@@ -65,14 +65,46 @@ Parameter settings are in the format of URL query parameter
 
 `workgroup(default)/dev?timeout=1m&polling=1ms`
 
-## Unsupported Features
+### Transaction Notes
 
-The following functions are not available
+The Redshift Data API does not have an interface for pasting transactions and querying sequentially.
+Therefore, we have implemented an unusual implementation.
 
-### [BeginTx](https://pkg.go.dev/database/sql#DB.BeginTx),[Begin](https://pkg.go.dev/database/sql#DB.Begin)  
 
-The Redshift Data API does not have an interface for pasting transactions and querying sequentially.
-Therefore, transactional functionality is not available.
+```go
+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](https://pkg.go.dev/database/sql#DB.Prepare)
 

client.go

@@ -11,6 +11,7 @@ type RedshiftDataClient interface {
 	ExecuteStatement(ctx context.Context, params *redshiftdata.ExecuteStatementInput, optFns ...func(*redshiftdata.Options)) (*redshiftdata.ExecuteStatementOutput, error)
 	DescribeStatement(ctx context.Context, params *redshiftdata.DescribeStatementInput, optFns ...func(*redshiftdata.Options)) (*redshiftdata.DescribeStatementOutput, error)
 	CancelStatement(ctx context.Context, params *redshiftdata.CancelStatementInput, optFns ...func(*redshiftdata.Options)) (*redshiftdata.CancelStatementOutput, error)
+	BatchExecuteStatement(ctx context.Context, params *redshiftdata.BatchExecuteStatementInput, optFns ...func(*redshiftdata.Options)) (*redshiftdata.BatchExecuteStatementOutput, error)
 	redshiftdata.GetStatementResultAPIClient
 }