Rename ArticleViewManager to ArticleViewRepository

2021-04-02 12:25:54 +02:00
parent 01c5b78325
commit db810ab0c6
5 changed files with 20 additions and 22 deletions
--- a/src/main/kotlin/fr/dcproject/component/article/ArticleViewRepository.kt
+++ b/src/main/kotlin/fr/dcproject/component/article/ArticleViewRepository.kt
@@ -0,0 +1,100 @@
+package fr.dcproject.component.article
+
+import fr.dcproject.common.entity.VersionableId
+import fr.dcproject.common.utils.contentToString
+import fr.dcproject.common.utils.getJsonField
+import fr.dcproject.common.utils.toIso
+import fr.dcproject.component.article.database.ArticleI
+import fr.dcproject.component.citizen.database.CitizenI
+import fr.dcproject.component.views.ViewRepository
+import fr.dcproject.component.views.entity.ViewAggregation
+import org.elasticsearch.client.Request
+import org.elasticsearch.client.RestClient
+import org.joda.time.DateTime
+import java.util.UUID
+
+/**
+ * Wrapper for manage views with elasticsearch
+ */
+class ArticleViewRepository <A> (private val restClient: RestClient) : ViewRepository<A> where A : VersionableId, A : ArticleI {
+    /**
+     * Add view on article to elasticsearch
+     */
+    override fun addView(ip: String, entity: A, citizen: CitizenI?, dateTime: DateTime) {
+        val isLogged = (citizen != null).toString()
+        val ref = citizen?.id ?: UUID.nameUUIDFromBytes(ip.toByteArray())!!
+        val request = Request(
+            "POST",
+            "/views/_doc/"
+        ).apply {
+            //language=JSON
+            setJsonEntity(
+                """
+                {
+                  "logged": $isLogged,
+                  "type": "article",
+                  "user_ref": "$ref",
+                  "ip": "$ip",
+                  "id": "${entity.id}",
+                  "version_id": "${entity.versionId}",
+                  "citizen_id": "${citizen?.id}",
+                  "view_at": "${dateTime.toIso()}"
+                }
+                """.trimIndent()
+            )
+        }
+
+        restClient.performRequest(request)
+    }
+
+    /**
+     * Get article views aggregations from elasticsearch
+     */
+    override fun getViewsCount(entity: A): ViewAggregation {
+        val request = Request(
+            "GET",
+            "/views/_search"
+        ).apply {
+            //language=JSON
+            setJsonEntity(
+                """
+                {
+                  "size": 0,
+                  "query": {
+                    "bool": {
+                      "must": {
+                        "term": {
+                          "version_id": "${entity.versionId}"
+                        }
+                      }
+                    }
+                  },
+                  "aggs" : {
+                    "total": {
+                      "composite" : {
+                        "sources" : [
+                          { "version_id": { "terms": {"field": "version_id" } } }
+                        ]
+                      }
+                    },
+                    "unique" : {
+                      "cardinality" : {
+                        "field" : "user_ref",
+                        "precision_threshold": 1
+                      }
+                    }
+                  }
+                }
+                """.trimIndent()
+            )
+        }
+
+        return restClient
+            .performRequest(request).contentToString().run {
+                ViewAggregation(
+                    getJsonField("$.aggregations.total.buckets[0].doc_count") ?: 0,
+                    getJsonField("$.aggregations.unique.value") ?: 0
+                )
+            }
+    }
+}