From 1df50093bd76315905a9aae880470e81b5e1d8f0 Mon Sep 17 00:00:00 2001
From: Keuin <keuinx@gmail.com>
Date: Sun, 24 Jan 2021 21:15:24 +0800
Subject: If incremental backup failed, unfinished copy will be fully reverted.

---
 .../incremental/ObjectCollectionIterator.java      | 50 ++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 src/main/java/com/keuin/kbackupfabric/backup/incremental/ObjectCollectionIterator.java

(limited to 'src/main/java/com/keuin/kbackupfabric/backup/incremental/ObjectCollectionIterator.java')

diff --git a/src/main/java/com/keuin/kbackupfabric/backup/incremental/ObjectCollectionIterator.java b/src/main/java/com/keuin/kbackupfabric/backup/incremental/ObjectCollectionIterator.java
new file mode 100644
index 0000000..248d36d
--- /dev/null
+++ b/src/main/java/com/keuin/kbackupfabric/backup/incremental/ObjectCollectionIterator.java
@@ -0,0 +1,50 @@
+package com.keuin.kbackupfabric.backup.incremental;
+
+import java.util.Iterator;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.NoSuchElementException;
+
+public class ObjectCollectionIterator implements Iterator<ObjectElement> {
+
+    // TODO: test this
+
+    private Iterator<ObjectElement> currentIterator;
+    private final List<ObjectCollection2> cols = new LinkedList<>();
+
+    public ObjectCollectionIterator(ObjectCollection2 collection) {
+        cols.addAll(collection.getSubCollectionSet());
+        currentIterator = collection.getElementSet().iterator();
+    }
+
+    @Override
+    public boolean hasNext() {
+        if (currentIterator != null) {
+            if (currentIterator.hasNext())
+                return true;
+            else {
+                currentIterator = null;
+                return hasNext();
+            }
+        } else {
+            if (cols.isEmpty())
+                return false;
+            else {
+                ObjectCollection2 consumedCollection = cols.remove(0);
+                cols.addAll(consumedCollection.getSubCollectionSet());
+                currentIterator = consumedCollection.getElementSet().iterator();
+                return hasNext();
+            }
+        }
+    }
+
+    @Override
+    public ObjectElement next() {
+        if (hasNext()) {
+            return currentIterator.next();
+        } else {
+            throw new NoSuchElementException();
+        }
+    }
+
+}
-- 
cgit v1.2.3