amoeba commented on code in PR #3204: URL: https://github.com/apache/arrow-adbc/pull/3204#discussion_r2254800000
########## go/adbc/README.md: ########## @@ -0,0 +1,308 @@ +<!--- + Licensed to the Apache Software Foundation (ASF) under one + or more contributor license agreements. See the NOTICE file + distributed with this work for additional information + regarding copyright ownership. The ASF licenses this file + to you under the Apache License, Version 2.0 (the + "License"); you may not use this file except in compliance + with the License. You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + + Unless required by applicable law or agreed to in writing, + software distributed under the License is distributed on an + "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY + KIND, either express or implied. See the License for the + specific language governing permissions and limitations + under the License. +--> + +# adbc + +The `adbc` package provides Go interface definitions for using Arrow data and databases, also known as [ADBC](https://arrow.apache.org/adbc/). + +## Getting Started + +To give a brief tour of what you can do with the `adbc` package, we'll show a brief example of using the [SQLite driver](https://arrow.apache.org/adbc/current/driver/sqlite.html). + +### Installation + +First, assuming you've already created a Go module, add the `drivermgr` package to your `go.mod` by running: + +```sh +go get github.com/apache/arrow-adbc/go/adbc/drivermgr +``` + +### Imports + +For imports, all we really need is `github.com/apache/arrow-adbc/go/adbc/drivermgr` to start but we'll import a few more modules now because we'll use them later. + +```go +package main + +import ( + "context" + "fmt" + "strings" + + "github.com/apache/arrow-adbc/go/adbc" + "github.com/apache/arrow-adbc/go/adbc/drivermgr" + "github.com/apache/arrow-go/v18/arrow" + "github.com/apache/arrow-go/v18/arrow/array" + "github.com/apache/arrow-go/v18/arrow/memory" +) + +func GettingStarted() error { + ctx := context.Background() + +``` + +### Setup + +Any program using ADBC will start with creating a `Database`, a `Connection` to that `Database`, and usually one or more `Statement` objects. + +First we create a `Database`, providing `adbc_driver_sqlite` as the name of the driver to load: + +```go + var drv drivermgr.Driver + db, err := drv.NewDatabase(map[string]string{ + "driver": "adbc_driver_sqlite", + }) + if err != nil { + return err + } + defer db.Close() +``` + +Using the `Database` instance we created above, we can now create a `Connection` with `Open`: + +```go + conn, err := db.Open(ctx) + if err != nil { + return err + } + defer conn.Close() +``` + +Before we can execute any queries, we need to create a `Statement` to manage them: + +```go + stmt, err := conn.NewStatement() + if err != nil { + return err + } + defer stmt.Close() +``` + +Now that we have a basic setup, we'll show some of the functionality. + +### Executing a Query + +We can execute a query and get the results as Arrow data: + +```go + err = stmt.SetSqlQuery("SELECT 1, 2.0, 'Hello, world!'") + if err != nil { + return err + } + + reader, _, err := stmt.ExecuteQuery(ctx) + if err != nil { + return err + } + defer reader.Release() + + for reader.Next() { + record := reader.Record() + + // Extract our three columns + col0 := record.Column(0) + col1 := record.Column(1) + col2 := record.Column(2) + + for i := 0; i < int(record.NumRows()); i++ { + fmt.Printf("Row %d: %v, %v, %v\n", i, + col0.ValueStr(i), col1.ValueStr(i), col2.ValueStr(i)) + } + } +``` + +This prints: + +```text +Row 0: 1, 2, Hello, world! +``` + +### Parameterized Queries + +We can bind Arrow records as parameters in our queries too: + +```go + // Create parameter schema and data + paramSchema := arrow.NewSchema([]arrow.Field{ + {Name: "param1", Type: arrow.PrimitiveTypes.Int64}, + }, nil) + + params, _, err := array.RecordFromJSON(memory.DefaultAllocator, paramSchema, + strings.NewReader(`[{"param1": 41}]`)) + if err != nil { + return err + } + defer params.Release() + + // Set query and bind parameters + err = stmt.SetSqlQuery("SELECT ? + 1 AS the_answer") + if err != nil { + return err + } + + err = stmt.Prepare(ctx) + if err != nil { + return err + } + + err = stmt.Bind(ctx, params) + if err != nil { + return err + } + + reader2, _, err := stmt.ExecuteQuery(ctx) + if err != nil { + return err + } + defer reader2.Release() +``` + +### Ingesting Bulk Data + +You may have noticed that it took a number of steps to bind Arrow data to a query. + +New in `adbc` 1.7.0, IngestStream is a utility that simplifies the five-step boilerplate of `NewStatement`, `SetOption`, `Bind`, `Execute`, and `Close`. Review Comment: I will catch on eventually, thanks for the catch. Fixed in 28ac6113aac713c1dc736438418d5a4260bf1a31. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: github-unsubscr...@arrow.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org
