working to generate query

Author: rustyrazorblade

src/main/java/com/datastax/cdm/CassandraDatasetManager.java

@@ -20,6 +20,7 @@
 //import com.datastax.loader.CqlDelimLoadTask;
 
 import java.io.*;
+import java.lang.reflect.Array;
 import java.math.BigDecimal;
 import java.net.MalformedURLException;
 import java.net.URL;
@@ -35,6 +36,10 @@ public class CassandraDatasetManager {
     private Map<String, Dataset> datasets;
     private Session session;
 
+    CassandraDatasetManager() {
+
+    }
+
     CassandraDatasetManager(Map<String, Dataset> datasets) {
         this.datasets = datasets;
     }
@@ -262,19 +267,19 @@ void install(String name) throws IOException, InterruptedException, GitAPIExcept
 
             HashMap types = new HashMap();
 
+            ArrayList<Field> fieldlist = new ArrayList<>();
+
             for(ColumnMetadata c: columns) {
                 fields.add(c.getName());
-                types.put(c.getName(), c.getType().getName().toString());
+                String ftype = c.getType().getName().toString();
+                types.put(c.getName(), ftype);
+                fieldlist.add(new Field(c.getName(), ftype));
             }
 
-            PreparedStatement p = session.prepare(insert_query.toString());
-            HashSet needs_quotes = new HashSet();
-            needs_quotes.add("text");
-            needs_quotes.add("datetime");
-
             for(CSVRecord record: records) {
                 // generate a CQL statement
-                String cql = generateCQL(record, types);
+                String cql = generateCQL(table, record,
+                                         fieldlist, types);
                 session.execute(cql);
             }
         }
@@ -289,8 +294,19 @@ Iterable<CSVRecord> openCSV(String path) throws IOException {
         return records;
     }
 
-    String generateCQL(CSVRecord record, HashMap<String, String> types) {
-        return "";
+    String generateCQL(String table,
+                       CSVRecord record,
+                       ArrayList<Field> fields,
+                       HashMap<String, String> types) {
+
+        HashSet needs_quotes = new HashSet();
+        needs_quotes.add("text");
+        needs_quotes.add("datetime");
+
+        StringBuilder query = new StringBuilder("INSERT INTO ");
+        query.append(table);
+
+        return query.toString();
     }
 
     void update() throws IOException {

src/main/java/com/datastax/cdm/Field.java

@@ -0,0 +1,14 @@
+package com.datastax.cdm;
+
+/**
+ * Created by jhaddad on 9/5/16.
+ */
+public class Field {
+    public String name;
+    public String type;
+
+    Field(String name, String type) {
+        this.name = name;
+        this.type = type;
+    }
+}

test/java/com/datastax/cdm/CassandraDatasetManagerTest.java

@@ -1,17 +1,35 @@
 package com.datastax.cdm;
 
 import static org.junit.Assert.assertEquals;
+import static org.hamcrest.CoreMatchers.containsString;
+import static org.junit.Assert.assertThat;
 
 import org.apache.commons.csv.CSVRecord;
 import org.junit.Test;
 
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.HashMap;
+
 /**
  * Created by jhaddad on 9/5/16.
  */
 public class CassandraDatasetManagerTest {
 
     @Test
-    public void testCQLStatementGeneration() {
+    public void testCQLStatementGeneration() throws IOException {
+        CassandraDatasetManager c = new CassandraDatasetManager();
+        Iterable<CSVRecord> records = c.openCSV("data/alltypes.csv");
+        CSVRecord r = records.iterator().next();
+
+        HashMap types = new HashMap();
+        ArrayList<Field> fieldList = new ArrayList<Field>();
+        String query = c.generateCQL("whatever",
+                                     r,
+                                     fieldList,
+                                     types);
+
+        assertThat(query, containsString("INSERT INTO whatever"));
 
     }