fix(dataTrasnfer): fail when a DataTransferErrorCommand is received

nrkno · Oct 4, 2018 · 441ef68 · 441ef68
1 parent d8b2d5f
commit 441ef68
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/src/dataTransfer/dataLock.ts b/src/dataTransfer/dataLock.ts
@@ -68,4 +68,13 @@ export default class DataLock {
 			this.commandQueue.push(command)
 		}
 	}
+
+	transferErrored (code: number) {
+		if (this.transfer) {
+			// @todo: dequeue any old commands
+			this.transfer.fail(code)
+		}
+		this.transfer = undefined
+		this.dequeueAndRun()
+	}
 }
diff --git a/src/dataTransfer/dataTransfer.ts b/src/dataTransfer/dataTransfer.ts
@@ -9,7 +9,7 @@ export default abstract class DataTransfer {
 	commandQueue: Array<Commands.AbstractCommand>
 
 	finish: (transfer: DataTransfer) => void
-	fail: () => void
+	fail: (errorCode?: number) => void
 	abstract start (): void
 
 	abstract handleCommand (command: Commands.AbstractCommand): void

diff --git a/src/dataTransfer/index.ts b/src/dataTransfer/index.ts
@@ -69,6 +69,9 @@ export class DataTransferManager {
 			if (!command.properties.locked) lock.lostLock()
 			else lock.updateLock(command.properties.locked)
 		}
+		if (command.constructor.name === Commands.DataTransferErrorCommand.name) {
+			lock.transferErrored(command.properties.errorCode)
+		}
 		if (lock.transfer) {
 			lock.transfer.handleCommand(command)
 			if (lock.transfer.state === Enums.TransferState.Finished) {