ExternalVendorCode/corda
1
0
Fork 0
mirror of https://github.com/corda/corda.git synced 2024-12-19 13:08:04 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fixed issue caused by race condition in process registration v node exit.

Browse Source 
See https://github.com/corda/corda/issues/88
This commit is contained in:
josecoll 2016-12-23 15:09:13 +00:00
parent 1bcabc8d41
commit b4a5a03992
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
node/src/main/kotlin/net/corda/node/driver/Driver.kt
View File

@ -264,7 +264,7 @@ open class DriverDSL(
private val networkMapAddress = portAllocation.nextHostAndPort()
class State {
val registeredProcesses = LinkedList<Process>()
val registeredProcesses = LinkedList<ListenableFuture<Process>>()
val clients = LinkedList<NodeMessagingClient>()
}
@ -279,12 +279,12 @@ open class DriverDSL(
Paths.get(quasarFileUrl.toURI()).toString()
}
fun registerProcess(process: Process) = state.locked { registeredProcesses.push(process) }
fun registerProcess(process: ListenableFuture<Process>) = state.locked { registeredProcesses.push(process) }
override fun waitForAllNodesToFinish() {
state.locked {
registeredProcesses.forEach {
it.waitFor()
it.getOrThrow().waitFor()
}
}
}
@ -294,7 +294,9 @@ open class DriverDSL(
clients.forEach {
it.stop()
}
registeredProcesses.forEach(Process::destroy)
registeredProcesses.forEach {
it.get().destroy()
}
}
/** Wait 5 seconds, then [Process.destroyForcibly] */
val finishedFuture = executorService.submit {
@ -306,7 +308,7 @@ open class DriverDSL(
finishedFuture.cancel(true)
state.locked {
registeredProcesses.forEach {
it.destroyForcibly()
it.get().destroyForcibly()
}
}
}
@ -368,8 +370,9 @@ open class DriverDSL(
configOverrides = configOverrides
)
return startNode(executorService, FullNodeConfiguration(config), quasarJarPath, debugPort).map {
registerProcess(it)
val startNode = startNode(executorService, FullNodeConfiguration(config), quasarJarPath, debugPort)
registerProcess(startNode)
return startNode.map {
NodeHandle(queryNodeInfo(apiAddress)!!, config, it)
}
}
@ -409,7 +412,7 @@ open class DriverDSL(
startNetworkMapService()
}
private fun startNetworkMapService(): ListenableFuture<Unit> {
private fun startNetworkMapService(): ListenableFuture<Process> {
val apiAddress = portAllocation.nextHostAndPort()
val debugPort = if (isDebug) debugPortAllocation.nextPort() else null
@ -428,9 +431,9 @@ open class DriverDSL(
)
log.info("Starting network-map-service")
return startNode(executorService, FullNodeConfiguration(config), quasarJarPath, debugPort).map {
registerProcess(it)
}
val startNode = startNode(executorService, FullNodeConfiguration(config), quasarJarPath, debugPort)
registerProcess(startNode)
return startNode
}
companion object {