Class: Sequel::Postgres::PGRange

Inherits:
Object
  • Object
show all
Includes:
Enumerable, SQL::AliasMethods
Defined in:
lib/sequel/extensions/pg_range.rb,
lib/sequel/extensions/pg_range_ops.rb

Defined Under Namespace

Modules: DatabaseMethods, DatasetMethods Classes: Parser

Constant Summary collapse

RANGE_TYPES =

Map of string database type names to type symbols (e.g. ‘int4range’ => :int4range), used in the schema parsing.

{}
EMPTY = 
'empty'.freeze
EMPTY_STRING = 
''.freeze
COMMA = 
','.freeze
QUOTED_EMPTY_STRING = 
'""'.freeze
OPEN_PAREN = 
"(".freeze
CLOSE_PAREN = 
")".freeze
OPEN_BRACKET = 
"[".freeze
CLOSE_BRACKET = 
"]".freeze
ESCAPE_RE = 
/("|,|\\|\[|\]|\(|\))/.freeze
ESCAPE_REPLACE = 
'\\\\\1'.freeze
CAST = 
'::'.freeze

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Methods included from SQL::AliasMethods

#as

Constructor Details

#initialize(beg, en, opts = OPTS) ⇒ PGRange

Initialize a new PGRange instance. Accepts the following options:

:db_type

The PostgreSQL database type for the range.

:empty

Whether the range is empty (has no points)

:exclude_begin

Whether the beginning element is excluded from the range.

:exclude_end

Whether the ending element is excluded from the range.



394
395
396
397
398
399
400
401
402
403
404
 
# File 'lib/sequel/extensions/pg_range.rb', line 394

def initialize(beg, en, opts=OPTS)
  @begin = beg
  @end = en
  @empty = !!opts[:empty]
  @exclude_begin = !!opts[:exclude_begin]
  @exclude_end = !!opts[:exclude_end]
  @db_type = opts[:db_type]
  if @empty
    raise(Error, 'cannot have an empty range with either a beginning or ending') unless @begin.nil? && @end.nil? && opts[:exclude_begin].nil? && opts[:exclude_end].nil?
  end
end

Instance Attribute Details

#begin ⇒ Object (readonly)

The beginning of the range. If nil, the range has an unbounded beginning.



369
370
371
 
# File 'lib/sequel/extensions/pg_range.rb', line 369

def begin
  @begin
end

#db_type ⇒ Object (readonly)

The PostgreSQL database type for the range (e.g. ‘int4range’).



375
376
377
 
# File 'lib/sequel/extensions/pg_range.rb', line 375

def db_type
  @db_type
end

#end ⇒ Object (readonly)

The end of the range. If nil, the range has an unbounded ending.



372
373
374
 
# File 'lib/sequel/extensions/pg_range.rb', line 372

def end
  @end
end

Class Method Details

.empty(db_type = nil) ⇒ Object

Create an empty PGRange with the given database type.



384
385
386
 
# File 'lib/sequel/extensions/pg_range.rb', line 384

def self.empty(db_type=nil)
  new(nil, nil, :empty=>true, :db_type=>db_type)
end

.from_range(range, db_type = nil) ⇒ Object

Create a new PGRange instance using the beginning and ending of the ruby Range, with the given db_type.



379
380
381
 
# File 'lib/sequel/extensions/pg_range.rb', line 379

def self.from_range(range, db_type=nil)
  new(range.begin, range.end, :exclude_end=>range.exclude_end?, :db_type=>db_type)
end

.register(db_type, opts = OPTS, &block) ⇒ Object

Registers a range type that the extension should handle. Makes a Database instance that has been extended with DatabaseMethods recognize the range type given and set up the appropriate typecasting. Also sets up automatic typecasting for the native postgres adapter, so that on retrieval, the values are automatically converted to PGRange instances. The db_type argument should be the name of the range type. Accepts the following options:

:converter

A callable object (e.g. Proc), that is called with the start or end of the range (usually a string), and should return the appropriate typecasted object.

:oid

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

:subtype_oid

Should be the PostgreSQL OID for the range’s subtype. If given, automatically sets the :converter option by looking for scalar conversion proc.

:type_procs

A hash mapping oids to conversion procs, used for setting the default :converter for :subtype_oid. Defaults to the global Sequel::Postgres::PG_TYPES.

:typecast_method_map

The map in which to place the database type string to type symbol mapping. Defaults to RANGE_TYPES.

:typecast_methods_module

If given, a module object to add the typecasting method to. Defaults to DatabaseMethods.

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



133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
 
# File 'lib/sequel/extensions/pg_range.rb', line 133

def self.register(db_type, opts=OPTS, &block)
  db_type = db_type.to_s.dup.freeze

  type_procs = opts[:type_procs] || PG_TYPES
  mod = opts[:typecast_methods_module] || DatabaseMethods
  typecast_method_map = opts[:typecast_method_map] || RANGE_TYPES

  if converter = opts[:converter]
    raise Error, "can't provide both a block and :converter option to register" if block
  else
    converter = block
  end

  if soid = opts[:subtype_oid]
    raise Error, "can't provide both a converter and :subtype_oid option to register" if converter 
    raise Error, "no conversion proc for :subtype_oid=>#{soid.inspect} in PG_TYPES" unless converter = type_procs[soid]
  end

  parser = Parser.new(db_type, converter)

  typecast_method_map[db_type] = db_type.to_sym

  define_range_typecast_method(mod, db_type, parser)

  if oid = opts[:oid]
    type_procs[oid] = parser
  end

  nil
end

Instance Method Details

#===(other) ⇒ Object

Allow PGRange values in case statements, where they return true if they are equal to each other using eql?, or if this PGRange can be converted to a Range, delegating to that range.



455
456
457
458
459
460
461
462
463
464
465
 
# File 'lib/sequel/extensions/pg_range.rb', line 455

def ===(other)
  if eql?(other)
    true
  else
    if valid_ruby_range?
      to_range === other 
    else
      false
    end
  end
end

#cover?(value) ⇒ Boolean

Return whether the value is inside the range.

Returns:

  • (Boolean) 


413
414
415
416
417
418
419
420
 
# File 'lib/sequel/extensions/pg_range.rb', line 413

def cover?(value)
  return false if empty?
  b = self.begin
  return false if b && b.send(exclude_begin? ? :>= : :>, value)
  e = self.end
  return false if e && e.send(exclude_end? ? :<= : :<, value)
  true
end

#empty? ⇒ Boolean

Whether this range is empty (has no points). Note that for manually created ranges (ones not retrieved from the database), this will only be true if the range was created using the :empty option.

Returns:

  • (Boolean) 


470
471
472
 
# File 'lib/sequel/extensions/pg_range.rb', line 470

def empty?
  @empty
end

#eql?(other) ⇒ Boolean Also known as: ==

Consider the receiver equal to other PGRange instances with the same beginning, ending, exclusions, and database type. Also consider it equal to Range instances if this PGRange can be converted to a a Range and those ranges are equal.

Returns:

  • (Boolean) 


426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
 
# File 'lib/sequel/extensions/pg_range.rb', line 426

def eql?(other)
  case other
  when PGRange
    if db_type == other.db_type
      if empty?
        other.empty?
      elsif other.empty?
        false
      else
        [:@begin, :@end, :@exclude_begin, :@exclude_end].all?{|v| instance_variable_get(v) == other.instance_variable_get(v)}
      end
    else
      false
    end
  when Range
    if valid_ruby_range?
      to_range.eql?(other)
    else
      false
    end
  else
    false
  end
end

#exclude_begin? ⇒ Boolean

Whether the beginning element is excluded from the range.

Returns:

  • (Boolean) 


475
476
477
 
# File 'lib/sequel/extensions/pg_range.rb', line 475

def exclude_begin?
  @exclude_begin
end

#exclude_end? ⇒ Boolean

Whether the ending element is excluded from the range.

Returns:

  • (Boolean) 


480
481
482
 
# File 'lib/sequel/extensions/pg_range.rb', line 480

def exclude_end?
  @exclude_end
end

#op ⇒ Object

Wrap the PGRange instance in an RangeOp, allowing you to easily use the PostgreSQL range functions and operators with literal ranges.



124
125
126
 
# File 'lib/sequel/extensions/pg_range_ops.rb', line 124

def op
  RangeOp.new(self)
end

#sql_literal_append(ds, sql) ⇒ Object

Append a literalize version of the receiver to the sql.



485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
 
# File 'lib/sequel/extensions/pg_range.rb', line 485

def sql_literal_append(ds, sql)
  if (s = @db_type) && !empty?
    sql << s.to_s << OPEN_PAREN
    ds.literal_append(sql, self.begin)
    sql << COMMA
    ds.literal_append(sql, self.end)
    sql << COMMA
    ds.literal_append(sql, "#{exclude_begin? ? OPEN_PAREN : OPEN_BRACKET}#{exclude_end? ? CLOSE_PAREN : CLOSE_BRACKET}")
    sql << CLOSE_PAREN
  else
    ds.literal_append(sql, unquoted_literal(ds))
    if s
      sql << CAST << s.to_s
    end
  end
end

#to_range ⇒ Object

Return a ruby Range object for this instance, if one can be created.

Raises:



503
504
505
506
507
508
509
510
 
# File 'lib/sequel/extensions/pg_range.rb', line 503

def to_range
  return @range if @range
  raise(Error, "cannot create ruby range for an empty PostgreSQL range") if empty?
  raise(Error, "cannot create ruby range when PostgreSQL range excludes beginning element") if exclude_begin?
  raise(Error, "cannot create ruby range when PostgreSQL range has unbounded beginning") unless self.begin
  raise(Error, "cannot create ruby range when PostgreSQL range has unbounded ending") unless self.end
  @range = Range.new(self.begin, self.end, exclude_end?)
end

#unbounded_begin? ⇒ Boolean

Whether the beginning of the range is unbounded.

Returns:

  • (Boolean) 


520
521
522
 
# File 'lib/sequel/extensions/pg_range.rb', line 520

def unbounded_begin?
  self.begin.nil? && !empty?
end

#unbounded_end? ⇒ Boolean

Whether the end of the range is unbounded.

Returns:

  • (Boolean) 


525
526
527
 
# File 'lib/sequel/extensions/pg_range.rb', line 525

def unbounded_end?
  self.end.nil? && !empty?
end

#unquoted_literal(ds) ⇒ Object

Return a string containing the unescaped version of the range. Separated out for use by the bound argument code.



531
532
533
534
535
536
537
 
# File 'lib/sequel/extensions/pg_range.rb', line 531

def unquoted_literal(ds)
  if empty?
    EMPTY
  else
    "#{exclude_begin? ? OPEN_PAREN : OPEN_BRACKET}#{escape_value(self.begin, ds)},#{escape_value(self.end, ds)}#{exclude_end? ? CLOSE_PAREN : CLOSE_BRACKET}"
  end
end

#valid_ruby_range? ⇒ Boolean

Whether or not this PGRange is a valid ruby range. In order to be a valid ruby range, it must have a beginning and an ending (no unbounded ranges), and it cannot exclude the beginning element.

Returns:

  • (Boolean) 


515
516
517
 
# File 'lib/sequel/extensions/pg_range.rb', line 515

def valid_ruby_range?
  !(empty? || exclude_begin? || !self.begin || !self.end)
end