Initial YQL support: SELECT logic, some convert tests, and YDB engine integration

- Added reserved keywords and parsing logic
- CREATE TABLE support and tests
- SELECT initial support and tests
- Initial YDB engine integration and examples

Author: 1NepuNep1

docker-compose.yml

@@ -19,3 +19,16 @@ services:
       POSTGRES_DB: postgres
       POSTGRES_PASSWORD: mysecretpassword
       POSTGRES_USER: postgres
+
+  ydb:
+    image: ydbplatform/local-ydb:latest
+    ports:
+      - "2135:2135"
+      - "2136:2136"
+      - "8765:8765"
+    restart: always
+    environment:
+      - YDB_USE_IN_MEMORY_PDISKS=true
+      - GRPC_TLS_PORT=2135
+      - GRPC_PORT=2136
+      - MON_PORT=8765

examples/authors/sqlc.yaml

@@ -2,51 +2,65 @@ version: '2'
 cloud:
   project: "01HAQMMECEYQYKFJN8MP16QC41"
 sql:
-- name: postgresql
-  schema: postgresql/schema.sql
-  queries: postgresql/query.sql
-  engine: postgresql
-  database:
-    uri: "${VET_TEST_EXAMPLES_POSTGRES_AUTHORS}"
-  analyzer:
-    database: false
-  rules:
-  - sqlc/db-prepare
-  - postgresql-query-too-costly
+# - name: postgresql
+#   schema: postgresql/schema.sql
+#   queries: postgresql/query.sql
+#   engine: postgresql
+#   database:
+#     uri: "${VET_TEST_EXAMPLES_POSTGRES_AUTHORS}"
+#   analyzer:
+#     database: false
+#   rules:
+#   - sqlc/db-prepare
+#   - postgresql-query-too-costly
+#   gen:
+#     go:
+#       package: authors
+#       sql_package: pgx/v5
+#       out: postgresql
+# - name: mysql
+#   schema: mysql/schema.sql
+#   queries: mysql/query.sql
+#   engine: mysql
+#   database:
+#     uri: "${VET_TEST_EXAMPLES_MYSQL_AUTHORS}"
+#   rules:
+#   - sqlc/db-prepare
+#   # - mysql-query-too-costly
+#   gen:
+#     go:
+#       package: authors
+#       out: mysql
+# - name: sqlite
+#   schema: sqlite/schema.sql
+#   queries: sqlite/query.sql
+#   engine: sqlite
+#   database:
+#     uri: file:authors?mode=memory&cache=shared
+#   rules:
+#   - sqlc/db-prepare
+#   gen:
+#     go:
+#       package: authors
+#       out: sqlite
+- name: ydb
+  schema: ydb/schema.sql
+  queries: ydb/query.sql
+  engine: ydb
+  # database:
+  #   uri: file:authors?mode=memory&cache=shared
+  # rules:
+  # - sqlc/db-prepare
   gen:
     go:
       package: authors
-      sql_package: pgx/v5
-      out: postgresql
-- name: mysql
-  schema: mysql/schema.sql
-  queries: mysql/query.sql
-  engine: mysql
-  database:
-    uri: "${VET_TEST_EXAMPLES_MYSQL_AUTHORS}"
-  rules:
-  - sqlc/db-prepare
-  # - mysql-query-too-costly
-  gen:
-    go:
-      package: authors
-      out: mysql
-- name: sqlite
-  schema: sqlite/schema.sql
-  queries: sqlite/query.sql
-  engine: sqlite
-  database:
-    uri: file:authors?mode=memory&cache=shared
-  rules:
-  - sqlc/db-prepare
-  gen:
-    go:
-      package: authors
-      out: sqlite
-rules:
-- name: postgresql-query-too-costly
-  message: "Too costly"
-  rule: "postgresql.explain.plan.total_cost > 300.0"
-- name: mysql-query-too-costly
-  message: "Too costly"
-  rule: "has(mysql.explain.query_block.cost_info) && double(mysql.explain.query_block.cost_info.query_cost) > 2.0"
+      out: ydb
+
+
+# rules:
+# - name: postgresql-query-too-costly
+#   message: "Too costly"
+#   rule: "postgresql.explain.plan.total_cost > 300.0"
+# - name: mysql-query-too-costly
+#   message: "Too costly"
+#   rule: "has(mysql.explain.query_block.cost_info) && double(mysql.explain.query_block.cost_info.query_cost) > 2.0"

examples/authors/ydb/db.go

@@ -0,0 +1,102 @@
+package authors
+
+import (
+	"context"
+	"testing"
+
+	"github.com/sqlc-dev/sqlc/internal/sqltest/local"
+	_ "github.com/ydb-platform/ydb-go-sdk/v3"
+)
+
+func TestAuthors(t *testing.T) {
+	ctx := context.Background()
+
+	test := local.YDB(t, []string{"schema.sql"})
+	defer test.DB.Close()
+
+	q := New(test.DB)
+
+	t.Run("ListAuthors", func(t *testing.T) {
+		authors, err := q.ListAuthors(ctx)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if len(authors) == 0 {
+			t.Fatal("expected at least one author, got none")
+		}
+		t.Log("Authors:")
+		for _, a := range authors {
+			bio := "NULL"
+			if a.Bio.Valid {
+				bio = a.Bio.String
+			}
+			t.Logf("- ID: %d | Name: %s | Bio: %s", a.ID, a.Name, bio)
+		}
+	})
+
+	t.Run("GetAuthor", func(t *testing.T) {
+		singleAuthor, err := q.GetAuthor(ctx, 10)
+		if err != nil {
+			t.Fatal(err)
+		}
+		bio := "NULL"
+		if singleAuthor.Bio.Valid {
+			bio = singleAuthor.Bio.String
+		}
+		t.Logf("- ID: %d | Name: %s | Bio: %s", singleAuthor.ID, singleAuthor.Name, bio)
+	})
+
+	t.Run("GetAuthorByName", func(t *testing.T) {
+		authors, err := q.GetAuthorsByName(ctx, "Александр Пушкин")
+		if err != nil {
+			t.Fatal(err)
+		}
+		if len(authors) == 0 {
+			t.Fatal("expected at least one author with this name, got none")
+		}
+		t.Log("Authors with this name:")
+		for _, a := range authors {
+			bio := "NULL"
+			if a.Bio.Valid {
+				bio = a.Bio.String
+			}
+			t.Logf("- ID: %d | Name: %s | Bio: %s", a.ID, a.Name, bio)
+		}
+	})
+
+	t.Run("ListAuthorsWithIdModulo", func(t *testing.T) {
+		authors, err := q.ListAuthorsWithIdModulo(ctx)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if len(authors) == 0 {
+			t.Fatal("expected at least one author with even ID, got none")
+		}
+		t.Log("Authors with even IDs:")
+		for _, a := range authors {
+			bio := "NULL"
+			if a.Bio.Valid {
+				bio = a.Bio.String
+			}
+			t.Logf("- ID: %d | Name: %s | Bio: %s", a.ID, a.Name, bio)
+		}
+	})
+
+	t.Run("ListAuthorsWithNullBio", func(t *testing.T) {
+		authors, err := q.ListAuthorsWithNullBio(ctx)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if len(authors) == 0 {
+			t.Fatal("expected at least one author with NULL bio, got none")
+		}
+		t.Log("Authors with NULL bio:")
+		for _, a := range authors {
+			bio := "NULL"
+			if a.Bio.Valid {
+				bio = a.Bio.String
+			}
+			t.Logf("- ID: %d | Name: %s | Bio: %s", a.ID, a.Name, bio)
+		}
+	})
+}

examples/authors/ydb/db_test.go

@@ -0,0 +1,19 @@
+-- name: ListAuthors :many 
+SELECT * FROM authors;
+
+-- name: GetAuthor :one
+SELECT * FROM authors
+WHERE id = $p0;
+
+-- name: ListAuthorsWithIdModulo :many
+SELECT * FROM authors
+WHERE id % 2 = 0;
+
+-- name: GetAuthorsByName :many
+SELECT * FROM authors
+WHERE name = $p0;
+
+-- name: ListAuthorsWithNullBio :many
+SELECT * FROM authors
+WHERE bio IS NULL;
+