feat(mysql): Add a test for VECTOR column type (#3734)

* feat(mysql): Add a test for VECTOR column type

* Fix invalid Go file

* Use MySQL 9.0 for tests

* Fix reference issue

* Fix last two test failures

* Add UNIQUE

Author: kyleconroy

.github/workflows/ci.yml

@@ -58,7 +58,7 @@ jobs:
     # Start a MySQL server
     - uses: shogo82148/actions-setup-mysql@v1
       with:
-        mysql-version: "8.1"
+        mysql-version: "9.0"
 
     - name: test ./...
       run: gotestsum --junitfile junit.xml -- --tags=examples -timeout 20m ./...

examples/ondeck/mysql/schema/0002_venue.sql

@@ -5,7 +5,7 @@ CREATE TABLE venues (
     statuses         text, -- status[],
     slug             text         not null COMMENT 'This value appears in public URLs',
     name             varchar(255) not null,
-    city             text         not null references city(slug),
+    city             varchar(255) not null references city(slug),
     spotify_playlist varchar(255) not null,
     songkick_id      text,
     tags             text  -- text[]

internal/endtoend/testdata/join_right/mysql/go/models.go

@@ -1,3 +1,8 @@
-CREATE TABLE foo (id serial not null, bar_id int references bar(id));
-CREATE TABLE bar (id serial not null);
+CREATE TABLE bar (
+    id integer not null,
+    UNIQUE(id)
+);
+
+CREATE TABLE foo (id integer not null, bar_id integer references bar(id));
+
 

internal/endtoend/testdata/join_right/mysql/go/query.sql.go

@@ -1,3 +1,7 @@
-CREATE TABLE bar (id serial not null);
-CREATE TABLE foo (id serial not null, bar integer references bar(id));
+CREATE TABLE bar (
+    id integer not null,
+    UNIQUE (id)
+);
+
+CREATE TABLE foo (id integer not null, bar integer references bar(id));
 

internal/endtoend/testdata/join_right/mysql/schema.sql

@@ -0,0 +1,7 @@
+-- name: InsertVector :exec
+INSERT INTO foo(embedding) VALUES (STRING_TO_VECTOR('[0.1, 0.2, 0.3, 0.4]'));
+
+-- name: SelectVector :many
+SELECT id FROM foo
+ORDER BY DISTANCE(STRING_TO_VECTOR('[1.2, 3.4, 5.6]'), embedding, 'L2_squared')
+LIMIT 10;

internal/endtoend/testdata/join_table_name/mysql/go/models.go

@@ -0,0 +1,4 @@
+CREATE TABLE foo(
+    id INT PRIMARY KEY auto_increment,
+    embedding VECTOR(4)
+);

internal/endtoend/testdata/join_table_name/mysql/go/query.sql.go

@@ -0,0 +1,14 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "sql_package": "database/sql",
+      "sql_driver": "github.com/go-sql-driver/mysql",
+      "engine": "mysql",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}