From ebb34cd802f44b3bc71b7985914b919ab780627b Mon Sep 17 00:00:00 2001
From: Swordsteel <swordsteel@gmail.com>
Date: Mon, 11 Nov 2024 11:41:34 +0100
Subject: [PATCH] add MeasurementRepository

---
 .../repository/MeasurementRepository.kt       | 44 ++++++++++
 .../repository/MeasurementRepositoryTest.kt   | 82 +++++++++++++++++++
 2 files changed, 126 insertions(+)
 create mode 100644 src/main/kotlin/ltd/hlaeja/repository/MeasurementRepository.kt
 create mode 100644 src/test/kotlin/ltd/hlaeja/repository/MeasurementRepositoryTest.kt

diff --git a/src/main/kotlin/ltd/hlaeja/repository/MeasurementRepository.kt b/src/main/kotlin/ltd/hlaeja/repository/MeasurementRepository.kt
new file mode 100644
index 0000000..85c6192
--- /dev/null
+++ b/src/main/kotlin/ltd/hlaeja/repository/MeasurementRepository.kt
@@ -0,0 +1,44 @@
+package ltd.hlaeja.repository
+
+import com.influxdb.client.InfluxDBClient
+import com.influxdb.client.write.Point
+import com.influxdb.query.FluxTable
+import java.util.UUID
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
+import ltd.hlaeja.properties.InfluxDbProperties
+import org.springframework.stereotype.Repository
+
+@Repository
+class MeasurementRepository(
+    private val influxDBClient: InfluxDBClient,
+    private val properties: InfluxDbProperties,
+) {
+
+    companion object {
+        const val BY_NODE_QUERY: String = """
+          from(bucket: "%s")
+            |> range(start: -1y)
+            |> filter(fn: (r) => r["_measurement"] == "%s")
+            |> filter(fn: (r) => r["node"] == "%s")
+            |> group(columns: ["_field"])
+            |> sort(columns: ["_time"], desc: true)
+            |> limit(n: 1)
+        """
+    }
+
+    suspend fun save(
+        point: Point,
+    ) = withContext(Dispatchers.IO) {
+        influxDBClient.makeWriteApi().writePoint(point)
+    }
+
+    suspend fun getByNode(
+        client: UUID,
+        device: UUID,
+    ): MutableList<FluxTable> = influxDBClient.queryApi
+        .query(
+            BY_NODE_QUERY.format(properties.bucket, client, device).trimIndent(),
+            properties.org,
+        )
+}
diff --git a/src/test/kotlin/ltd/hlaeja/repository/MeasurementRepositoryTest.kt b/src/test/kotlin/ltd/hlaeja/repository/MeasurementRepositoryTest.kt
new file mode 100644
index 0000000..66b1286
--- /dev/null
+++ b/src/test/kotlin/ltd/hlaeja/repository/MeasurementRepositoryTest.kt
@@ -0,0 +1,82 @@
+package ltd.hlaeja.repository
+
+import com.influxdb.client.InfluxDBClient
+import com.influxdb.client.QueryApi
+import com.influxdb.client.WriteApi
+import com.influxdb.client.write.Point
+import com.influxdb.query.FluxTable
+import io.mockk.Runs
+import io.mockk.coVerify
+import io.mockk.every
+import io.mockk.just
+import io.mockk.mockk
+import io.mockk.verify
+import java.util.UUID
+import kotlin.test.Test
+import kotlinx.coroutines.test.runTest
+import ltd.hlaeja.properties.InfluxDbProperties
+import org.junit.jupiter.api.Assertions.assertEquals
+import org.junit.jupiter.api.BeforeEach
+
+class MeasurementRepositoryTest {
+
+    private val client: InfluxDBClient = mockk()
+    private val properties: InfluxDbProperties = mockk()
+    private val writeApi: WriteApi = mockk()
+    private val queryApi: QueryApi = mockk()
+
+    private lateinit var repository: MeasurementRepository
+
+    @BeforeEach
+    fun setUp() {
+        repository = MeasurementRepository(client, properties)
+    }
+
+    @Test
+    fun `save point to influxdb`() = runTest {
+        // given
+        val point = Point.measurement("test").addField("value", 12.3)
+
+        every { client.makeWriteApi() } returns writeApi
+        every { writeApi.writePoint(any()) } just Runs
+
+        // when
+        repository.save(point)
+
+        // then
+        coVerify { writeApi.writePoint(point) }
+    }
+
+    @Test
+    fun `load fields from influxdb by client and device`() = runTest {
+        // given
+        val clientUuid = UUID.fromString("00000000-0000-0000-0000-000000000000")
+        val deviceUuid = UUID.fromString("00000000-0000-0000-0000-000000000001")
+        val mockResult = mutableListOf(FluxTable())
+
+        every { client.queryApi } returns queryApi
+        properties.apply {
+            every { bucket } returns "bucket"
+            every { org } returns "organization"
+        }
+        every { queryApi.query(any(String::class), any(String::class)) } returns mockResult
+
+        // when
+        val result = repository.getByNode(clientUuid, deviceUuid)
+
+        // then
+        assertEquals(mockResult, result)
+        verify {
+            queryApi.query(
+                "from(bucket: \"bucket\")\n" +
+                    "  |> range(start: -1y)\n" +
+                    "  |> filter(fn: (r) => r[\"_measurement\"] == \"00000000-0000-0000-0000-000000000000\")\n" +
+                    "  |> filter(fn: (r) => r[\"node\"] == \"00000000-0000-0000-0000-000000000001\")\n" +
+                    "  |> group(columns: [\"_field\"])\n" +
+                    "  |> sort(columns: [\"_time\"], desc: true)\n" +
+                    "  |> limit(n: 1)",
+                "organization",
+            )
+        }
+    }
+}