fix: update log references to onStart onStop

pelotoncycle · Oct 6, 2024 · 9808b73 · 9808b73
1 parent 76ebb7a
commit 9808b73
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/src/main/kotlin/com/onepeloton/locust4k/LocustWorker.kt b/src/main/kotlin/com/onepeloton/locust4k/LocustWorker.kt
@@ -333,11 +333,11 @@ class LocustWorker(
                             try {
                                 taskInstance.beforeExecuteLoop(taskContext)
                             } catch (e: CancellationException) {
-                                logger.info { "Task (${taskInstance.name()}) cancelled (onStart)" }
+                                logger.info { "Task (${taskInstance.name()}) cancelled (beforeExecuteLoop)" }
                                 return@launch
                             } catch (e: Exception) {
-                                logger.error(e) { "Task (${taskInstance.name()}) onStart exception caught" }
-                                stats.failure(0, e.message ?: "", taskInstance.name(), "onStart")
+                                logger.error(e) { "Task (${taskInstance.name()}) beforeExecuteLoop exception caught" }
+                                stats.failure(0, e.message ?: "", taskInstance.name(), "beforeExecuteLoop")
                                 return@launch
                             }
                             try {
@@ -360,9 +360,9 @@ class LocustWorker(
                                 try {
                                     taskInstance.afterExecuteLoop(taskContext)
                                 } catch (e: CancellationException) {
-                                    logger.info { "Task (${taskInstance.name()}) cancelled (onStop)" }
+                                    logger.info { "Task (${taskInstance.name()}) cancelled (afterExecuteLoop)" }
                                 } catch (e: Exception) {
-                                    logger.warn(e) { "Task (${taskInstance.name()}) onStop exception caught" }
+                                    logger.warn(e) { "Task (${taskInstance.name()}) afterExecuteLoop exception caught" }
                                 }
                             }
                         } finally {

diff --git a/src/main/kotlin/com/onepeloton/locust4k/examples/ExampleApp.kt b/src/main/kotlin/com/onepeloton/locust4k/examples/ExampleApp.kt
@@ -48,11 +48,11 @@ private class ExampleLocustTask : LocustTask {
     override fun name(): String = "exampleTask"
 
     override suspend fun beforeExecuteLoop(context: CoroutineContext) {
-        logger.info { "onStart invoked" }
+        logger.info { "beforeExecuteLoop invoked" }
     }
 
     override suspend fun afterExecuteLoop(context: CoroutineContext) {
-        logger.info { "onStop invoked" }
+        logger.info { "afterExecuteLoop invoked" }
     }
 
     override suspend fun execute(

diff --git a/src/test/kotlin/com/onepeloton/locust4k/examples/ExampleAppTest.kt b/src/test/kotlin/com/onepeloton/locust4k/examples/ExampleAppTest.kt
@@ -30,11 +30,11 @@ class ExampleAppTest : TestcontainersBase() {
                     "Spawn message from controller",
                     "Stats reset",
                     "+1 (-0) user-units",
-                    "onStart invoked",
+                    "beforeExecuteLoop invoked",
                     "Spawning Complete",
                     "Quit message",
                     "Task (exampleTask) cancelled",
-                    "onStop invoked",
+                    "afterExecuteLoop invoked",
                     "Stats consumer stopped",
                     "Stats reporter stopped",
                     "Shutting down Locust Worker",