Add routing param

lambdaworks · Nov 29, 2022 · 64e4afd · 64e4afd
1 parent 61415ec
commit 64e4afd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/modules/library/src/main/scala/zio/elasticsearch/ElasticRequest.scala b/modules/library/src/main/scala/zio/elasticsearch/ElasticRequest.scala
@@ -65,7 +65,8 @@ object ElasticRequest {
 
   private[elasticsearch] final case class Exists(
     index: IndexName,
-    id: DocumentId
+    id: DocumentId,
+    routing: Option[Routing] = None
   ) extends ElasticRequest[Boolean]
 
   private[elasticsearch] final case class GetById(

diff --git a/modules/library/src/main/scala/zio/elasticsearch/HttpElasticExecutor.scala b/modules/library/src/main/scala/zio/elasticsearch/HttpElasticExecutor.scala
@@ -66,9 +66,9 @@ private[elasticsearch] final class HttpElasticExecutor private (config: ElasticC
   }
 
   private def executeExists(r: Exists): Task[Boolean] = {
-    val uri = uri"$basePath/${r.index}/$Doc/${r.id}"
+    val uri = uri"$basePath/${r.index}/$Doc/${r.id}".withParam("routing", r.routing.map(_.value))
     request
-      .get(uri)
+      .head(uri)
       .send(client)
       .map(_.code.equals(Ok))
   }