Class: ActiveRecord::Tasks::SQLiteDatabaseTasks

Inherits:
AbstractTasks show all
Defined in:
activerecord/lib/active_record/tasks/sqlite_database_tasks.rb

Overview

:nodoc:

Instance Method Summary collapse

Methods inherited from AbstractTasks

#charset, #collation, using_database_configurations?

Constructor Details

#initialize(db_config, root = ActiveRecord::Tasks::DatabaseTasks.root) ⇒ SQLiteDatabaseTasks

Returns a new instance of SQLiteDatabaseTasks.



6
7
8
9
# File 'activerecord/lib/active_record/tasks/sqlite_database_tasks.rb', line 6

def initialize(db_config, root = ActiveRecord::Tasks::DatabaseTasks.root)
  @db_config = db_config
  @root = root
end

Instance Method Details

#check_current_protected_environment!(db_config, migration_class) ⇒ Object



60
61
62
63
64
65
66
67
68
# File 'activerecord/lib/active_record/tasks/sqlite_database_tasks.rb', line 60

def check_current_protected_environment!(db_config, migration_class)
  super
rescue ActiveRecord::StatementInvalid => e
  case e.cause
  when SQLite3::ReadOnlyException
  else
    raise e
  end
end

#createObject



11
12
13
14
15
16
17
# File 'activerecord/lib/active_record/tasks/sqlite_database_tasks.rb', line 11

def create
  file = ConnectionAdapters::SQLite3Adapter.resolve_path(db_config.database)
  raise DatabaseAlreadyExists if File.exist?(file)

  establish_connection
  connection
end

#dropObject



19
20
21
22
23
24
25
# File 'activerecord/lib/active_record/tasks/sqlite_database_tasks.rb', line 19

def drop
  file = ConnectionAdapters::SQLite3Adapter.resolve_path(db_config.database, root: root)
  FileUtils.rm(file)
  FileUtils.rm_f(["#{file}-shm", "#{file}-wal"])
rescue Errno::ENOENT => error
  raise NoDatabaseError.new(error.message)
end

#purgeObject



27
28
29
30
31
32
33
34
# File 'activerecord/lib/active_record/tasks/sqlite_database_tasks.rb', line 27

def purge
  connection.disconnect!
  drop
rescue NoDatabaseError
ensure
  create
  connection.reconnect!
end

#structure_dump(filename, extra_flags) ⇒ Object



36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
# File 'activerecord/lib/active_record/tasks/sqlite_database_tasks.rb', line 36

def structure_dump(filename, extra_flags)
  args = []
  args.concat(Array(extra_flags)) if extra_flags
  args << db_config.database

  ignore_tables = ActiveRecord::SchemaDumper.ignore_tables
  if ignore_tables.any?
    ignore_tables = connection.data_sources.select { |table| ignore_tables.any? { |pattern| pattern === table } }
    condition = ignore_tables.map { |table| connection.quote(table) }.join(", ")
    args << "SELECT sql || ';' FROM sqlite_master WHERE tbl_name NOT IN (#{condition}) ORDER BY tbl_name, type DESC, name"
  else
    args << ".schema --nosys"
  end

  run_cmd("sqlite3", *args, out: filename)
end

#structure_load(filename, extra_flags) ⇒ Object



53
54
55
56
57
58
# File 'activerecord/lib/active_record/tasks/sqlite_database_tasks.rb', line 53

def structure_load(filename, extra_flags)
  args = []
  args.concat(extra_flags) if extra_flags
  args << db_config.database
  run_cmd("sqlite3", *args, in: filename)
end