EPMDJ-3407 Fixed restoring of build diff (parent build reference) aft…

…er server restart.
Drill4J · Sep 7, 2020 · b1940f4 · b1940f4
1 parent 3f39396
commit b1940f4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/core/src/main/kotlin/agent/State.kt b/core/src/main/kotlin/agent/State.kt
@@ -52,11 +52,17 @@ internal class AgentData(
     )
 
     suspend fun initBuild(version: String) {
-        val agentBuild = buildManager.init(version)
-        store(agentBuild)
+        if (buildManager.builds.none()) {
+            loadStoredData()
+        }
+        buildManager[version] ?: run {
+            val agentBuild = buildManager.init(version)
+            store(agentBuild)
+        }
     }
 
     internal suspend fun initClasses() {
+        kotlin.runCatching {  }
         val addedClasses: List<ByteArray> = buildManager.collectClasses()
         logger.debug { "Saving ${addedClasses.size} classes..." }
         measureTime {
@@ -102,7 +108,7 @@ internal class AgentData(
         logger.debug { "Saved build ${agentBuild.id}." }
     }
 
-    suspend fun loadStoredData() = storeClient.findById<AgentDataSummary>(agentId)?.let { summary ->
+    private suspend fun loadStoredData() = storeClient.findById<AgentDataSummary>(agentId)?.let { summary ->
         logger.debug { "Loading data for $agentId..." }
         _settings.value = summary.settings
         val classBytes: Map<String, ByteArray> = storeClient.findById<StoredCodeData>(agentId)?.let {

diff --git a/core/src/main/kotlin/endpoints/AgentManager.kt b/core/src/main/kotlin/endpoints/AgentManager.kt
@@ -85,7 +85,6 @@ class AgentManager(override val kodein: Kodein) : KodeinAware {
             val info = storedInfo ?: config.toAgentInfo()
             val entry = AgentEntry(info, session)
             agentStorage.put(id, entry)
-            adminData.loadStoredData()
             storedInfo?.initPlugins(entry)
             app.launch {
                 storedInfo?.sync(needSync) // sync only existing info!