Methods enabling Database object integration with the inet/cidr types.
Reset the conversion procs when extending the Database object, so it will pick up the inet/cidr converter. Also, extend the datasets with support for literalizing the IPAddr types.
# File lib/sequel/extensions/pg_inet.rb, line 42 def self.extended(db) db.instance_eval do extend_datasets(InetDatasetMethods) copy_conversion_procs([869, 650, 1041, 651, 1040]) @schema_type_classes[:ipaddr] = IPAddr end end
Convert an IPAddr arg to a string. Probably not necessary, but done for safety.
# File lib/sequel/extensions/pg_inet.rb, line 52 def bound_variable_arg(arg, conn) case arg when IPAddr "#{arg.to_s}/#{arg.instance_variable_get(:@mask_addr).to_s(2).count('1')}" else super end end
Handle inet[]/cidr types in bound variables.
# File lib/sequel/extensions/pg_inet.rb, line 64 def bound_variable_array(a) case a when IPAddr "\"#{a.to_s}/#{a.instance_variable_get(:@mask_addr).to_s(2).count('1')}\"" else super end end
Make the column type detection recognize the inet and cidr types.
# File lib/sequel/extensions/pg_inet.rb, line 74 def schema_column_type(db_type) case db_type when 'inet', 'cidr' :ipaddr else super end end
Typecast the given value to an IPAddr object.
# File lib/sequel/extensions/pg_inet.rb, line 84 def typecast_value_ipaddr(value) case value when IPAddr value when String IPAddr.new(value) else raise Sequel::InvalidValue, "invalid value for inet/cidr: #{value.inspect}" end end