|
1 | 1 | package com.datastax.cdm; |
2 | 2 |
|
3 | | -import com.datastax.driver.core.Cluster; |
4 | | -import com.datastax.driver.core.Session; |
| 3 | +import com.datastax.driver.core.*; |
5 | 4 | import com.fasterxml.jackson.core.type.TypeReference; |
6 | 5 | import com.fasterxml.jackson.databind.ObjectMapper; |
7 | 6 | import com.fasterxml.jackson.dataformat.yaml.YAMLFactory; |
|
20 | 19 | import java.io.*; |
21 | 20 | import java.net.MalformedURLException; |
22 | 21 | import java.net.URL; |
| 22 | +import java.util.List; |
23 | 23 | import java.util.Map; |
24 | 24 |
|
25 | 25 | /** |
@@ -229,28 +229,29 @@ void install(String name) throws IOException, InterruptedException, GitAPIExcept |
229 | 229 | session.execute("DROP KEYSPACE IF EXISTS " + config.keyspace); |
230 | 230 | session.execute(createKeyspace.toString()); |
231 | 231 |
|
232 | | -// Runtime.getRuntime().exec(new String[]{"bash", "-c", createKeyspace}).waitFor(); |
233 | | - |
234 | | - |
235 | 232 | System.out.println("Schema: " + schema); |
236 | 233 | String loadSchema = "cqlsh -k " + config.keyspace + " -f " + schema; |
237 | 234 | Runtime.getRuntime().exec(new String[]{"bash", "-c", loadSchema}).waitFor(); |
238 | 235 |
|
239 | 236 | System.out.println("Loading data"); |
240 | 237 |
|
| 238 | + Cluster cluster2 = Cluster.builder() |
| 239 | + .addContactPoint("127.0.0.1") |
| 240 | + .build(); |
| 241 | + |
241 | 242 | for(String table: config.tables) { |
242 | 243 | String dataFile = dataPath + table + ".csv"; |
243 | 244 |
|
244 | 245 | Reader in = new FileReader(dataFile); |
245 | 246 | Iterable<CSVRecord> records = CSVFormat.RFC4180.parse(in); |
| 247 | + KeyspaceMetadata keyspaceMetadata = cluster2.getMetadata() |
| 248 | + .getKeyspace(config.keyspace); |
| 249 | + TableMetadata tableMetadata = keyspaceMetadata.getTable(table); |
| 250 | +// PreparedStatement p = session.prepare(); |
246 | 251 | for(CSVRecord record: records) { |
247 | | - |
| 252 | + // generate a CQL statement |
| 253 | + System.out.println("Blah"); |
248 | 254 | } |
249 | | - |
250 | | -// String command = "COPY " + table + " FROM " + "'" + dataFile + "'"; |
251 | | -// String loadData= "cqlsh -k " + config.keyspace + " -e \"" + command + "\""; |
252 | | -// System.out.println(loadData); |
253 | | -// Runtime.getRuntime().exec(new String[]{"bash", "-c", loadData}).waitFor(); |
254 | 255 | } |
255 | 256 |
|
256 | 257 |
|
|
0 commit comments