module Sequel::Postgres::PGArray::DatabaseMethods

  1. lib/sequel/extensions/pg_array.rb

Constants

BLOB_RANGE = 1...-1  

Public Class methods

extended(db)

Create the local hash of database type strings to schema type symbols, used for array types local to this database.

[show source]
    # File lib/sequel/extensions/pg_array.rb
 87 def self.extended(db)
 88   db.instance_exec do
 89     @pg_array_schema_types ||= {}
 90     register_array_type('timestamp without time zone', :oid=>1115, :scalar_oid=>1114, :type_symbol=>:datetime)
 91     register_array_type('timestamp with time zone', :oid=>1185, :scalar_oid=>1184, :type_symbol=>:datetime_timezone, :scalar_typecast=>:datetime)
 92 
 93     register_array_type('text', :oid=>1009, :scalar_oid=>25, :type_symbol=>:string)
 94     register_array_type('integer', :oid=>1007, :scalar_oid=>23)
 95     register_array_type('bigint', :oid=>1016, :scalar_oid=>20, :scalar_typecast=>:integer)
 96     register_array_type('numeric', :oid=>1231, :scalar_oid=>1700, :type_symbol=>:decimal)
 97     register_array_type('double precision', :oid=>1022, :scalar_oid=>701, :type_symbol=>:float)
 98 
 99     register_array_type('boolean', :oid=>1000, :scalar_oid=>16)
100     register_array_type('bytea', :oid=>1001, :scalar_oid=>17, :type_symbol=>:blob)
101     register_array_type('date', :oid=>1182, :scalar_oid=>1082)
102     register_array_type('time without time zone', :oid=>1183, :scalar_oid=>1083, :type_symbol=>:time)
103     register_array_type('time with time zone', :oid=>1270, :scalar_oid=>1266, :type_symbol=>:time_timezone, :scalar_typecast=>:time)
104 
105     register_array_type('smallint', :oid=>1005, :scalar_oid=>21, :scalar_typecast=>:integer)
106     register_array_type('oid', :oid=>1028, :scalar_oid=>26, :scalar_typecast=>:integer)
107     register_array_type('real', :oid=>1021, :scalar_oid=>700, :scalar_typecast=>:float)
108     register_array_type('character', :oid=>1014, :converter=>nil, :array_type=>:text, :scalar_typecast=>:string)
109     register_array_type('character varying', :oid=>1015, :converter=>nil, :scalar_typecast=>:string, :type_symbol=>:varchar)
110 
111     register_array_type('xml', :oid=>143, :scalar_oid=>142)
112     register_array_type('money', :oid=>791, :scalar_oid=>790)
113     register_array_type('bit', :oid=>1561, :scalar_oid=>1560)
114     register_array_type('bit varying', :oid=>1563, :scalar_oid=>1562, :type_symbol=>:varbit)
115     register_array_type('uuid', :oid=>2951, :scalar_oid=>2950)
116 
117     register_array_type('xid', :oid=>1011, :scalar_oid=>28)
118     register_array_type('cid', :oid=>1012, :scalar_oid=>29)
119 
120     register_array_type('name', :oid=>1003, :scalar_oid=>19)
121     register_array_type('tid', :oid=>1010, :scalar_oid=>27)
122     register_array_type('int2vector', :oid=>1006, :scalar_oid=>22)
123     register_array_type('oidvector', :oid=>1013, :scalar_oid=>30)
124 
125     [:string_array, :integer_array, :decimal_array, :float_array, :boolean_array, :blob_array, :date_array, :time_array, :datetime_array].each do |v|
126       @schema_type_classes[v] = PGArray
127     end
128   end
129 end

Public Instance methods

add_named_conversion_proc(name, &block)
[show source]
    # File lib/sequel/extensions/pg_array.rb
131 def add_named_conversion_proc(name, &block)
132   ret = super
133   name = name.to_s if name.is_a?(Symbol)
134   from(:pg_type).where(:typname=>name).select_map([:oid, :typarray]).each do |scalar_oid, array_oid|
135     register_array_type(name, :oid=>array_oid.to_i, :scalar_oid=>scalar_oid.to_i)
136   end
137   ret
138 end
bound_variable_arg(arg, conn)

Handle arrays in bound variables

[show source]
    # File lib/sequel/extensions/pg_array.rb
141 def bound_variable_arg(arg, conn)
142   case arg
143   when PGArray
144     bound_variable_array(arg.to_a)
145   when Array
146     bound_variable_array(arg)
147   else
148     super
149   end
150 end
freeze()

Freeze the pg array schema types to prevent adding new ones.

[show source]
    # File lib/sequel/extensions/pg_array.rb
153 def freeze
154   @pg_array_schema_types.freeze
155   super
156 end
register_array_type(db_type, opts=OPTS, &block)

Register a database specific array type. Options:

:array_type

The type to automatically cast the array to when literalizing the array. Usually the same as db_type.

:converter

A callable object (e.g. Proc), that is called with each element of the array (usually a string), and should return the appropriate typecasted object.

:oid

The PostgreSQL OID for the array type. This is used by the Sequel postgres adapter to set up automatic type conversion on retrieval from the database.

:scalar_oid

Should be the PostgreSQL OID for the scalar version of this array type. If given, automatically sets the :converter option by looking for scalar conversion proc.

:scalar_typecast

Should be a symbol indicating the typecast method that should be called on each element of the array, when a plain array is passed into a database typecast method. For example, for an array of integers, this could be set to :integer, so that the typecast_value_integer method is called on all of the array elements. Defaults to :type_symbol option.

:type_symbol

The base of the schema type symbol for this type. For example, if you provide :integer, Sequel will recognize this type as :integer_array during schema parsing. Defaults to the db_type argument.

If a block is given, it is treated as the :converter option.

[show source]
    # File lib/sequel/extensions/pg_array.rb
179 def register_array_type(db_type, opts=OPTS, &block)
180   oid = opts[:oid]
181   soid = opts[:scalar_oid]
182 
183   if has_converter = opts.has_key?(:converter)
184     raise Error, "can't provide both a block and :converter option to register_array_type" if block
185     converter = opts[:converter]
186   else
187     has_converter = true if block
188     converter = block
189   end
190 
191   unless (soid || has_converter) && oid
192     array_oid, scalar_oid = from(:pg_type).where(:typname=>db_type.to_s).get([:typarray, :oid])
193     soid ||= scalar_oid unless has_converter
194     oid ||= array_oid
195   end
196 
197   db_type = db_type.to_s
198   type = (opts[:type_symbol] || db_type).to_sym
199   typecast_method_map = @pg_array_schema_types
200 
201   if soid
202     raise Error, "can't provide both a converter and :scalar_oid option to register" if has_converter 
203     converter = conversion_procs[soid]
204   end
205 
206   array_type = (opts[:array_type] || db_type).to_s.dup.freeze
207   creator = Creator.new(array_type, converter)
208   add_conversion_proc(oid, creator)
209 
210   typecast_method_map[db_type] = :"#{type}_array"
211 
212   singleton_class.class_eval do
213     meth = :"typecast_value_#{type}_array"
214     scalar_typecast_method = :"typecast_value_#{opts.fetch(:scalar_typecast, type)}"
215     define_method(meth){|v| typecast_value_pg_array(v, creator, scalar_typecast_method)}
216     private meth
217     alias_method(meth, meth)
218   end
219 
220   @schema_type_classes[:"#{type}_array"] = PGArray
221   nil
222 end