Add Go support.

README.md

@@ -177,6 +177,44 @@ console.log(version);
 
 See [`deno/sqlite-vss/README.md`](./deno/README.md) for more details.
 
+### Go
+
+For Go developers, use the [Go](./go) module:
+
+```go
+package main
+
+import (
+	"fmt"
+	"log"
+
+	vss "github.com/asg017/sqlite-vss/go"
+)
+
+func main() {
+	// /usr/local/lib/sqlite/ext/sqlite-vss is where vss0 and vector0 are stored.
+	db, err := vss.Open("vss-example.db", "/usr/local/lib/sqlite/ext/sqlite-vss")
+	if err != nil {
+		log.Fatal()
+	}
+	defer db.Close()
+
+	r := db.QueryRow("select vss_version();")
+	if err := r.Err(); err != nil {
+		log.Fatal(err)
+	}
+
+	var versions any
+	if err := r.Scan(&versions); err != nil {
+		log.Fatal(err)
+	}
+
+	fmt.Println(versions)
+}
+```
+
+See [`go/README.md`](./go/README.md) for more details.
+
 ### Datasette
 
 And for [Datasette](https://datasette.io/), install the [`datasette-sqlite-vss` plugin](https://datasette.io/plugins/datasette-sqlite-vss) with:

go/README.md

@@ -0,0 +1,43 @@
+# sqlite-vss Go package
+
+
+This package wraps the [go-sqlite3](https://github.com/mattn/go-sqlite3) driver and automatically loads extensions when opening a database connection. 
+
+```go
+package main
+
+import (
+	"fmt"
+	"log"
+
+	vss "github.com/asg017/sqlite-vss/go"
+)
+
+func main() {
+	// /usr/local/lib/sqlite/ext/sqlite-vss is where vss0 and vector0 are stored.
+	db, err := vss.Open("vss-example.db", "/usr/local/lib/sqlite/ext/sqlite-vss")
+	if err != nil {
+		log.Fatal()
+	}
+	defer db.Close()
+
+	r := db.QueryRow("select vss_version();")
+	if err := r.Err(); err != nil {
+		log.Fatal(err)
+	}
+
+	var versions any
+	if err := r.Scan(&versions); err != nil {
+		log.Fatal(err)
+	}
+
+	fmt.Println(versions)
+}
+```
+
+## Installing
+
+
+```bash
+go get github.com/asg017/sqlite-vss/go
+```

go/example/main.go

@@ -0,0 +1,29 @@
+package main
+
+import (
+	"fmt"
+	"log"
+
+	vss "github.com/asg017/sqlite-vss/go"
+)
+
+func main() {
+	// /usr/local/lib/sqlite/ext/sqlite-vss is where vss0 and vector0 are stored.
+	db, err := vss.Open("vss-example.db", "/usr/local/lib/sqlite/ext/sqlite-vss")
+	if err != nil {
+		log.Fatal()
+	}
+	defer db.Close()
+
+	r := db.QueryRow("select vss_version();")
+	if err := r.Err(); err != nil {
+		log.Fatal(err)
+	}
+
+	var versions any
+	if err := r.Scan(&versions); err != nil {
+		log.Fatal(err)
+	}
+
+	fmt.Println(versions)
+}

go/go.mod

@@ -0,0 +1,5 @@
+module github.com/asg017/sqlite-vss/go
+
+go 1.20
+
+require github.com/mattn/go-sqlite3 v1.14.16 // indirect

go/go.sum

@@ -0,0 +1,2 @@
+github.com/mattn/go-sqlite3 v1.14.16 h1:yOQRA0RpS5PFz/oikGwBEqvAWhWg5ufRz4ETLjwpU1Y=
+github.com/mattn/go-sqlite3 v1.14.16/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg=

go/vss.go

@@ -0,0 +1,20 @@
+package vss
+
+import (
+	"database/sql"
+	"path"
+
+	"github.com/mattn/go-sqlite3"
+)
+
+// Open returns a sql.DB pre-configured with sqlite-vss extensions,
+// to work vss0 and vector0 files should be stored under extPath.
+func Open(dbName string, extPath string) (*sql.DB, error) {
+	sql.Register("sqlite-vss", &sqlite3.SQLiteDriver{
+		Extensions: []string{
+			path.Join(extPath, "vector0"),
+			path.Join(extPath, "vss0"),
+		},
+	})
+	return sql.Open("sqlite-vss", dbName)
+}