Module: Seabright::Matchers::ClassMethods

Defined in:
lib/redis_object/matchers.rb

Constant Summary collapse

NilPattern = 
'nilpattern:'
GetKeyList = 
"local itms
if KEYS[2] then
  itms = {}
  for n=2,#KEYS do
    local kys = redis.call('SMEMBERS',KEYS[n])
    for k=1,#kys do
      local rk
      if kys[k]:match('.*_h') then
        rk = kys[k]
      else
        rk = kys[k]..'_h'
      end
      table.insert(itms, rk)
    end
  end
else
  itms = {}
  local kys = redis.call('SMEMBERS',KEYS[1])
  for k=1,#kys do
    local rk
    if kys[k]:match('.*_h') then
      rk = kys[k]
    else
      rk = kys[k]..'_h'
    end
    table.insert(itms, rk)
  end
end".gsub(/\t/,'').freeze

Instance Method Summary collapse

Instance Method Details

#convert_regex_to_lua(reg) ⇒ Object 



450
451
452
 
# File 'lib/redis_object/matchers.rb', line 450

def convert_regex_to_lua(reg)
  "#{reg.casefold? ? "i" : ""}pattern:#{reg.source.gsub("\\","")}"
end

#extract_usable_indices(pkt) ⇒ Object 



373
374
375
376
377
378
379
380
 
# File 'lib/redis_object/matchers.rb', line 373

def extract_usable_indices(pkt)
  pkt.inject([]) do |acc,(k,v)|
    if self.has_index?(k)
      acc << index_key(k,v)
    end
    acc
  end
end

#find(ident) ⇒ Object 



472
473
474
 
# File 'lib/redis_object/matchers.rb', line 472

def find(ident)
  grab(ident)
end

#find_first(ident) ⇒ Object 



476
477
478
 
# File 'lib/redis_object/matchers.rb', line 476

def find_first(ident)
  match_first(ident)
end

#grab(ident) ⇒ Object 



454
455
456
457
458
459
460
461
462
 
# File 'lib/redis_object/matchers.rb', line 454

def grab(ident)
  case ident
  when String, Symbol
    return store.exists(self.hkey(ident.to_s)) ? self.new(ident.to_s) : nil
  when Hash
    return match(ident)
  end
  nil
end

#inject_key(key, list) ⇒ Object 



437
438
439
440
441
442
443
444
445
446
447
448
 
# File 'lib/redis_object/matchers.rb', line 437

def inject_key(key,list)
  out = []
  list.each do |i|
    if i == list.first
      out << i
    else
      out << key
      out << i
    end
  end
  out
end

#match(pkt, use_or = false) ⇒ Object 



192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
 
# File 'lib/redis_object/matchers.rb', line 192

def match(pkt, use_or=false)
  if use_or
    mtchr = :OrMatcher
  else
    mtchr = pkt.keys.count > 1 ? :MultiMatcher : :Matcher
  end
  if (ids = pkt[id_sym] || pkt[id_sym.to_s]) and !ids.is_a?(Regexp)
    return match_by_id(ids, pkt, use_or)
  end
  indcs = [plname] + extract_usable_indices(pkt)
  pkt = pkt.flatten.reduce([]) do |i,v|
    x = case v
    when Regexp
      convert_regex_to_lua(v)
    when Array
      raise ArgumentError.new("An array can only be used with the find_or method") unless use_or
      inject_key(i.last, v)
    when NilClass
      NilPattern
    else
      v.to_s
    end
    i << x
    i
  end
  kys = run_script(mtchr, indcs, pkt.flatten)
  ListEnumerator.new(kys) do |y|
    kys.each do |k|
      y << find(k)
    end
  end
end

#match_by_id(ids, pkt, use_or = false) ⇒ Object 



382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
 
# File 'lib/redis_object/matchers.rb', line 382

def match_by_id(ids, pkt, use_or = false)
  case ids
  when Array
    raise ArgumentError.new("An array can only be used with the or_find_first method") unless use_or
    ids.map do |i|
      match_by_id(i, pkt, use_or)
    end.flatten
  when String, Symbol
    if obj = find(ids)
      pkt.each do |k,v|
        case v
        when Regexp
          return nil unless v.match(obj.get(k))
        when Array
          raise ArgumentError.new("An array can only be used with the or_find_first method") unless use_or
          return nil unless v.any? { |val| obj.get(k) == v }
        when String, Symbol
          return nil unless obj.get(k) == v
        when NilClass
          return nil unless obj.get(k) == nil
        end
      end
    end
    [obj]
  end
end

#match_first(pkt, use_or = false) ⇒ Object 



409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
 
# File 'lib/redis_object/matchers.rb', line 409

def match_first(pkt, use_or=false)
  if use_or
    mtchr = :FirstOrMatcher
  else
    mtchr = pkt.keys.count > 1 ? :FirstMultiMatcher : :FirstMatcher
  end
  if (ids = pkt[id_sym] || pkt[id_sym.to_s]) and !ids.is_a?(Regexp)
    return match_by_id(ids, pkt, use_or).first
  end
  indcs = [plname] + extract_usable_indices(pkt)
  pkt = pkt.flatten.reduce([]) do |i,v|
    x = case v
    when Regexp
      convert_regex_to_lua(v)
    when Array
      raise ArgumentError.new("An array can only be used with the or_find_first method") unless use_or
      inject_key(i.last, v)
    when NilClass
      NilPattern
    else
      v.to_s
    end
    i << x
    i
  end
  find_by_key(run_script(mtchr,indcs, pkt.flatten))
end

#or_find(ident) ⇒ Object 



480
481
482
 
# File 'lib/redis_object/matchers.rb', line 480

def or_find(ident)
  or_grab(ident)
end

#or_find_first(ident) ⇒ Object 



484
485
486
 
# File 'lib/redis_object/matchers.rb', line 484

def or_find_first(ident)
  match_first(ident)
end

#or_grab(ident) ⇒ Object 



464
465
466
467
468
469
470
 
# File 'lib/redis_object/matchers.rb', line 464

def or_grab(ident)
  case ident
  when Hash
    return match(ident, true)
  end
  nil
end