pg_tasks 1.1.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: fc265a9b51988c66412de5651644fe6427761bd0
-  data.tar.gz: 7326fe644b0d418c17707e6f80600c3554ba9c61
+  metadata.gz: 12e52f1af8a55af227b0925b29096958f20826bc
+  data.tar.gz: 4e8258d91b992df3c6827c460422e464523b21e0
 SHA512:
-  metadata.gz: b005994e668eb75f856f12ad35b17d671cea87a31779ee6bd519a7035147b01dd478f95d150f800e0b4faee7920858e50dd1289403c08ea21c4e0d7e99a826a7
-  data.tar.gz: 5bfd6120421f4f5606a274e7840c23c474846eaed9150d04e60b898c594d068f9bcb02b1d0415c959e504c84a5df5317d8e709f5ccb68f50ce19b36b4c974462
+  metadata.gz: 4df1c4897e695827b180621f4b187b95cbea3c8efbd476d302fef310df75652428664e8dacee6fc664acdf40cc2ccaffb9f1fe53f8828406d3aa842355cb66ce
+  data.tar.gz: 3b584e9d0df49843bea70d23bc6b05808d23a443dc0173f2ad9ec6897c7098795fadf89aaaff992e4c956ffa554fa44b6e53c8f68ce0a19bf114becc3fd7d48a

data/lib/pg_tasks.rb

@@ -43,6 +43,10 @@ module PgTasks
       end
     end
 
+    def terminate_connections
+      ActiveRecord::Tasks::DatabaseTasks.terminate_connections current_config
+    end
+
     private
 
     def current_config
@@ -118,6 +122,20 @@ module ActiveRecord
             "has been restored to '#{filename}'"
         end
       end
+
+      def terminate_connections
+        set_psql_env
+        database = configuration['database']
+        command = "psql -c \"SELECT pg_terminate_backend(pg_stat_activity.pid) " \
+          ' FROM pg_stat_activity ' \
+          "WHERE pg_stat_activity.datname = '#{database}' " \
+          " AND pid <> pg_backend_pid();\" '#{database}'"
+        unless Kernel.system(command)
+          raise 'Error during terminate_connections'
+        else
+          $stdout.puts "Connections to '#{database}' have been terminated."
+        end
+      end
     end
   end
 end
@@ -135,6 +153,12 @@ module ActiveRecord
       rescue Exception => error
         $stderr.puts error, *(error.backtrace)
       end
+
+      def terminate_connections(*arguments)
+        configuration = arguments.first
+        class_for_adapter(configuration['adapter']) \
+          .new(*arguments).send :terminate_connections
+      end
     end
   end
 end

data/lib/pg_tasks/version.rb

@@ -7,5 +7,5 @@
 #++
 
 module PgTasks
-  VERSION = '1.1.0'
+  VERSION = '1.2.0'
 end

data/lib/tasks/pg_tasks.rake

@@ -14,6 +14,10 @@ namespace :db do
       PgTasks.truncate_tables
     end
 
+    task terminate_connections: [:environment, :load_config] do
+      PgTasks.terminate_connections
+    end
+
     namespace :data do
       task dump: [:environment, :load_config] do
         PgTasks.data_dump ENV['FILE']

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pg_tasks
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.2.0
 platform: ruby
 authors:
 - Thomas Schank
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-01-27 00:00:00.000000000 Z
+date: 2015-02-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails