Added UPDATE and DELETE full logic. Updated examples in example/authors

Author: 1NepuNep1

examples/authors/ydb/db_test.go

@@ -46,13 +46,13 @@ func TestAuthors(t *testing.T) {
 
 	t.Run("CreateOrUpdateAuthorReturningBio", func(t *testing.T) {
 		newBio := "Обновленная биография автора"
-		arg := CreateOrUpdateAuthorRetunringBioParams{
+		arg := CreateOrUpdateAuthorReturningBioParams{
 			P0: 3,
 			P1: "Тестовый Автор",
 			P2: &newBio,
 		}
 
-		returnedBio, err := q.CreateOrUpdateAuthorRetunringBio(ctx, arg)
+		returnedBio, err := q.CreateOrUpdateAuthorReturningBio(ctx, arg)
 		if err != nil {
 			t.Fatalf("failed to create or update author: %v", err)
 		}
@@ -67,6 +67,24 @@ func TestAuthors(t *testing.T) {
 		t.Logf("Author created or updated successfully with bio: %s", *returnedBio)
 	})
 
+	t.Run("Update Author", func(t *testing.T) {
+		arg := UpdateAuthorByIDParams{
+			P0: "Максим Горький",
+			P1: ptr("Обновленная биография"),
+			P2: 10,
+		}
+
+		singleAuthor, err := q.UpdateAuthorByID(ctx, arg)
+		if err != nil {
+			t.Fatal(err)
+		}
+		bio := "Null"
+		if singleAuthor.Bio != nil {
+			bio = *singleAuthor.Bio
+		}
+		t.Logf("- ID: %d | Name: %s | Bio: %s", singleAuthor.ID, singleAuthor.Name, bio)
+	})
+
 	t.Run("ListAuthors", func(t *testing.T) {
 		authors, err := q.ListAuthors(ctx)
 		if err != nil {
@@ -115,24 +133,6 @@ func TestAuthors(t *testing.T) {
 		}
 	})
 
-	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 != nil {
-				bio = *a.Bio
-			}
-			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 {

examples/authors/ydb/query.sql

@@ -5,10 +5,6 @@ SELECT * FROM authors;
 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;
@@ -20,10 +16,11 @@ WHERE bio IS NULL;
 -- name: CreateOrUpdateAuthor :execresult 
 UPSERT INTO authors (id, name, bio) VALUES ($p0, $p1, $p2);
 
--- name: CreateOrUpdateAuthorRetunringBio :one
+-- name: CreateOrUpdateAuthorReturningBio :one
 UPSERT INTO authors (id, name, bio) VALUES ($p0, $p1, $p2) RETURNING bio;
 
 -- name: DeleteAuthor :exec 
-DELETE FROM authors
-WHERE id = $p0;
+DELETE FROM authors WHERE id = $p0;
 
+-- name: UpdateAuthorByID :one
+UPDATE authors SET name = $p0, bio = $p1 WHERE id = $p2 RETURNING *;

examples/authors/ydb/query.sql.go

@@ -24,6 +24,8 @@ require (
 	github.com/tetratelabs/wazero v1.9.0
 	github.com/wasilibs/go-pgquery v0.0.0-20250409022910-10ac41983c07
 	github.com/xeipuuv/gojsonschema v1.2.0
+	github.com/ydb-platform/ydb-go-sdk/v3 v3.108.0
+	github.com/ydb-platform/yql-parsers v0.0.0-20250309001738-7d693911f333
 	golang.org/x/sync v0.13.0
 	google.golang.org/grpc v1.71.1
 	google.golang.org/protobuf v1.36.6
@@ -35,6 +37,7 @@ require (
 	cel.dev/expr v0.19.1 // indirect
 	filippo.io/edwards25519 v1.1.0 // indirect
 	github.com/dustin/go-humanize v1.0.1 // indirect
+	github.com/golang-jwt/jwt/v4 v4.5.0 // indirect
 	github.com/google/uuid v1.6.0 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 	github.com/jackc/chunkreader/v2 v2.0.1 // indirect
@@ -45,6 +48,7 @@ require (
 	github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
 	github.com/jackc/pgtype v1.14.0 // indirect
 	github.com/jackc/puddle/v2 v2.2.2 // indirect
+	github.com/jonboulle/clockwork v0.3.0 // indirect
 	github.com/mattn/go-isatty v0.0.20 // indirect
 	github.com/ncruces/go-strftime v0.1.9 // indirect
 	github.com/pingcap/errors v0.11.5-0.20240311024730-e056997136bb // indirect
@@ -56,6 +60,7 @@ require (
 	github.com/wasilibs/wazero-helpers v0.0.0-20240620070341-3dff1577cd52 // indirect
 	github.com/xeipuuv/gojsonpointer v0.0.0-20180127040702-4e3ac2762d5f // indirect
 	github.com/xeipuuv/gojsonreference v0.0.0-20180127040603-bd5ef7bd5415 // indirect
+	github.com/ydb-platform/ydb-go-genproto v0.0.0-20241112172322-ea1f63298f77 // indirect
 	go.uber.org/atomic v1.11.0 // indirect
 	go.uber.org/multierr v1.11.0 // indirect
 	go.uber.org/zap v1.27.0 // indirect