Merge pull request #2 from rhappe/async-sum-util

Add utility for calculating sums async and blocking

Author: jsoberg

build.gradle.kts

@@ -7,6 +7,7 @@ group = Publishing.GroupId
 version = Publishing.Version
 
 dependencies {
+    implementation(libs.kotlin.coroutines)
     implementation(libs.kotlin.stdlib)
     implementation(libs.soberg.aoc.api)
 

gradle/libs.versions.toml

@@ -1,6 +1,6 @@
 [versions]
 assertk = "0.28.1"
-kotlin = "2.0.21"
+kotlin = "2.1.0"
 kotlinCoroutines = "1.9.0"
 junitJupiter = "5.11.3"
 mockk = "1.13.13"

src/main/kotlin/com/soberg/aoc/utlities/extensions/AsyncSum.kt

@@ -0,0 +1,56 @@
+package com.soberg.aoc.utlities.extensions
+
+import kotlinx.coroutines.*
+import java.util.concurrent.atomic.AtomicInteger
+import java.util.concurrent.atomic.AtomicLong
+import kotlin.experimental.ExperimentalTypeInference
+
+@OptIn(ExperimentalTypeInference::class)
+@OverloadResolutionByLambdaReturnType
+@JvmName("asyncSumOfIntBlocking")
+inline fun <T> Iterable<T>.asyncSumOfBlocking(
+    dispatcher: CoroutineDispatcher = Dispatchers.Default,
+    crossinline selector: (T) -> Int,
+): Int = runBlocking(dispatcher) {
+    asyncSumOf(selector)
+}
+
+@OptIn(ExperimentalTypeInference::class)
+@OverloadResolutionByLambdaReturnType
+@JvmName("asyncSumOfInt")
+suspend inline fun <T> Iterable<T>.asyncSumOf(crossinline selector: (T) -> Int): Int {
+    val sum = AtomicInteger()
+    coroutineScope {
+        forEach {
+            launch {
+                sum.getAndAdd(selector(it))
+            }
+        }
+    }
+    return sum.get()
+}
+
+@OptIn(ExperimentalTypeInference::class)
+@OverloadResolutionByLambdaReturnType
+@JvmName("asyncSumOfLongBlocking")
+inline fun <T> Iterable<T>.asyncSumOfBlocking(
+    dispatcher: CoroutineDispatcher = Dispatchers.Default,
+    crossinline selector: (T) -> Long,
+): Long = runBlocking(dispatcher){
+    asyncSumOf(selector)
+}
+
+@OptIn(ExperimentalTypeInference::class)
+@OverloadResolutionByLambdaReturnType
+@JvmName("asyncSumOfLong")
+suspend inline fun <T> Iterable<T>.asyncSumOf(crossinline selector: (T) -> Long): Long {
+    val sum = AtomicLong()
+    coroutineScope {
+        forEach {
+            launch {
+                sum.getAndAdd(selector(it))
+            }
+        }
+    }
+    return sum.get()
+}

src/test/kotlin/com/soberg/aoc/utlities/extensions/AsyncSumTest.kt

@@ -0,0 +1,46 @@
+package com.soberg.aoc.utlities.extensions
+
+import assertk.assertThat
+import assertk.assertions.isEqualTo
+import kotlinx.coroutines.test.runTest
+import org.junit.jupiter.api.Test
+
+class AsyncSumTest {
+    private val testData = "The quick brown fox jumped over the lazy dog".split(" ")
+
+    @Test
+    fun `return sum of integers - blocking`() {
+        val letterCount = testData.asyncSumOfBlocking {
+            it.length
+        }
+
+        assertThat(letterCount).isEqualTo(36)
+    }
+
+    @Test
+    fun `return sum of integers - coroutine`() = runTest{
+        val letterCount = testData.asyncSumOf {
+            it.length
+        }
+
+        assertThat(letterCount).isEqualTo(36)
+    }
+
+    @Test
+    fun `return sum of longs - blocking`() {
+        val letterCount = testData.asyncSumOfBlocking {
+            it.length.toLong()
+        }
+
+        assertThat(letterCount).isEqualTo(36)
+    }
+
+    @Test
+    fun `return sum of longs - coroutine`() = runTest{
+        val letterCount = testData.asyncSumOf {
+            it.length.toLong()
+        }
+
+        assertThat(letterCount).isEqualTo(36)
+    }
+}