Merge pull request #5 from magro/shutdown-cluster

Shutdown cluster in migrate & initialize
comeara · Mar 25, 2014 · abe7950 · abe7950
2 parents cf0e20b + 0bf5af2
commit abe7950
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/main/scala/com/streamsend/pillar/CassandraMigrator.scala b/src/main/scala/com/streamsend/pillar/CassandraMigrator.scala
@@ -12,6 +12,8 @@ class CassandraMigrator(registry: Registry) extends Migrator {
 
     selectMigrationsToReverse(dateRestriction, appliedMigrations).foreach(_.executeDownStatement(session))
     selectMigrationsToApply(dateRestriction, appliedMigrations).foreach(_.executeUpStatement(session))
+
+    cluster.close
   }
 
   def initialize(dataStore: DataStore, replicationOptions: ReplicationOptions = ReplicationOptions.default) {
@@ -28,6 +30,7 @@ class CassandraMigrator(registry: Registry) extends Migrator {
         |  )
       """.stripMargin.format(dataStore.keyspace)
     )
+    cluster.close
   }
 
   private def executeIdempotentCommand(session: Session, statement: String) {