docs: thinking in language worker

unit-mesh · Dec 6, 2023 · cf829d8 · cf829d8
1 parent 614ef7c
commit cf829d8
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 10 deletions.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/picker/CodePicker.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/picker/CodePicker.kt
@@ -1,15 +1,14 @@
 package cc.unitmesh.pick.picker
 
+import cc.unitmesh.pick.worker.WorkerDispatch
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.channels.Channel
-import kotlinx.coroutines.coroutineScope
 import kotlinx.coroutines.launch
 import kotlinx.coroutines.runBlocking
 import org.archguard.action.checkout.GitSourceSettings
 import org.archguard.action.checkout.executeGitCheckout
 import java.nio.file.Files
 import java.nio.file.Path
-import kotlin.coroutines.coroutineContext
 
 class CodePicker(val config: PickerConfig) {
     private val logger = org.slf4j.LoggerFactory.getLogger(javaClass)
@@ -31,13 +30,13 @@ class CodePicker(val config: PickerConfig) {
                 }
                 launch {
                     for (fileJob in walkdirChannel) {
-                        // todo call by language worker
-                        println(fileJob)
+                        WorkerDispatch.worker(fileJob)
                     }
                 }
             }
 
             logger.info("stop picker")
+
             // 3. generate tree to jsonl
         }
     }

diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/AbstractWorker.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/AbstractWorker.kt
@@ -1,5 +1,5 @@
 package cc.unitmesh.pick.worker
 
 abstract class AbstractWorker {
-    abstract suspend fun start(filePath: String)
+    abstract suspend fun start()
 }
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/JavaFileWorker.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/JavaFileWorker.kt
@@ -1,5 +1,6 @@
 package cc.unitmesh.pick.worker
 
+import cc.unitmesh.pick.picker.PickJob
 import kotlinx.coroutines.coroutineScope
 import org.archguard.scanner.analyser.count.FileJob
 
@@ -17,11 +18,11 @@ import org.archguard.scanner.analyser.count.FileJob
  * - by Horizontal (with Import File):
  * - by Vertical (with History Change):
  */
-class JavaFileWorker : AbstractWorker() {
+class JavaFileWorker(val job: PickJob) : AbstractWorker() {
     val packageTree: Map<String, FileJob> = mapOf()
     /// [Coco](https://github.com/inherd/coco) high change and long line, means is important file, and need to be checked.
 
-    override suspend fun start(filePath: String) = coroutineScope {
+    override suspend fun start() = coroutineScope {
         // 1. read directory to a collection of files for FileJob
 
         // 2. check package information from line 1?

diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/WorkerDispatch.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/WorkerDispatch.kt
@@ -1,12 +1,14 @@
 package cc.unitmesh.pick.worker
 
+import cc.unitmesh.pick.picker.PickJob
+
 object WorkerDispatch {
     /**
      * Dispatch language worker by language
      */
-    fun worker(language: String): AbstractWorker? {
-        return when (language) {
-            "Java" -> JavaFileWorker()
+    fun worker(job: PickJob): AbstractWorker? {
+        return when (job.language.lowercase()) {
+            "java" -> JavaFileWorker(job)
             else -> null
         }
     }