Added almost full INSERT support and basic DELETE (without ON) support

Author: 1NepuNep1

examples/authors/sqlc.yaml

@@ -2,47 +2,47 @@ 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
-  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: 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

examples/authors/ydb/db_test.go

@@ -8,6 +8,10 @@ import (
 	_ "github.com/ydb-platform/ydb-go-sdk/v3"
 )
 
+func ptr(s string) *string {
+	return &s
+}
+
 func TestAuthors(t *testing.T) {
 	ctx := context.Background()
 
@@ -16,6 +20,53 @@ func TestAuthors(t *testing.T) {
 
 	q := New(test.DB)
 
+	t.Run("InsertAuthors", func(t *testing.T) {
+		authorsToInsert := []CreateOrUpdateAuthorParams{
+			{P0: 1, P1: "Лев Толстой", P2: ptr("Русский писатель, автор \"Война и мир\"")},
+			{P0: 2, P1: "Александр Пушкин", P2: ptr("Автор \"Евгения Онегина\"")},
+			{P0: 3, P1: "Александр Пушкин", P2: ptr("Русский поэт, драматург и прозаик")},
+			{P0: 4, P1: "Фёдор Достоевский", P2: ptr("Автор \"Преступление и наказание\"")},
+			{P0: 5, P1: "Николай Гоголь", P2: ptr("Автор \"Мёртвые души\"")},
+			{P0: 6, P1: "Антон Чехов", P2: nil},
+			{P0: 7, P1: "Иван Тургенев", P2: ptr("Автор \"Отцы и дети\"")},
+			{P0: 8, P1: "Михаил Лермонтов", P2: nil},
+			{P0: 9, P1: "Даниил Хармс", P2: ptr("Абсурдист, писатель и поэт")},
+			{P0: 10, P1: "Максим Горький", P2: ptr("Автор \"На дне\"")},
+			{P0: 11, P1: "Владимир Маяковский", P2: nil},
+			{P0: 12, P1: "Сергей Есенин", P2: ptr("Русский лирик")},
+			{P0: 13, P1: "Борис Пастернак", P2: ptr("Автор \"Доктор Живаго\"")},
+		}
+
+		for _, author := range authorsToInsert {
+			if _, err := q.CreateOrUpdateAuthor(ctx, author); err != nil {
+				t.Fatalf("failed to insert author %q: %v", author.P1, err)
+			}
+		}
+	})
+
+	t.Run("CreateOrUpdateAuthorReturningBio", func(t *testing.T) {
+		newBio := "Обновленная биография автора"
+		arg := CreateOrUpdateAuthorRetunringBioParams{
+			P0: 3,
+			P1: "Тестовый Автор",
+			P2: &newBio,
+		}
+
+		returnedBio, err := q.CreateOrUpdateAuthorRetunringBio(ctx, arg)
+		if err != nil {
+			t.Fatalf("failed to create or update author: %v", err)
+		}
+
+		if returnedBio == nil {
+			t.Fatal("expected non-nil bio, got nil")
+		}
+		if *returnedBio != newBio {
+			t.Fatalf("expected bio %q, got %q", newBio, *returnedBio)
+		}
+
+		t.Logf("Author created or updated successfully with bio: %s", *returnedBio)
+	})
+
 	t.Run("ListAuthors", func(t *testing.T) {
 		authors, err := q.ListAuthors(ctx)
 		if err != nil {
@@ -26,9 +77,9 @@ func TestAuthors(t *testing.T) {
 		}
 		t.Log("Authors:")
 		for _, a := range authors {
-			bio := "NULL"
-			if a.Bio.Valid {
-				bio = a.Bio.String
+			bio := "Null"
+			if a.Bio != nil {
+				bio = *a.Bio
 			}
 			t.Logf("- ID: %d | Name: %s | Bio: %s", a.ID, a.Name, bio)
 		}
@@ -39,9 +90,9 @@ func TestAuthors(t *testing.T) {
 		if err != nil {
 			t.Fatal(err)
 		}
-		bio := "NULL"
-		if singleAuthor.Bio.Valid {
-			bio = singleAuthor.Bio.String
+		bio := "Null"
+		if singleAuthor.Bio != nil {
+			bio = *singleAuthor.Bio
 		}
 		t.Logf("- ID: %d | Name: %s | Bio: %s", singleAuthor.ID, singleAuthor.Name, bio)
 	})
@@ -56,9 +107,9 @@ func TestAuthors(t *testing.T) {
 		}
 		t.Log("Authors with this name:")
 		for _, a := range authors {
-			bio := "NULL"
-			if a.Bio.Valid {
-				bio = a.Bio.String
+			bio := "Null"
+			if a.Bio != nil {
+				bio = *a.Bio
 			}
 			t.Logf("- ID: %d | Name: %s | Bio: %s", a.ID, a.Name, bio)
 		}
@@ -74,9 +125,9 @@ func TestAuthors(t *testing.T) {
 		}
 		t.Log("Authors with even IDs:")
 		for _, a := range authors {
-			bio := "NULL"
-			if a.Bio.Valid {
-				bio = a.Bio.String
+			bio := "Null"
+			if a.Bio != nil {
+				bio = *a.Bio
 			}
 			t.Logf("- ID: %d | Name: %s | Bio: %s", a.ID, a.Name, bio)
 		}
@@ -92,11 +143,27 @@ func TestAuthors(t *testing.T) {
 		}
 		t.Log("Authors with NULL bio:")
 		for _, a := range authors {
-			bio := "NULL"
-			if a.Bio.Valid {
-				bio = a.Bio.String
+			bio := "Null"
+			if a.Bio != nil {
+				bio = *a.Bio
 			}
 			t.Logf("- ID: %d | Name: %s | Bio: %s", a.ID, a.Name, bio)
 		}
 	})
+
+	t.Run("Delete All Authors", func(t *testing.T) {
+		var i uint64
+		for i = 1; i <= 13; i++ {
+			if err := q.DeleteAuthor(ctx, i); err != nil {
+				t.Fatalf("failed to delete authors: %v", err)
+			}
+		}
+		authors, err := q.ListAuthors(ctx)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if len(authors) != 0 {
+			t.Fatalf("expected no authors, got %d", len(authors))
+		}
+	})
 }

examples/authors/ydb/models.go

@@ -17,3 +17,13 @@ WHERE name = $p0;
 SELECT * FROM authors
 WHERE bio IS NULL;
 
+-- name: CreateOrUpdateAuthor :execresult 
+UPSERT INTO authors (id, name, bio) VALUES ($p0, $p1, $p2);
+
+-- name: CreateOrUpdateAuthorRetunringBio :one
+UPSERT INTO authors (id, name, bio) VALUES ($p0, $p1, $p2) RETURNING bio;
+
+-- name: DeleteAuthor :exec 
+DELETE FROM authors
+WHERE id = $p0;
+