SchedulerTaskContext.afterTask() extension function

Author: elizarov

common/kotlinx-coroutines-core-common/src/Dispatched.kt

@@ -239,7 +239,7 @@ internal abstract class DispatchedTask<in T>(
         } catch (e: Throwable) {
             throw DispatchException("Unexpected exception running $this", e)
         } finally {
-            afterTask(taskContext)
+            taskContext.afterTask()
         }
     }
 }

common/kotlinx-coroutines-core-common/src/SchedulerTask.common.kt

@@ -10,4 +10,4 @@ internal expect interface SchedulerTaskContext
 
 internal expect val SchedulerTask.taskContext: SchedulerTaskContext
 
-internal expect inline fun SchedulerTask.afterTask(taskContext: SchedulerTaskContext)
+internal expect inline fun SchedulerTaskContext.afterTask()

core/kotlinx-coroutines-core/src/SchedulerTask.kt

@@ -13,6 +13,6 @@ internal actual typealias SchedulerTaskContext = TaskContext
 @Suppress("EXTENSION_SHADOWED_BY_MEMBER")
 internal actual val SchedulerTask.taskContext: SchedulerTaskContext get() = taskContext
 
-@Suppress("NOTHING_TO_INLINE")
-internal actual inline fun SchedulerTask.afterTask(taskContext: SchedulerTaskContext) =
-    taskContext.afterTask()
+@Suppress("NOTHING_TO_INLINE", "EXTENSION_SHADOWED_BY_MEMBER")
+internal actual inline fun SchedulerTaskContext.afterTask() =
+    afterTask()

core/kotlinx-coroutines-core/src/scheduling/Tasks.kt

@@ -100,7 +100,7 @@ internal class TaskImpl(
         try {
             block.run()
         } finally {
-            afterTask(taskContext)
+            taskContext.afterTask()
         }
     }
 

js/kotlinx-coroutines-core-js/src/SchedulerTask.kt

@@ -12,5 +12,5 @@ internal actual typealias SchedulerTaskContext = Unit
 internal actual val SchedulerTask.taskContext: SchedulerTaskContext get() = Unit
 
 @Suppress("NOTHING_TO_INLINE")
-internal actual inline fun SchedulerTask.afterTask(taskContext: SchedulerTaskContext) {}
+internal actual inline fun SchedulerTaskContext.afterTask() {}
 

native/kotlinx-coroutines-core-native/src/SchedulerTask.kt

@@ -12,4 +12,4 @@ internal actual typealias SchedulerTaskContext = Unit
 internal actual val SchedulerTask.taskContext: SchedulerTaskContext get() = kotlin.Unit
 
 @Suppress("NOTHING_TO_INLINE")
-internal actual inline fun SchedulerTask.afterTask(taskContext: SchedulerTaskContext) {}
+internal actual inline fun SchedulerTaskContext.afterTask() {}