Introduce EngineProcessTask

Author: IlyaMuravjov

utbot-framework/src/main/kotlin/org/utbot/framework/process/EngineProcessMain.kt

@@ -273,6 +273,11 @@ private fun EngineProcessModel.setup(kryoHelper: KryoHelper, watchdog: IdleWatch
         }
         GenerateTestReportResult(notifyMessage, statistics, hasWarnings)
     }
+    watchdog.measureTimeForActiveCall(perform, "Performing dynamic task") { params ->
+        val task = kryoHelper.readObject<EngineProcessTask<Any?>>(params.engineProcessTask)
+        val result = task.perform(kryoHelper)
+        kryoHelper.writeObject(result)
+    }
 }
 
 private fun processInitialWarnings(report: TestsGenerationReport, params: GenerateTestReportArgs) {

utbot-framework/src/main/kotlin/org/utbot/framework/process/EngineProcessTask.kt

@@ -0,0 +1,16 @@
+package org.utbot.framework.process
+
+import org.utbot.framework.process.kryo.KryoHelper
+
+/**
+ * Implementations of this interface can be passed to engine process for execution and should
+ * be used for adding feature-specific (e.g. Spring-specific) tasks without inflating core UtBot codebase.
+ *
+ * Such tasks are serialised with kryo when passed between processes, meaning that for successful execution same
+ * implementation of [EngineProcessTask] should be present on the classpath of both parent process and engine process.
+ *
+ * @param R result type of the task (should be present on the classpath of both processes).
+ */
+interface EngineProcessTask<R> {
+    fun perform(kryoHelper: KryoHelper): R
+}

utbot-framework/src/main/kotlin/org/utbot/framework/process/generated/EngineProcessModel.Generated.kt

@@ -30,7 +30,8 @@ class EngineProcessModel private constructor(
     private val _findMethodsInClassMatchingSelected: RdCall<FindMethodsInClassMatchingSelectedArguments, FindMethodsInClassMatchingSelectedResult>,
     private val _findMethodParamNames: RdCall<FindMethodParamNamesArguments, FindMethodParamNamesResult>,
     private val _writeSarifReport: RdCall<WriteSarifReportArguments, String>,
-    private val _generateTestReport: RdCall<GenerateTestReportArgs, GenerateTestReportResult>
+    private val _generateTestReport: RdCall<GenerateTestReportArgs, GenerateTestReportResult>,
+    private val _perform: RdCall<PerformParams, ByteArray>
 ) : RdExtBase() {
     //companion
 
@@ -58,6 +59,7 @@ class EngineProcessModel private constructor(
             serializers.register(BeanAdditionalData)
             serializers.register(BeanDefinitionData)
             serializers.register(SpringAnalyzerResult)
+            serializers.register(PerformParams)
         }
 
 
@@ -78,7 +80,7 @@ class EngineProcessModel private constructor(
         }
 
 
-        const val serializationHash = -271702162812013897L
+        const val serializationHash = 1643215834533745881L
 
     }
     override val serializersOwner: ISerializersOwner get() = EngineProcessModel
@@ -97,6 +99,7 @@ class EngineProcessModel private constructor(
     val findMethodParamNames: RdCall<FindMethodParamNamesArguments, FindMethodParamNamesResult> get() = _findMethodParamNames
     val writeSarifReport: RdCall<WriteSarifReportArguments, String> get() = _writeSarifReport
     val generateTestReport: RdCall<GenerateTestReportArgs, GenerateTestReportResult> get() = _generateTestReport
+    val perform: RdCall<PerformParams, ByteArray> get() = _perform
     //methods
     //initializer
     init {
@@ -112,6 +115,7 @@ class EngineProcessModel private constructor(
         _findMethodParamNames.async = true
         _writeSarifReport.async = true
         _generateTestReport.async = true
+        _perform.async = true
     }
 
     init {
@@ -127,6 +131,7 @@ class EngineProcessModel private constructor(
         bindableChildren.add("findMethodParamNames" to _findMethodParamNames)
         bindableChildren.add("writeSarifReport" to _writeSarifReport)
         bindableChildren.add("generateTestReport" to _generateTestReport)
+        bindableChildren.add("perform" to _perform)
     }
 
     //secondary constructor
@@ -143,7 +148,8 @@ class EngineProcessModel private constructor(
         RdCall<FindMethodsInClassMatchingSelectedArguments, FindMethodsInClassMatchingSelectedResult>(FindMethodsInClassMatchingSelectedArguments, FindMethodsInClassMatchingSelectedResult),
         RdCall<FindMethodParamNamesArguments, FindMethodParamNamesResult>(FindMethodParamNamesArguments, FindMethodParamNamesResult),
         RdCall<WriteSarifReportArguments, String>(WriteSarifReportArguments, FrameworkMarshallers.String),
-        RdCall<GenerateTestReportArgs, GenerateTestReportResult>(GenerateTestReportArgs, GenerateTestReportResult)
+        RdCall<GenerateTestReportArgs, GenerateTestReportResult>(GenerateTestReportArgs, GenerateTestReportResult),
+        RdCall<PerformParams, ByteArray>(PerformParams, FrameworkMarshallers.ByteArray)
     )
 
     //equals trait
@@ -164,6 +170,7 @@ class EngineProcessModel private constructor(
             print("findMethodParamNames = "); _findMethodParamNames.print(printer); println()
             print("writeSarifReport = "); _writeSarifReport.print(printer); println()
             print("generateTestReport = "); _generateTestReport.print(printer); println()
+            print("perform = "); _perform.print(printer); println()
         }
         printer.print(")")
     }
@@ -181,7 +188,8 @@ class EngineProcessModel private constructor(
             _findMethodsInClassMatchingSelected.deepClonePolymorphic(),
             _findMethodParamNames.deepClonePolymorphic(),
             _writeSarifReport.deepClonePolymorphic(),
-            _generateTestReport.deepClonePolymorphic()
+            _generateTestReport.deepClonePolymorphic(),
+            _perform.deepClonePolymorphic()
         )
     }
     //contexts
@@ -1099,6 +1107,63 @@ data class MethodDescription (
 }
 
 
+/**
+ * #### Generated from [EngineProcessModel.kt:149]
+ */
+data class PerformParams (
+    val engineProcessTask: ByteArray
+) : IPrintable {
+    //companion
+    
+    companion object : IMarshaller<PerformParams> {
+        override val _type: KClass<PerformParams> = PerformParams::class
+        
+        @Suppress("UNCHECKED_CAST")
+        override fun read(ctx: SerializationCtx, buffer: AbstractBuffer): PerformParams  {
+            val engineProcessTask = buffer.readByteArray()
+            return PerformParams(engineProcessTask)
+        }
+        
+        override fun write(ctx: SerializationCtx, buffer: AbstractBuffer, value: PerformParams)  {
+            buffer.writeByteArray(value.engineProcessTask)
+        }
+        
+        
+    }
+    //fields
+    //methods
+    //initializer
+    //secondary constructor
+    //equals trait
+    override fun equals(other: Any?): Boolean  {
+        if (this === other) return true
+        if (other == null || other::class != this::class) return false
+        
+        other as PerformParams
+        
+        if (!(engineProcessTask contentEquals other.engineProcessTask)) return false
+        
+        return true
+    }
+    //hash code trait
+    override fun hashCode(): Int  {
+        var __r = 0
+        __r = __r*31 + engineProcessTask.contentHashCode()
+        return __r
+    }
+    //pretty print
+    override fun print(printer: PrettyPrinter)  {
+        printer.println("PerformParams (")
+        printer.indent {
+            print("engineProcessTask = "); engineProcessTask.print(printer); println()
+        }
+        printer.print(")")
+    }
+    //deepClone
+    //contexts
+}
+
+
 /**
  * #### Generated from [EngineProcessModel.kt:69]
  */

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/process/EngineProcess.kt

@@ -24,6 +24,7 @@ import org.utbot.framework.plugin.api.BeanDefinitionData
 import org.utbot.framework.plugin.services.JdkInfo
 import org.utbot.framework.plugin.services.WorkingDirService
 import org.utbot.framework.process.AbstractRDProcessCompanion
+import org.utbot.framework.process.EngineProcessTask
 import org.utbot.framework.process.generated.*
 import org.utbot.framework.process.generated.MethodDescription
 import org.utbot.framework.process.kryo.KryoHelper
@@ -416,6 +417,13 @@ class EngineProcess private constructor(val project: Project, private val classN
         return Triple(result.notifyMessage, result.statistics, result.hasWarnings)
     }
 
+    fun <R> perform(engineProcessTask: EngineProcessTask<R>): R {
+        assertReadAccessNotAllowed()
+        return kryoHelper.readObject(
+            engineModel.perform.startBlocking(PerformParams(kryoHelper.writeObject(engineProcessTask)))
+        )
+    }
+
     init {
         lifetime.onTermination {
             protocol.synchronizationModel.stopProcess.fire(Unit)

utbot-rd/src/main/rdgen/org/utbot/rd/models/EngineProcessModel.kt

@@ -146,6 +146,9 @@ object EngineProcessModel : Ext(EngineProcessRoot) {
     val springAnalyzerResult = structdef {
         field("beanDefinitions", array(beanDefinitionData))
     }
+    val performParams = structdef {
+        field("engineProcessTask", array(PredefinedType.byte))
+    }
 
     init {
         call("setupUtContext", setupContextParams, PredefinedType.void).async
@@ -160,5 +163,6 @@ object EngineProcessModel : Ext(EngineProcessRoot) {
         call("findMethodParamNames", findMethodParamNamesArguments, findMethodParamNamesResult).async
         call("writeSarifReport", writeSarifReportArguments, PredefinedType.string).async
         call("generateTestReport", generateTestReportArgs, generateTestReportResult).async
+        call("perform", performParams, array(PredefinedType.byte)).async
     }
 }