A Database object represents a virtual connection to a database. The Database class is meant to be subclassed by database adapters in order to provide the functionality needed for executing queries.
Whether the schema should be cached for this database. True by default for performance, can be set to false to always issue a database query to get the schema.
The prepared statement object hash for this database, keyed by name symbol
Runs the supplied SQL statement string on the database server. Returns self so it can be safely chained:
DB << "UPDATE albums SET artist_id = NULL" << "DROP TABLE artists"
# File lib/sequel/database/query.rb, line 25 def <<(sql) run(sql) self end
Call the prepared statement with the given name with the given hash of arguments.
DB[:items].filter(:id=>1).prepare(:first, :sa) DB.call(:sa) # SELECT * FROM items WHERE id = 1
# File lib/sequel/database/query.rb, line 35 def call(ps_name, hash={}, &block) prepared_statement(ps_name).call(hash, &block) end
Executes the given SQL on the database. This method should be overridden in descendants. This method should not be called directly by user code.
# File lib/sequel/database/query.rb, line 41 def execute(sql, opts={}) raise NotImplemented, "#execute should be overridden by adapters" end
Method that should be used when submitting any DDL (Data Definition
Language) SQL, such as create_table
.
By default, calls execute_dui
. This method should not be
called directly by user code.
# File lib/sequel/database/query.rb, line 48 def execute_ddl(sql, opts={}, &block) execute_dui(sql, opts, &block) end
Method that should be used when issuing a DELETE, UPDATE, or INSERT statement. By default, calls execute. This method should not be called directly by user code.
# File lib/sequel/database/query.rb, line 55 def execute_dui(sql, opts={}, &block) execute(sql, opts, &block) end
Method that should be used when issuing a INSERT statement. By default, calls execute_dui. This method should not be called directly by user code.
# File lib/sequel/database/query.rb, line 62 def execute_insert(sql, opts={}, &block) execute_dui(sql, opts, &block) end
Returns an array of hashes containing foreign key information from the table. Each hash will contain at least the following fields:
An array of columns in the given table
The table referenced by the columns
An array of columns referenced (in the table specified by :table), but can be nil on certain adapters if the primary key is referenced.
The hash may also contain entries for:
Whether the constraint is deferrable
The name of the constraint
The action to take ON DELETE
The action to take ON UPDATE
# File lib/sequel/database/query.rb, line 80 def foreign_key_list(table, opts={}) raise NotImplemented, "#foreign_key_list should be overridden by adapters" end
Returns a single value from the database, e.g.:
DB.get(1) # SELECT 1 # => 1 DB.get{server_version{}} # SELECT server_version()
# File lib/sequel/database/query.rb, line 89 def get(*args, &block) dataset.get(*args, &block) end
Return a hash containing index information for the table. Hash keys are index name symbols. Values are subhashes with two keys, :columns and :unique. The value of :columns is an array of symbols of column names. The value of :unique is true or false depending on if the index is unique.
Should not include the primary key index, functional indexes, or partial indexes.
DB.indexes(:artists) # => {:artists_name_ukey=>{:columns=>[:name], :unique=>true}}
# File lib/sequel/database/query.rb, line 102 def indexes(table, opts={}) raise NotImplemented, "#indexes should be overridden by adapters" end
Returns the schema for the given table as an array with all members being arrays of length 2, the first member being the column name, and the second member being a hash of column information. The table argument can also be a dataset, as long as it only has one table. Available options are:
Ignore any cached results, and get fresh information from the database.
An explicit schema to use. It may also be implicitly provided via the table name.
If schema parsing is supported by the database, the column information should hash at least contain the following entries:
Whether NULL is an allowed value for the column.
The database type for the column, as a database specific string.
The database default for the column, as a database specific string.
Whether the columns is a primary key column. If this column is not present, it means that primary key information is unavailable, not that the column is not a primary key.
The database default for the column, as a ruby object. In many cases, complex database defaults cannot be parsed into ruby objects, in which case nil will be used as the value.
A symbol specifying the type, such as :integer or :string.
Example:
DB.schema(:artists) # [[:id, # {:type=>:integer, # :primary_key=>true, # :default=>"nextval('artist_id_seq'::regclass)", # :ruby_default=>nil, # :db_type=>"integer", # :allow_null=>false}], # [:name, # {:type=>:string, # :primary_key=>false, # :default=>nil, # :ruby_default=>nil, # :db_type=>"text", # :allow_null=>false}]]
# File lib/sequel/database/query.rb, line 156 def schema(table, opts={}) raise(Error, 'schema parsing is not implemented on this database') unless supports_schema_parsing? opts = opts.dup tab = if table.is_a?(Dataset) o = table.opts from = o[:from] raise(Error, "can only parse the schema for a dataset with a single from table") unless from && from.length == 1 && !o.include?(:join) && !o.include?(:sql) table.first_source_table else table end qualifiers = split_qualifiers(tab) table_name = qualifiers.pop sch = qualifiers.pop information_schema_schema = case qualifiers.length when 1 Sequel.identifier(*qualifiers) when 2 Sequel.qualify(*qualifiers) end if table.is_a?(Dataset) quoted_name = table.literal(tab) opts[:dataset] = table else quoted_name = schema_utility_dataset.literal(table) end opts[:schema] = sch if sch && !opts.include?(:schema) opts[:information_schema_schema] = information_schema_schema if information_schema_schema && !opts.include?(:information_schema_schema) Sequel.synchronize{@schemas.delete(quoted_name)} if opts[:reload] if v = Sequel.synchronize{@schemas[quoted_name]} return v end cols = schema_parse_table(table_name, opts) raise(Error, 'schema parsing returned no columns, table probably doesn\t exist') if cols.nil? || cols.empty? cols.each{|_,c| c[:ruby_default] = column_schema_to_ruby_default(c[:default], c[:type])} Sequel.synchronize{@schemas[quoted_name] = cols} if cache_schema cols end
Returns true if a table with the given name exists. This requires a query to the database.
DB.table_exists?(:foo) # => false # SELECT NULL FROM foo LIMIT 1
Note that since this does a SELECT from the table, it can give false negatives if you don't have permission to SELECT from the table.
# File lib/sequel/database/query.rb, line 209 def table_exists?(name) sch, table_name = schema_and_table(name) name = SQL::QualifiedIdentifier.new(sch, table_name) if sch _table_exists?(from(name)) true rescue DatabaseError false end
Return all tables in the database as an array of symbols.
DB.tables # => [:albums, :artists]
# File lib/sequel/database/query.rb, line 221 def tables(opts={}) raise NotImplemented, "#tables should be overridden by adapters" end
Return all views in the database as an array of symbols.
DB.views # => [:gold_albums, :artists_with_many_albums]
# File lib/sequel/database/query.rb, line 228 def views(opts={}) raise NotImplemented, "#views should be overridden by adapters" end
Should raise an error if the table doesn't not exist, and not raise an error if the table does exist.
# File lib/sequel/database/query.rb, line 236 def _table_exists?(ds) ds.get(Sequel::NULL) end
Whether the type should be treated as a string type when parsing the column schema default value.
# File lib/sequel/database/query.rb, line 242 def column_schema_default_string_type?(type) COLUMN_SCHEMA_STRING_TYPES.include?(type) end
Transform the given normalized default string into a ruby object for the given type.
# File lib/sequel/database/query.rb, line 248 def column_schema_default_to_ruby_value(default, type) case type when :boolean case default when /[f0]/ false when /[t1]/ true end when :string, :enum, :set, :interval default when :blob Sequel::SQL::Blob.new(default) when :integer Integer(default) when :float Float(default) when :date Sequel.string_to_date(default) when :datetime DateTime.parse(default) when :time Sequel.string_to_time(default) when :decimal BigDecimal.new(default) end end
Normalize the default value string for the given type and return the normalized value.
# File lib/sequel/database/query.rb, line 278 def column_schema_normalize_default(default, type) if column_schema_default_string_type?(type) return unless m = STRING_DEFAULT_RE.match(default) m[1].gsub("''", "'") else default end end
Convert the given default, which should be a database specific string, into a ruby object.
# File lib/sequel/database/query.rb, line 289 def column_schema_to_ruby_default(default, type) return default unless default.is_a?(String) if COLUMN_SCHEMA_DATETIME_TYPES.include?(type) if CURRENT_TIMESTAMP_RE.match(default) if type == :date return Sequel::CURRENT_DATE else return Sequel::CURRENT_TIMESTAMP end end end default = column_schema_normalize_default(default, type) column_schema_default_to_ruby_value(default, type) rescue nil end
Return a Method object for the dataset's output_identifier_method. Used in metadata parsing to make sure the returned information is in the correct format.
# File lib/sequel/database/query.rb, line 307 def input_identifier_meth(ds=nil) (ds || dataset).method(:input_identifier) end
Return a dataset that uses the default identifier input and output methods for this database. Used when parsing metadata so that column symbols are returned as expected.
# File lib/sequel/database/query.rb, line 314 def metadata_dataset return @metadata_dataset if defined?(@metadata_dataset) && @metadata_dataset ds = dataset ds.identifier_input_method = identifier_input_method_default ds.identifier_output_method = identifier_output_method_default @metadata_dataset = ds end
Return a Method object for the dataset's output_identifier_method. Used in metadata parsing to make sure the returned information is in the correct format.
# File lib/sequel/database/query.rb, line 325 def output_identifier_meth(ds=nil) (ds || dataset).method(:output_identifier) end
Remove the cached schema for the given schema name
# File lib/sequel/database/query.rb, line 330 def remove_cached_schema(table) Sequel.synchronize{@schemas.delete(quote_schema_table(table))} if @schemas end
Match the database's column type to a ruby type via a regular expression, and return the ruby type as a symbol such as :integer or :string.
# File lib/sequel/database/query.rb, line 337 def schema_column_type(db_type) case db_type when /\A(character( varying)?|n?(var)?char|n?text|string|clob)/o :string when /\A(int(eger)?|(big|small|tiny)int)/o :integer when /\Adate\z/o :date when /\A((small)?datetime|timestamp( with(out)? time zone)?)(\(\d+\))?\z/o :datetime when /\Atime( with(out)? time zone)?\z/o :time when /\A(bool(ean)?)\z/o :boolean when /\A(real|float|double( precision)?|double\(\d+,\d+\)( unsigned)?)\z/o :float when /\A(?:(?:(?:num(?:ber|eric)?|decimal)(?:\(\d+,\s*(\d+|false|true)\))?))\z/o $1 && ['0', 'false'].include?($1) ? :integer : :decimal when /bytea|blob|image|(var)?binary/o :blob when /\Aenum/o :enum end end
The order of column modifiers to use when defining a column.
The alter table operations that are combinable.
The default options for join table columns.
Adds a column to the specified table. This method expects a column name, a datatype and optionally a hash with additional constraints and options:
DB.add_column :items, :name, :text, :unique => true, :null => false DB.add_column :items, :category, :text, :default => 'ruby'
See alter_table
.
# File lib/sequel/database/schema_methods.rb, line 36 def add_column(table, *args) alter_table(table) {add_column(*args)} end
Adds an index to a table for the given columns:
DB.add_index :posts, :title DB.add_index :posts, [:author, :title], :unique => true
Options:
Ignore any DatabaseErrors that are raised
See alter_table
.
# File lib/sequel/database/schema_methods.rb, line 49 def add_index(table, columns, options={}) e = options[:ignore_errors] begin alter_table(table){add_index(columns, options)} rescue DatabaseError raise unless e end end
Alters the given table with the specified block. Example:
DB.alter_table :items do add_column :category, :text, :default => 'ruby' drop_column :category rename_column :cntr, :counter set_column_type :value, :float set_column_default :value, :float add_index [:group, :category] drop_index [:group, :category] end
Note that add_column
accepts all the options available for
column definitions using create_table
, and
add_index
accepts all the options available for index
definition.
See Schema::AlterTableGenerator
and the “Migrations and Schema
Modification” guide.
# File lib/sequel/database/schema_methods.rb, line 75 def alter_table(name, generator=nil, &block) generator ||= alter_table_generator(&block) remove_cached_schema(name) apply_alter_table_generator(name, generator) nil end
Return a new Schema::AlterTableGenerator instance with the receiver as the database and the given block.
# File lib/sequel/database/schema_methods.rb, line 84 def alter_table_generator(&block) alter_table_generator_class.new(self, &block) end
Create a join table using a hash of foreign keys to referenced table names. Example:
create_join_table(:cat_id=>:cats, :dog_id=>:dogs) # CREATE TABLE cats_dogs ( # cat_id integer NOT NULL REFERENCES cats, # dog_id integer NOT NULL REFERENCES dogs, # PRIMARY KEY (cat_id, dog_id) # ) # CREATE INDEX cats_dogs_dog_id_cat_id_index ON cats_dogs(dog_id, cat_id)
The primary key and index are used so that almost all operations on the table can benefit from one of the two indexes, and the primary key ensures that entries in the table are unique, which is the typical desire for a join table.
You can provide column options by making the values in the hash be option hashes, so long as the option hashes have a :table entry giving the table referenced:
create_join_table(:cat_id=>{:table=>:cats, :type=>Bignum}, :dog_id=>:dogs)
You can provide a second argument which is a table options hash:
create_join_table({:cat_id=>:cats, :dog_id=>:dogs}, :temp=>true)
Some table options are handled specially:
The options to pass to the index
The name of the table to create
Set to true not to create the second index.
Set to true to not create the primary key.
# File lib/sequel/database/schema_methods.rb, line 120 def create_join_table(hash, options={}) keys = hash.keys.sort_by{|k| k.to_s} create_table(join_table_name(hash, options), options) do keys.each do |key| v = hash[key] unless v.is_a?(Hash) v = {:table=>v} end v = DEFAULT_JOIN_TABLE_COLUMN_OPTIONS.merge(v) foreign_key(key, v) end primary_key(keys) unless options[:no_primary_key] index(keys.reverse, options[:index_options] || {}) unless options[:no_index] end end
Creates a view, replacing a view with the same name if one already exists.
DB.create_or_replace_view(:some_items, "SELECT * FROM items WHERE price < 100") DB.create_or_replace_view(:some_items, DB[:items].filter(:category => 'ruby'))
For databases where replacing a view is not natively supported, support is emulated by dropping a view with the same name before creating the view.
# File lib/sequel/database/schema_methods.rb, line 212 def create_or_replace_view(name, source, options = {}) if supports_create_or_replace_view? options = options.merge(:replace=>true) else drop_view(name) rescue nil end create_view(name, source, options) end
Creates a table with the columns given in the provided block:
DB.create_table :posts do primary_key :id column :title, :text String :content index :title end
General options:
Create the table using the value, which should be either a dataset or a literal SQL string. If this option is used, a block should not be given to the method.
Ignore any errors when creating indexes.
Create the table as a temporary table.
MySQL specific options:
The character set to use for the table.
The collation to use for the table.
The table engine to use for the table.
PostgreSQL specific options:
Create the table as an unlogged table.
See Schema::Generator
and the “Schema Modification”
guide.
# File lib/sequel/database/schema_methods.rb, line 161 def create_table(name, options={}, &block) remove_cached_schema(name) options = {:generator=>options} if options.is_a?(Schema::CreateTableGenerator) if sql = options[:as] raise(Error, "can't provide both :as option and block to create_table") if block create_table_as(name, sql, options) else generator = options[:generator] || create_table_generator(&block) create_table_from_generator(name, generator, options) create_table_indexes_from_generator(name, generator, options) nil end end
Forcibly create a table, attempting to drop it if it already exists, then creating it.
DB.create_table!(:a){Integer :a} # SELECT NULL FROM a LIMIT 1 -- check existence # DROP TABLE a -- drop table if already exists # CREATE TABLE a (a integer)
# File lib/sequel/database/schema_methods.rb, line 181 def create_table!(name, options={}, &block) drop_table?(name) create_table(name, options, &block) end
Creates the table unless the table already exists.
DB.create_table?(:a){Integer :a} # SELECT NULL FROM a LIMIT 1 -- check existence # CREATE TABLE a (a integer) -- if it doesn't already exist
# File lib/sequel/database/schema_methods.rb, line 191 def create_table?(name, options={}, &block) if supports_create_table_if_not_exists? create_table(name, options.merge(:if_not_exists=>true), &block) elsif !table_exists?(name) create_table(name, options, &block) end end
Return a new Schema::CreateTableGenerator instance with the receiver as the database and the given block.
# File lib/sequel/database/schema_methods.rb, line 201 def create_table_generator(&block) create_table_generator_class.new(self, &block) end
Creates a view based on a dataset or an SQL string:
DB.create_view(:cheap_items, "SELECT * FROM items WHERE price < 100") DB.create_view(:ruby_items, DB[:items].filter(:category => 'ruby'))
PostgreSQL/SQLite specific option:
Create a temporary view, automatically dropped on disconnect.
# File lib/sequel/database/schema_methods.rb, line 229 def create_view(name, source, options = {}) execute_ddl(create_view_sql(name, source, options)) remove_cached_schema(name) nil end
Removes a column from the specified table:
DB.drop_column :items, :category
See alter_table
.
# File lib/sequel/database/schema_methods.rb, line 240 def drop_column(table, *args) alter_table(table) {drop_column(*args)} end
Removes an index for the given table and column/s:
DB.drop_index :posts, :title DB.drop_index :posts, [:author, :title]
See alter_table
.
# File lib/sequel/database/schema_methods.rb, line 250 def drop_index(table, columns, options={}) alter_table(table){drop_index(columns, options)} end
Drop the join table that would have been created with the same arguments to #create_join_table:
drop_join_table(:cat_id=>:cats, :dog_id=>:dogs) # DROP TABLE cats_dogs
# File lib/sequel/database/schema_methods.rb, line 259 def drop_join_table(hash, options={}) drop_table(join_table_name(hash, options), options) end
Drops one or more tables corresponding to the given names:
DB.drop_table(:posts) # DROP TABLE posts DB.drop_table(:posts, :comments) DB.drop_table(:posts, :comments, :cascade=>true)
# File lib/sequel/database/schema_methods.rb, line 268 def drop_table(*names) options = names.last.is_a?(Hash) ? names.pop : {} names.each do |n| execute_ddl(drop_table_sql(n, options)) remove_cached_schema(n) end nil end
Drops the table if it already exists. If it doesn't exist, does nothing.
DB.drop_table?(:a) # SELECT NULL FROM a LIMIT 1 -- check existence # DROP TABLE a -- if it already exists
# File lib/sequel/database/schema_methods.rb, line 283 def drop_table?(*names) options = names.last.is_a?(Hash) ? names.pop : {} if supports_drop_table_if_exists? options = options.merge(:if_exists=>true) names.each do |name| drop_table(name, options) end else names.each do |name| drop_table(name, options) if table_exists?(name) end end end
Drops one or more views corresponding to the given names:
DB.drop_view(:cheap_items) DB.drop_view(:cheap_items, :pricey_items) DB.drop_view(:cheap_items, :pricey_items, :cascade=>true)
# File lib/sequel/database/schema_methods.rb, line 302 def drop_view(*names) options = names.last.is_a?(Hash) ? names.pop : {} names.each do |n| execute_ddl(drop_view_sql(n, options)) remove_cached_schema(n) end nil end
Renames a column in the specified table. This method expects the current column name and the new column name:
DB.rename_column :items, :cntr, :counter
See alter_table
.
# File lib/sequel/database/schema_methods.rb, line 328 def rename_column(table, *args) alter_table(table) {rename_column(*args)} end
Renames a table:
DB.tables #=> [:items] DB.rename_table :items, :old_items DB.tables #=> [:old_items]
# File lib/sequel/database/schema_methods.rb, line 316 def rename_table(name, new_name) execute_ddl(rename_table_sql(name, new_name)) remove_cached_schema(name) nil end
Sets the default value for the given column in the given table:
DB.set_column_default :items, :category, 'perl!'
See alter_table
.
# File lib/sequel/database/schema_methods.rb, line 337 def set_column_default(table, *args) alter_table(table) {set_column_default(*args)} end
Set the data type for the given column in the given table:
DB.set_column_type :items, :price, :float
See alter_table
.
# File lib/sequel/database/schema_methods.rb, line 346 def set_column_type(table, *args) alter_table(table) {set_column_type(*args)} end
The class used for #alter_table generators.
# File lib/sequel/database/schema_methods.rb, line 363 def alter_table_generator_class Schema::AlterTableGenerator end
SQL fragment for given alter table operation.
# File lib/sequel/database/schema_methods.rb, line 368 def alter_table_op_sql(table, op) quoted_name = quote_identifier(op[:name]) if op[:name] case op[:op] when :add_column "ADD COLUMN #{column_definition_sql(op)}" when :drop_column "DROP COLUMN #{quoted_name}#{' CASCADE' if op[:cascade]}" when :rename_column "RENAME COLUMN #{quoted_name} TO #{quote_identifier(op[:new_name])}" when :set_column_type "ALTER COLUMN #{quoted_name} TYPE #{type_literal(op)}" when :set_column_default "ALTER COLUMN #{quoted_name} SET DEFAULT #{literal(op[:default])}" when :set_column_null "ALTER COLUMN #{quoted_name} #{op[:null] ? 'DROP' : 'SET'} NOT NULL" when :add_constraint "ADD #{constraint_definition_sql(op)}" when :drop_constraint if op[:type] == :foreign_key quoted_name ||= quote_identifier(foreign_key_name(table, op[:columns])) end "DROP CONSTRAINT #{quoted_name}#{' CASCADE' if op[:cascade]}" else raise Error, "Unsupported ALTER TABLE operation: #{op[:op]}" end end
The SQL to execute to modify the DDL for the given table name. op should be one of the operations returned by the AlterTableGenerator.
# File lib/sequel/database/schema_methods.rb, line 397 def alter_table_sql(table, op) case op[:op] when :add_index index_definition_sql(table, op) when :drop_index drop_index_sql(table, op) else "ALTER TABLE #{quote_schema_table(table)} #{alter_table_op_sql(table, op)}" end end
Array of SQL DDL modification statements for the given table, corresponding to the DDL changes specified by the operations.
# File lib/sequel/database/schema_methods.rb, line 410 def alter_table_sql_list(table, operations) if supports_combining_alter_table_ops? grouped_ops = [] last_combinable = false operations.each do |op| if combinable_alter_table_op?(op) if sql = alter_table_op_sql(table, op) grouped_ops << [] unless last_combinable grouped_ops.last << sql last_combinable = true end elsif sql = alter_table_sql(table, op) Array(sql).each{|s| grouped_ops << s} last_combinable = false end end grouped_ops.map do |gop| if gop.is_a?(Array) "ALTER TABLE #{quote_schema_table(table)} #{gop.join(', ')}" else gop end end else operations.map{|op| alter_table_sql(table, op)}.flatten.compact end end
Apply the changes in the given alter table ops to the table given by name.
# File lib/sequel/database/schema_methods.rb, line 353 def apply_alter_table(name, ops) alter_table_sql_list(name, ops).each{|sql| execute_ddl(sql)} end
Apply the operations in the given generator to the table given by name.
# File lib/sequel/database/schema_methods.rb, line 358 def apply_alter_table_generator(name, generator) apply_alter_table(name, generator.operations) end
The SQL string specify the autoincrement property, generally used by primary keys.
# File lib/sequel/database/schema_methods.rb, line 440 def auto_increment_sql AUTOINCREMENT end
The order of the column definition, as an array of symbols.
# File lib/sequel/database/schema_methods.rb, line 445 def column_definition_order self.class.const_get(:COLUMN_DEFINITION_ORDER) end
Add primary key SQL fragment to column creation SQL.
# File lib/sequel/database/schema_methods.rb, line 479 def column_definition_primary_key_sql(sql, column) if column[:primary_key] if name = column[:primary_key_constraint_name] sql << " CONSTRAINT #{quote_identifier(name)}" end sql << PRIMARY_KEY end end
Add foreign key reference SQL fragment to column creation SQL.
# File lib/sequel/database/schema_methods.rb, line 489 def column_definition_references_sql(sql, column) if column[:table] if name = column[:foreign_key_constraint_name] sql << " CONSTRAINT #{quote_identifier(name)}" end sql << column_references_column_constraint_sql(column) end end
SQL DDL fragment containing the column creation SQL for the given column.
# File lib/sequel/database/schema_methods.rb, line 450 def column_definition_sql(column) sql = "#{quote_identifier(column[:name])} #{type_literal(column)}" column_definition_order.each{|m| send(:"column_definition_#{m}_sql", sql, column)} sql end
SQL for all given columns, used inside a CREATE TABLE block.
# File lib/sequel/database/schema_methods.rb, line 509 def column_list_sql(generator) (generator.columns.map{|c| column_definition_sql(c)} + generator.constraints.map{|c| constraint_definition_sql(c)}).join(COMMA_SEPARATOR) end
SQL DDL fragment for column foreign key references (column constraints)
# File lib/sequel/database/schema_methods.rb, line 514 def column_references_column_constraint_sql(column) column_references_sql(column) end
SQL DDL fragment for column foreign key references
# File lib/sequel/database/schema_methods.rb, line 519 def column_references_sql(column) sql = " REFERENCES #{quote_schema_table(column[:table])}" sql << "(#{Array(column[:key]).map{|x| quote_identifier(x)}.join(COMMA_SEPARATOR)})" if column[:key] sql << " ON DELETE #{on_delete_clause(column[:on_delete])}" if column[:on_delete] sql << " ON UPDATE #{on_update_clause(column[:on_update])}" if column[:on_update] constraint_deferrable_sql_append(sql, column[:deferrable]) sql end
SQL DDL fragment for table foreign key references (table constraints)
# File lib/sequel/database/schema_methods.rb, line 529 def column_references_table_constraint_sql(constraint) "FOREIGN KEY #{literal(constraint[:columns])}#{column_references_sql(constraint)}" end
Whether the given alter table operation is combinable.
# File lib/sequel/database/schema_methods.rb, line 534 def combinable_alter_table_op?(op) # Use a dynamic lookup for easier overriding in adapters COMBINABLE_ALTER_TABLE_OPS.include?(op[:op]) end
SQL DDL fragment specifying the deferrable constraint attributes.
# File lib/sequel/database/schema_methods.rb, line 560 def constraint_deferrable_sql_append(sql, defer) case defer when nil when false sql << ' NOT DEFERRABLE' when :immediate sql << ' DEFERRABLE INITIALLY IMMEDIATE' else sql << ' DEFERRABLE INITIALLY DEFERRED' end end
SQL DDL fragment specifying a constraint on a table.
# File lib/sequel/database/schema_methods.rb, line 540 def constraint_definition_sql(constraint) sql = constraint[:name] ? "CONSTRAINT #{quote_identifier(constraint[:name])} " : "" case constraint[:type] when :check check = constraint[:check] sql << "CHECK #{filter_expr((check.is_a?(Array) && check.length == 1) ? check.first : check)}" when :primary_key sql << "PRIMARY KEY #{literal(constraint[:columns])}" when :foreign_key sql << column_references_table_constraint_sql(constraint.merge(:deferrable=>nil)) when :unique sql << "UNIQUE #{literal(constraint[:columns])}" else raise Error, "Invalid constraint type #{constraint[:type]}, should be :check, :primary_key, :foreign_key, or :unique" end constraint_deferrable_sql_append(sql, constraint[:deferrable]) sql end
Run a command to create the table with the given name from the given SELECT sql statement.
# File lib/sequel/database/schema_methods.rb, line 618 def create_table_as(name, sql, options) sql = sql.sql if sql.is_a?(Sequel::Dataset) run(create_table_as_sql(name, sql, options)) end
DDL statement for creating a table from the result of a SELECT statement.
sql
should be a string representing a SELECT query.
# File lib/sequel/database/schema_methods.rb, line 625 def create_table_as_sql(name, sql, options) "#{create_table_prefix_sql(name, options)} AS #{sql}" end
Execute the create table statements using the generator.
# File lib/sequel/database/schema_methods.rb, line 573 def create_table_from_generator(name, generator, options) execute_ddl(create_table_sql(name, generator, options)) end
The class used for #create_table generators.
# File lib/sequel/database/schema_methods.rb, line 578 def create_table_generator_class Schema::CreateTableGenerator end
Execute the create index statements using the generator.
# File lib/sequel/database/schema_methods.rb, line 583 def create_table_indexes_from_generator(name, generator, options) e = options[:ignore_index_errors] || options[:if_not_exists] generator.indexes.each do |index| begin index_sql_list(name, [index]).each{|sql| execute_ddl(sql)} rescue Error raise unless e end end end
DDL fragment for initial part of CREATE TABLE statement
# File lib/sequel/database/schema_methods.rb, line 630 def create_table_prefix_sql(name, options) "CREATE #{temporary_table_sql if options[:temp]}TABLE#{' IF NOT EXISTS' if options[:if_not_exists]} #{options[:temp] ? quote_identifier(name) : quote_schema_table(name)}" end
DDL statement for creating a table with the given name, columns, and options
# File lib/sequel/database/schema_methods.rb, line 595 def create_table_sql(name, generator, options) unless supports_named_column_constraints? # Split column constraints into table constraints if they have a name generator.columns.each do |c| if (constraint_name = c.delete(:foreign_key_constraint_name)) && (table = c.delete(:table)) opts = {} opts[:name] = constraint_name [:key, :on_delete, :on_update, :deferrable].each{|k| opts[k] = c[k]} generator.foreign_key([c[:name]], table, opts) end if (constraint_name = c.delete(:unique_constraint_name)) && c.delete(:unique) generator.unique(c[:name], :name=>constraint_name) end if (constraint_name = c.delete(:primary_key_constraint_name)) && c.delete(:primary_key) generator.primary_key([c[:name]], :name=>constraint_name) end end end "#{create_table_prefix_sql(name, options)} (#{column_list_sql(generator)})" end
DDL fragment for initial part of CREATE VIEW statement
# File lib/sequel/database/schema_methods.rb, line 641 def create_view_prefix_sql(name, options) "CREATE #{'OR REPLACE 'if options[:replace]}VIEW #{quote_schema_table(name)}" end
DDL statement for creating a view.
# File lib/sequel/database/schema_methods.rb, line 635 def create_view_sql(name, source, options) source = source.sql if source.is_a?(Dataset) "#{create_view_prefix_sql(name, options)} AS #{source}" end
Default index name for the table and columns, may be too long for certain databases.
# File lib/sequel/database/schema_methods.rb, line 647 def default_index_name(table_name, columns) schema, table = schema_and_table(table_name) "#{"#{schema}_" if schema and schema != default_schema}#{table}_#{columns.map{|c| [String, Symbol].any?{|cl| c.is_a?(cl)} ? c : literal(c).gsub(/\W/, '_')}.join(UNDERSCORE)}_index" end
The SQL to drop an index for the table.
# File lib/sequel/database/schema_methods.rb, line 660 def drop_index_sql(table, op) "DROP INDEX #{quote_identifier(op[:name] || default_index_name(table, op[:columns]))}" end
SQL DDL statement to drop the table with the given name.
# File lib/sequel/database/schema_methods.rb, line 665 def drop_table_sql(name, options) "DROP TABLE#{' IF EXISTS' if options[:if_exists]} #{quote_schema_table(name)}#{' CASCADE' if options[:cascade]}" end
SQL DDL statement to drop a view with the given name.
# File lib/sequel/database/schema_methods.rb, line 670 def drop_view_sql(name, options) "DROP VIEW #{quote_schema_table(name)}#{' CASCADE' if options[:cascade]}" end
Proxy the #filter_expr call to the dataset, used for creating constraints.
# File lib/sequel/database/schema_methods.rb, line 675 def filter_expr(*args, &block) schema_utility_dataset.literal(schema_utility_dataset.send(:filter_expr, *args, &block)) end
Get foreign key name for given table and columns.
# File lib/sequel/database/schema_methods.rb, line 653 def foreign_key_name(table_name, columns) keys = foreign_key_list(table_name).select{|key| key[:columns] == columns} raise(Error, "#{keys.empty? ? 'Missing' : 'Ambiguous'} foreign key for #{columns.inspect}") unless keys.size == 1 keys.first[:name] end
SQL DDL statement for creating an index for the table with the given name and index specifications.
# File lib/sequel/database/schema_methods.rb, line 681 def index_definition_sql(table_name, index) index_name = index[:name] || default_index_name(table_name, index[:columns]) if index[:type] raise Error, "Index types are not supported for this database" elsif index[:where] raise Error, "Partial indexes are not supported for this database" else "CREATE #{'UNIQUE ' if index[:unique]}INDEX #{quote_identifier(index_name)} ON #{quote_schema_table(table_name)} #{literal(index[:columns])}" end end
Extract the join table name from the arguments given to create_join_table. Also does argument validation for the #create_join_table method.
# File lib/sequel/database/schema_methods.rb, line 700 def join_table_name(hash, options) entries = hash.values raise Error, "must have 2 entries in hash given to (create|drop)_join_table" unless entries.length == 2 if options[:name] options[:name] else table_names = entries.map{|e| join_table_name_extract(e)} table_names.map{|t| t.to_s}.sort.join('_') end end
Extract an individual join table name, which should either be a string or symbol, or a hash containing one of those as the value for :table.
# File lib/sequel/database/schema_methods.rb, line 713 def join_table_name_extract(entry) case entry when Symbol, String entry when Hash join_table_name_extract(entry[:table]) else raise Error, "can't extract table name from #{entry.inspect}" end end
SQL DDL ON DELETE fragment to use, based on the given action. The following actions are recognized:
:cascade - Delete rows referencing this row.
:no_action (default) - Raise an error if other rows reference this row, allow deferring of the integrity check.
:restrict - Raise an error if other rows reference this row, but do not allow deferring the integrity check.
:set_default - Set columns referencing this row to their default value.
:set_null - Set columns referencing this row to NULL.
Any other object given is just converted to a string, with “_” converted to “ ” and upcased.
# File lib/sequel/database/schema_methods.rb, line 736 def on_delete_clause(action) action.to_s.gsub("_", " ").upcase end
Alias of on_delete_clause, since the two usually behave the same.
# File lib/sequel/database/schema_methods.rb, line 741 def on_update_clause(action) on_delete_clause(action) end
Proxy the #quote_schema_table method to the dataset
# File lib/sequel/database/schema_methods.rb, line 746 def quote_schema_table(table) schema_utility_dataset.quote_schema_table(table) end
SQL DDL statement for renaming a table.
# File lib/sequel/database/schema_methods.rb, line 751 def rename_table_sql(name, new_name) "ALTER TABLE #{quote_schema_table(name)} RENAME TO #{quote_schema_table(new_name)}" end
Remove the cached #schema_utility_dataset, because the identifier quoting has changed.
# File lib/sequel/database/schema_methods.rb, line 757 def reset_schema_utility_dataset @schema_utility_dataset = nil end
Split the schema information from the table
# File lib/sequel/database/schema_methods.rb, line 762 def schema_and_table(table_name) schema_utility_dataset.schema_and_table(table_name) end
Return true if the given column schema represents an autoincrementing primary key.
# File lib/sequel/database/schema_methods.rb, line 767 def schema_autoincrementing_primary_key?(schema) !!(schema[:primary_key] && schema[:db_type] =~ /int/o) end
The dataset to use for proxying certain schema methods.
# File lib/sequel/database/schema_methods.rb, line 772 def schema_utility_dataset @schema_utility_dataset ||= dataset end
Split the schema information from the table
# File lib/sequel/database/schema_methods.rb, line 777 def split_qualifiers(table_name) schema_utility_dataset.split_qualifiers(table_name) end
SQL DDL fragment for temporary table
# File lib/sequel/database/schema_methods.rb, line 782 def temporary_table_sql self.class.const_get(:TEMPORARY) end
SQL fragment specifying the type of a given column.
# File lib/sequel/database/schema_methods.rb, line 787 def type_literal(column) column[:type].is_a?(Class) ? type_literal_generic(column) : type_literal_specific(column) end
SQL fragment specifying the full type of a column, consider the type with possible modifiers.
# File lib/sequel/database/schema_methods.rb, line 793 def type_literal_generic(column) meth = "type_literal_generic_#{column[:type].name.to_s.downcase}" if respond_to?(meth, true) send(meth, column) else raise Error, "Unsupported ruby class used as database type: #{column[:type]}" end end
Alias for #type_literal_generic_numeric, to make overriding in a subclass easier.
# File lib/sequel/database/schema_methods.rb, line 803 def type_literal_generic_bigdecimal(column) type_literal_generic_numeric(column) end
Sequel uses the bigint type by default for Bignums.
# File lib/sequel/database/schema_methods.rb, line 808 def type_literal_generic_bignum(column) :bigint end
Sequel uses the date type by default for Dates.
# File lib/sequel/database/schema_methods.rb, line 813 def type_literal_generic_date(column) :date end
Sequel uses the timestamp type by default for DateTimes.
# File lib/sequel/database/schema_methods.rb, line 818 def type_literal_generic_datetime(column) :timestamp end
Alias for #type_literal_generic_trueclass, to make overriding in a subclass easier.
# File lib/sequel/database/schema_methods.rb, line 823 def type_literal_generic_falseclass(column) type_literal_generic_trueclass(column) end
Sequel uses the blob type by default for Files.
# File lib/sequel/database/schema_methods.rb, line 828 def type_literal_generic_file(column) :blob end
Alias for #type_literal_generic_integer, to make overriding in a subclass easier.
# File lib/sequel/database/schema_methods.rb, line 833 def type_literal_generic_fixnum(column) type_literal_generic_integer(column) end
Sequel uses the double precision type by default for Floats.
# File lib/sequel/database/schema_methods.rb, line 838 def type_literal_generic_float(column) :"double precision" end
Sequel uses the integer type by default for integers
# File lib/sequel/database/schema_methods.rb, line 843 def type_literal_generic_integer(column) :integer end
Sequel uses the numeric type by default for Numerics and BigDecimals. If a size is given, it is used, otherwise, it will default to whatever the database default is for an unsized value.
# File lib/sequel/database/schema_methods.rb, line 850 def type_literal_generic_numeric(column) column[:size] ? "numeric(#{Array(column[:size]).join(', ')})" : :numeric end
Sequel uses the varchar type by default for Strings. If a size isn't present, Sequel assumes a size of 255. If the :fixed option is used, Sequel uses the char type. If the :text option is used, Sequel uses the :text type.
# File lib/sequel/database/schema_methods.rb, line 858 def type_literal_generic_string(column) if column[:text] uses_clob_for_text? ? :clob : :text elsif column[:fixed] "char(#{column[:size]||default_string_column_size})" else "varchar(#{column[:size]||default_string_column_size})" end end
Sequel uses the timestamp type by default for Time values. If the :only_time option is used, the time type is used.
# File lib/sequel/database/schema_methods.rb, line 870 def type_literal_generic_time(column) column[:only_time] ? :time : :timestamp end
Sequel uses the boolean type by default for TrueClass and FalseClass.
# File lib/sequel/database/schema_methods.rb, line 875 def type_literal_generic_trueclass(column) :boolean end
SQL fragment for the given type of a column if the column is not one of the generic types specified with a ruby class.
# File lib/sequel/database/schema_methods.rb, line 881 def type_literal_specific(column) type = column[:type] type = "double precision" if type.to_s == 'double' column[:size] ||= default_string_column_size if type.to_s == 'varchar' elements = column[:size] || column[:elements] "#{type}#{literal(Array(elements)) if elements}#{UNSIGNED if column[:unsigned]}" end
Whether clob should be used for String :text=>true columns.
# File lib/sequel/database/schema_methods.rb, line 890 def uses_clob_for_text? false end
Returns a dataset for the database. If the first argument is a string, the method acts as an alias for #fetch, returning a dataset for arbitrary SQL, with or without placeholders:
DB['SELECT * FROM items'].all DB['SELECT * FROM items WHERE name = ?', my_name].all
Otherwise, acts as an alias for #from, setting the primary table for the dataset:
DB[:items].sql #=> "SELECT * FROM items"
# File lib/sequel/database/dataset.rb, line 19 def [](*args) (String === args.first) ? fetch(*args) : from(*args) end
Returns a blank dataset for this database.
DB.dataset # SELECT * DB.dataset.from(:items) # SELECT * FROM items
# File lib/sequel/database/dataset.rb, line 27 def dataset(opts=nil) @dataset_class.new(self, opts) end
Fetches records for an arbitrary SQL statement. If a block is given, it is used to iterate over the records:
DB.fetch('SELECT * FROM items'){|r| p r}
The fetch
method returns a dataset instance:
DB.fetch('SELECT * FROM items').all
fetch
can also perform parameterized queries for protection
against SQL injection:
DB.fetch('SELECT * FROM items WHERE name = ?', my_name).all
# File lib/sequel/database/dataset.rb, line 44 def fetch(sql, *args, &block) ds = dataset.with_sql(sql, *args) ds.each(&block) if block ds end
Returns a new dataset with the from
method invoked. If a block
is given, it is used as a filter on the dataset.
DB.from(:items) # SELECT * FROM items DB.from(:items){id > 2} # SELECT * FROM items WHERE (id > 2)
# File lib/sequel/database/dataset.rb, line 55 def from(*args, &block) ds = dataset.from(*args) block ? ds.filter(&block) : ds end
Returns a new dataset with the select method invoked.
DB.select(1) # SELECT 1 DB.select{server_version{}} # SELECT server_version() DB.select(:id).from(:items) # SELECT id FROM items
# File lib/sequel/database/dataset.rb, line 65 def select(*args, &block) dataset.select(*args, &block) end
Array of supported database adapters
The Database subclass for the given adapter scheme. Raises Sequel::AdapterNotFound if the adapter could not be loaded.
# File lib/sequel/database/connecting.rb, line 17 def self.adapter_class(scheme) return scheme if scheme.is_a?(Class) scheme = scheme.to_s.gsub('-', '_').to_sym unless klass = ADAPTER_MAP[scheme] # attempt to load the adapter file begin Sequel.tsk_require "sequel/adapters/#{scheme}" rescue LoadError => e raise Sequel.convert_exception_class(e, AdapterNotFound) end # make sure we actually loaded the adapter unless klass = ADAPTER_MAP[scheme] raise AdapterNotFound, "Could not load #{scheme} adapter: adapter class not registered in ADAPTER_MAP" end end klass end
Returns the scheme symbol for the Database class.
# File lib/sequel/database/connecting.rb, line 39 def self.adapter_scheme @scheme if defined?(@scheme) end
Connects to a database. See Sequel.connect.
# File lib/sequel/database/connecting.rb, line 44 def self.connect(conn_string, opts = {}) case conn_string when String if match = /\A(jdbc|do):/.match(conn_string) c = adapter_class(match[1].to_sym) opts = opts.merge(:orig_opts=>opts.dup) opts = {:uri=>conn_string}.merge(opts) else uri = URI.parse(conn_string) scheme = uri.scheme scheme = :dbi if scheme =~ /\Adbi-/ c = adapter_class(scheme) uri_options = c.send(:uri_to_options, uri) uri.query.split('&').collect{|s| s.split('=')}.each{|k,v| uri_options[k.to_sym] = v if k && !k.empty?} unless uri.query.to_s.strip.empty? uri_options.to_a.each{|k,v| uri_options[k] = (defined?(URI::DEFAULT_PARSER) ? URI::DEFAULT_PARSER : URI).unescape(v) if v.is_a?(String)} opts = uri_options.merge(opts).merge!(:orig_opts=>opts.dup, :uri=>conn_string, :adapter=>scheme) end when Hash opts = conn_string.merge(opts) opts = opts.merge(:orig_opts=>opts.dup) c = adapter_class(opts[:adapter_class] || opts[:adapter] || opts['adapter']) else raise Error, "Sequel::Database.connect takes either a Hash or a String, given: #{conn_string.inspect}" end # process opts a bit opts = opts.inject({}) do |m, (k,v)| k = :user if k.to_s == 'username' m[k.to_sym] = v m end begin db = c.new(opts) db.test_connection if opts[:test] && db.send(:typecast_value_boolean, opts[:test]) if block_given? return yield(db) end ensure if block_given? db.disconnect if db Sequel.synchronize{::Sequel::DATABASES.delete(db)} end end db end
Sets the default single_threaded mode for new databases. See Sequel.single_threaded=.
# File lib/sequel/database/connecting.rb, line 91 def self.single_threaded=(value) @@single_threaded = value end
Returns the scheme symbol for this instance's class, which reflects
which adapter is being used. In some cases, this can be the same as the
database_type
(for native adapters), in others (i.e. adapters
with subadapters), it will be different.
Sequel.connect('jdbc:postgres://...').adapter_scheme # => :jdbc
# File lib/sequel/database/connecting.rb, line 123 def adapter_scheme self.class.adapter_scheme end
Dynamically add new servers or modify server options at runtime. Also adds new servers to the connection pool. Intended for use with master/slave or shard configurations where it is useful to add new server hosts at runtime.
servers argument should be a hash with server name symbol keys and hash or proc values. If a servers key is already in use, it's value is overridden with the value provided.
DB.add_servers(:f=>{:host=>"hash_host_f"})
# File lib/sequel/database/connecting.rb, line 136 def add_servers(servers) if h = @opts[:servers] Sequel.synchronize{h.merge!(servers)} @pool.add_servers(servers.keys) end end
Connects to the database. This method should be overridden by descendants.
# File lib/sequel/database/connecting.rb, line 144 def connect(server) raise NotImplemented, "#connect should be overridden by adapters" end
The database type for this database object, the same as the adapter scheme by default. Should be overridden in adapters (especially shared adapters) to be the correct type, so that even if two separate Database objects are using different adapters you can tell that they are using the same database type. Even better, you can tell that two Database objects that are using the same adapter are connecting to different database types (think JDBC or DataObjects).
Sequel.connect('jdbc:postgres://...').database_type # => :postgres
# File lib/sequel/database/connecting.rb, line 157 def database_type adapter_scheme end
Disconnects all available connections from the connection pool. Any connections currently in use will not be disconnected. Options:
Should be a symbol specifing the server to disconnect from, or an array of symbols to specify multiple servers.
Example:
DB.disconnect # All servers DB.disconnect(:servers=>:server1) # Single server DB.disconnect(:servers=>[:server1, :server2]) # Multiple servers
# File lib/sequel/database/connecting.rb, line 171 def disconnect(opts = {}) pool.disconnect(opts) end
Should only be called by the connection pool code to disconnect a connection. By default, calls the close method on the connection object, since most adapters use that, but should be overwritten on other adapters.
# File lib/sequel/database/connecting.rb, line 178 def disconnect_connection(conn) conn.close end
Yield a new Database instance for every server in the connection pool. Intended for use in sharded environments where there is a need to make schema modifications (DDL queries) on each shard.
DB.each_server{|db| db.create_table(:users){primary_key :id; String :name}}
# File lib/sequel/database/connecting.rb, line 187 def each_server(&block) raise(Error, "Database#each_server must be passed a block") unless block servers.each{|s| self.class.connect(server_opts(s), &block)} end
Dynamically remove existing servers from the connection pool. Intended for use with master/slave or shard configurations where it is useful to remove existing server hosts at runtime.
servers should be symbols or arrays of symbols. If a nonexistent server is specified, it is ignored. If no servers have been specified for this database, no changes are made. If you attempt to remove the :default server, an error will be raised.
DB.remove_servers(:f1, :f2)
# File lib/sequel/database/connecting.rb, line 202 def remove_servers(*servers) if h = @opts[:servers] servers.flatten.each{|s| Sequel.synchronize{h.delete(s)}} @pool.remove_servers(servers) end end
An array of servers/shards for this Database object.
DB.servers # Unsharded: => [:default] DB.servers # Sharded: => [:default, :server1, :server2]
# File lib/sequel/database/connecting.rb, line 213 def servers pool.servers end
Returns true if the database is using a single-threaded connection pool.
# File lib/sequel/database/connecting.rb, line 218 def single_threaded? @single_threaded end
Acquires a database connection, yielding it to the passed block. This is useful if you want to make sure the same connection is used for all database queries in the block. It is also useful if you want to gain direct access to the underlying connection object if you need to do something Sequel does not natively support.
If a server option is given, acquires a connection for that specific server, instead of the :default server.
DB.synchronize do |conn| ... end
# File lib/sequel/database/connecting.rb, line 235 def synchronize(server=nil) @pool.hold(server || :default){|conn| yield conn} end
Attempts to acquire a database connection. Returns true if successful. Will probably raise an Error if unsuccessful. If a server argument is given, attempts to acquire a database connection to the given server/shard.
# File lib/sequel/database/connecting.rb, line 250 def test_connection(server=nil) synchronize(server){|conn|} true end
Check whether the given connection is currently valid, by running a query against it. If the query fails, the connection should probably be removed from the connection pool.
# File lib/sequel/database/connecting.rb, line 259 def valid_connection?(conn) sql = valid_connection_sql begin log_connection_execute(conn, sql) rescue Sequel::DatabaseError, *database_error_classes false else true end end
The default options for the connection pool.
# File lib/sequel/database/connecting.rb, line 273 def connection_pool_default_options {} end
Return the options for the given server by merging the generic options for all server with the specific options for the given server specified in the :servers option.
# File lib/sequel/database/connecting.rb, line 280 def server_opts(server) opts = if @opts[:servers] and server_options = @opts[:servers][server] case server_options when Hash @opts.merge(server_options) when Proc @opts.merge(server_options.call(self)) else raise Error, 'Server opts should be a hash or proc' end elsif server.is_a?(Hash) @opts.merge(server) else @opts.dup end opts.delete(:servers) opts end
The SQL query to issue to check if a connection is valid.
# File lib/sequel/database/connecting.rb, line 300 def valid_connection_sql @valid_connection_sql ||= select(nil).sql end
The default class to use for datasets
The class to use for creating datasets. Should respond to new with the Database argument as the first argument, and an optional options hash.
The default schema to use, generally should be nil. This sets the default schema used for some schema modification and introspection queries, but does not effect most dataset code.
The method to call on identifiers going into the database
# File lib/sequel/database/dataset_defaults.rb, line 21 def self.identifier_input_method @@identifier_input_method end
Set the method to call on identifiers going into the database See Sequel.identifier_input_method=.
# File lib/sequel/database/dataset_defaults.rb, line 27 def self.identifier_input_method=(v) @@identifier_input_method = v || "" end
The method to call on identifiers coming from the database
# File lib/sequel/database/dataset_defaults.rb, line 32 def self.identifier_output_method @@identifier_output_method end
Set the method to call on identifiers coming from the database See Sequel.identifier_output_method=.
# File lib/sequel/database/dataset_defaults.rb, line 38 def self.identifier_output_method=(v) @@identifier_output_method = v || "" end
Sets the default quote_identifiers mode for new databases. See Sequel.quote_identifiers=.
# File lib/sequel/database/dataset_defaults.rb, line 44 def self.quote_identifiers=(value) @@quote_identifiers = value end
If the database has any dataset modules associated with it, use a subclass of the given class that includes the modules as the dataset class.
# File lib/sequel/database/dataset_defaults.rb, line 61 def dataset_class=(c) unless @dataset_modules.empty? c = Class.new(c) @dataset_modules.each{|m| c.send(:include, m)} end @dataset_class = c end
Equivalent to extending all datasets produced by the database with a module. What it actually does is use a subclass of the current #dataset_class as the new #dataset_class, and include the module in the subclass. Instead of a module, you can provide a block that is used to create an anonymous module.
This allows you to override any of the dataset methods even if they are defined directly on the dataset class that this Database object uses.
Examples:
# Introspec columns for all of DB's datasets DB.extend_datasets(Sequel::ColumnsIntrospection) # Trace all SELECT queries by printing the SQL and the full backtrace DB.extend_datasets do def fetch_rows(sql) puts sql puts caller super end end
# File lib/sequel/database/dataset_defaults.rb, line 91 def extend_datasets(mod=nil, &block) raise(Error, "must provide either mod or block, not both") if mod && block reset_schema_utility_dataset mod = Module.new(&block) if block if @dataset_modules.empty? @dataset_modules = [mod] @dataset_class = Class.new(@dataset_class) else @dataset_modules << mod end @dataset_class.send(:include, mod) end
The method to call on identifiers going into the database
# File lib/sequel/database/dataset_defaults.rb, line 105 def identifier_input_method case @identifier_input_method when nil @identifier_input_method = @opts.fetch(:identifier_input_method, (@@identifier_input_method.nil? ? identifier_input_method_default : @@identifier_input_method)) @identifier_input_method == "" ? nil : @identifier_input_method when "" nil else @identifier_input_method end end
Set the method to call on identifiers going into the database:
DB[:items] # SELECT * FROM items DB.identifier_input_method = :upcase DB[:items] # SELECT * FROM ITEMS
# File lib/sequel/database/dataset_defaults.rb, line 122 def identifier_input_method=(v) reset_schema_utility_dataset @identifier_input_method = v || "" end
The method to call on identifiers coming from the database
# File lib/sequel/database/dataset_defaults.rb, line 128 def identifier_output_method case @identifier_output_method when nil @identifier_output_method = @opts.fetch(:identifier_output_method, (@@identifier_output_method.nil? ? identifier_output_method_default : @@identifier_output_method)) @identifier_output_method == "" ? nil : @identifier_output_method when "" nil else @identifier_output_method end end
Set the method to call on identifiers coming from the database:
DB[:items].first # {:id=>1, :name=>'foo'} DB.identifier_output_method = :upcase DB[:items].first # {:ID=>1, :NAME=>'foo'}
# File lib/sequel/database/dataset_defaults.rb, line 145 def identifier_output_method=(v) reset_schema_utility_dataset @identifier_output_method = v || "" end
Set whether to quote identifiers (columns and tables) for this database:
DB[:items] # SELECT * FROM items DB.quote_identifiers = true DB[:items] # SELECT * FROM "items"
# File lib/sequel/database/dataset_defaults.rb, line 155 def quote_identifiers=(v) reset_schema_utility_dataset @quote_identifiers = v end
Returns true if the database quotes identifiers.
# File lib/sequel/database/dataset_defaults.rb, line 161 def quote_identifiers? return @quote_identifiers unless @quote_identifiers.nil? @quote_identifiers = @opts.fetch(:quote_identifiers, (@@quote_identifiers.nil? ? quote_identifiers_default : @@quote_identifiers)) end
The default dataset class to use for the database
# File lib/sequel/database/dataset_defaults.rb, line 169 def dataset_class_default self.class.const_get(:DatasetClass) end
The default value for default_schema.
# File lib/sequel/database/dataset_defaults.rb, line 174 def default_schema_default nil end
The method to apply to identifiers going into the database by default. Should be overridden in subclasses for databases that fold unquoted identifiers to lower case instead of uppercase, such as MySQL, PostgreSQL, and SQLite.
# File lib/sequel/database/dataset_defaults.rb, line 182 def identifier_input_method_default :upcase end
The method to apply to identifiers coming the database by default. Should be overridden in subclasses for databases that fold unquoted identifiers to lower case instead of uppercase, such as MySQL, PostgreSQL, and SQLite.
# File lib/sequel/database/dataset_defaults.rb, line 190 def identifier_output_method_default :downcase end
Whether to quote identifiers by default for this database, true by default.
# File lib/sequel/database/dataset_defaults.rb, line 196 def quote_identifiers_default true end
Numeric specifying the duration beyond which queries are logged at warn level instead of info level.
Log level at which to log SQL queries. This is actually the method sent to the logger, so it should be the method name symbol. The default is :info, it can be set to :debug to log at DEBUG level.
Log a message at error level, with information about the exception.
# File lib/sequel/database/logging.rb, line 21 def log_exception(exception, message) log_each(:error, "#{exception.class}: #{exception.message.strip if exception.message}: #{message}") end
Log a message at level info to all loggers.
# File lib/sequel/database/logging.rb, line 26 def log_info(message, args=nil) log_each(:info, args ? "#{message}; #{args.inspect}" : message) end
Yield to the block, logging any errors at error level to all loggers, and all other queries with the duration at warn or info level.
# File lib/sequel/database/logging.rb, line 32 def log_yield(sql, args=nil) return yield if @loggers.empty? sql = "#{sql}; #{args.inspect}" if args start = Time.now begin yield rescue => e log_exception(e, sql) raise ensure log_duration(Time.now - start, sql) unless e end end
Remove any existing loggers and just use the given logger:
DB.logger = Logger.new($stdout)
# File lib/sequel/database/logging.rb, line 49 def logger=(logger) @loggers = Array(logger) end
Log the given SQL and then execute it on the connection, used by the transaction code.
# File lib/sequel/database/logging.rb, line 57 def log_connection_execute(conn, sql) log_yield(sql){conn.send(connection_execute_method, sql)} end
Log message with message prefixed by duration at info level, or warn level if duration is greater than log_warn_duration.
# File lib/sequel/database/logging.rb, line 63 def log_duration(duration, message) log_each((lwd = log_warn_duration and duration >= lwd) ? :warn : sql_log_level, "(#{sprintf('%0.6fs', duration)}) #{message}") end
Log message at level (which should be :error, :warn, or :info) to all loggers.
# File lib/sequel/database/logging.rb, line 69 def log_each(level, message) @loggers.each{|logger| logger.send(level, message)} end
Empty exception regexp to class map, used by default if Sequel doesn't have specific support for the database in use.
The general default size for string columns for all Sequel::Database instances.
Hash of extension name symbols to callable objects to load the extension into the Database object (usually by extending it with a module defined in the extension).
Used for checking/removing leading zeroes from strings so they don't get interpreted as octal.
:nocov: Replacement string when replacing leading zeroes.
Mapping of schema type symbols to class or arrays of classes for that symbol.
The specific default size of string columns for this Sequel::Database, usually 255 by default.
The options hash for this database
Set the timezone to use for this database, overridding
Sequel.database_timezone
.
Register a hook that will be run when a new Database is instantiated. It is called with the new database handle.
# File lib/sequel/database/misc.rb, line 32 def self.after_initialize(&block) raise Error, "must provide block to after_initialize" unless block Sequel.synchronize do previous = @initialize_hook @initialize_hook = Proc.new do |db| previous.call(db) block.call(db) end end end
Apply an extension to all Database objects created in the future.
# File lib/sequel/database/misc.rb, line 44 def self.extension(*extensions) after_initialize{|db| db.extension(*extensions)} end
Constructs a new instance of a database connection with the specified options hash.
Accepts the following options:
The default schema to use, see default_schema.
The default size of string columns, 255 by default.
A string method symbol to call on identifiers going into the database
A string method symbol to call on identifiers coming from the database
A specific logger to use
An array of loggers to use
Whether to quote identifiers
A hash specifying a server/shard specific options, keyed by shard symbol
Whether to use a single-threaded connection pool
Method to use to log SQL to a logger, :info by default.
All options given are also passed to the connection pool.
# File lib/sequel/database/misc.rb, line 106 def initialize(opts = {}, &block) @opts ||= opts @opts = connection_pool_default_options.merge(@opts) @loggers = Array(@opts[:logger]) + Array(@opts[:loggers]) self.log_warn_duration = @opts[:log_warn_duration] block ||= proc{|server| connect(server)} @opts[:servers] = {} if @opts[:servers].is_a?(String) @opts[:adapter_class] = self.class @opts[:single_threaded] = @single_threaded = typecast_value_boolean(@opts.fetch(:single_threaded, @@single_threaded)) @schemas = {} @default_schema = @opts.fetch(:default_schema, default_schema_default) @default_string_column_size = @opts[:default_string_column_size] || DEFAULT_STRING_COLUMN_SIZE @prepared_statements = {} @transactions = {} @identifier_input_method = nil @identifier_output_method = nil @quote_identifiers = nil @timezone = nil @dataset_class = dataset_class_default @cache_schema = typecast_value_boolean(@opts.fetch(:cache_schema, true)) @dataset_modules = [] @schema_type_classes = SCHEMA_TYPE_CLASSES.dup self.sql_log_level = @opts[:sql_log_level] ? @opts[:sql_log_level].to_sym : :info @pool = ConnectionPool.get_pool(self, @opts) unless typecast_value_boolean(@opts[:keep_reference]) == false Sequel.synchronize{::Sequel::DATABASES.push(self)} end Sequel::Database.run_after_initialize(self) end
Register an extension callback for Database objects. ext should be the extension name symbol, and mod should either be a Module that the database is extended with, or a callable object called with the database object. If mod is not provided, a block can be provided and is treated as the mod object.
# File lib/sequel/database/misc.rb, line 53 def self.register_extension(ext, mod=nil, &block) if mod raise(Error, "cannot provide both mod and block to Database.register_extension") if block if mod.is_a?(Module) block = proc{|db| db.extend(mod)} else block = mod end end Sequel.synchronize{EXTENSIONS[ext] = block} end
Run the ::after_initialize hook
for the given instance
.
# File lib/sequel/database/misc.rb, line 66 def self.run_after_initialize(instance) @initialize_hook.call(instance) end
Converts a uri to an options hash. These options are then passed to a newly created database object.
# File lib/sequel/database/misc.rb, line 72 def self.uri_to_options(uri) { :user => uri.user, :password => uri.password, :host => uri.host, :port => uri.port, :database => (m = /\/(.*)/.match(uri.path)) && (m[1]) } end
If a transaction is not currently in process, yield to the block immediately. Otherwise, add the block to the list of blocks to call after the currently in progress transaction commits (and only if it commits). Options:
The server/shard to use.
# File lib/sequel/database/misc.rb, line 143 def after_commit(opts={}, &block) raise Error, "must provide block to after_commit" unless block synchronize(opts[:server]) do |conn| if h = _trans(conn) raise Error, "cannot call after_commit in a prepared transaction" if h[:prepare] (h[:after_commit] ||= []) << block else yield end end end
If a transaction is not currently in progress, ignore the block. Otherwise, add the block to the list of the blocks to call after the currently in progress transaction rolls back (and only if it rolls back). Options:
The server/shard to use.
# File lib/sequel/database/misc.rb, line 160 def after_rollback(opts={}, &block) raise Error, "must provide block to after_rollback" unless block synchronize(opts[:server]) do |conn| if h = _trans(conn) raise Error, "cannot call after_rollback in a prepared transaction" if h[:prepare] (h[:after_rollback] ||= []) << block end end end
Cast the given type to a literal type
DB.cast_type_literal(Float) # double precision DB.cast_type_literal(:foo) # foo
# File lib/sequel/database/misc.rb, line 174 def cast_type_literal(type) type_literal(:type=>type) end
Load an extension into the receiver. In addition to requiring the extension file, this also modifies the database to work with the extension (usually extending it with a module defined in the extension file). If no related extension file exists or the extension does not have specific support for Database objects, an Error will be raised. Returns self.
# File lib/sequel/database/misc.rb, line 183 def extension(*exts) Sequel.extension(*exts) exts.each do |ext| if pr = Sequel.synchronize{EXTENSIONS[ext]} pr.call(self) else raise(Error, "Extension #{ext} does not have specific support handling individual databases") end end self end
Convert the given timestamp from the application's timezone, to the databases's timezone or the default database timezone if the database does not have a timezone.
# File lib/sequel/database/misc.rb, line 198 def from_application_timestamp(v) Sequel.convert_output_timestamp(v, timezone) end
Return true if already in a transaction given the options, false otherwise. Respects the :server option for selecting a shard.
# File lib/sequel/database/misc.rb, line 205 def in_transaction?(opts={}) synchronize(opts[:server]){|conn| !!_trans(conn)} end
Returns a string representation of the database object including the class name and connection URI and options used when connecting (if any).
# File lib/sequel/database/misc.rb, line 211 def inspect a = [] a << uri.inspect if uri if (oo = opts[:orig_opts]) && !oo.empty? a << oo.inspect end "#<#{self.class}: #{a.join(' ')}>" end
Proxy the literal call to the dataset.
DB.literal(1) # 1 DB.literal(:a) # a DB.literal('a') # 'a'
# File lib/sequel/database/misc.rb, line 225 def literal(v) schema_utility_dataset.literal(v) end
Synchronize access to the prepared statements cache.
# File lib/sequel/database/misc.rb, line 230 def prepared_statement(name) Sequel.synchronize{prepared_statements[name]} end
Proxy the #quote_identifier method to the dataset, useful for quoting unqualified identifiers for use outside of datasets.
# File lib/sequel/database/misc.rb, line 237 def quote_identifier(v) schema_utility_dataset.quote_identifier(v) end
Return ruby class or array of classes for the given type symbol.
# File lib/sequel/database/misc.rb, line 242 def schema_type_class(type) @schema_type_classes[type] end
Default serial primary key options, used by the table creation code.
# File lib/sequel/database/misc.rb, line 248 def serial_primary_key_options {:primary_key => true, :type => Integer, :auto_increment => true} end
Cache the prepared statement object at the given name.
# File lib/sequel/database/misc.rb, line 253 def set_prepared_statement(name, ps) ps.prepared_sql Sequel.synchronize{prepared_statements[name] = ps} end
The timezone to use for this database, defaulting to
Sequel.database_timezone
.
# File lib/sequel/database/misc.rb, line 259 def timezone @timezone || Sequel.database_timezone end
Convert the given timestamp to the application's timezone, from the databases's timezone or the default database timezone if the database does not have a timezone.
# File lib/sequel/database/misc.rb, line 266 def to_application_timestamp(v) Sequel.convert_timestamp(v, timezone) end
Typecast the value to the given column_type. Calls typecast_value_#{column_type} if the method exists, otherwise returns the value. This method should raise Sequel::InvalidValue if assigned value is invalid.
# File lib/sequel/database/misc.rb, line 275 def typecast_value(column_type, value) return nil if value.nil? meth = "typecast_value_#{column_type}" begin respond_to?(meth, true) ? send(meth, value) : value rescue ArgumentError, TypeError => e raise Sequel.convert_exception_class(e, InvalidValue) end end
Returns the URI use to connect to the database. If a URI was not used when connecting, returns nil.
# File lib/sequel/database/misc.rb, line 287 def uri opts[:uri] end
Explicit alias of uri for easier subclassing.
# File lib/sequel/database/misc.rb, line 292 def url uri end
Returns true when the object is considered blank. The only objects that are blank are nil, false, strings with all whitespace, and ones that respond true to empty?
# File lib/sequel/database/misc.rb, line 302 def blank_object?(obj) return obj.blank? if obj.respond_to?(:blank?) case obj when NilClass, FalseClass true when Numeric, TrueClass false when String obj.strip.empty? else obj.respond_to?(:empty?) ? obj.empty? : false end end
Return the Sequel::DatabaseError subclass to wrap the given exception in.
# File lib/sequel/database/misc.rb, line 330 def database_error_class(exception, opts) database_specific_error_class(exception, opts) || DatabaseError end
Which transaction errors to translate, blank by default.
# File lib/sequel/database/misc.rb, line 317 def database_error_classes [] end
An enumerable yielding pairs of regexps and exception classes, used to match against underlying driver exception messages in order to raise a more specific Sequel::DatabaseError subclass.
# File lib/sequel/database/misc.rb, line 324 def database_error_regexps DEFAULT_DATABASE_ERROR_REGEXPS end
Return the SQLState for the given exception, if one can be determined
# File lib/sequel/database/misc.rb, line 336 def database_exception_sqlstate(exception, opts) nil end
Return a specific Sequel::DatabaseError exception class if one is appropriate for the underlying exception, or nil if there is no specific exception class.
# File lib/sequel/database/misc.rb, line 343 def database_specific_error_class(exception, opts) return DatabaseDisconnectError if disconnect_error?(exception, opts) if sqlstate = database_exception_sqlstate(exception, opts) if klass = database_specific_error_class_from_sqlstate(sqlstate) return klass end else database_error_regexps.each do |regexp, klss| return klss if exception.message =~ regexp end end nil end
Given the SQLState, return the appropriate DatabaseError subclass.
# File lib/sequel/database/misc.rb, line 365 def database_specific_error_class_from_sqlstate(sqlstate) case sqlstate when *NOT_NULL_CONSTRAINT_SQLSTATES NotNullConstraintViolation when *FOREIGN_KEY_CONSTRAINT_SQLSTATES ForeignKeyConstraintViolation when *UNIQUE_CONSTRAINT_SQLSTATES UniqueConstraintViolation when *CHECK_CONSTRAINT_SQLSTATES CheckConstraintViolation when *SERIALIZATION_CONSTRAINT_SQLSTATES SerializationFailure end end
Return true if exception represents a disconnect error, false otherwise.
# File lib/sequel/database/misc.rb, line 381 def disconnect_error?(exception, opts) opts[:disconnect] end
Convert the given exception to an appropriate Sequel::DatabaseError subclass, keeping message and traceback.
# File lib/sequel/database/misc.rb, line 387 def raise_error(exception, opts={}) if !opts[:classes] || Array(opts[:classes]).any?{|c| exception.is_a?(c)} raise Sequel.convert_exception_class(exception, database_error_class(exception, opts)) else raise exception end end
Typecast the value to an SQL::Blob
# File lib/sequel/database/misc.rb, line 396 def typecast_value_blob(value) value.is_a?(Sequel::SQL::Blob) ? value : Sequel::SQL::Blob.new(value) end
Typecast the value to true, false, or nil
# File lib/sequel/database/misc.rb, line 401 def typecast_value_boolean(value) case value when false, 0, "0", /\Af(alse)?\z/, /\Ano?\z/ false else blank_object?(value) ? nil : true end end
Typecast the value to a Date
# File lib/sequel/database/misc.rb, line 411 def typecast_value_date(value) case value when DateTime, Time Date.new(value.year, value.month, value.day) when Date value when String Sequel.string_to_date(value) when Hash Date.new(*[:year, :month, :day].map{|x| (value[x] || value[x.to_s]).to_i}) else raise InvalidValue, "invalid value for Date: #{value.inspect}" end end
Typecast the value to a DateTime or Time depending on Sequel.datetime_class
# File lib/sequel/database/misc.rb, line 427 def typecast_value_datetime(value) Sequel.typecast_to_application_timestamp(value) end
Typecast the value to a BigDecimal
# File lib/sequel/database/misc.rb, line 432 def typecast_value_decimal(value) case value when BigDecimal value when Numeric BigDecimal.new(value.to_s) when String d = BigDecimal.new(value) if d.zero? # BigDecimal parsing is loose by default, returning a 0 value for # invalid input. If a zero value is received, use Float to check # for validity. begin Float(value) rescue ArgumentError raise InvalidValue, "invalid value for BigDecimal: #{value.inspect}" end end d else raise InvalidValue, "invalid value for BigDecimal: #{value.inspect}" end end
Typecast the value to a Float
# File lib/sequel/database/misc.rb, line 457 def typecast_value_float(value) Float(value) end
Typecast the value to an Integer
# File lib/sequel/database/misc.rb, line 466 def typecast_value_integer(value) (value.is_a?(String) && value =~ LEADING_ZERO_RE) ? Integer(value, 10) : Integer(value) end
Typecast the value to a String
# File lib/sequel/database/misc.rb, line 481 def typecast_value_string(value) value.to_s end
Typecast the value to a Time
# File lib/sequel/database/misc.rb, line 486 def typecast_value_time(value) case value when Time if value.is_a?(SQLTime) value else # specifically check for nsec == 0 value to work around JRuby 1.6 ruby 1.9 mode bug SQLTime.create(value.hour, value.min, value.sec, (value.respond_to?(:nsec) && value.nsec != 0) ? value.nsec/1000.0 : value.usec) end when String Sequel.string_to_time(value) when Hash SQLTime.create(*[:hour, :minute, :second].map{|x| (value[x] || value[x.to_s]).to_i}) else raise Sequel::InvalidValue, "invalid value for Time: #{value.inspect}" end end
Whether the database uses a global namespace for the index. If false, the indexes are going to be namespaced per table.
# File lib/sequel/database/features.rb, line 11 def global_index_namespace? true end
Whether the database supports CREATE TABLE IF NOT EXISTS syntax, false by default.
# File lib/sequel/database/features.rb, line 17 def supports_create_table_if_not_exists? false end
Whether the database supports deferrable constraints, false by default as few databases do.
# File lib/sequel/database/features.rb, line 23 def supports_deferrable_constraints? false end
Whether the database supports deferrable foreign key constraints, false by default as few databases do.
# File lib/sequel/database/features.rb, line 29 def supports_deferrable_foreign_key_constraints? supports_deferrable_constraints? end
Whether the database supports DROP TABLE IF EXISTS syntax, default is the same as supports_create_table_if_not_exists?.
# File lib/sequel/database/features.rb, line 35 def supports_drop_table_if_exists? supports_create_table_if_not_exists? end
Whether the database and adapter support prepared transactions (two-phase commit), false by default.
# File lib/sequel/database/features.rb, line 41 def supports_prepared_transactions? false end
Whether the database and adapter support savepoints, false by default.
# File lib/sequel/database/features.rb, line 46 def supports_savepoints? false end
Whether the database and adapter support savepoints inside prepared transactions (two-phase commit), default is false.
# File lib/sequel/database/features.rb, line 52 def supports_savepoints_in_prepared_transactions? supports_prepared_transactions? && supports_savepoints? end
Whether the database supports schema parsing via #schema.
# File lib/sequel/database/features.rb, line 57 def supports_schema_parsing? respond_to?(:schema_parse_table, true) end
Whether the database and adapter support transaction isolation levels, false by default.
# File lib/sequel/database/features.rb, line 62 def supports_transaction_isolation_levels? false end
Whether DDL statements work correctly in transactions, false by default.
# File lib/sequel/database/features.rb, line 67 def supports_transactional_ddl? false end
Whether the database supports combining multiple alter table operations into a single query, false by default.
# File lib/sequel/database/features.rb, line 75 def supports_combining_alter_table_ops? false end
Whether the database supports CREATE OR REPLACE VIEW. If not, support will be emulated by dropping the view first. false by default.
# File lib/sequel/database/features.rb, line 81 def supports_create_or_replace_view? false end
Whether the database supports named column constraints. True by default. Those that don't support named column constraints have to have column constraints converted to table constraints if the column constraints have names.
# File lib/sequel/database/features.rb, line 89 def supports_named_column_constraints? true end