Move Document to separate file

lambdaworks · Nov 25, 2022 · 027b50a · 027b50a
1 parent cc2d8e0
commit 027b50a
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 20 deletions.
diff --git a/modules/library/src/main/scala/zio/elasticsearch/Document.scala b/modules/library/src/main/scala/zio/elasticsearch/Document.scala
@@ -0,0 +1,15 @@
+package zio.elasticsearch
+
+import zio.schema.Schema
+import zio.schema.codec.JsonCodec.JsonDecoder
+import zio.schema.codec.{DecodeError, JsonCodec}
+
+private[elasticsearch] final case class Document(json: String) {
+  def decode[A](implicit schema: Schema[A]): Either[DecodeError, A] = JsonDecoder.decode(schema, json)
+}
+
+private[elasticsearch] object Document {
+  def apply[A](doc: A)(implicit schema: Schema[A]): Document = Document(
+    JsonCodec.jsonEncoder(schema).encodeJson(a = doc, indent = None).toString
+  )
+}
diff --git a/modules/library/src/main/scala/zio/elasticsearch/ElasticRequest.scala b/modules/library/src/main/scala/zio/elasticsearch/ElasticRequest.scala
@@ -24,13 +24,13 @@ object ElasticRequest {
       case None           => Left(DocumentNotFound)
     }
 
-  def put[A](
+  def put[A: Schema](
     index: IndexName,
     id: DocumentId,
     doc: A,
     routing: Option[Routing] = None
-  )(implicit schema: Schema[A]): ElasticRequest[Unit] =
-    Put(index, id, Document.apply(doc), routing)
+  ): ElasticRequest[Unit] =
+    Put(index, id, Document(doc), routing)
 
   private[elasticsearch] final case class GetById(
     index: IndexName,

diff --git a/modules/library/src/main/scala/zio/elasticsearch/package.scala b/modules/library/src/main/scala/zio/elasticsearch/package.scala