fixing all the types

Author: rustyrazorblade

cdm.yaml

@@ -1,5 +1,5 @@
 keyspace: cdmtest
 tables:
     - users
-    - collection
+    - alltypes
 version: 2.2

data/alltypes.csv

@@ -0,0 +1 @@
+49e37fdb-49e7-4dec-8fd6-3da78273de3c,4.5,500.01,{1: 3},1,01eb1ed0-739b-11e6-928a-15b0d640b4c2

data/collection.csv

@@ -1,8 +1,8 @@
-Jimmy O'Hare,5
 Rachel,10
-Luke,5
-Dani,15
+Duy Hai,9
 Jon,50
 Patrick,2
+Dani,15
+Jimmy O'Hare,5
+Luke,5
 Rebecca,8
-Duy Hai,9

data/users.csv

@@ -1,2 +1,10 @@
 create table users ( name text primary key, num_tacos int );
-create table movies (name text primary key, year int);
+
+CREATE TABLE alltypes (
+    id uuid PRIMARY KEY,
+    avg float,
+    cash decimal,
+    intmap map<int, float>,
+    num int,
+    ts timeuuid
+);

schema.cql

@@ -1,6 +1,8 @@
 package com.datastax.cdm;
 
 import com.datastax.driver.core.*;
+import com.datastax.driver.core.querybuilder.Insert;
+import com.datastax.driver.core.querybuilder.QueryBuilder;
 import com.fasterxml.jackson.core.type.TypeReference;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
@@ -12,6 +14,8 @@
 import org.eclipse.jgit.api.errors.GitAPIException;
 import org.eclipse.jgit.api.errors.InvalidRemoteException;
 import org.eclipse.jgit.api.errors.TransportException;
+
+import javax.management.Query;
 import java.lang.StringBuilder;
 
 //import com.datastax.loader.CqlDelimLoadTask;
@@ -244,13 +248,35 @@ void install(String name) throws IOException, InterruptedException, GitAPIExcept
 
             Reader in = new FileReader(dataFile);
             Iterable<CSVRecord> records = CSVFormat.RFC4180.parse(in);
+            System.out.println("Importing " + table);
             KeyspaceMetadata keyspaceMetadata = cluster2.getMetadata()
                                                         .getKeyspace(config.keyspace);
             TableMetadata tableMetadata = keyspaceMetadata.getTable(table);
 //            PreparedStatement p = session.prepare();
+            List<ColumnMetadata> columns = tableMetadata.getColumns();
+
             for(CSVRecord record: records) {
                 // generate a CQL statement
-                System.out.println("Blah");
+                Insert insert = QueryBuilder.insertInto(tableMetadata);
+
+                int i = 0;
+                for(ColumnMetadata cm: columns) {
+                    String t = cm.getType().getName().toString().toLowerCase();
+                    System.out.println(t);
+                    if(t.equals("int")) {
+                        insert.value(cm.getName(), new Integer(record.get(i)));
+                    }
+                    else if(t.equals("float")) {
+                        insert.value(cm.getName(), new Float(record.get(i)));
+                    }
+                    else {
+                        insert.value(cm.getName(), record.get(i));
+                    }
+                    i++;
+                }
+                String query = insert.toString();
+                System.out.println(query);
+                session.execute(query);
             }
         }