pgreset 0.3 → 0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 97f6d30c6fcbbba220f065297f7eb6441101609a99d0b39e176e7f9a61be7637
-  data.tar.gz: 6d886b10d4e0da04137c0ec8086aea356228fcbe86d8702218921325e6f6b445
+  metadata.gz: de67aa02241019195553648e33408586563205a64baf3fb219b26de23fb9e180
+  data.tar.gz: db83454a328df654b9000b5b08bb8c378ed36ddb020e8dd786329dacfdff920e
 SHA512:
-  metadata.gz: f3c218c0d55321c25a5b1679902c06cfa878ce5bc51ff09fbd60325554afbecf797c258920b7cf335f8edb9524f73368de8b6e78348d7c7144d9408f6e3acb0b
-  data.tar.gz: d22de6723053d9306598d6689834b6855b3b43247292405a40622d085447bb511f75b3c6a2f716f3cc00fa37e9c1c3cf54cf54a9bddaa0295cc8a9aa26c9097a
+  metadata.gz: ceb11b130b41efa4eebb8af64e7a6787b694b795a10d206e5157de20caedb0a9b1db102e6b8d9dcaf8573767c5af8949e6fca03069130f73378e3310209fcef5
+  data.tar.gz: bdc44ff256602541c9434be34f0860881c6baf1211163656167f58cace2ff41531f9e43f88cf6ff512b4031bd7528e4c7bb9b15b2adb42e35a01b5f965fd3480

data/README.md

@@ -31,7 +31,20 @@ Now you can run rails db:reset as normal, and rails will kill active connections
 
 After checking out the repo, run `bin/setup` to install dependencies. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
 
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+To install this gem onto your local machine, run `bundle exec rake install`.
+
+## Testing
+
+There's a script called pgreset-test.sh that can test a local copy of pgreset against various versions of ruby and rails.  The script assumes postgres is listening on a socket at /tmp/.s.PGSQL.5432 and accepts connections for a user with no password, and that rvm is installed.  You should at least test against these versions, since there were breaking changes to pgreset in them (replace PG_USER with your postgres username):
+
+    cd /path/to/your/pgreset/clone
+    ./pgreset-test.sh 3.0.4 6.0.4 PG_USER
+    ./pgreset-test.sh 3.1.2 7.0.8 PG_USER
+    ./pgreset-test.sh 3.1.2 7.1.1 PG_USER
+
+## Releasing
+
+To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
 
 ## Contributing
 

data/lib/pgreset/version.rb

@@ -1,3 +1,3 @@
 module Pgreset
-  VERSION = "0.3"
+  VERSION = "0.4"
 end

data/lib/pgreset.rb

@@ -4,15 +4,16 @@ module ActiveRecord
   module Tasks
     class PostgreSQLDatabaseTasks
       def drop
-        establish_master_connection
-
-        # `configuration_hash` was introduced in 6.1.0, prior versions have the database name in `configuration`.
-        database_name =
-          if respond_to?(:configuration_hash, true)
-            configuration_hash.with_indifferent_access.fetch(:database)
-          else
-            configuration['database']
-          end
+        # Extract the name of the application database from the configuration and
+        # establish a connection to the "postgres" database in the public schema.
+        # `configuration_hash` was introduced in rails 6.1.0, prior versions use `configuration`.
+        if respond_to?(:configuration_hash, true)
+          database_name = configuration_hash.with_indifferent_access.fetch(:database)
+          establish_connection(configuration_hash.merge(database: "postgres", schema_search_path: "public"))
+        else
+          database_name = configuration['database']
+          establish_connection(configuration.merge(database: "postgres", schema_search_path: "public"))
+        end
 
         pid_column = 'pid'       # Postgresql 9.2 and newer
         if 0 == connection.select_all("SELECT column_name FROM information_schema.columns WHERE table_name = 'pg_stat_activity' AND column_name = 'pid';").count

data/pgreset-test.sh

@@ -0,0 +1,76 @@
+#!/usr/bin/env bash
+
+# This is a simple script to test pgreset against different versions of ruby and rails.
+# Right now it expects to connect to postgres through a socket in /tmp as a given user
+# with no password.
+
+function die
+{
+  echo "$*" >&2
+  exit 1
+}
+
+if [[ ! -f "pgreset.gemspec" ]]; then
+  die "This script must be run from the pgreset source directory"
+fi
+
+if [[ -z "$3" ]]; then
+  die "Usage: $0 RUBY_VERSION RAILS_VERSION POSTGRES_USERNAME"
+fi
+
+
+source "$HOME/.rvm/scripts/rvm"
+
+set -e
+
+PGRESET_DIR="$(pwd)"
+RUBY_VERSION="ruby-$1"
+RAILS_VERSION="$2"
+PG_USER="$3"
+
+# can't have periods in the database name
+TEST_NAME="$(echo pgreset-test-rails-$RAILS_VERSION | sed 's/\./-/g')"
+
+rvm install "$RUBY_VERSION"
+rvm use "$RUBY_VERSION"
+rvm --force gemset delete "$TEST_NAME"
+rvm gemset create "$TEST_NAME"
+rvm gemset use "$TEST_NAME"
+cd /tmp
+rm -rf "$TEST_NAME"
+gem install rails -N -v "$RAILS_VERSION"
+rails new "$TEST_NAME"
+cd "$TEST_NAME"
+echo "gem 'pg'" >> Gemfile
+echo "gem 'pgreset', path: \"${PGRESET_DIR}\"" >> Gemfile
+bundle install
+rails g model user name:string
+
+echo "development: 
+  adapter: postgresql
+  encoding: UTF8
+  username: $PG_USER
+  host: /tmp
+  database: $TEST_NAME
+" > config/database.yml
+
+dropdb --if-exists -f "$TEST_NAME"
+rails db:create
+rails db:migrate
+
+
+# Start a connection to the database.
+# If pgreset works properly, this will automatically be disconnected
+# with an error saying the server closed the connection unexpectedly.
+psql -U "$PG_USER" "$TEST_NAME" -c 'select pg_sleep(120);' >log/psql.log 2>&1 &  
+sleep 1 # make sure psql is fully connected
+
+rails db:reset
+
+if grep 'terminating connection due to administrator command' log/psql.log >/dev/null ; then
+  echo "pgreset successfully terminated the postgres connection!"
+else
+  echo "Hmm something doesn't look right.  Here's the psql output:"
+  echo
+  cat log/psql.log
+fi

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pgreset
 version: !ruby/object:Gem::Version
-  version: '0.3'
+  version: '0.4'
 platform: ruby
 authors:
 - Dan Falcone
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2021-01-24 00:00:00.000000000 Z
+date: 2023-10-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -55,6 +55,7 @@ files:
 - bin/setup
 - lib/pgreset.rb
 - lib/pgreset/version.rb
+- pgreset-test.sh
 - pgreset.gemspec
 homepage: https://github.com/dafalcon/pgreset
 licenses:
@@ -75,7 +76,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.6
+rubygems_version: 3.2.33
 signing_key:
 specification_version: 4
 summary: Automatically kill postgres connections during db:reset