module Sequel::Postgres::InetDatabaseMethods
Methods enabling Database
object integration with the inet/cidr types.
Public Class Methods
extended(db)
click to toggle source
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 40 def self.extended(db) 41 db.instance_exec do 42 extend_datasets(InetDatasetMethods) 43 44 # :nocov: 45 if !defined?(SEQUEL_PG_VERSION_INTEGER) || SEQUEL_PG_VERSION_INTEGER >= 11300 46 # :nocov: 47 # sequel_pg 1.13.0+ will use inet/cidr conversion procs, but doing so is 48 # slower, so don't add the conversion procs if using sequel_pg 1.13.0+. 49 meth = IPAddr.method(:new) 50 add_conversion_proc(869, meth) 51 add_conversion_proc(650, meth) 52 if respond_to?(:register_array_type) 53 register_array_type('inet', :oid=>1041, :scalar_oid=>869) 54 register_array_type('cidr', :oid=>651, :scalar_oid=>650) 55 end 56 end 57 58 if respond_to?(:register_array_type) 59 register_array_type('macaddr', :oid=>1040, :scalar_oid=>829) 60 end 61 @schema_type_classes[:ipaddr] = IPAddr 62 end 63 end
Public Instance Methods
bound_variable_arg(arg, conn)
click to toggle source
Convert an IPAddr arg to a string. Probably not necessary, but done for safety.
Calls superclass method
# File lib/sequel/extensions/pg_inet.rb 67 def bound_variable_arg(arg, conn) 68 case arg 69 when IPAddr 70 "#{arg.to_s}/#{arg.instance_variable_get(:@mask_addr).to_s(2).count('1')}" 71 else 72 super 73 end 74 end
Private Instance Methods
bound_variable_array(a)
click to toggle source
Handle inet[]/cidr types in bound variables.
Calls superclass method
# File lib/sequel/extensions/pg_inet.rb 79 def bound_variable_array(a) 80 case a 81 when IPAddr 82 "\"#{a.to_s}/#{a.instance_variable_get(:@mask_addr).to_s(2).count('1')}\"" 83 else 84 super 85 end 86 end
schema_column_type(db_type)
click to toggle source
Make the column type detection recognize the inet and cidr types.
Calls superclass method
# File lib/sequel/extensions/pg_inet.rb 89 def schema_column_type(db_type) 90 case db_type 91 when 'inet', 'cidr' 92 :ipaddr 93 else 94 super 95 end 96 end
schema_post_process(_)
click to toggle source
Set the :ruby_default value if the default value is recognized as an ip address.
Calls superclass method
# File lib/sequel/extensions/pg_inet.rb 99 def schema_post_process(_) 100 super.each do |a| 101 h = a[1] 102 if h[:type] == :ipaddr && h[:default] =~ /\A'([:a-fA-F0-9\.\/]+)'::(?:inet|cidr)\z/ 103 h[:ruby_default] = IPAddr.new($1) 104 end 105 end 106 end
typecast_value_ipaddr(value)
click to toggle source
Typecast the given value to an IPAddr object.
# File lib/sequel/extensions/pg_inet.rb 109 def typecast_value_ipaddr(value) 110 case value 111 when IPAddr 112 value 113 when String 114 IPAddr.new(value) 115 else 116 raise Sequel::InvalidValue, "invalid value for inet/cidr: #{value.inspect}" 117 end 118 end