Add ListDataContainer

2 files changed, 51 insertions, 0 deletions

diff --git a/src/main/com/jantuomi/tunkki/core/parser/datatype/DataContainer.java b/src/main/com/jantuomi/tunkki/core/parser/datatype/DataContainer.java
index 1f3cda2..de18278 100644
--- a/src/main/com/jantuomi/tunkki/core/parser/datatype/DataContainer.java
+++ b/src/main/com/jantuomi/tunkki/core/parser/datatype/DataContainer.java
@@ -8,6 +8,7 @@ import com.jantuomi.tunkki.exception.TunkkiError;
 abstract public class DataContainer<T> {
 
     public enum Type {
+        List,
         Integer,
         Double,
         String,
diff --git a/src/main/com/jantuomi/tunkki/core/parser/datatype/ListDataContainer.java b/src/main/com/jantuomi/tunkki/core/parser/datatype/ListDataContainer.java
new file mode 100644
index 0000000..5cd7e90
--- /dev/null
+++ b/src/main/com/jantuomi/tunkki/core/parser/datatype/ListDataContainer.java
@@ -0,0 +1,50 @@
+package com.jantuomi.tunkki.core.parser.datatype;
+
+import com.jantuomi.tunkki.exception.TunkkiError;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.stream.Collectors;
+
+/**
+ * Created by jan on 27.7.2016.
+ */
+public class ListDataContainer extends DataContainer{
+    @Override
+    public Type getType() {
+        return Type.List;
+    }
+
+    private List<DataContainer> elements = new ArrayList<>();
+
+    @Override
+    public String toString() {
+        List<String> strings = elements.stream().map( dataContainer -> dataContainer.toString() ).collect(Collectors.toList());
+        return String.join(", ", strings);
+    }
+
+    @Override
+    public DataContainer add(DataContainer other) throws TunkkiError {
+        switch (other.getType()) {
+            case List:
+                elements.addAll(((ListDataContainer) other).elements);
+                break;
+        }
+        return null;
+    }
+
+    @Override
+    public DataContainer subtract(DataContainer other) throws TunkkiError {
+        return null;
+    }
+
+    @Override
+    public DataContainer multiply(DataContainer other) throws TunkkiError {
+        return null;
+    }
+
+    @Override
+    public DataContainer divide(DataContainer other) throws TunkkiError {
+        return null;
+    }
+}