Replace code only used in 1 test with existing general mechanism. (#600)

This commit is contained in:
Andrzej Cichocki
2017-04-28 15:50:24 +01:00
committed by GitHub
parent 9dde0db407
commit f2d138cdab
8 changed files with 92 additions and 85 deletions

View File

@ -7,6 +7,8 @@ open class StorageServiceImpl(override val attachments: AttachmentStorage,
override val validatedTransactions: TransactionStorage,
override val stateMachineRecordedTransactionMapping: StateMachineRecordedTransactionMappingStorage)
: SingletonSerializeAsToken(), TxWritableStorageService {
override val attachmentsClassLoaderEnabled = false
lateinit override var uploaders: List<FileUploader>
fun initUploaders(uploadersList: List<FileUploader>) {

View File

@ -374,16 +374,18 @@ class StateMachineManager(val serviceHub: ServiceHubInternal,
private fun serializeFiber(fiber: FlowStateMachineImpl<*>): SerializedBytes<FlowStateMachineImpl<*>> {
return quasarKryo().run { kryo ->
// add the map of tokens -> tokenizedServices to the kyro context
SerializeAsTokenSerializer.setContext(kryo, serializationContext)
fiber.serialize(kryo)
kryo.withSerializationContext(serializationContext) {
fiber.serialize(kryo)
}
}
}
private fun deserializeFiber(checkpoint: Checkpoint): FlowStateMachineImpl<*> {
return quasarKryo().run { kryo ->
// put the map of token -> tokenized into the kryo context
SerializeAsTokenSerializer.setContext(kryo, serializationContext)
checkpoint.serializedFiber.deserialize(kryo).apply { fromCheckpoint = true }
kryo.withSerializationContext(serializationContext) {
checkpoint.serializedFiber.deserialize(kryo)
}.apply { fromCheckpoint = true }
}
}