Merge pull request #605 from Benky/master

Introduce functionality for data deletion during ETL

Author: liry

src/main/java/com/gooddata/dataset/DatasetManifest.java

@@ -91,7 +91,7 @@ public void setFile(String file) {
     public void setUploadMode(final UploadMode uploadMode) {
         notNull(uploadMode, "uploadMode");
         for (Part part : parts) {
-            part.setUploadMode(uploadMode.name());
+            part.setUploadMode(uploadMode.toString());
         }
     }
 

src/main/java/com/gooddata/dataset/UploadMode.java

@@ -5,6 +5,60 @@
  */
 package com.gooddata.dataset;
 
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * Upload modes enum.
+ */
 public enum UploadMode {
-	INCREMENTAL, FULL
+    /**
+     * Dataset input for loading in full mode (reload from scratch)
+     */
+    FULL("FULL"),
+    /**
+     * Dataset input for loading data incrementally
+     */
+    INCREMENTAL("INCREMENTAL"),
+    /**
+     * Dataset input for loading data by calculating based on its attributes.
+     * Example: last sync, max timestamp, having timestamp field,...
+     */
+    DEFAULT("DEFAULT"),
+    /**
+     * Dataset input for deleting
+     */
+    DELETE_CENTER("DELETE-CENTER");
+
+    private static final Map<String, UploadMode> lookup = new HashMap<>();
+
+    static {
+        for (UploadMode m : UploadMode.values()) {
+            lookup.put(m.modeStr, m);
+        }
+    }
+
+    private final String modeStr;
+
+    /**
+     * @param modeStr string representation of the upload mode
+     */
+    UploadMode(final String modeStr) {
+        this.modeStr = modeStr;
+    }
+
+    /**
+     * Lookup method.
+     *
+     * @param mode a string value passed in the constructor
+     * @return UploadMode enum instance according to the given mode string
+     */
+    public static UploadMode get(String mode) {
+        return lookup.get(mode);
+    }
+
+    @Override
+    public String toString() {
+        return this.modeStr;
+    }
 }

src/test/java/com/gooddata/dataset/DatasetManifestTest.java

@@ -61,6 +61,18 @@ public void shouldSetUploadModeForAllParts() throws Exception {
         assertThat(manifest.getParts().get(1).getUploadMode(), is("INCREMENTAL"));
     }
 
+    @Test
+    public void shouldSetUploadModeDeleteForAllParts() throws Exception {
+        final DatasetManifest manifest = new DatasetManifest("dataset", "file.csv", asList(
+                new DatasetManifest.Part("FULL", "col1", asList("attr1"), true, null),
+                new DatasetManifest.Part("FULL", "col2", asList("attr2"), true, null)
+        ));
+        manifest.setUploadMode(UploadMode.DELETE_CENTER);
+
+        assertThat(manifest.getParts().get(0).getUploadMode(), is("DELETE-CENTER"));
+        assertThat(manifest.getParts().get(1).getUploadMode(), is("DELETE-CENTER"));
+    }
+
     @Test
     public void shouldSetMapping() throws Exception {
         final DatasetManifest manifest = new DatasetManifest("dataset", "file.csv", asList(