avm-eac_postgresql_base0 0.2.4 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 03a08a9048593955e5cc62a604c6dfc94e21abe79e995d1933a3de00b908f18a
-  data.tar.gz: d29c537ded1c570897830df081f2c36f5101f5e3f508392e7279e21743465709
+  metadata.gz: 87fd0b8fc37e185451163ffce33b07b4e9dae7ac8a4c9b0e8b9b345abe31bd92
+  data.tar.gz: aa7ca433255e681a839e3d66f94b3cc15f59316aa5358f1d342c1068bc1c6ea4
 SHA512:
-  metadata.gz: 55b5a9708048a4d3fc467db99a3960e0dca5c505fdc8e07d3933c542fe0665532f5006ba3faa8bdd3f58ae0cdeedac3bea1c210056a59e7ec45a759d9c8de69d
-  data.tar.gz: e3d579fc45bda1bea8666ca036989b6bd684348668edebdffb5b699add72e38e20e15a55d0b911d52d5d4fc9fdc084d3d9512fcceacb99edfe8da43353515d46
+  metadata.gz: 79a4baf2e58a6d56b7046badcc10a8b6d2456f2e42c459c8356668942fa104f256d18f3a80996a54bc8f706624be4b6e21002871e83632119165903bb55fca5c
+  data.tar.gz: f73ad27f21f5a99fb84782504b891e20f3f659cbea4c4afc793709f37c855c25f46ba20158524811623e7980ecac5e608836ae92858270381115c0fec6e362d5

data/lib/avm/eac_postgresql_base0/instance/commands.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+
+require 'avm/eac_postgresql_base0/instance/data_unit'
+require 'eac_ruby_utils/core_ext'
+
+module Avm
+  module EacPostgresqlBase0
+    class Instance
+      module Commands
+        DUMP_EXCLUDE_PATTERN = '^(CREATE|COMMENT ON) EXTENSION'
+
+        # @return [EacRubyUtils::Envs::Command]
+        def dump_command
+          env.command('pg_dump', '--no-privileges', '--no-owner', *common_command_args)
+             .envvar('PGPASSWORD', password)
+             .pipe(remove_extensions_ddl)
+        end
+
+        # @return [EacRubyUtils::Envs::Command]
+        def dump_gzip_command
+          dump_command.pipe(gzip_compress_command)
+        end
+
+        # @return [EacRubyUtils::Envs::Command]
+        def gzip_compress_command
+          env.command('gzip', '-9', '-c')
+        end
+
+        # @return [EacRubyUtils::Envs::Command]
+        def gzip_decompress_command
+          env.command('gzip', '-d')
+        end
+
+        # @return [EacRubyUtils::Envs::Command]
+        def load_gzip_command
+          gzip_decompress_command.pipe(psql_command)
+        end
+
+        # @return [String]
+        def password_command_argument
+          "@ESC_PGPASSWORD=#{password}"
+        end
+
+        def psql_command(database = true)
+          env.command(password_command_argument, 'psql', '--variable', 'ON_ERROR_STOP=t',
+                      *common_command_args(database))
+        end
+
+        def psql_command_command(sql, database = true)
+          psql_command(database).append(['--quiet', '--tuples-only', '--command', sql])
+        end
+
+        def root_psql_command(sql = nil)
+          args = ['sudo', '-u', 'postgres', 'psql']
+          args += ['--quiet', '--tuples-only', '--command', sql] if sql.present?
+          env.command(*args)
+        end
+
+        def common_command_args(database = true)
+          ['--host', host, '--username', user, '--port', port,
+           (database ? name : MAINTENANCE_DATABASE)]
+        end
+
+        private
+
+        # @return [EacRubyUtils::Envs::Command]
+        def remove_extensions_ddl
+          env.command('grep', '--invert-match', '--extended-regexp', DUMP_EXCLUDE_PATTERN)
+        end
+      end
+    end
+  end
+end

data/lib/avm/eac_postgresql_base0/instance/data_unit.rb

@@ -7,9 +7,23 @@ module Avm
     class Instance
       class DataUnit < ::Avm::Instances::Data::Unit
         EXTENSION = '.pgdump.gz'
-        TABLES_SQL = 'select tablename from pg_tables where schemaname = \'public\''
+        SCHEMA_VAR = '%%SCHEMA%%'
+        TABLE_PARTS_SEPARATOR = '/'
+        TABLES_SQL = "select schemaname || '#{TABLE_PARTS_SEPARATOR}' || tablename from " \
+          "pg_tables where schemaname = '#{SCHEMA_VAR}'"
 
-        before_load :clear_database
+        before_load :clear
+
+        def clear
+          info 'Clearing database (Dropping all tables)...'
+          ts = tables
+          if ts.empty?
+            info 'Database has no tables'
+          else
+            info "Removing #{ts.count} table(s)..."
+            run_sql(drop_tables_sql(ts))
+          end
+        end
 
         def dump_command
           instance.dump_gzip_command
@@ -21,23 +35,32 @@ module Avm
 
         private
 
-        def clear_database
-          info 'Clearing database (Dropping all tables)...'
-          ts = tables
-          if ts.empty?
-            info 'Database has no tables'
-          else
-            info "Removing #{ts.count} table(s)..."
-            run_sql('drop table if exists ' + ts.map { |t| "\"#{t}\"" }.join(', ') + ' cascade')
-          end
+        # @param table_list [Array<String>]
+        # @return [String]
+        def drop_tables_sql(table_list)
+          'drop table ' + table_list.map(&:to_s).join(', ') + ' cascade'
+        end
+
+        # @param parts [Array<String>, Strings]
+        # @return [String]
+        def join_table_parts(parts)
+          parts = parts.to_s.split(TABLE_PARTS_SEPARATOR) unless parts.is_a?(::Enumerable)
+          parts.map { |p| "\"#{p}\"" }.join('.')
         end
 
         def run_sql(sql)
           instance.psql_command_command(sql).execute!
         end
 
+        # @return [Array<String>]
         def tables
-          run_sql(TABLES_SQL).each_line.map(&:strip).reject(&:blank?)
+          run_sql(tables_sql).each_line.map(&:strip).reject(&:blank?)
+                             .map { |line| join_table_parts(line) }
+        end
+
+        # @return [String]
+        def tables_sql
+          TABLES_SQL.gsub(SCHEMA_VAR, instance.schema)
         end
       end
     end

data/lib/avm/eac_postgresql_base0/instance.rb

@@ -6,8 +6,11 @@ require 'eac_ruby_utils/core_ext'
 module Avm
   module EacPostgresqlBase0
     class Instance
-      require_sub __FILE__
+      require_sub __FILE__, include_modules: true
 
+      DEFAULT_HOSTNAME = '127.0.0.1'
+      DEFAULT_PORT = 5432
+      DEFAULT_SCHEMA = 'public'
       MAINTENANCE_DATABASE = 'postgres'
 
       common_constructor :env, :connection_params do
@@ -22,61 +25,12 @@ module Avm
         ::Avm::EacPostgresqlBase0::Instance::DataUnit.new(self)
       end
 
-      def dump_command
-        env.command('pg_dump', '--no-privileges', '--clean', '--no-owner', *common_command_args)
-           .envvar('PGPASSWORD', password)
-      end
-
-      # @return [EacRubyUtils::Envs::Command]
-      def dump_gzip_command
-        dump_command.pipe(gzip_compress_command)
-      end
-
-      # @return [EacRubyUtils::Envs::Command]
-      def gzip_compress_command
-        env.command('gzip', '-9', '-c')
-      end
-
-      # @return [EacRubyUtils::Envs::Command]
-      def gzip_decompress_command
-        env.command('gzip', '-d')
-      end
-
-      # @return [EacRubyUtils::Envs::Command]
-      def load_gzip_command
-        gzip_decompress_command.pipe(psql_command)
-      end
-
-      # @return [String]
-      def password_command_argument
-        "@ESC_PGPASSWORD=#{password}"
-      end
-
-      def psql_command(database = true)
-        env.command(password_command_argument, 'psql', *common_command_args(database))
-      end
-
-      def psql_command_command(sql, database = true)
-        psql_command(database).append(['--quiet', '--tuples-only', '--command', sql])
-      end
-
-      def root_psql_command(sql = nil)
-        args = ['sudo', '-u', 'postgres', 'psql']
-        args += ['--quiet', '--tuples-only', '--command', sql] if sql.present?
-        env.command(*args)
-      end
-
-      def common_command_args(database = true)
-        ['--host', host, '--username', user, '--port', port,
-         (database ? name : MAINTENANCE_DATABASE)]
-      end
-
       def host
-        connection_params[:host] || '127.0.0.1'
+        connection_params[:host] || DEFAULT_HOSTNAME
       end
 
       def port
-        connection_params[:port] || '5432'
+        connection_params[:port] || DEFAULT_PORT
       end
 
       def user
@@ -90,6 +44,10 @@ module Avm
       def name
         connection_params.fetch(:name)
       end
+
+      def schema
+        connection_params[:schema] || DEFAULT_SCHEMA
+      end
     end
   end
 end

data/lib/avm/eac_postgresql_base0/version.rb

@@ -2,6 +2,6 @@
 
 module Avm
   module EacPostgresqlBase0
-    VERSION = '0.2.4'
+    VERSION = '0.4.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: avm-eac_postgresql_base0
 version: !ruby/object:Gem::Version
-  version: 0.2.4
+  version: 0.4.0
 platform: ruby
 authors:
 - Put here the authors
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-04-14 00:00:00.000000000 Z
+date: 2023-05-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: avm
@@ -16,48 +16,42 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.66'
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 0.66.1
+        version: '0.69'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.66'
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 0.66.1
+        version: '0.69'
 - !ruby/object:Gem::Dependency
   name: eac_ruby_utils
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.112'
+        version: '0.113'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.112'
+        version: '0.113'
 - !ruby/object:Gem::Dependency
   name: eac_ruby_gem_support
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.4'
+        version: 0.5.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.4'
+        version: 0.5.1
 description: 
 email: 
 executables: []
@@ -67,6 +61,7 @@ files:
 - lib/avm/eac_postgresql_base0.rb
 - lib/avm/eac_postgresql_base0/instance.rb
 - lib/avm/eac_postgresql_base0/instance/assert.rb
+- lib/avm/eac_postgresql_base0/instance/commands.rb
 - lib/avm/eac_postgresql_base0/instance/data_unit.rb
 - lib/avm/eac_postgresql_base0/instance_with.rb
 - lib/avm/eac_postgresql_base0/version.rb