potluck-postgres 0.0.1 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5350781163dc218410b42e790bedb5c2ca246572e1334672ea9471c81efb432c
-  data.tar.gz: fdd55c1e5b645f6f0c99df93b9761e8830776b52b32e5310cd0def002e5c4f51
+  metadata.gz: 667d6b357e4b021bba60a415b90f8623d9b72683ed3b5869ce87f2e43e033325
+  data.tar.gz: 656fd69e5541b414c9e67f750daabaf072c94fbf1237042424a3a5848cfedfa8
 SHA512:
-  metadata.gz: 9e9475b41d894b36972aa8481eb06dfc6d1d0497d88a6841dfe92b5179bbb205db7f1cf48995df077aed00dbbc49371c553ccc5914154e15ca9c5684ed4bb9fe
-  data.tar.gz: d6df0284807374eafa9d9e48bebed6f79f00cbeb1398f6564831c2e1be703720ca059fbc5a09e0d84819f4f63f9fa5e8ff4519fc3f113a54f6696fd069c8fe86
+  metadata.gz: 03b852e0ce06ed33d0c7010779eb5bd6d1804e0151df4814a3ad218a50ffcaa27daebf47e7b7288ac7755c75ef4c11b003a82516f6a383e724dbbcdc4269759c
+  data.tar.gz: c3cfc79e9c846ca36e2da1015871cac23928bca1291bb7ee7c05ec8b459c71c4918bf057f70516da5dec9a6ab6942a4fc9d1428af970ab79bbfb20936039f5d9

data/lib/potluck/postgres.rb

@@ -1,17 +1,68 @@
 # frozen_string_literal: true
 
 require('potluck')
+require('sequel')
 
 module Potluck
-  class Postgres < Dish
+  ##
+  # Error class used to wrap errors encountered while connecting to or setting up a database.
+  #
+  class PostgresError < ServiceError
+    attr_reader(:wrapped_error)
+
+    ##
+    # Creates a new instance.
+    #
+    # * +message+ - Error message.
+    # * +wrapped_error+ - Original error that was rescued and is being wrapped by this one (optional).
+    #
+    def initialize(message, wrapped_error = nil)
+      super(message)
+
+      @wrapped_error = wrapped_error
+    end
+  end
+
+  ##
+  # A Ruby interface for controlling and connecting to Postgres. Uses
+  # [Sequel](https://github.com/jeremyevans/sequel) to connect and perform automatic role and database
+  # creation, as well as for utility methods such as database schema migration.
+  #
+  class Postgres < Service
+    ROLE_NOT_FOUND_REGEX = /role .* does not exist/.freeze
+    DATABASE_NOT_FOUND_REGEX = /database .* does not exist/.freeze
+
+    STARTING_UP_STRING = 'the database system is starting up'
+    STARTING_UP_TIMEOUT = 30
+
+    CONNECTION_REFUSED_STRING = 'connection refused'
+    CONNECTION_REFUSED_TIMEOUT = 3
+
     attr_reader(:database)
 
+    ##
+    # Creates a new instance.
+    #
+    # * +config+ - Configuration hash to pass to <tt>Sequel.connect</tt>.
+    # * +args+ - Arguments to pass to Potluck::Service.new (optional).
+    #
     def initialize(config, **args)
       super(**args)
 
       @config = config
     end
 
+    ##
+    # Disconnects and stops the Postgres process.
+    #
+    def stop
+      disconnect
+      super
+    end
+
+    ##
+    # Connects to the configured Postgres database.
+    #
     def connect
       (tries ||= 0) && (tries += 1)
       @database = Sequel.connect(@config, logger: @logger)
@@ -21,61 +72,43 @@ module Potluck
         Sequel.synchronize { Sequel::DATABASES.delete(dud) }
       end
 
-      if e.message =~ /role .* does not exist/ && tries == 1
+      message = e.message.downcase
+
+      if message =~ ROLE_NOT_FOUND_REGEX && tries == 1
         create_database_role
         create_database
         retry
-      elsif e.message =~ /database .* does not exist/ && tries == 1
+      elsif message =~ DATABASE_NOT_FOUND_REGEX && tries == 1
         create_database
         retry
-      elsif (@is_local && tries < 3) && (e.message.include?('could not connect') ||
-          e.message.include?('the database system is starting up'))
+      elsif message.include?(STARTING_UP_STRING) && tries < STARTING_UP_TIMEOUT
         sleep(1)
         retry
-      elsif e.message.include?('could not connect')
-        abort("#{e.class}: #{e.message.strip}")
+      elsif message.include?(CONNECTION_REFUSED_STRING) && tries < CONNECTION_REFUSED_TIMEOUT && manage?
+        sleep(1)
+        retry
+      elsif message.include?(CONNECTION_REFUSED_STRING)
+        raise(PostgresError.new(e.message.strip, e))
       else
-        abort("#{e.class}: #{e.message.strip}\n  #{e.backtrace.join("\n  ")}")
+        raise
       end
     end
 
+    ##
+    # Disconnects from the database if a connection was made.
+    #
     def disconnect
       @database&.disconnect
     end
 
-    def create_database_role
-      tmp_config = @config.dup
-      tmp_config[:database] = 'postgres'
-      tmp_config[:username] = ENV['USER']
-      tmp_config[:password] = nil
-
-      begin
-        Sequel.connect(tmp_config, logger: @logger) do |database|
-          database.execute("CREATE ROLE #{@config[:username]} WITH LOGIN CREATEDB REPLICATION PASSWORD "\
-            "'#{@config[:password]}'")
-        end
-      rescue => e
-        @logger.error("#{e.class}: #{e.message.strip}\n  #{e.backtrace.join("\n  ")}\n")
-        abort("Could not create role '#{@config[:username]}'. Make sure database user '#{ENV['USER']}' "\
-          'has permission to do so, or create it manually.')
-      end
-    end
-
-    def create_database
-      tmp_config = @config.dup
-      tmp_config[:database] = 'postgres'
-
-      begin
-        Sequel.connect(tmp_config, logger: @logger) do |database|
-          database.execute("CREATE DATABASE #{@config[:database]}")
-        end
-      rescue => e
-        @logger.error("#{e.class}: #{e.message.strip}\n  #{e.backtrace.join("\n  ")}\n")
-        abort("Could not create database '#{@config[:database]}'. Make sure database user "\
-          "'#{@config[:username]}' has permission to do so, or create it manually.")
-      end
-    end
-
+    ##
+    # Runs database migrations by way of Sequel's migration extension. Migration files must use the
+    # timestamp naming strategy as opposed to integers.
+    #
+    # * +dir+ - Directory where migration files are located.
+    # * +steps+ - Number of steps forward or backward to migrate from the current migration, otherwise will
+    #   migrate to latest (optional).
+    #
     def migrate(dir, steps = nil)
       return unless File.directory?(dir)
 
@@ -83,7 +116,7 @@ module Potluck
 
       # Suppress Sequel schema migration table queries.
       original_level = @logger.level
-      @logger.level = Logger::WARN
+      @logger.level = Logger::WARN if @logger.level == Logger::INFO
 
       args = [Sequel::Model.db, dir, {allow_missing_migration_files: true}]
       migrator = Sequel::TimestampMigrator.new(*args)
@@ -106,10 +139,55 @@ module Potluck
       migrator = Sequel::TimestampMigrator.new(*args)
       @logger.level = original_level
       migrator.run
+    ensure
+      @logger.level = original_level if original_level
     end
 
     private
 
+    ##
+    # Attempts to connect to the 'postgres' database as the system user with no password and create the
+    # configured role. Useful in development.
+    #
+    def create_database_role
+      tmp_config = @config.dup
+      tmp_config[:database] = 'postgres'
+      tmp_config[:username] = ENV['USER']
+      tmp_config[:password] = nil
+
+      begin
+        Sequel.connect(tmp_config, logger: @logger) do |database|
+          database.execute("CREATE ROLE #{@config[:username]} WITH LOGIN CREATEDB REPLICATION PASSWORD "\
+            "'#{@config[:password]}'")
+        end
+      rescue => e
+        raise(PostgresError.new("Database role #{@config[:username].inspect} could not be created using "\
+          "system user #{tmp_config[:username].inspect}. Please create the role manually.", e))
+      end
+    end
+
+    ##
+    # Attempts to connect to the 'postgres' database with the configured user and password and create the
+    # configured database. Useful in development.
+    #
+    def create_database
+      tmp_config = @config.dup
+      tmp_config[:database] = 'postgres'
+
+      begin
+        Sequel.connect(tmp_config, logger: @logger) do |database|
+          database.execute("CREATE DATABASE #{@config[:database]}")
+        end
+      rescue => e
+        raise(PostgresError.new("Database #{@config[:database].inspect} could not be created by "\
+          "connecting to system database #{tmp_config[:database].inspect}. Please create the database "\
+          'manually.', e))
+      end
+    end
+
+    ##
+    # Content of the launchctl plist file.
+    #
     def self.plist
       super(
         <<~EOS

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: potluck-postgres
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.5
 platform: ruby
 authors:
 - Nate Pickens
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-03-27 00:00:00.000000000 Z
+date: 2021-12-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: potluck
@@ -16,14 +16,28 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 0.0.1
+        version: 0.0.5
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 0.0.1
+        version: 0.0.5
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
 - !ruby/object:Gem::Dependency
   name: sequel
   requirement: !ruby/object:Gem::Requirement
@@ -105,7 +119,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.3
+rubygems_version: 3.2.32
 signing_key:
 specification_version: 4
 summary: A Ruby manager for Postgres.