ExternalVendorCode/corda
1
0
Fork 0
mirror of https://github.com/corda/corda.git synced 2025-01-31 00:24:59 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Shutdown copycat server and h2 database on Node shutdown

Browse Source
This commit is contained in:
Andras Slemmer 2017-04-19 12:24:38 +01:00
parent 66890d845a
commit 201c558245
2 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
node/src/main/kotlin/net/corda/node/internal/Node.kt
View File

@ -196,7 +196,10 @@ class Node(override val configuration: FullNodeConfiguration,
override fun makeUniquenessProvider(type: ServiceType): UniquenessProvider {
return when (type) {
RaftValidatingNotaryService.type, RaftNonValidatingNotaryService.type -> with(configuration) {
RaftUniquenessProvider(baseDirectory, notaryNodeAddress!!, notaryClusterAddresses, database, configuration)
val provider = RaftUniquenessProvider(baseDirectory, notaryNodeAddress!!, notaryClusterAddresses, database, configuration)
provider.start()
runOnStop += Runnable { provider.stop() }
provider
}
else -> PersistentUniquenessProvider()
}
@ -224,6 +227,7 @@ class Node(override val configuration: FullNodeConfiguration,
"-tcpAllowOthers",
"-tcpDaemon",
"-key", "node", databaseName)
runOnStop += Runnable { server.stop() }
val url = server.start().url
printBasicNodeInfo("Database connection url is", "jdbc:h2:$url/node")
}

20
node/src/main/kotlin/net/corda/node/services/transactions/RaftUniquenessProvider.kt
View File

@ -42,14 +42,20 @@ import javax.annotation.concurrent.ThreadSafe
* @param config SSL configuration
*/
@ThreadSafe
class RaftUniquenessProvider(storagePath: Path, myAddress: HostAndPort, clusterAddresses: List<HostAndPort>,
db: Database, config: SSLConfiguration) : UniquenessProvider, SingletonSerializeAsToken() {
class RaftUniquenessProvider(
val storagePath: Path,
val myAddress: HostAndPort,
val clusterAddresses: List<HostAndPort>,
val db: Database,
val config: SSLConfiguration
) : UniquenessProvider, SingletonSerializeAsToken() {
companion object {
private val log = loggerFor<RaftUniquenessProvider>()
private val DB_TABLE_NAME = "notary_committed_states"
}
private val _clientFuture: CompletableFuture<CopycatClient>
private lateinit var _clientFuture: CompletableFuture<CopycatClient>
private lateinit var server: CopycatServer
/**
* Copycat clients are responsible for connecting to the cluster and submitting commands and queries that operate
* on the cluster's replicated state machine.
@ -57,7 +63,7 @@ class RaftUniquenessProvider(storagePath: Path, myAddress: HostAndPort, clusterA
private val client: CopycatClient
get() = _clientFuture.get()
init {
fun start() {
log.info("Creating Copycat server, log stored in: ${storagePath.toFile()}")
val stateMachineFactory = { DistributedImmutableMap<String, ByteArray>(db, DB_TABLE_NAME) }
val address = Address(myAddress.host, myAddress.port)
@ -65,7 +71,7 @@ class RaftUniquenessProvider(storagePath: Path, myAddress: HostAndPort, clusterA
val transport = buildTransport(config)
val serializer = Serializer()
val server = CopycatServer.builder(address)
server = CopycatServer.builder(address)
.withStateMachine(stateMachineFactory)
.withStorage(storage)
.withServerTransport(transport)
@ -89,6 +95,10 @@ class RaftUniquenessProvider(storagePath: Path, myAddress: HostAndPort, clusterA
_clientFuture = serverFuture.thenCompose { client.connect(address) }
}
fun stop() {
server.shutdown()
}
private fun buildStorage(storagePath: Path): Storage? {
return Storage.builder()
.withDirectory(storagePath.toFile())