database_tasks.rb 9.7 KB
Newer Older
1 2
module ActiveRecord
  module Tasks # :nodoc:
3
    class DatabaseAlreadyExists < StandardError; end # :nodoc:
4
    class DatabaseNotSupported < StandardError; end # :nodoc:
5

6 7 8
    # <tt>ActiveRecord::Tasks::DatabaseTasks</tt> is a utility class, which encapsulates
    # logic behind common tasks used to manage database and migrations.
    #
9
    # The tasks defined here are used with Rake tasks provided by Active Record.
10 11
    #
    # In order to use DatabaseTasks, a few config values need to be set. All the needed
12 13 14 15
    # config values are set by Rails already, so it's necessary to do it only if you
    # want to change the defaults or when you want to use Active Record outside of Rails
    # (in such case after configuring the database tasks, you can also use the rake tasks
    # defined in Active Record).
16
    #
17
    # The possible config values are:
18
    #
19 20 21 22 23 24
    #   * +env+: current environment (like Rails.env).
    #   * +database_configuration+: configuration of your databases (as in +config/database.yml+).
    #   * +db_dir+: your +db+ directory.
    #   * +fixtures_path+: a path to fixtures directory.
    #   * +migrations_paths+: a list of paths to directories with migrations.
    #   * +seed_loader+: an object which will load seeds, it needs to respond to the +load_seed+ method.
25
    #   * +root+: a path to the root of the application.
26
    #
27
    # Example usage of +DatabaseTasks+ outside Rails could look as such:
28
    #
29
    #   include ActiveRecord::Tasks
30
    #   DatabaseTasks.database_configuration = YAML.load_file('my_database_config.yml')
31 32
    #   DatabaseTasks.db_dir = 'db'
    #   # other settings...
33
    #
34
    #   DatabaseTasks.create_current('production')
35
    module DatabaseTasks
36
      extend self
P
Pat Allan 已提交
37

38 39
      attr_writer :current_config, :db_dir, :migrations_paths, :fixtures_path, :root, :env, :seed_loader
      attr_accessor :database_configuration
40

41
      LOCAL_HOSTS    = ['127.0.0.1', 'localhost']
42

43 44 45 46 47
      def register_task(pattern, task)
        @tasks ||= {}
        @tasks[pattern] = task
      end

K
kennyj 已提交
48 49 50
      register_task(/mysql/,        ActiveRecord::Tasks::MySQLDatabaseTasks)
      register_task(/postgresql/,   ActiveRecord::Tasks::PostgreSQLDatabaseTasks)
      register_task(/sqlite/,       ActiveRecord::Tasks::SQLiteDatabaseTasks)
K
kennyj 已提交
51

52 53 54 55 56 57 58 59 60
      def db_dir
        @db_dir ||= Rails.application.config.paths["db"].first
      end

      def migrations_paths
        @migrations_paths ||= Rails.application.paths['db/migrate'].to_a
      end

      def fixtures_path
61 62 63 64 65
        @fixtures_path ||= if ENV['FIXTURES_PATH']
                             File.join(root, ENV['FIXTURES_PATH'])
                           else
                             File.join(root, 'test', 'fixtures')
                           end
66 67 68 69 70 71 72 73 74 75 76 77 78 79
      end

      def root
        @root ||= Rails.root
      end

      def env
        @env ||= Rails.env
      end

      def seed_loader
        @seed_loader ||= Rails.application
      end

80
      def current_config(options = {})
81
        options.reverse_merge! :env => env
82 83 84
        if options.has_key?(:config)
          @current_config = options[:config]
        else
85
          @current_config ||= ActiveRecord::Base.configurations[options[:env]]
86 87 88
        end
      end

89
      def create(*arguments)
90 91
        configuration = arguments.first
        class_for_adapter(configuration['adapter']).new(*arguments).create
92 93
      rescue DatabaseAlreadyExists
        $stderr.puts "#{configuration['database']} already exists"
94 95 96 97
      rescue Exception => error
        $stderr.puts error, *(error.backtrace)
        $stderr.puts "Couldn't create database for #{configuration.inspect}"
      end
P
Pat Allan 已提交
98

99
      def create_all
100 101
        each_local_configuration { |configuration| create configuration }
      end
P
Pat Allan 已提交
102

103
      def create_current(environment = env)
104 105 106
        each_current_configuration(environment) { |configuration|
          create configuration
        }
107
        ActiveRecord::Base.establish_connection(environment.to_sym)
108
      end
P
Pat Allan 已提交
109

110
      def drop(*arguments)
111 112
        configuration = arguments.first
        class_for_adapter(configuration['adapter']).new(*arguments).drop
113
      rescue ActiveRecord::NoDatabaseError
114
        $stderr.puts "Database '#{configuration['database']}' does not exist"
115 116 117 118
      rescue Exception => error
        $stderr.puts error, *(error.backtrace)
        $stderr.puts "Couldn't drop #{configuration['database']}"
      end
P
Pat Allan 已提交
119

120
      def drop_all
121 122
        each_local_configuration { |configuration| drop configuration }
      end
P
Pat Allan 已提交
123

124
      def drop_current(environment = env)
125 126 127 128
        each_current_configuration(environment) { |configuration|
          drop configuration
        }
      end
P
Pat Allan 已提交
129

130 131 132 133
      def migrate
        verbose = ENV["VERBOSE"] ? ENV["VERBOSE"] == "true" : true
        version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
        scope   = ENV['SCOPE']
134
        verbose_was, Migration.verbose = Migration.verbose, verbose
135 136 137
        Migrator.migrate(Migrator.migrations_paths, version) do |migration|
          scope.blank? || scope == migration.scope
        end
138 139
      ensure
        Migration.verbose = verbose_was
140 141
      end

142
      def charset_current(environment = env)
S
Simon Jefford 已提交
143 144 145 146 147 148 149 150
        charset ActiveRecord::Base.configurations[environment]
      end

      def charset(*arguments)
        configuration = arguments.first
        class_for_adapter(configuration['adapter']).new(*arguments).charset
      end

151
      def collation_current(environment = env)
152 153 154 155 156 157 158 159
        collation ActiveRecord::Base.configurations[environment]
      end

      def collation(*arguments)
        configuration = arguments.first
        class_for_adapter(configuration['adapter']).new(*arguments).collation
      end

160
      def purge(configuration)
161 162
        class_for_adapter(configuration['adapter']).new(configuration).purge
      end
P
Pat Allan 已提交
163

164 165 166 167 168 169 170 171 172 173 174 175
      def purge_all
        each_local_configuration { |configuration|
          purge configuration
        }
      end

      def purge_current(environment = env)
        each_current_configuration(environment) { |configuration|
          purge configuration
        }
      end

K
kennyj 已提交
176 177 178 179 180 181
      def structure_dump(*arguments)
        configuration = arguments.first
        filename = arguments.delete_at 1
        class_for_adapter(configuration['adapter']).new(*arguments).structure_dump(filename)
      end

K
kennyj 已提交
182 183 184 185 186 187
      def structure_load(*arguments)
        configuration = arguments.first
        filename = arguments.delete_at 1
        class_for_adapter(configuration['adapter']).new(*arguments).structure_load(filename)
      end

188
      def load_schema(format = ActiveRecord::Base.schema_format, file = nil)
189 190 191 192
        ActiveSupport::Deprecation.warn \
          "This method will act on a specific connection in the future. " \
          "To act on the current connection, use `load_schema_current` instead."

193 194 195 196 197 198
        load_schema_current(format, file)
      end

      # This method is the successor of +load_schema+. We should rename it
      # after +load_schema+ went through a deprecation cycle. (Rails > 4.2)
      def load_schema_for(configuration, format = ActiveRecord::Base.schema_format, file = nil) # :nodoc:
199 200 201 202
        case format
        when :ruby
          file ||= File.join(db_dir, "schema.rb")
          check_schema_file(file)
203 204
          purge(configuration)
          ActiveRecord::Base.establish_connection(configuration)
205 206 207 208
          load(file)
        when :sql
          file ||= File.join(db_dir, "structure.sql")
          check_schema_file(file)
209 210
          purge(configuration)
          structure_load(configuration, file)
211 212 213 214 215
        else
          raise ArgumentError, "unknown format #{format.inspect}"
        end
      end

216 217 218 219 220 221
      def load_schema_current(format = ActiveRecord::Base.schema_format, file = nil, environment = env)
        each_current_configuration(environment) { |configuration|
          load_schema_for configuration, format, file
        }
      end

222
      def check_schema_file(filename)
A
Arun Agrawal 已提交
223
        unless File.exist?(filename)
224 225 226 227 228 229
          message = %{#{filename} doesn't exist yet. Run `rake db:migrate` to create it, then try again.}
          message << %{ If you do not intend to use a database, you should instead alter #{Rails.root}/config/application.rb to limit the frameworks that will be loaded.} if defined?(::Rails)
          Kernel.abort message
        end
      end

230 231 232 233 234 235 236 237 238 239
      def load_seed
        if seed_loader
          seed_loader.load_seed
        else
          raise "You tried to load seed data, but no seed loader is specified. Please specify seed " +
                "loader with ActiveRecord::Tasks::DatabaseTasks.seed_loader = your_seed_loader\n" +
                "Seed loader should respond to load_seed method"
        end
      end

240
      private
P
Pat Allan 已提交
241

242
      def class_for_adapter(adapter)
243
        key = @tasks.keys.detect { |pattern| adapter[pattern] }
244 245 246
        unless key
          raise DatabaseNotSupported, "Rake tasks not supported by '#{adapter}' adapter"
        end
247
        @tasks[key]
248
      end
P
Pat Allan 已提交
249

250
      def each_current_configuration(environment)
251
        environments = [environment]
252 253
        # add test environment only if no RAILS_ENV was specified.
        environments << 'test' if environment == 'development' && ENV['RAILS_ENV'].nil?
P
Pat Allan 已提交
254

255 256 257 258 259 260
        configurations = ActiveRecord::Base.configurations.values_at(*environments)
        configurations.compact.each do |configuration|
          yield configuration unless configuration['database'].blank?
        end
      end

261
      def each_local_configuration
262 263
        ActiveRecord::Base.configurations.each_value do |configuration|
          next unless configuration['database']
P
Pat Allan 已提交
264

265 266 267 268 269 270
          if local_database?(configuration)
            yield configuration
          else
            $stderr.puts "This task only modifies local databases. #{configuration['database']} is on a remote host."
          end
        end
P
Pat Allan 已提交
271 272
      end

273
      def local_database?(configuration)
274
        configuration['host'].blank? || LOCAL_HOSTS.include?(configuration['host'])
275 276
      end
    end
P
Pat Allan 已提交
277
  end
278
end