Module: Sequel::Plugins::TstzrangeFields

Defined in:
lib/sequel/plugins/tstzrange_fields.rb

Defined Under Namespace

Modules: ClassMethods, DatasetMethods

Constant Summary collapse

VERSION = 
SequelTstzrangeFields::VERSION

Class Method Summary collapse

Class Method Details

.apply(_model, _opts = {}) ⇒ Object 



38
39
40
 
# File 'lib/sequel/plugins/tstzrange_fields.rb', line 38

def self.apply(_model, _opts={})
  Sequel.extension :pg_range_ops
end

.configure(model, *args) ⇒ Object 



42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
 
# File 'lib/sequel/plugins/tstzrange_fields.rb', line 42

def self.configure(model, *args)
  unless model.db.schema_type_class(:tstzrange)
    msg = "tstzrange_fields plugin requires pg_range db extension to be installed. " \
          "Use db.extension(:pg_range) after the db = Sequel.connect call."
    raise msg
  end
  args << :period if args.empty?
  args = args.flatten

  setup_model(model)
  model.tstzrange_columns = args.flatten
  model.tstzrange_columns.each do |column|
    create_accessors(model, column)
  end
end

.create_accessors(model, column) ⇒ Object 



75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
 
# File 'lib/sequel/plugins/tstzrange_fields.rb', line 75

def self.create_accessors(model, column)
  get_column_method = column.to_sym
  set_column_method = :"#{column}="
  get_begin_method = :"#{column}_begin"
  set_begin_method = :"#{column}_begin="
  get_end_method = :"#{column}_end"
  set_end_method = :"#{column}_end="

  model.define_method(get_column_method) do
    self[column]
  end

  model.define_method(set_column_method) do |value|
    case value
      when Sequel::Postgres::PGRange
        self[column] = value
      when Float::INFINITY
        range = Sequel::Postgres::PGRange.new(nil, nil, empty: false, db_type: :tstzrange)
        self[column] = range
      when "empty", nil
        self[column] = Sequel::Postgres::PGRange.empty(:tstzrange)
    else
        beg = value.respond_to?(:begin) ? value.begin : (value[:begin] || value["begin"])
        en = value.respond_to?(:end) ? value.end : (value[:end] || value["end"])
        self[column] = self.class.new_tstzrange(beg, en)
    end
  end

  model.define_method(get_begin_method) do
    r = send(get_column_method)
    return r&.begin
  end

  model.define_method(set_begin_method) do |new_time|
    new_range = self.class.new_tstzrange(new_time, send(get_end_method))
    send(set_column_method, new_range)
  end

  model.define_method(get_end_method) do
    r = send(get_column_method)
    return r&.end
  end

  model.define_method(set_end_method) do |new_time|
    new_range = self.class.new_tstzrange(send(get_begin_method), new_time)
    send(set_column_method, new_range)
  end
end

.setup_model(model) ⇒ Object 



58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
 
# File 'lib/sequel/plugins/tstzrange_fields.rb', line 58

def self.setup_model(model)
  model.class.define_method(:new_tstzrange) do |b, e|
    b = value_to_time(b)
    e = value_to_time(e)
    return Sequel::Postgres::PGRange.empty(:tstzrange) if b.nil? && e.nil?

    return Sequel::Postgres::PGRange.new(b&.to_time, e&.to_time, db_type: :tstzrange, exclude_end: true)
  end

  model.class.define_method(:value_to_time) do |v|
    return v if v.nil?
    return v if v.respond_to?(:to_time)

    return Time.parse(v)
  end
end