mysql_db_tool 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b7ba1c8784d5842dc7f1212c89ccaf8cd3fbd31793be8c3dd1c168bf75366ff7
-  data.tar.gz: e2eb44bd513921e7957f293076386ce63f614ee02bafbd242ee8f71bb8599145
+  metadata.gz: 072265e57ad114a3724d8be02c24656ebe8aadb4dca3f0bfa397bdafafb00451
+  data.tar.gz: 0cb1d78f8b4144669f27c97000419b25c7d3fe3a1a724fd03750b13720528af3
 SHA512:
-  metadata.gz: fbe7207993d444b6cbf7c408f2860702bd0c63af6f09776865905598ab1c54aa7b4217e64921c5a742f6a1f7db1a19d4381d33aa8488eae2a31aa3fae0e03552
-  data.tar.gz: f90055728240fa5754b159b81e2798964956d299376c56eac7d31edad5e280483b739e6d3c73e47da5b11193c10f927625c503fe3bbff2437faded8d363a4f2b
+  metadata.gz: 2cac1bb6cb7de94e1a6415c5980f3ec8067dc3849c164aa36a167b3dd77a901b3391f8aac81da9357dd695675ce8fd1a687df0a09ed8c4d77ea2b58a43f188dd
+  data.tar.gz: ea432104fb0514200af032ccc619eebbc316b530263678fb0990e8d90dcf60b3b4ea20b07f9d5f7959070592ad62fea53207b65d7e947ce7c21b3073a21ce788

data/README.md

@@ -36,7 +36,7 @@ Create a config-<env>.json file under the current directory according to the dat
 ## Data backup
 
 ```shell
-./bin/backup {env} {backup id} {run?} {gzip?}
+./bin/backup -e {env} -i {backup id} -r {run?} --gzip
 ```
 
 * env - default (local), key to find the configuration file. e.g.) config-local.json
@@ -52,7 +52,7 @@ After execution, a directory named "backup-{backup id}" will be created under th
 ## restore backup data
 
 ```shell
-./bin/restore <env> {backup id} {run?} {drop all tables?}
+./bin/restore -e {env} -i {backup id} -r {run?} --drop-all-tables
 ```
 
 * drop all tables? - Default (false), to keep existing tables, or true, which may cause integration check error if not set to true
@@ -80,4 +80,3 @@ ruby -v
 ## TODO
 
 * [ ] Add option to get db configuration information from spring cloud config
-* [ ] Change linux standard command line argument input format to --backup-id=1 when inputting arguments

data/bin/mysql_backup

@@ -2,23 +2,43 @@
 
 require "bundler/setup"
 require "mysql_db_tool"
+require "optparse"
+
+# Define the options
+options = {
+  env: "local",
+  id: "0",
+  run: false,
+  gzip: true
+}
 
 # Parse command line arguments
-options = {}
-options[:env] = ARGV[0] || "local"
-options[:id] = ARGV[1] || "0"
-options[:run] = ARGV[2] == "true"
-options[:gzip] = ARGV.count > 3 ? ARGV[3] == "true" : true
+OptionParser.new do |opts|
+  opts.banner = "Usage: mysql_backup [options]\n\n#{MySQLDBTool::ENV_VAR_DESCRIPTION}\n"
+
+  MySQLDBTool.env_opt(options, opts)
+  MySQLDBTool.id_opt(options, opts)
+  MySQLDBTool.run_opt(options, opts)
+
+  opts.on("-g", "--[no-]gzip", "Enable or disable gzip (default: enabled)") do |gzip|
+    options[:gzip] = gzip
+  end
+
+  opts.on("-h", "--help", "Show this help message") do
+    puts opts
+    exit
+  end
+end.parse!
 
 # Print the options for verification
 puts "options=#{options}"
 
 # Perform the backup
-# begin
+begin
   MySQLDBTool.backup(options)
-# rescue => e
-#   puts "Error during backup: #{e.message}"
-#   exit 1
-# end
+rescue => e
+  puts "Error during backup: #{e.message}"
+  exit 1
+end
 
-puts "Backup completed successfully."
+puts "Backup completed successfully."

data/bin/mysql_restore

@@ -2,26 +2,43 @@
 
 require "bundler/setup"
 require "mysql_db_tool"
+require "optparse"
+
+# Define the options with default values
+options = {
+  env: "local",
+  id: "0",
+  run: false,
+  drop_all_tables: true
+}
 
 # Parse command line arguments
-options = {}
-options[:env] = ARGV[0] || "local"
-options[:id] = ARGV[1] || "0"
-options[:run] = ARGV[2] == "true"
-options[:drop_all_tables] = ARGV.count > 3 ? ARGV[3] == "true" : true
+OptionParser.new do |opts|
+  opts.banner = "Usage: mysql_restore [options]"
+
+  MySQLDBTool.env_opt(options, opts)
+  MySQLDBTool.id_opt(options, opts)
+  MySQLDBTool.run_opt(options, opts)
+
+  opts.on("-d", "--[no-]drop-all-tables", "Drop all tables before restoring (default: enabled)") do |drop_all_tables|
+    options[:drop_all_tables] = drop_all_tables
+  end
+
+  opts.on("-h", "--help", "Show this help message") do
+    puts opts
+    exit
+  end
+end.parse!
 
 # Print the options for verification
-puts "Env: #{options[:env]}"
-puts "Backup ID: #{options[:id]}"
-puts "Run: #{options[:run]}"
-puts "DropAllTables: #{options[:drop_all_tables]}"
+puts "options=#{options}"
 
-# Perform the backup
-# begin
+# Perform the restore
+begin
   MySQLDBTool.restore(options)
-# rescue => e
-#   puts "Error during backup: #{e.message}"
-#   exit 1
-# end
+rescue => e
+  puts "Error during restore: #{e.message}"
+  exit 1
+end
 
-puts "Restore completed successfully."
+puts "Restore completed successfully."

data/lib/mysql_db_tool/backup.rb

@@ -42,7 +42,7 @@ module MySQLDBTool
 
         ignoreTablesOption = @ignore_tables.map { |e| "--ignore-table=#{@db_info[:database]}.#{e}"  }.join(' ')
 
-        commands.push gzipCommand("mysqldump --no-data #{ignoreTablesOption} #{defaultOptions}", isGzip, "#{backupFile}-schema.sql")
+        commands.push gzipCommand("mysqldump --no-data #{ignoreTablesOption} #{defaultOptions}", isGzip, "#{backupFile}-schema.sql#{isGzip ? '.gz' : ''}")
 
         backupTables = []
 
@@ -52,11 +52,11 @@ module MySQLDBTool
             backupTables.push(table[:name])
             next
           else
-            commands.push(gzipCommand("mysqldump --no-create-info #{options} #{where} #{defaultOptions} #{table[:name]}", isGzip, "#{backupFile}-#{table[:name]}.sql"))
+            commands.push(gzipCommand("mysqldump --no-create-info #{options} #{where} #{defaultOptions} #{table[:name]}", isGzip, "#{backupFile}-#{table[:name]}.sql#{isGzip ? '.gz' : ''}"))
           end
         }
 
-        commands.push(gzipCommand("mysqldump --no-create-info #{options} #{defaultOptions} #{backupTables.join(' ')}", isGzip, "#{backupFile}-all-other-tables.sql"))
+        commands.push(gzipCommand("mysqldump --no-create-info #{options} #{defaultOptions} #{backupTables.join(' ')}", isGzip, "#{backupFile}-all-other-tables.sql#{isGzip ? '.gz' : ''}"))
         commands
       end
     end

data/lib/mysql_db_tool/version.rb

@@ -1,5 +1,5 @@
 # lib/mysql_db_tool/version.rb
 
 module MySQLDBTool
-  VERSION = "0.1.0"
+  VERSION = "0.2.0"
 end

data/lib/mysql_db_tool.rb

@@ -7,17 +7,40 @@ require_relative 'mysql_db_tool/db_backup_base'
 module MySQLDBTool
   class Error < StandardError; end
 
+  # Environment variable description
+  ENV_VAR_DESCRIPTION = <<~DESC
+    Environment Variables:
+      DUMP_OPTIONS - Additional options to pass to the MySQL dump command during the backup process.
+                     This can include any options supported by mysqldump, such as --skip-lock-tables.
+  DESC
+
+  def self.env_opt(options, opts)
+    opts.on("-e", "--env ENVIRONMENT", "Environment (default: local) - key to find the configuration file. e.g.) config-local.json") do |env|
+      options[:env] = env
+    end
+  end
+
+  def self.id_opt(options, opts)
+    opts.on("-i", "--backup-id BACKUP_ID", "BACKUP_ID (default: 0)") do |id|
+      options[:id] = id
+    end
+  end
+
+  def self.run_opt(options, opts)
+    opts.on("-r", "--run", "Run the backup (default: false) or Just show the commands") do
+      options[:run] = true
+    end
+  end
+
   # You might want to add methods to easily access your main functionalities
   def self.backup(options = {})
-    isRun = options[:run]
     commands = Backup.new(options).perform
-    commands.each { |command| run(isRun, command) }
+    commands.each { |command| run(options[:run], command) }
   end
 
   def self.restore(options = {})
-    isRun = options[:run]
     commands = Restore.new(options).perform
-    commands.each { |command| run(isRun, command) }
+    commands.each { |command| run(options[:run], command) }
   end
 
   def self.find_resource(relative_path)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mysql_db_tool
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Soonoh Jung
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-13 00:00:00.000000000 Z
+date: 2024-08-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -88,7 +88,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: '2.6'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="