Initial YDB 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

@@ -43,6 +43,16 @@ sql:
     go:
       package: authors
       out: sqlite
+- name: ydb
+  schema: ydb/schema.sql
+  queries: ydb/query.sql
+  engine: ydb
+  gen:
+    go:
+      package: authors
+      out: ydb
+
+
 rules:
 - name: postgresql-query-too-costly
   message: "Too costly"

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;
+

examples/authors/ydb/models.go

@@ -0,0 +1,6 @@
+CREATE TABLE authors (
+				id Uint64,
+                name Utf8 NOT NULL,
+                bio Utf8,
+				PRIMARY KEY (id)
+);

examples/authors/ydb/query.sql

@@ -86,6 +86,8 @@ func goInnerType(req *plugin.GenerateRequest, options *opts.Options, col *plugin
 		return postgresType(req, options, col)
 	case "sqlite":
 		return sqliteType(req, options, col)
+	case "ydb":
+		return YDBType(req, options, col)
 	default:
 		return "interface{}"
 	}