diff --git a/lib/dm-cassandra-adapter/adapter.rb b/lib/dm-cassandra-adapter/adapter.rb index c71f0d0..be9a4f3 100644 --- a/lib/dm-cassandra-adapter/adapter.rb +++ b/lib/dm-cassandra-adapter/adapter.rb @@ -49,6 +49,12 @@ def execute(*args) nil end + def reset + @pool.shutdown(&:disconnect!) + @pool = new_pool + self + end + private def setup_keyspace