add function setval

add test
cashapp · Aug 16, 2023 · b6a3c34 · b6a3c34
1 parent 1a74ee5
commit b6a3c34
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/...tgresql/src/main/kotlin/app/cash/sqldelight/dialects/postgresql/PostgreSqlTypeResolver.kt b/...tgresql/src/main/kotlin/app/cash/sqldelight/dialects/postgresql/PostgreSqlTypeResolver.kt
@@ -109,7 +109,7 @@ class PostgreSqlTypeResolver(private val parentResolver: TypeResolver) : TypeRes
     "regr_count" -> IntermediateType(BIG_INT).asNullable()
     "gen_random_uuid" -> IntermediateType(PostgreSqlType.UUID)
     "length", "character_length", "char_length" -> IntermediateType(PostgreSqlType.INTEGER).nullableIf(resolvedType(exprList[0]).javaType.isNullable)
-    "currval", "lastval", "nextval" -> IntermediateType(BIG_INT)
+    "currval", "lastval", "nextval", "setval" -> IntermediateType(BIG_INT)
     else -> null
   }
 

diff --git a/...on-postgresql/src/main/sqldelight/app/cash/sqldelight/postgresql/integration/Sequences.sq b/...on-postgresql/src/main/sqldelight/app/cash/sqldelight/postgresql/integration/Sequences.sq
@@ -16,3 +16,6 @@ SELECT currval('serial');
 
 selectLastVal:
 SELECT lastval();
+
+selectSetVal:
+SELECT setval('serial', 200, TRUE);
diff --git a/...n-postgresql/src/test/kotlin/app/cash/sqldelight/postgresql/integration/PostgreSqlTest.kt b/...n-postgresql/src/test/kotlin/app/cash/sqldelight/postgresql/integration/PostgreSqlTest.kt
@@ -336,5 +336,8 @@ class PostgreSqlTest {
     val selectNextVal = database.sequencesQueries.selectNextVal().executeAsOne()
     val lastVal = database.sequencesQueries.selectLastVal().executeAsOne()
     assertThat(selectNextVal).isEqualTo(lastVal)
+
+    val selectSetVal = database.sequencesQueries.selectSetVal().executeAsOne()
+    assertThat(selectSetVal).isEqualTo(nextVal)
   }
 }