Fix custom headers for RPC requests

supabase-community · Sep 8, 2024 · 24f87c9 · 24f87c9
1 parent 825aa59
commit 24f87c9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/Postgrest/src/commonMain/kotlin/io/github/jan/supabase/postgrest/PostgrestRpc.kt b/Postgrest/src/commonMain/kotlin/io/github/jan/supabase/postgrest/PostgrestRpc.kt
@@ -57,7 +57,8 @@ suspend inline fun <reified T : Any> Postgrest.rpc(
         count = requestBuilder.count,
         urlParams = urlParams,
         body = encodedParameters,
-        schema = requestBuilder.schema
+        schema = requestBuilder.schema,
+        headers = requestBuilder.headers.build()
     )
     return RestRequestExecutor.execute(postgrest = this, path = "rpc/$function", request = rpcRequest)
 }
@@ -78,7 +79,8 @@ suspend inline fun Postgrest.rpc(
         method = requestBuilder.method.httpMethod,
         count = requestBuilder.count,
         urlParams = requestBuilder.params.mapToFirstValue(),
-        schema = requestBuilder.schema
+        schema = requestBuilder.schema,
+        headers = requestBuilder.headers.build()
     )
     return RestRequestExecutor.execute(postgrest = this, path = "rpc/$function", request = rpcRequest)
 }
diff --git a/Postgrest/src/commonMain/kotlin/io/github/jan/supabase/postgrest/request/RpcRequest.kt b/Postgrest/src/commonMain/kotlin/io/github/jan/supabase/postgrest/request/RpcRequest.kt
@@ -1,7 +1,7 @@
 package io.github.jan.supabase.postgrest.request
 
 import io.github.jan.supabase.postgrest.query.Count
-import io.ktor.http.HttpMethod
+import io.ktor.http.*
 import kotlinx.serialization.json.JsonElement
 
 @PublishedApi
@@ -10,7 +10,8 @@ internal class RpcRequest(
     val count: Count? = null,
     override val urlParams: Map<String, String>,
     override val body: JsonElement? = null,
-    override val schema: String = "public"
+    override val schema: String = "public",
+    override val headers: Headers
 ) : PostgrestRequest {
 
     override val prefer = if (count != null) listOf("count=${count.identifier}") else listOf()