feat: update for context

unit-mesh · Dec 18, 2023 · ad760f0 · ad760f0
1 parent 104ed32
commit ad760f0
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/WorkerManager.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/worker/WorkerManager.kt
@@ -5,6 +5,7 @@ import cc.unitmesh.pick.prompt.Instruction
 import cc.unitmesh.pick.worker.worker.JavaWorker
 import cc.unitmesh.pick.worker.worker.TypescriptWorker
 import org.archguard.rule.common.Language
+import org.slf4j.Logger
 
 class WorkerManager(workerContext: WorkerContext) {
     private val workers: Map<Language, LangWorker> = mapOf(
@@ -13,17 +14,30 @@ class WorkerManager(workerContext: WorkerContext) {
         Language.JAVASCRIPT to TypescriptWorker(workerContext),
     )
 
+    private val logger: Logger = org.slf4j.LoggerFactory.getLogger(WorkerManager::class.java)
+
     fun addJob(job: InstructionFileJob) {
-        if (job.fileSummary.binary || job.fileSummary.generated || job.fileSummary.weightedComplexity > 1000) {
+        if (job.fileSummary.complexity > 100) {
+            logger.info("skip file ${job.fileSummary.location} for complexity ${job.fileSummary.complexity}")
+            return;
+        }
+        if (job.fileSummary.binary || job.fileSummary.generated || job.fileSummary.minified) {
             return
         }
 
+        // if the file size is too large, we just try 64k
+        if (job.fileSummary.bytes > 1024 * 64) {
+            logger.info("skip file ${job.fileSummary.location} for size ${job.fileSummary.bytes}")
+            return;
+        }
+
         val language = job.fileSummary.language.toSupportLanguage()
-        val worker = workers[language]
-        worker?.addJob(job)
+        val worker = workers[language] ?: return
+        logger.info("add file:" + job.fileSummary.location)
+        worker.addJob(job)
     }
 
-    suspend fun runAll() : List<Instruction> {
+    suspend fun runAll(): List<Instruction> {
         return workers.map { (_, worker) ->
             try {
                 worker.start()