pg_export 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 69622bfd641c2024a474d24d95c53e2baf552a2a
-  data.tar.gz: 7123b34af0f29d2b36b9690efb4b64afcdb60130
+  metadata.gz: ed370ed90634fe7db175d5e9f93ffb9d423dad07
+  data.tar.gz: 901040b143cef0587fcf9d14f9a677fbbf8fcb54
 SHA512:
-  metadata.gz: 44241da6e0c062993be0990996cefeb784fe1bc9d2df71aa56743f262f123e8f6f8a068ee320e0a1d43922201cab3639e7f3aae46b748fceca38efa2ae1709bf
-  data.tar.gz: f69d004ff3a44aa4e014bf230f78ba9bd91445149736f412ec3ed13b5b1647a81f13f7f6d532c484d67cdce7fc2ee0982fcf482992c8d11a05822afd90edac6b
+  metadata.gz: ae0a892e79679675d220fc0a91b9c579e4083f9e1c5791c66bd0ab4fcb6d4097da8c242567a0be935d61ab084d7b392f05a4bbb19ac1f63564aee36e8441fae9
+  data.tar.gz: 19fd6f4cd3eb192dd8b6a0a93f27badb704c9116fdb1ed6c9fd54efb08214e18f801b54d1a6ee5286606fe13453adff294cfcf2b91a6f1438807d6e7ff0047d1

data/.rubocop.yml

@@ -19,11 +19,7 @@ Lint/NestedMethodDefinition:
 
 Lint/HandleExceptions:
   Exclude:
-    - 'spec/actions/create_dump_spec.rb'
-
-Lint/AssignmentInCondition:
-  Exclude:
-    - 'lib/pg_export/actions/compress_dump.rb'
+    - 'spec/services/utils_spec.rb'
 
 Lint/AmbiguousOperator:
   Exclude:

data/README.md

@@ -1,15 +1,26 @@
 # PgExport
 
-CLI for exporting PostgreSQL database dump to FTP.
+CLI for creating and exporting PostgreSQL dumps to FTP.
 
 Can be used for backups or synchronizing databases between production and development environments.
 
-Configurable by ENV variables and (partially) by command line options.
+Example:
+
+    pg_export --database database_name -keep 5
+    
+Above command will perform database dump, gzip it, upload it to FTP and remove old dumps from FTP, keeping newest 5.
+
+FTP connection params are configured by env variables.
+
+Features:
+
+- uses shell command pg_dump
+- no external gem dependencies
+- uses ruby tempfiles, so local dumps are garbage collected automatically
 
 ## Dependencies
 
   * Ruby >= 2.1
-  * PostgreSQL >= 9.1
   * $ pg_dump
 
 ## Installation
@@ -33,12 +44,14 @@ Or install it yourself as:
     $ pg_export -h
 
     Usage: pg_export [options]
-        -c, --configuration              Prints the configuration
-        -f, --ftp                        Tries connecting to FTP to verify the connection
         -d, --database DATABASE          [Required] Name of the database to export
-        -k, --keep [KEEP]                [Optional] Number of dump files to keep locally and on FTP (default: 10)
+        -k, --keep [KEEP]                [Optional] Number of dump files to keep on FTP (default: 10)
         -t, --timestamped                [Optional] Enables log messages with timestamps
         -h, --help                       Show this message
+    
+    Setting can be verified by running following commands:
+        -c, --configuration              Prints the configuration
+        -f, --ftp                        Tries connecting to FTP to verify the connection
 
 ## How to start
 
@@ -54,17 +67,13 @@ Password cannot include `#` sign, [more info](http://serverfault.com/questions/5
 __Step 2.__ Configure other variables (optional).
 
     # /etc/environment
-    DUMPFILE_DIR="/var/dumps"  # default: "tmp/dumps"
-    KEEP_DUMPS=3               # default: 10
-    KEEP_FTP_DUMPS=5           # default: 10
+    KEEP_DUMPS=5   # default: 10
 
-__Step 3.__ Print the configuration to verify if env variables has been loaded.
+__Step 3.__ Print the configuration to verify whether env variables has been loaded.
 
     $ pg_export --configuration
     => database: 
-       dumpfile_dir: /var/dumps
-       keep_dumps: 3
-       keep_ftp_dumps: 5
+       keep_dumps: 5
        ftp_host: yourftp.example.com
        ftp_user: user
        ftp_password: pass***
@@ -78,10 +87,10 @@ __Step 4.__ Try connecting to FTP to verify the connection.
 __Step 5.__ Perform database export.
 
     $ pg_export -d your_database
-    => Dump your_database to /var/dumps/your_database_20161020_125747 (892.38kB)
-       Zip dump your_database_20161020_125747.gz (874.61kB)
+    => Create Dump Tempfile (1.36MB)
+       Create Compressed Dump Tempfile (1.34MB)
        Connect to yourftp.example.com
-       Export your_database_20161020_125747.gz to FTP
+       Export Compressed Dump Tempfile (1.34MB) your_database_20161020_125747.gz to yourftp.example.com
        Close FTP
 
 ## Development

data/bin/pg_export

@@ -15,7 +15,7 @@ option_parser = OptionParser.new do |opts|
     options.database = database
   end
 
-  opts.on('-k', '--keep [KEEP]', Integer, '[Optional] Number of dump files to keep locally and on FTP (default: 10)') do |keep|
+  opts.on('-k', '--keep [KEEP]', Integer, '[Optional] Number of dump files to keep on FTP (default: 10)') do |keep|
     options.keep = keep
   end
 
@@ -50,11 +50,10 @@ begin
   PgExport.new do |config|
     config.database = options.database if options.database
     config.keep_dumps = options.keep if options.keep
-    config.keep_ftp_dumps = options.keep if options.keep
   end.call
 rescue OptionParser::MissingArgument, PgExport::InvalidConfigurationError => e
   puts "Error: #{e}"
   puts option_parser
-rescue PgExport::DatabaseDoesNotExistError, PgExport::DependencyRequiredError => e
+rescue PgExport::PgDumpError => e
   puts e
 end

data/lib/pg_export.rb

@@ -1,22 +1,24 @@
 require 'logger'
-require 'pathname'
-require 'fileutils'
+require 'tempfile'
 require 'zlib'
 require 'net/ftp'
 
-require 'pg'
-
 require 'pg_export/version'
 require 'pg_export/logging'
-require 'pg_export/configuration'
 require 'pg_export/errors'
-require 'pg_export/dump'
-require 'pg_export/actions'
-require 'pg_export/ftp_service'
-require 'pg_export/ftp_service/connection'
+require 'pg_export/configuration'
+require 'pg_export/concurrency'
+require 'pg_export/entities/dump/size_human'
+require 'pg_export/entities/dump/base'
+require 'pg_export/entities/sql_dump'
+require 'pg_export/entities/compressed_dump'
+require 'pg_export/services/ftp_service'
+require 'pg_export/services/ftp_service/connection'
+require 'pg_export/services/utils'
+require 'pg_export/services/dump_storage'
 
 class PgExport
-  include Logging
+  include Concurrency
 
   def initialize
     @config = Configuration.new
@@ -25,42 +27,28 @@ class PgExport
   end
 
   def call
-    initialize_dump
     concurrently do |job|
-      job << Thread.new { perform_local_job }
-      job << Thread.new { initialize_ftp_service }
+      job << create_dump
+      job << initialize_dump_storage
     end
-    perform_ftp_job
+    dump_storage.upload(dump)
+    dump_storage.remove_old(keep: config.keep_dumps)
     self
   end
 
   private
 
   attr_reader :config
-  attr_accessor :ftp_service, :dump
-
-  def initialize_dump
-    self.dump = Dump.new(config.database, config.dumpfile_dir)
-  end
-
-  def initialize_ftp_service
-    self.ftp_service = FtpService.new(config.ftp_params)
-  end
-
-  def concurrently
-    t = []
-    yield t
-    t.each(&:join)
-  end
+  attr_accessor :dump, :dump_storage
 
-  def perform_local_job
-    CreateDump.new(dump).call
-    CompressDump.new(dump).call
-    RemoveOldDumps.new(dump, config.keep_dumps).call
+  def initialize_dump_storage
+    ftp_service = FtpService.new(config.ftp_params)
+    self.dump_storage = DumpStorage.new(ftp_service, config.database)
   end
 
-  def perform_ftp_job
-    SendDumpToFtp.new(dump, ftp_service).call
-    RemoveOldDumpsFromFtp.new(dump, ftp_service, config.keep_ftp_dumps).call
+  def create_dump
+    sql_dump = Utils.create_dump(config.database)
+    compressed_dump = Utils.compress(sql_dump)
+    self.dump = compressed_dump
   end
 end

data/lib/pg_export/concurrency.rb

@@ -0,0 +1,21 @@
+class PgExport
+  module Concurrency
+    class ThreadsArray < Array
+      def <<(job)
+        super Thread.new { job }
+      end
+
+      alias push <<
+    end
+
+    def self.included(*)
+      Thread.abort_on_exception = true
+    end
+
+    def concurrently
+      t = ThreadsArray.new
+      yield t
+      t.each(&:join)
+    end
+  end
+end

data/lib/pg_export/configuration.rb

@@ -2,9 +2,7 @@ class PgExport
   class Configuration
     DEFAULTS = {
       database: nil,
-      dumpfile_dir: ENV['DUMPFILE_DIR'] || 'tmp/dumps',
       keep_dumps: ENV['KEEP_DUMPS'] || 10,
-      keep_ftp_dumps: ENV['KEEP_FTP_DUMPS'] || 10,
       ftp_host: ENV['BACKUP_FTP_HOST'],
       ftp_user: ENV['BACKUP_FTP_USER'],
       ftp_password: ENV['BACKUP_FTP_PASSWORD']

data/lib/pg_export/entities/compressed_dump.rb

@@ -0,0 +1,19 @@
+class PgExport
+  class CompressedDump < Dump::Base
+    def name
+      'Compressed Dump'
+    end
+
+    def ext
+      '.gz'
+    end
+
+    def open(operation_type, &block)
+      case operation_type.to_sym
+        when :read then Zlib::GzipReader.open(path, &block)
+        when :write then Zlib::GzipWriter.open(path, &block)
+        else raise ArgumentError, 'Operation type can be only :read or :write'
+      end
+    end
+  end
+end

data/lib/pg_export/entities/dump/base.rb

@@ -0,0 +1,32 @@
+class PgExport
+  module Dump
+    class Base
+      extend Forwardable
+      include SizeHuman
+
+      CHUNK_SIZE = (2**16).freeze
+
+      def_delegators :file, :path, :read, :write, :rewind, :size, :eof?
+
+      def initialize
+        @file = Tempfile.new('dump')
+      end
+
+      def ext
+        raise 'Overwrite it'
+      end
+
+      def read_chunk
+        raise 'Overwrite it'
+      end
+
+      def to_s
+        "#{name || self.class} #{file.class} (#{size_human})"
+      end
+
+      private
+
+      attr_reader :file
+    end
+  end
+end

data/lib/pg_export/entities/dump/size_human.rb

@@ -0,0 +1,15 @@
+class PgExport
+  module Dump
+    module SizeHuman
+      def size_human
+        {
+          'B'  => 1024,
+          'kB' => 1024 * 1024,
+          'MB' => 1024 * 1024 * 1024,
+          'GB' => 1024 * 1024 * 1024 * 1024,
+          'TB' => 1024 * 1024 * 1024 * 1024 * 1024
+        }.each_pair { |e, s| return "#{(size.to_f / (s / 1024)).round(2)}#{e}" if size < s }
+      end
+    end
+  end
+end

data/lib/pg_export/entities/sql_dump.rb

@@ -0,0 +1,15 @@
+class PgExport
+  class SqlDump < Dump::Base
+    def name
+      'Dump'
+    end
+
+    def ext
+      ''
+    end
+
+    def read_chunk
+      read(CHUNK_SIZE)
+    end
+  end
+end

data/lib/pg_export/errors.rb

@@ -1,6 +1,4 @@
 class PgExport
-  class DependencyRequiredError < StandardError; end
-  class DatabaseDoesNotExistError < StandardError; end
-  class DumpFileDoesNotExistError < StandardError; end
   class InvalidConfigurationError < StandardError; end
+  class PgDumpError < StandardError; end
 end

data/lib/pg_export/services/dump_storage.rb

@@ -0,0 +1,37 @@
+class PgExport
+  class DumpStorage
+    include Logging
+
+    TIMESTAMP = '_%Y%m%d_%H%M%S'.freeze
+    TIMESTAMP_REGEX = '[0-9]{8}_[0-9]{6}'.freeze
+
+    def initialize(ftp_service, name)
+      @ftp_service, @name = ftp_service, name
+    end
+
+    def upload(dump)
+      dump_name = timestamped_name(dump)
+      ftp_service.upload_file(dump.path, dump_name)
+      logger.info "Export #{dump} #{dump_name} to #{ftp_service}"
+    end
+
+    def remove_old(keep:)
+      ftp_service.list(ftp_regex).drop(keep.to_i).each do |filename|
+        ftp_service.delete(filename)
+        logger.info "Remove #{filename} from FTP"
+      end
+    end
+
+    private
+
+    attr_reader :ftp_service, :name
+
+    def timestamped_name(dump)
+      name + Time.now.strftime(TIMESTAMP) + dump.ext
+    end
+
+    def ftp_regex
+      name + '_*'
+    end
+  end
+end

data/lib/pg_export/{ftp_service.rb → services/ftp_service.rb}

@@ -1,6 +1,9 @@
 class PgExport
   class FtpService
+    CHUNK_SIZE = (2**16).freeze
+
     def initialize(params)
+      @host = params.fetch(:host)
       connection = Connection.new(params)
       @ftp = connection.ftp
       ObjectSpace.define_finalizer(self, proc { connection.close })
@@ -14,12 +17,16 @@ class PgExport
       ftp.delete(filename)
     end
 
-    def upload_file(path)
-      ftp.putbinaryfile(path.to_s)
+    def upload_file(path, name)
+      ftp.putbinaryfile(path.to_s, name, CHUNK_SIZE)
+    end
+
+    def to_s
+      host
     end
 
     private
 
-    attr_reader :ftp
+    attr_reader :ftp, :host
   end
 end

data/lib/pg_export/services/utils.rb

@@ -0,0 +1,34 @@
+class PgExport
+  class Utils
+    extend Logging
+
+    def self.create_dump(database_name)
+      dump = SqlDump.new
+      out = `pg_dump -Fc --file #{dump.path} #{database_name} 2>&1`
+      raise PgDumpError, out if /FATAL/ =~ out
+      logger.info "Create #{dump}"
+      dump
+    end
+
+    def self.compress(dump)
+      dump_gz = CompressedDump.new
+      dump_gz.open(:write) do |gz|
+        gz.write(dump.read_chunk) until dump.eof?
+      end
+
+      logger.info "Create #{dump_gz}"
+      dump_gz
+    end
+
+    def self.decompress(dump_gz)
+      dump = SqlDump.new
+      dump_gz.open(:read) do |gz|
+        dump.write(gz.readpartial(Dump::Base::CHUNK_SIZE)) until gz.eof?
+      end
+      dump.rewind
+
+      logger.info "Create #{dump}"
+      dump
+    end
+  end
+end

data/lib/pg_export/version.rb

@@ -1,3 +1,3 @@
 class PgExport
-  VERSION = '0.2.0'.freeze
+  VERSION = '0.3.0'.freeze
 end

data/pg_export.gemspec

@@ -9,9 +9,9 @@ Gem::Specification.new do |spec|
   spec.authors       = ['Krzysztof Maicher']
   spec.email         = ['krzysztof.maicher@gmail.com']
 
-  spec.summary       = 'CLI for exporting Rails Postgres database dump to FTP'
-  spec.description   = "CLI for exporting Rails Postgres database dump to FTP.\
-                        Can be used for backups or synchronizing databases between production and development environments"
+  spec.summary       = 'CLI for creating and exporting PostgreSQL dumps to FTP.'
+  spec.description   = "CLI for creating and exporting PostgreSQL dumps to FTP.\
+                        Can be used for backups or synchronizing databases between production and development environments."
   spec.homepage      = 'https://github.com/maicher/pg_export'
   spec.license       = 'MIT'
 
@@ -20,11 +20,10 @@ Gem::Specification.new do |spec|
   spec.require_paths = ['lib']
   spec.required_ruby_version = '>= 2.1.0'
 
-  spec.add_dependency 'pg', '>= 0.16'
-
   spec.add_development_dependency 'bundler', '~> 1.10'
   spec.add_development_dependency 'rubocop', '~> 0.44'
   spec.add_development_dependency 'rake', '~> 10.0'
-  spec.add_development_dependency 'rspec', '~>3.4'
+  spec.add_development_dependency 'rspec', '~> 3.4'
+  spec.add_development_dependency 'pg', '~> 0.19'
   spec.add_development_dependency 'pry'
 end

metadata

@@ -1,29 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: pg_export
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Krzysztof Maicher
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-10-20 00:00:00.000000000 Z
+date: 2016-10-25 00:00:00.000000000 Z
 dependencies:
-- !ruby/object:Gem::Dependency
-  name: pg
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0.16'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0.16'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -80,6 +66,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.4'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.19'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.19'
 - !ruby/object:Gem::Dependency
   name: pry
   requirement: !ruby/object:Gem::Requirement
@@ -94,9 +94,9 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: CLI for exporting Rails Postgres database dump to FTP.                        Can
+description: CLI for creating and exporting PostgreSQL dumps to FTP.                        Can
   be used for backups or synchronizing databases between production and development
-  environments
+  environments.
 email:
 - krzysztof.maicher@gmail.com
 executables:
@@ -117,18 +117,18 @@ files:
 - bin/pg_export
 - bin/setup
 - lib/pg_export.rb
-- lib/pg_export/actions.rb
-- lib/pg_export/actions/compress_dump.rb
-- lib/pg_export/actions/create_dump.rb
-- lib/pg_export/actions/remove_old_dumps.rb
-- lib/pg_export/actions/remove_old_dumps_from_ftp.rb
-- lib/pg_export/actions/send_dump_to_ftp.rb
+- lib/pg_export/concurrency.rb
 - lib/pg_export/configuration.rb
-- lib/pg_export/dump.rb
+- lib/pg_export/entities/compressed_dump.rb
+- lib/pg_export/entities/dump/base.rb
+- lib/pg_export/entities/dump/size_human.rb
+- lib/pg_export/entities/sql_dump.rb
 - lib/pg_export/errors.rb
-- lib/pg_export/ftp_service.rb
-- lib/pg_export/ftp_service/connection.rb
 - lib/pg_export/logging.rb
+- lib/pg_export/services/dump_storage.rb
+- lib/pg_export/services/ftp_service.rb
+- lib/pg_export/services/ftp_service/connection.rb
+- lib/pg_export/services/utils.rb
 - lib/pg_export/version.rb
 - pg_export.gemspec
 homepage: https://github.com/maicher/pg_export
@@ -154,5 +154,5 @@ rubyforge_project:
 rubygems_version: 2.4.8
 signing_key: 
 specification_version: 4
-summary: CLI for exporting Rails Postgres database dump to FTP
+summary: CLI for creating and exporting PostgreSQL dumps to FTP.
 test_files: []

data/lib/pg_export/actions.rb

@@ -1,5 +0,0 @@
-require 'pg_export/actions/create_dump'
-require 'pg_export/actions/compress_dump'
-require 'pg_export/actions/remove_old_dumps'
-require 'pg_export/actions/send_dump_to_ftp'
-require 'pg_export/actions/remove_old_dumps_from_ftp'

data/lib/pg_export/actions/compress_dump.rb

@@ -1,39 +0,0 @@
-class PgExport
-  class CompressDump
-    include Logging
-
-    def initialize(dump)
-      @dump = dump
-    end
-
-    def call
-      validate_dumpfile_exists
-      compress_dumpfile
-      remove_dumpfile
-      self
-    end
-
-    private
-
-    attr_reader :dump
-
-    def validate_dumpfile_exists
-      File.exist?(dump.pathname) or raise DumpFileDoesNotExistError, "#{dump.pathname} does not exist"
-    end
-
-    def compress_dumpfile
-      Zlib::GzipWriter.open(dump.pathname_gz) do |gz|
-        File.open(dump.pathname) do |fp|
-          while chunk = fp.read(16 * 1024)
-            gz.write chunk
-          end
-        end
-      end
-      logger.info "Zip dump #{dump.basename_gz} (#{dump.size_gz})"
-    end
-
-    def remove_dumpfile
-      File.delete(dump.pathname)
-    end
-  end
-end

data/lib/pg_export/actions/create_dump.rb

@@ -1,35 +0,0 @@
-class PgExport
-  class CreateDump
-    include Logging
-
-    def initialize(dump)
-      @dump = dump
-    end
-
-    def call
-      validate_pg_dump_exists
-      validate_db_exists(dump.database)
-      execute_dump_command
-      logger.info "Dump #{dump.database} to #{dump.pathname} (#{dump.size})"
-    end
-
-    private
-
-    attr_reader :dump
-
-    def validate_pg_dump_exists
-      out = `pg_dump -V`
-      /pg_dump \(PostgreSQL\)/ =~ out or raise DependencyRequiredError, 'Shell command "pg_dump" is required. Pleas install "pg_dump" and try again.'
-    end
-
-    def validate_db_exists(database)
-      PG.connect(dbname: database)
-    rescue PG::ConnectionBad => e
-      raise DatabaseDoesNotExistError, e
-    end
-
-    def execute_dump_command
-      `pg_dump -Fc --file #{dump.pathname} #{dump.database}`
-    end
-  end
-end

data/lib/pg_export/actions/remove_old_dumps.rb

@@ -1,25 +0,0 @@
-class PgExport
-  class RemoveOldDumps
-    include Logging
-
-    def initialize(dump, keep_dumps_count)
-      @dump = dump
-      @keep_dumps_count = keep_dumps_count.to_i
-    end
-
-    def call
-      files.sort.reverse.drop(keep_dumps_count).each do |filename|
-        File.delete("#{dump.dirname}/#{filename}")
-        logger.info "Remove file #{filename}"
-      end
-    end
-
-    private
-
-    attr_reader :dump, :keep_dumps_count
-
-    def files
-      Dir.entries(dump.dirname).grep(dump.regexp)
-    end
-  end
-end

data/lib/pg_export/actions/remove_old_dumps_from_ftp.rb

@@ -1,22 +0,0 @@
-class PgExport
-  class RemoveOldDumpsFromFtp
-    include Logging
-
-    def initialize(dump, ftp_service, keep_dumps)
-      @dump = dump
-      @ftp_service = ftp_service
-      @keep_dumps = keep_dumps.to_i
-    end
-
-    def call
-      ftp_service.list(dump.ftp_regexp).drop(keep_dumps).each do |filename|
-        ftp_service.delete(filename)
-        logger.info "Remove file #{filename} from FTP"
-      end
-    end
-
-    private
-
-    attr_accessor :dump, :ftp_service, :keep_dumps
-  end
-end

data/lib/pg_export/actions/send_dump_to_ftp.rb

@@ -1,19 +0,0 @@
-class PgExport
-  class SendDumpToFtp
-    include Logging
-
-    def initialize(dump, ftp_service)
-      @dump = dump
-      @ftp_service = ftp_service
-    end
-
-    def call
-      ftp_service.upload_file(dump.pathname_gz)
-      logger.info "Export #{dump.basename_gz} to FTP"
-    end
-
-    private
-
-    attr_reader :dump, :ftp_service
-  end
-end

data/lib/pg_export/dump.rb

@@ -1,61 +0,0 @@
-class PgExport
-  class Dump
-    TIMESTAMP_REGEX = '[0-9]{8}_[0-9]{6}'.freeze
-
-    attr_reader :database, :dir, :dirname, :basename, :basename_gz, :regexp, :ftp_regexp, :pathname, :pathname_gz
-
-    def initialize(a_database, a_dir)
-      @database = a_database
-      @dir = a_dir
-      @dirname = absolute_path(dir)
-      @basename = append_to_database(Time.now.strftime('%Y%m%d_%H%M%S'))
-      @basename_gz = basename + '.gz'
-      @regexp = Regexp.new(append_to_database(TIMESTAMP_REGEX))
-      @ftp_regexp = append_to_database('*')
-      @pathname = [dirname, basename].join('/')
-      @pathname_gz = pathname + '.gz'
-      create_dir_if_necessary
-    end
-
-    def size
-      file_size(pathname)
-    end
-
-    def size_gz
-      file_size(pathname_gz)
-    end
-
-    private
-
-    def file_size(path)
-      return unless File.exist?(path)
-      bytes2human(File.size(path))
-    end
-
-    def bytes2human(bytes)
-      {
-        'B'  => 1024,
-        'kB' => 1024 * 1024,
-        'MB' => 1024 * 1024 * 1024,
-        'GB' => 1024 * 1024 * 1024 * 1024,
-        'TB' => 1024 * 1024 * 1024 * 1024 * 1024
-      }.each_pair { |e, s| return "#{(bytes.to_f / (s / 1024)).round(2)}#{e}" if bytes < s }
-    end
-
-    def create_dir_if_necessary
-      FileUtils.mkdir_p(dirname)
-    end
-
-    def append_to_database(segment)
-      [database, segment].join('_')
-    end
-
-    def absolute_path(dir)
-      if Pathname.new(dir).absolute?
-        dir
-      else
-        [Pathname.pwd, dir].join('/')
-      end
-    end
-  end
-end