add coroutine versions of asyncSumOf

Author: rhappe

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

@@ -1,22 +1,26 @@
 package com.soberg.aoc.utlities.extensions
 
-import kotlinx.coroutines.CoroutineDispatcher
-import kotlinx.coroutines.Dispatchers
-import kotlinx.coroutines.launch
-import kotlinx.coroutines.runBlocking
+import kotlinx.coroutines.*
 import java.util.concurrent.atomic.AtomicInteger
 import java.util.concurrent.atomic.AtomicLong
 import kotlin.experimental.ExperimentalTypeInference
 
 @OptIn(ExperimentalTypeInference::class)
 @OverloadResolutionByLambdaReturnType
-@JvmName("asyncSumOfInt")
-inline fun <T> Iterable<T>.asyncSumOf(
+@JvmName("asyncSumOfIntBlocking")
+inline fun <T> Iterable<T>.asyncSumOfBlocking(
     dispatcher: CoroutineDispatcher = Dispatchers.Default,
     crossinline selector: (T) -> Int,
-): 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()
-    runBlocking(dispatcher) {
+    coroutineScope {
         forEach {
             launch {
                 sum.getAndAdd(selector(it))
@@ -28,13 +32,20 @@ inline fun <T> Iterable<T>.asyncSumOf(
 
 @OptIn(ExperimentalTypeInference::class)
 @OverloadResolutionByLambdaReturnType
-@JvmName("asyncSumOfLong")
-inline fun <T> Iterable<T>.asyncSumOf(
+@JvmName("asyncSumOfLongBlocking")
+inline fun <T> Iterable<T>.asyncSumOfBlocking(
     dispatcher: CoroutineDispatcher = Dispatchers.Default,
     crossinline selector: (T) -> Long,
-): 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()
-    runBlocking(dispatcher) {
+    coroutineScope {
         forEach {
             launch {
                 sum.getAndAdd(selector(it))

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

@@ -2,13 +2,23 @@ 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`() {
+    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
         }
@@ -17,7 +27,16 @@ class AsyncSumTest {
     }
 
     @Test
-    fun `return sum of longs`() {
+    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()
         }