[minor] initial Scheduler POC

copper-leaf · Feb 6, 2024 · 7a23838 · 7a23838
1 parent 37ff255
commit 7a23838
Show file tree

Hide file tree

Showing 109 changed files with 5,011 additions and 411 deletions.
diff --git a/ballast-api/api/android/ballast-api.api b/ballast-api/api/android/ballast-api.api
@@ -391,18 +391,21 @@ public final class com/copperleaf/ballast/Queued$HandleInput : com/copperleaf/ba
 	public fun <init> (Lkotlinx/coroutines/CompletableDeferred;Ljava/lang/Object;)V
 	public fun getDeferred ()Lkotlinx/coroutines/CompletableDeferred;
 	public final fun getInput ()Ljava/lang/Object;
+	public fun toString ()Ljava/lang/String;
 }
 
 public final class com/copperleaf/ballast/Queued$RestoreState : com/copperleaf/ballast/Queued {
 	public fun <init> (Lkotlinx/coroutines/CompletableDeferred;Ljava/lang/Object;)V
 	public fun getDeferred ()Lkotlinx/coroutines/CompletableDeferred;
 	public final fun getState ()Ljava/lang/Object;
+	public fun toString ()Ljava/lang/String;
 }
 
 public final class com/copperleaf/ballast/Queued$ShutDownGracefully : com/copperleaf/ballast/Queued {
 	public synthetic fun <init> (Lkotlinx/coroutines/CompletableDeferred;JLkotlin/jvm/internal/DefaultConstructorMarker;)V
 	public fun getDeferred ()Lkotlinx/coroutines/CompletableDeferred;
 	public final fun getGracePeriod-UwyO8pc ()J
+	public fun toString ()Ljava/lang/String;
 }
 
 public abstract interface class com/copperleaf/ballast/SideJobScope : kotlinx/coroutines/CoroutineScope {
@@ -597,6 +600,23 @@ public final class com/copperleaf/ballast/internal/BallastViewModelImpl : com/co
 	public fun trySend-JP2dKIU (Ljava/lang/Object;)Ljava/lang/Object;
 }
 
+public abstract interface class com/copperleaf/ballast/internal/RestartableJob : kotlinx/coroutines/Job {
+	public abstract fun restart (Ljava/lang/Object;)V
+}
+
+public final class com/copperleaf/ballast/internal/RestartableJob$DefaultImpls {
+	public static synthetic fun cancel (Lcom/copperleaf/ballast/internal/RestartableJob;)V
+	public static fun fold (Lcom/copperleaf/ballast/internal/RestartableJob;Ljava/lang/Object;Lkotlin/jvm/functions/Function2;)Ljava/lang/Object;
+	public static fun get (Lcom/copperleaf/ballast/internal/RestartableJob;Lkotlin/coroutines/CoroutineContext$Key;)Lkotlin/coroutines/CoroutineContext$Element;
+	public static fun minusKey (Lcom/copperleaf/ballast/internal/RestartableJob;Lkotlin/coroutines/CoroutineContext$Key;)Lkotlin/coroutines/CoroutineContext;
+	public static fun plus (Lcom/copperleaf/ballast/internal/RestartableJob;Lkotlin/coroutines/CoroutineContext;)Lkotlin/coroutines/CoroutineContext;
+	public static fun plus (Lcom/copperleaf/ballast/internal/RestartableJob;Lkotlinx/coroutines/Job;)Lkotlinx/coroutines/Job;
+}
+
+public final class com/copperleaf/ballast/internal/RestartableJobKt {
+	public static final fun restartableJob (Lkotlinx/coroutines/CoroutineScope;Lkotlin/jvm/functions/Function3;)Lcom/copperleaf/ballast/internal/RestartableJob;
+}
+
 public abstract interface class com/copperleaf/ballast/internal/Status {
 	public abstract fun checkCanClear ()V
 	public abstract fun checkCanShutDown ()V

diff --git a/ballast-api/api/jvm/ballast-api.api b/ballast-api/api/jvm/ballast-api.api
@@ -391,18 +391,21 @@ public final class com/copperleaf/ballast/Queued$HandleInput : com/copperleaf/ba
 	public fun <init> (Lkotlinx/coroutines/CompletableDeferred;Ljava/lang/Object;)V
 	public fun getDeferred ()Lkotlinx/coroutines/CompletableDeferred;
 	public final fun getInput ()Ljava/lang/Object;
+	public fun toString ()Ljava/lang/String;
 }
 
 public final class com/copperleaf/ballast/Queued$RestoreState : com/copperleaf/ballast/Queued {
 	public fun <init> (Lkotlinx/coroutines/CompletableDeferred;Ljava/lang/Object;)V
 	public fun getDeferred ()Lkotlinx/coroutines/CompletableDeferred;
 	public final fun getState ()Ljava/lang/Object;
+	public fun toString ()Ljava/lang/String;
 }
 
 public final class com/copperleaf/ballast/Queued$ShutDownGracefully : com/copperleaf/ballast/Queued {
 	public synthetic fun <init> (Lkotlinx/coroutines/CompletableDeferred;JLkotlin/jvm/internal/DefaultConstructorMarker;)V
 	public fun getDeferred ()Lkotlinx/coroutines/CompletableDeferred;
 	public final fun getGracePeriod-UwyO8pc ()J
+	public fun toString ()Ljava/lang/String;
 }
 
 public abstract interface class com/copperleaf/ballast/SideJobScope : kotlinx/coroutines/CoroutineScope {
@@ -597,6 +600,23 @@ public final class com/copperleaf/ballast/internal/BallastViewModelImpl : com/co
 	public fun trySend-JP2dKIU (Ljava/lang/Object;)Ljava/lang/Object;
 }
 
+public abstract interface class com/copperleaf/ballast/internal/RestartableJob : kotlinx/coroutines/Job {
+	public abstract fun restart (Ljava/lang/Object;)V
+}
+
+public final class com/copperleaf/ballast/internal/RestartableJob$DefaultImpls {
+	public static synthetic fun cancel (Lcom/copperleaf/ballast/internal/RestartableJob;)V
+	public static fun fold (Lcom/copperleaf/ballast/internal/RestartableJob;Ljava/lang/Object;Lkotlin/jvm/functions/Function2;)Ljava/lang/Object;
+	public static fun get (Lcom/copperleaf/ballast/internal/RestartableJob;Lkotlin/coroutines/CoroutineContext$Key;)Lkotlin/coroutines/CoroutineContext$Element;
+	public static fun minusKey (Lcom/copperleaf/ballast/internal/RestartableJob;Lkotlin/coroutines/CoroutineContext$Key;)Lkotlin/coroutines/CoroutineContext;
+	public static fun plus (Lcom/copperleaf/ballast/internal/RestartableJob;Lkotlin/coroutines/CoroutineContext;)Lkotlin/coroutines/CoroutineContext;
+	public static fun plus (Lcom/copperleaf/ballast/internal/RestartableJob;Lkotlinx/coroutines/Job;)Lkotlinx/coroutines/Job;
+}
+
+public final class com/copperleaf/ballast/internal/RestartableJobKt {
+	public static final fun restartableJob (Lkotlinx/coroutines/CoroutineScope;Lkotlin/jvm/functions/Function3;)Lcom/copperleaf/ballast/internal/RestartableJob;
+}
+
 public abstract interface class com/copperleaf/ballast/internal/Status {
 	public abstract fun checkCanClear ()V
 	public abstract fun checkCanShutDown ()V

diff --git a/ballast-api/src/commonMain/kotlin/com/copperleaf/ballast/Queued.kt b/ballast-api/src/commonMain/kotlin/com/copperleaf/ballast/Queued.kt
@@ -19,15 +19,23 @@ public sealed class Queued<Inputs : Any, Events : Any, State : Any> {
     public class RestoreState<Inputs : Any, Events : Any, State : Any>(
         public override val deferred: CompletableDeferred<Unit>?,
         public val state: State,
-    ) : Queued<Inputs, Events, State>()
+    ) : Queued<Inputs, Events, State>() {
+        override fun toString(): String {
+            return "RestoreState(state=$state)"
+        }
+    }
 
     /**
      * A request to handle an Input by sending it to the [InputHandler].
      */
     public class HandleInput<Inputs : Any, Events : Any, State : Any>(
         public override val deferred: CompletableDeferred<Unit>?,
         public val input: Inputs,
-    ) : Queued<Inputs, Events, State>()
+    ) : Queued<Inputs, Events, State>() {
+        override fun toString(): String {
+            return "HandleInput(input=$input)"
+        }
+    }
 
     /**
      * A request to gracefully shut down the ViewModel. This will attempt to let it finish processing any Inputs
@@ -42,5 +50,9 @@ public sealed class Queued<Inputs : Any, Events : Any, State : Any> {
     public class ShutDownGracefully<Inputs : Any, Events : Any, State : Any>(
         public override val deferred: CompletableDeferred<Unit>?,
         public val gracePeriod: Duration,
-    ) : Queued<Inputs, Events, State>()
+    ) : Queued<Inputs, Events, State>() {
+        override fun toString(): String {
+            return "ShutDownGracefully(gracePeriod=$gracePeriod)"
+        }
+    }
 }
diff --git a/ballast-api/src/commonMain/kotlin/com/copperleaf/ballast/internal/RestartableJob.kt b/ballast-api/src/commonMain/kotlin/com/copperleaf/ballast/internal/RestartableJob.kt
@@ -0,0 +1,26 @@
+package com.copperleaf.ballast.internal
+
+import kotlinx.coroutines.CoroutineScope
+import kotlinx.coroutines.Job
+import kotlinx.coroutines.channels.Channel
+import kotlinx.coroutines.flow.collectLatest
+import kotlinx.coroutines.flow.consumeAsFlow
+import kotlinx.coroutines.launch
+
+public interface RestartableJob<T> : Job {
+    public fun restart(context: T)
+}
+
+public fun <T> CoroutineScope.restartableJob(
+    block: suspend CoroutineScope.(T) -> Unit,
+): RestartableJob<T> {
+    val channel = Channel<T>(Channel.CONFLATED)
+    val job = launch {
+        channel.consumeAsFlow().collectLatest { block(it) }
+    }
+    return object : RestartableJob<T>, Job by job {
+        override fun restart(context: T) {
+            channel.trySend(context)
+        }
+    }
+}