Require all transactions are added via the ServiceHub.recordTransaction() function

node/src/main/kotlin/com/r3corda/node/internal/AbstractNode.kt

@@ -4,6 +4,7 @@ import com.codahale.metrics.MetricRegistry
 import com.google.common.util.concurrent.ListenableFuture
 import com.google.common.util.concurrent.SettableFuture
 import com.r3corda.core.RunOnCallerThread
+import com.r3corda.core.contracts.SignedTransaction
 import com.r3corda.core.crypto.Party
 import com.r3corda.core.messaging.MessagingService
 import com.r3corda.core.messaging.runOnNextMessage
@@ -80,15 +81,18 @@ abstract class AbstractNode(val dir: Path, val configuration: NodeConfiguration,
     protected val _servicesThatAcceptUploads = ArrayList<AcceptsFileUpload>()
     val servicesThatAcceptUploads: List<AcceptsFileUpload> = _servicesThatAcceptUploads
 
-    val services = object : ServiceHubInternal {
+    val services = object : ServiceHubInternal() {
         override val networkService: MessagingService get() = net
         override val networkMapCache: NetworkMapCache = InMemoryNetworkMapCache()
-        override val storageService: StorageService get() = storage
+        override val storageService: TxWritableStorageService get() = storage
         override val walletService: WalletService get() = wallet
         override val keyManagementService: KeyManagementService get() = keyManagement
         override val identityService: IdentityService get() = identity
         override val monitoringService: MonitoringService = MonitoringService(MetricRegistry())
         override val clock: Clock = platformClock
+
+        override fun recordTransactions(txs: Iterable<SignedTransaction>) =
+                recordTransactionsInternal(storage, txs)
     }
 
     val info: NodeInfo by lazy {
@@ -97,7 +101,7 @@ abstract class AbstractNode(val dir: Path, val configuration: NodeConfiguration,
 
     open fun findMyLocation(): PhysicalLocation? = CityDatabase[configuration.nearestCity]
 
-    lateinit var storage: StorageService
+    lateinit var storage: TxWritableStorageService
     lateinit var checkpointStorage: CheckpointStorage
     lateinit var smm: StateMachineManager
     lateinit var wallet: WalletService
@@ -265,7 +269,7 @@ abstract class AbstractNode(val dir: Path, val configuration: NodeConfiguration,
 
     protected abstract fun startMessagingService()
 
-    protected open fun initialiseStorageService(dir: Path): Pair<StorageService, CheckpointStorage> {
+    protected open fun initialiseStorageService(dir: Path): Pair<TxWritableStorageService, CheckpointStorage> {
         val attachments = makeAttachmentStorage(dir)
         val checkpointStorage = PerFileCheckpointStorage(dir.resolve("checkpoints"))
         val transactionStorage = PerFileTransactionStorage(dir.resolve("transactions"))

node/src/main/kotlin/com/r3corda/node/internal/testing/TradeSimulation.kt

@@ -38,7 +38,7 @@ class TradeSimulation(runAsync: Boolean, latencyInjector: InMemoryMessagingNetwo
             tx.signWith(seller.storage.myLegalIdentityKey)
             tx.toSignedTransaction(true)
         }
-        seller.services.storageService.validatedTransactions.addTransaction(issuance)
+        seller.services.recordTransactions(issuance)
 
         val cashIssuerKey = generateKeyPair()
         val amount = 1000.DOLLARS `issued by` Party("Big friendly bank", cashIssuerKey.public).ref(1)

node/src/main/kotlin/com/r3corda/node/services/api/ServiceHubInternal.kt

@@ -1,7 +1,21 @@
 package com.r3corda.node.services.api
 
+import com.r3corda.core.contracts.SignedTransaction
 import com.r3corda.core.node.ServiceHub
+import com.r3corda.core.node.services.TxWritableStorageService
 
-interface ServiceHubInternal : ServiceHub {
-    val monitoringService: MonitoringService
+abstract class ServiceHubInternal : ServiceHub {
+    abstract val monitoringService: MonitoringService
+
+    /**
+     * Given a list of [SignedTransaction]s, writes them to the given storage for validated transactions and then
+     * sends them to the wallet for further processing. This is intended for implementations to call from
+     * [recordTransactions].
+     *
+     * @param txs The transactions to record
+     */
+    internal fun recordTransactionsInternal(writableStorageService: TxWritableStorageService, txs: Iterable<SignedTransaction>) {
+        txs.forEach { writableStorageService.validatedTransactions.addTransaction(it) }
+        walletService.notifyAll(txs.map { it.tx })
+    }
 }

node/src/main/kotlin/com/r3corda/node/services/persistence/StorageServiceImpl.kt

@@ -4,6 +4,7 @@ import com.r3corda.core.crypto.Party
 import com.r3corda.core.node.services.AttachmentStorage
 import com.r3corda.core.node.services.StorageService
 import com.r3corda.core.node.services.TransactionStorage
+import com.r3corda.core.node.services.TxWritableStorageService
 import com.r3corda.core.serialization.SingletonSerializeAsToken
 import java.security.KeyPair
 
@@ -11,4 +12,4 @@ open class StorageServiceImpl(override val attachments: AttachmentStorage,
                               override val validatedTransactions: TransactionStorage,
                               override val myLegalIdentityKey: KeyPair,
                               override val myLegalIdentity: Party = Party("Unit test party", myLegalIdentityKey.public))
-: SingletonSerializeAsToken(), StorageService
+: SingletonSerializeAsToken(), TxWritableStorageService

node/src/test/kotlin/com/r3corda/node/services/MockServices.kt

@@ -1,6 +1,7 @@
 package com.r3corda.node.services
 
 import com.codahale.metrics.MetricRegistry
+import com.r3corda.core.contracts.SignedTransaction
 import com.r3corda.core.messaging.MessagingService
 import com.r3corda.core.node.services.*
 import com.r3corda.core.node.services.testing.MockStorageService
@@ -19,11 +20,11 @@ open class MockServices(
         val keyManagement: KeyManagementService? = null,
         val net: MessagingService? = null,
         val identity: IdentityService? = MOCK_IDENTITY_SERVICE,
-        val storage: StorageService? = MockStorageService(),
+        val storage: TxWritableStorageService? = MockStorageService(),
         val mapCache: NetworkMapCache? = MockNetworkMapCache(),
         val mapService: NetworkMapService? = null,
         val overrideClock: Clock? = NodeClock()
-) : ServiceHubInternal {
+) : ServiceHubInternal() {
     override val walletService: WalletService = customWallet ?: NodeWalletService(this)
 
     override val keyManagementService: KeyManagementService
@@ -40,6 +41,12 @@ open class MockServices(
         get() = overrideClock ?: throw UnsupportedOperationException()
 
     override val monitoringService: MonitoringService = MonitoringService(MetricRegistry())
+    // We isolate the storage service with writable TXes so that it can't be accessed except via recordTransactions()
+    private val txStorageService: TxWritableStorageService
+        get() = storage ?: throw UnsupportedOperationException()
+
+    override fun recordTransactions(txs: Iterable<SignedTransaction>) =
+            recordTransactionsInternal(txStorageService, txs)
 
     init {
         if (net != null && storage != null) {