diff options
| -rw-r--r-- | lib/dbshell/database_client.rb | 9 | ||||
| -rw-r--r-- | lib/dbshell/rails/tasks/dbshell.rake | 4 |
2 files changed, 11 insertions, 2 deletions
diff --git a/lib/dbshell/database_client.rb b/lib/dbshell/database_client.rb index 8d55095..edea777 100644 --- a/lib/dbshell/database_client.rb +++ b/lib/dbshell/database_client.rb @@ -1,3 +1,6 @@ +require 'dbshell/sqlite3_client' +require 'dbshell/postgres_client' + class DBShell::DatabaseClient def self.handler(connection_params) case connection_params['adapter'] @@ -9,6 +12,12 @@ class DBShell::DatabaseClient raise DBShell::InvalidDatabaseAdapter end end + + def self.runshell(connection_params) + self + .handler(connection_params) + .runshell(connection_params) + end end diff --git a/lib/dbshell/rails/tasks/dbshell.rake b/lib/dbshell/rails/tasks/dbshell.rake index 0505e38..7b192c3 100644 --- a/lib/dbshell/rails/tasks/dbshell.rake +++ b/lib/dbshell/rails/tasks/dbshell.rake @@ -1,10 +1,10 @@ -require 'dbshell/sqlite3_client' +require 'dbshell/database_client' task :dbshell do # {"default"=>{"adapter"=>"sqlite3", "pool"=>5, "timeout"=>5000}, "development"=>{"adapter"=>"sqlite3", "pool"=>5, "timeout"=>5000, "database"=>"db/development.sqlite3"}, "test"=>{"adapter"=>"sqlite3", "pool"=>5, "timeout"=>5000, "database"=>"db/test.sqlite3"}, "production"=>{"adapter"=>"sqlite3", "pool"=>5, "timeout"=>5000, "database"=>"db/production.sqlite3"}} env = ENV.fetch('RAILS_ENV', 'development') - DBShell::Sqlite3Client.runshell( + DBShell::DatabaseClient.runshell( Rails.application.config.database_configuration[env] ) end |