active_postgres 0.7.0 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 503ec5557f2365458cfe544af059bb5636e82be153a58926978c62ff43384b6f
-  data.tar.gz: 20d9bad0c0cbde26d214d304f379854d8d8dd548f06f86462840dad56f0f4580
+  metadata.gz: 0266b3f7dabc2d844b30aa6410cfd7c59131a7fb77a2dea85f01f9a66c2a2443
+  data.tar.gz: fd720306229f1ddea1d4e01a9dc3a5df3b7a60a620c38e86779e82f9698ebcbc
 SHA512:
-  metadata.gz: 26ad8c310458b4e4ecdce1586b98cce9b062260b060e8779481e965c3d2c3d092a93691956372dbf62bd3e138d2b68e97b9dc135748da8a5f4481b683d739c1b
-  data.tar.gz: c9ee9f54f725cf43398815a79c621579ac1dad70b36ab997a413905a8cd5222eea42ca0e3397928f26ccf48fc1f1c10da18d9aef31cabfd62eff4ab3aadd0f84
+  metadata.gz: bb3bc17966f9ebe98a55f09f678fe2bc2e57b92615c45848c347a7568f6e3bb61d15788d6bb16008e4498a9b944a3f53241f2e2579b204d28679341b041f4978
+  data.tar.gz: 3863cace22d1ba3aea901d312d26152de067c7ca898ddff5b654a75702a9e6b4c01f6ffbac820fed338fc5db8ce011ad39676143f81f785f66f7fec72cd72ae5

data/lib/active_postgres/configuration.rb

@@ -8,7 +8,6 @@ module ActivePostgres
       @environment = environment
       env_config = config_hash[environment] || {}
 
-      # Check if deployment should be skipped (e.g., for development)
       @skip_deployment = env_config['skip_deployment'] == true
 
       @version = env_config['version'] || 18
@@ -65,6 +64,16 @@ module ActivePostgres
       private_ip_for(node) || host
     end
 
+    # Returns the host to use for direct PostgreSQL connections (private_ip preferred)
+    def connection_host_for(host)
+      node = node_config_for(host)
+      private_ip_for(node) || host
+    end
+
+    def primary_connection_host
+      connection_host_for(primary_host)
+    end
+
     def standby_config_for(host)
       @standbys.find { |s| s['host'] == host }
     end

data/lib/active_postgres/direct_executor.rb

@@ -0,0 +1,92 @@
+require 'pg'
+
+module ActivePostgres
+  class DirectExecutor
+    attr_reader :config
+
+    def initialize(config, quiet: false)
+      @config = config
+      @quiet = quiet
+    end
+
+    def quiet?
+      @quiet
+    end
+
+    def postgres_running?(host)
+      with_connection(host) do |conn|
+        conn.exec('SELECT 1')
+        true
+      end
+    rescue PG::Error
+      false
+    end
+
+    def run_sql(host, sql)
+      with_connection(host) do |conn|
+        result = conn.exec(sql)
+        result.values.flatten.join("\n")
+      end
+    rescue PG::Error => e
+      raise Error, "Failed to execute SQL on #{host}: #{e.message}"
+    end
+
+    def get_postgres_status(host)
+      run_sql(host, 'SELECT version();')
+    end
+
+    private
+
+    def with_connection(host)
+      connection_host = config.connection_host_for(host)
+      superuser_password = resolve_secret(config.secrets_config['superuser_password'])
+
+      conn = PG.connect(
+        host: connection_host,
+        port: 5432,
+        dbname: 'postgres',
+        user: config.postgres_user,
+        password: superuser_password,
+        connect_timeout: 10,
+        sslmode: config.component_enabled?(:ssl) ? 'require' : 'prefer'
+      )
+
+      begin
+        yield conn
+      ensure
+        conn.close
+      end
+    end
+
+    def resolve_secret(value)
+      return nil if value.nil?
+
+      # Handle Rails credentials
+      if value.start_with?('rails_credentials:')
+        return resolve_rails_credentials(value)
+      end
+
+      # Handle environment variables
+      if value.start_with?('$')
+        return ENV[value[1..]]
+      end
+
+      # Handle shell commands
+      if value.start_with?('$(') && value.end_with?(')')
+        return `#{value[2..-2]}`.strip
+      end
+
+      value
+    end
+
+    def resolve_rails_credentials(value)
+      path = value.sub('rails_credentials:', '')
+      keys = path.split('.').map(&:to_sym)
+
+      credentials = Rails.application.credentials
+      keys.reduce(credentials) { |obj, key| obj&.dig(key) }
+    rescue NameError
+      nil
+    end
+  end
+end

data/lib/active_postgres/health_checker.rb

@@ -1,10 +1,19 @@
 module ActivePostgres
   class HealthChecker
-    attr_reader :config, :ssh_executor
+    attr_reader :config, :executor
 
     def initialize(config)
       @config = config
-      @ssh_executor = SSHExecutor.new(config, quiet: true)
+      @executor = create_executor
+    end
+
+    # Backwards compatibility alias
+    def ssh_executor
+      @executor
+    end
+
+    private def create_executor
+      DirectExecutor.new(config, quiet: true)
     end
 
     def show_status
@@ -287,31 +296,29 @@ module ActivePostgres
     def check_pgbouncer_status(host)
       return '-' unless config.component_enabled?(:pgbouncer)
 
-      ssh_executor.execute_on_host(host) do
-        result = capture(:sudo, 'systemctl', 'is-active', 'pgbouncer').strip
-        result == 'active' ? '✓ running' : '✗ down'
-      end
+      check_pgbouncer_direct(host) ? '✓ running' : '✗ down'
     rescue StandardError
       '✗ down'
     end
 
     def check_pgbouncer_running(host)
-      ssh_executor.execute_on_host(host) do
-        result = capture(:sudo, 'systemctl', 'is-active', 'pgbouncer').strip
-        result == 'active'
-      end
+      check_pgbouncer_direct(host)
     rescue StandardError
       false
     end
 
-    def check_pgbouncer_userlist(host)
-      app_user = config.app_user
-      return true unless app_user # No app user configured, skip check
+    def check_pgbouncer_userlist(_host)
+      true
+    end
+
+    def check_pgbouncer_direct(host)
+      require 'socket'
+      connection_host = config.connection_host_for(host)
+      pgbouncer_port = config.component_config(:pgbouncer)[:listen_port] || 6432
 
-      ssh_executor.execute_on_host(host) do
-        userlist = capture(:sudo, 'cat', '/etc/pgbouncer/userlist.txt').strip
-        userlist.include?(app_user)
-      end
+      socket = TCPSocket.new(connection_host, pgbouncer_port)
+      socket.close
+      true
     rescue StandardError
       false
     end

data/lib/active_postgres/version.rb

@@ -1,3 +1,3 @@
 module ActivePostgres
-  VERSION = '0.7.0'.freeze
+  VERSION = '0.8.0'.freeze
 end

data/lib/active_postgres.rb

@@ -21,6 +21,7 @@ require_relative 'active_postgres/standby_deployment_flow'
 require_relative 'active_postgres/cli'
 require_relative 'active_postgres/installer'
 require_relative 'active_postgres/ssh_executor'
+require_relative 'active_postgres/direct_executor'
 require_relative 'active_postgres/health_checker'
 require_relative 'active_postgres/failover'
 require_relative 'active_postgres/performance_tuner'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: active_postgres
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.8.0
 platform: ruby
 authors:
 - BoringCache
@@ -169,6 +169,7 @@ files:
 - lib/active_postgres/connection_pooler.rb
 - lib/active_postgres/credentials.rb
 - lib/active_postgres/deployment_flow.rb
+- lib/active_postgres/direct_executor.rb
 - lib/active_postgres/error_handler.rb
 - lib/active_postgres/failover.rb
 - lib/active_postgres/generators/active_postgres/install_generator.rb
@@ -220,7 +221,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.7.2
+rubygems_version: 3.6.9
 specification_version: 4
 summary: PostgreSQL High Availability for Rails, made simple
 test_files: []