Update inline incremental compilation for compile daemon

compiler/rmi/kotlinr/kotlinr.iml

@@ -10,5 +10,6 @@
     <orderEntry type="library" name="kotlin-runtime" level="project" />
     <orderEntry type="module" module-name="rmi-interface" />
     <orderEntry type="module" module-name="frontend.java" />
+    <orderEntry type="module" module-name="util" />
   </component>
 </module>

compiler/rmi/kotlinr/src/org/jetbrains/kotlin/rmi/kotlinr/KotlinCompilerClient.kt

@@ -17,6 +17,7 @@
 package org.jetbrains.kotlin.rmi.kotlinr
 
 import org.jetbrains.kotlin.load.kotlin.incremental.components.IncrementalCache
+import org.jetbrains.kotlin.modules.Module
 import org.jetbrains.kotlin.rmi.*
 import java.io.File
 import java.io.OutputStream
@@ -175,10 +176,10 @@ public class KotlinCompilerClient {
             }
         }
 
-        public fun incrementalCompile(compiler: CompileService, args: Array<out String>, caches: Map<String, IncrementalCache>, out: OutputStream): Int {
+        public fun incrementalCompile(compiler: CompileService, args: Array<out String>, caches: Map<Module, IncrementalCache>, out: OutputStream): Int {
 
             val outStrm = RemoteOutputStreamServer(out)
-            val cacheServers = hashMapOf<String, RemoteIncrementalCacheServer>()
+            val cacheServers = hashMapOf<Module, RemoteIncrementalCacheServer>()
             try {
                 caches.mapValuesTo(cacheServers, { RemoteIncrementalCacheServer( it.getValue()) })
                 return compiler.remoteIncrementalCompile(args, cacheServers, outStrm, CompileService.OutputFormat.XML)

compiler/rmi/kotlinr/src/org/jetbrains/kotlin/rmi/kotlinr/RemoteIncrementalCacheServer.kt

@@ -17,6 +17,7 @@
 package org.jetbrains.kotlin.rmi.kotlinr
 
 import org.jetbrains.kotlin.load.kotlin.incremental.components.IncrementalCache
+import org.jetbrains.kotlin.load.kotlin.incremental.components.InlineRegistering
 import org.jetbrains.kotlin.rmi.CompileService
 import java.rmi.server.UnicastRemoteObject
 
@@ -33,6 +34,10 @@ public class RemoteIncrementalCacheServer(val cache: IncrementalCache) : Compile
 
     override fun getModuleMappingData(): ByteArray? = cache.getModuleMappingData()
 
+    override fun getInlineRegistering(): InlineRegistering = cache.getInlineRegistering()
+
+    override fun getClassFilePath(internalClassName: String): String = cache.getClassFilePath(internalClassName)
+
     override fun close() {
         cache.close()
     }

compiler/rmi/rmi-interface/rmi-interface.iml

@@ -10,5 +10,6 @@
     <orderEntry type="library" name="kotlin-runtime" level="project" />
     <orderEntry type="module" module-name="cli-common" />
     <orderEntry type="module" module-name="frontend.java" />
+    <orderEntry type="module" module-name="util" />
   </component>
 </module>

compiler/rmi/rmi-interface/src/org/jetbrains/kotlin/rmi/CompileService.kt

@@ -16,6 +16,8 @@
 
 package org.jetbrains.kotlin.rmi
 
+import org.jetbrains.kotlin.load.kotlin.incremental.components.InlineRegistering
+import org.jetbrains.kotlin.modules.Module
 import java.rmi.Remote
 import java.rmi.RemoteException
 
@@ -36,6 +38,12 @@ public interface CompileService : Remote {
         throws(RemoteException::class)
         public fun getModuleMappingData(): ByteArray?
 
+        throws(RemoteException::class)
+        public fun getInlineRegistering(): InlineRegistering
+
+        throws(RemoteException::class)
+        fun getClassFilePath(internalClassName: String): String
+
         throws(RemoteException::class)
         public fun close()
     }
@@ -55,7 +63,7 @@ public interface CompileService : Remote {
     throws(RemoteException::class)
     public fun remoteIncrementalCompile(
             args: Array<out String>,
-            caches: Map<String, RemoteIncrementalCache>,
+            caches: Map<Module, RemoteIncrementalCache>,
             outputStream: RemoteOutputStream,
             outputFormat: OutputFormat): Int
 }

compiler/rmi/rmi-server/src/org/jetbrains/kotlin/rmi/service/CompileServiceImpl.kt

@@ -22,6 +22,7 @@ import org.jetbrains.kotlin.config.Services
 import org.jetbrains.kotlin.incremental.components.LookupTracker
 import org.jetbrains.kotlin.load.kotlin.incremental.components.IncrementalCache
 import org.jetbrains.kotlin.load.kotlin.incremental.components.IncrementalCompilationComponents
+import org.jetbrains.kotlin.modules.Module
 import org.jetbrains.kotlin.rmi.*
 import org.jetbrains.kotlin.rmi.service.RemoteIncrementalCacheClient
 import org.jetbrains.kotlin.rmi.service.RemoteOutputStreamClient
@@ -80,14 +81,14 @@ class CompileServiceImpl<Compiler: CLICompiler<*>>(
         alive = true
     }
 
-    public class IncrementalCompilationComponentsImpl(val idToCache: Map<String, CompileService.RemoteIncrementalCache>): IncrementalCompilationComponents {
+    public class IncrementalCompilationComponentsImpl(val targetToCache: Map<Module, CompileService.RemoteIncrementalCache>): IncrementalCompilationComponents {
         // perf: cheap object, but still the pattern may be costly if there are too many calls to cache with the same id (which seems not to be the case now)
-        override fun getIncrementalCache(moduleId: String): IncrementalCache = RemoteIncrementalCacheClient(idToCache[moduleId]!!)
+        override fun getIncrementalCache(target: Module): IncrementalCache = RemoteIncrementalCacheClient(targetToCache[target]!!)
         // TODO: add appropriate proxy into interaction when lookup tracker is needed
         override fun getLookupTracker(): LookupTracker = LookupTracker.DO_NOTHING
     }
 
-    private fun createCompileServices(incrementalCaches: Map<String, CompileService.RemoteIncrementalCache>): Services =
+    private fun createCompileServices(incrementalCaches: Map<Module, CompileService.RemoteIncrementalCache>): Services =
         Services.Builder()
                 .register(IncrementalCompilationComponents::class.java, IncrementalCompilationComponentsImpl(incrementalCaches))
                 // TODO: add remote proxy for cancellation status tracking
@@ -186,7 +187,7 @@ class CompileServiceImpl<Compiler: CLICompiler<*>>(
             }
         }
 
-    override fun remoteIncrementalCompile(args: Array<out String>, caches: Map<String, CompileService.RemoteIncrementalCache>, outputStream: RemoteOutputStream, outputFormat: CompileService.OutputFormat): Int =
+    override fun remoteIncrementalCompile(args: Array<out String>, caches: Map<Module, CompileService.RemoteIncrementalCache>, outputStream: RemoteOutputStream, outputFormat: CompileService.OutputFormat): Int =
         doCompile(args, outputStream) { printStream ->
             when (outputFormat) {
                 CompileService.OutputFormat.PLAIN -> throw NotImplementedError("Only XML output is supported in remote incremental compilation")

compiler/rmi/rmi-server/src/org/jetbrains/kotlin/rmi/service/RemoteIncrementalCacheClient.kt

@@ -17,14 +17,19 @@
 package org.jetbrains.kotlin.rmi.service
 
 import org.jetbrains.kotlin.load.kotlin.incremental.components.IncrementalCache
+import org.jetbrains.kotlin.load.kotlin.incremental.components.InlineRegistering
 import org.jetbrains.kotlin.rmi.CompileService
 
 public class RemoteIncrementalCacheClient(val cache: CompileService.RemoteIncrementalCache): IncrementalCache {
+    override fun getInlineRegistering(): InlineRegistering = cache.getInlineRegistering()
+
     override fun getObsoletePackageParts(): Collection<String> = cache.getObsoletePackageParts()
 
     override fun getPackagePartData(fqName: String): ByteArray? = cache.getPackagePartData(fqName)
 
     override fun getModuleMappingData(): ByteArray? = cache.getModuleMappingData()
 
+    override fun getClassFilePath(internalClassName: String): String = cache.getClassFilePath(internalClassName)
+
     override fun close(): Unit = cache.close()
 }