Merge pull request #2093 from lossyrob/backport/1.1/2083

[BACKPORT]: Ensure keyspace exists in CassandraRDDWriter
locationtech · Mar 25, 2017 · 3a9493e · 3a9493e
2 parents 2a2c628 + 161d881
commit 3a9493e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cassandra/src/main/scala/geotrellis/spark/io/cassandra/CassandraRDDWriter.scala b/cassandra/src/main/scala/geotrellis/spark/io/cassandra/CassandraRDDWriter.scala
@@ -49,10 +49,10 @@ object CassandraRDDWriter {
     implicit val sc = raster.sparkContext
 
     val codec = KeyValueRecordCodec[K, V]
-    val schema = codec.schema
 
-    instance.withSessionDo {
-      _.execute(
+    instance.withSessionDo { session =>
+      instance.ensureKeyspaceExists(keyspace, session)
+      session.execute(
         SchemaBuilder.createTable(keyspace, table).ifNotExists()
           .addPartitionKey("key", bigint)
           .addClusteringColumn("name", text)