From 476dbfdeafbc33f0881b9cefb8f74055ae2cbd38 Mon Sep 17 00:00:00 2001
From: Atos1337 <victor.samoilov.2002@mail.ru>
Date: Thu, 16 Dec 2021 18:18:47 +0300
Subject: [PATCH] Add getPropertyHistory for xodus

---
 .../mipt/npm/controls/xodus/util/queries.kt   | 28 +++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/controls-xodus/src/main/kotlin/ru/mipt/npm/controls/xodus/util/queries.kt b/controls-xodus/src/main/kotlin/ru/mipt/npm/controls/xodus/util/queries.kt
index ee9361f..5d7f959 100644
--- a/controls-xodus/src/main/kotlin/ru/mipt/npm/controls/xodus/util/queries.kt
+++ b/controls-xodus/src/main/kotlin/ru/mipt/npm/controls/xodus/util/queries.kt
@@ -1,9 +1,14 @@
 package ru.mipt.npm.controls.xodus.util
 
+import jetbrains.exodus.entitystore.PersistentEntityStore
 import jetbrains.exodus.entitystore.StoreTransaction
 import kotlinx.datetime.Instant
 import ru.mipt.npm.controls.api.PropertyChangedMessage
+import ru.mipt.npm.controls.xodus.defaultPersistentStoreFactory
 import ru.mipt.npm.controls.xodus.toPropertyChangedMessage
+import ru.mipt.npm.xodus.serialization.json.decodeFromEntity
+import space.kscience.dataforge.context.Factory
+import space.kscience.dataforge.meta.Meta
 
 public fun StoreTransaction.selectPropertyChangedMessagesFromRange(
     range: ClosedRange<Instant>
@@ -13,3 +18,26 @@ public fun StoreTransaction.selectPropertyChangedMessagesFromRange(
     range.start.toEpochMilliseconds(),
     range.endInclusive.toEpochMilliseconds()
 ).mapNotNull { it.toPropertyChangedMessage() }
+
+/**
+ * @return the list of deviceMessages that describes changes of specified property of specified device sorted by time
+ * @param sourceDeviceName a name of device, history of which property we want to get
+ * @param propertyName a name of property, history of which we want to get
+ * @param factory a factory that produce mongo clients
+ */
+public fun getPropertyHistory(
+    sourceDeviceName: String,
+    propertyName: String,
+    factory: Factory<PersistentEntityStore> = defaultPersistentStoreFactory,
+    meta: Meta = Meta.EMPTY
+): List<PropertyChangedMessage> {
+    return factory(meta).use { store ->
+        store.computeInTransaction { txn ->
+            txn.find("DeviceMessage", "type", "property.changed").asSequence()
+                .filter { it?.getProperty("sourceDevice")?.let { it == sourceDeviceName } ?: false &&
+                        it?.getProperty("property")?.let { it == propertyName } ?: false
+                }.sortedByDescending { it?.getProperty("time")?.let { timeStr -> Instant.parse(timeStr as String) } }
+                .toList().map { txn.decodeFromEntity(it) }
+        }
+    }
+}