Class: Seabright::Collection

Inherits:
Array
  • Object
show all
Includes:
CachedScripts
Defined in:
lib/redis_object/collection.rb

Constant Summary collapse

NilPattern = 
'nilpattern:'

Class Method Summary collapse

Instance Method Summary collapse

Methods included from CachedScripts

included, #run_script

Constructor Details

#initialize(name, owner) ⇒ Collection

Returns a new instance of Collection.



217
218
219
220
 
# File 'lib/redis_object/collection.rb', line 217

def initialize(name,owner)
	@name = name.to_s
	@owner = owner
end

Class Method Details

.class_const_for(name) ⇒ Object 



603
604
605
606
607
608
 
# File 'lib/redis_object/collection.rb', line 603

def class_const_for(name)
	if cls = RedisObject.constant_lookups[name.to_s.classify.to_sym]
		return cls
	end
	Object.const_get(name.to_s.classify.to_sym) rescue RedisObject
end

.load(name, owner) ⇒ Object 



597
598
599
600
601
 
# File 'lib/redis_object/collection.rb', line 597

def load(name,owner)
	out = new(name,owner)
	out.replace class_const_for(name).store.zrange(out.key,0,-1)
	out
end

Instance Method Details

#<<(obj) ⇒ Object 



573
574
575
576
577
 
# File 'lib/redis_object/collection.rb', line 573

def <<(obj)
	k = obj.class == String ? obj : obj.hkey
	store.zadd(key,store.zcount(key,"-inf", "+inf"),k)
	super(k)
end

#[](k) ⇒ Object 



289
290
291
 
# File 'lib/redis_object/collection.rb', line 289

def [](k)
	find k
end

#class_const ⇒ Object 



583
584
585
 
# File 'lib/redis_object/collection.rb', line 583

def class_const
	self.class.class_const_for(@name)
end

#cleanup! ⇒ Object 



535
536
537
538
539
540
541
542
543
544
545
546
 
# File 'lib/redis_object/collection.rb', line 535

def cleanup!
	each_index do |key|
		unless a = class_const.find_by_key(at(key))
			Log.debug "Deleting #{key} because not #{a.inspect}"
			delete at(key)
		end
	end
	if size < 1
		Log.debug "Deleting collection #{@name} because empty"
		remove!
	end
end

#clear! ⇒ Object 



569
570
571
 
# File 'lib/redis_object/collection.rb', line 569

def clear!
	store.zrem(key,self.join(" "))
end

#convert_regex_to_lua(reg) ⇒ Object 



488
489
490
 
# File 'lib/redis_object/collection.rb', line 488

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

#delete(obj) ⇒ Object 



563
564
565
566
567
 
# File 'lib/redis_object/collection.rb', line 563

def delete(obj)
	k = obj.class == String ? obj : obj.hkey
	store.zrem(key,k)
	super(k)
end

#each ⇒ Object 



518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
 
# File 'lib/redis_object/collection.rb', line 518

def each
	out = Enumerator.new do |y|
		each_index do |key|
			if a = class_const.find_by_key(at(key))
				y << a
			end
		end
	end
	if block_given?
		out.each do |a|
			yield a
		end
	else
		out
	end
end

#find(k) ⇒ Object 



278
279
280
281
282
283
284
285
286
287
 
# File 'lib/redis_object/collection.rb', line 278

def find(k)
	if k.is_a? String
		return real_at(item_key(k))
	elsif k.is_a? Hash
		return match(k)
	elsif k.is_a? Integer
		return real_at(at(k))
	end
	return nil
end

#first ⇒ Object 



510
511
512
 
# File 'lib/redis_object/collection.rb', line 510

def first
	class_const.find_by_key(super)
end

#indexed(idx, num = -1,, reverse = false) ⇒ Object 



230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
 
# File 'lib/redis_object/collection.rb', line 230

def indexed(idx,num=-1,reverse=false)
	keys = keys_by_index(idx,num,reverse)
	out = ListEnumerator.new(keys) do |y|
		keys.each do |member|
			if a = class_const.find_by_key(member)
				y << a
			end
		end
	end
	if block_given?
		out.each do |itm|
			yield itm
		end
	else
		out
	end
end

#inject_key(key, list) ⇒ Object 



475
476
477
478
479
480
481
482
483
484
485
486
 
# File 'lib/redis_object/collection.rb', line 475

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

#item_key(k) ⇒ Object 



274
275
276
 
# File 'lib/redis_object/collection.rb', line 274

def item_key(k)
	"#{class_const}:#{k}_h"
end

#key ⇒ Object 



591
592
593
 
# File 'lib/redis_object/collection.rb', line 591

def key
	"#{@owner ? "#{@owner.key}:" : ""}COLLECTION:#{@name}"
end

#keys_by_index(idx, num = -1,, reverse = false) ⇒ Object 



261
262
263
264
265
266
267
268
 
# File 'lib/redis_object/collection.rb', line 261

def keys_by_index(idx,num=-1,reverse=false)
	keys = run_script(reverse ? :RevScript : :FwdScript, [temp_key, sort_index_key(idx), key, num>0 ? num - 1 : -1])
	ListEnumerator.new(keys.uniq) do |y|
		keys.each do |member|
			y << member
		end
	end
end

#last ⇒ Object 



514
515
516
 
# File 'lib/redis_object/collection.rb', line 514

def last
	class_const.find_by_key(super)
end

#latest ⇒ Object 



226
227
228
 
# File 'lib/redis_object/collection.rb', line 226

def latest
	indexed(:created_at,5,true).first || first
end

#map(&block) ⇒ Object 



548
549
550
 
# File 'lib/redis_object/collection.rb', line 548

def map(&block)
	each.map(&block)
end

#match(pkt, use_or = false) ⇒ Object 



446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
 
# File 'lib/redis_object/collection.rb', line 446

def match(pkt, use_or=false)
	if use_or
		mtchr = :ColOrMatcher
	else
		mtchr = pkt.keys.count > 1 ? :ColMultiMatcher : :ColMatcher
	end
	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,self,pkt.flatten)
	ListEnumerator.new(kys) do |y|
		kys.each do |k|
			y << class_const.find_by_key(k)
		end
	end
end

#objects ⇒ Object 



506
507
508
 
# File 'lib/redis_object/collection.rb', line 506

def objects
	each.to_a
end

#push(obj) ⇒ Object 



579
580
581
 
# File 'lib/redis_object/collection.rb', line 579

def push(obj)
	self << obj
end

#real_at(key) ⇒ Object

def match(pkt) Enumerator.new do |y| each do |i| if pkt.all? {|hk,va| i.get(hk)==va } y << i end end end end



502
503
504
 
# File 'lib/redis_object/collection.rb', line 502

def real_at(key)
	class_const.find_by_key(key)
end

#remove! ⇒ Object 



222
223
224
 
# File 'lib/redis_object/collection.rb', line 222

def remove!
	@owner.remove_collection! @name
end

#select(&block) ⇒ Object 



552
553
554
555
556
557
558
559
560
561
 
# File 'lib/redis_object/collection.rb', line 552

def select(&block)
	return nil unless block_given?
	Enumerator.new do |y|
		each_index do |key|
			if (a = class_const.find_by_key(at(key))) && block.call(a)
				y << a
			end
		end
	end
end

#sort_index_key(idx) ⇒ Object 



270
271
272
 
# File 'lib/redis_object/collection.rb', line 270

def sort_index_key(idx)
	class_const.sort_index_key(idx)
end

#store ⇒ Object 



587
588
589
 
# File 'lib/redis_object/collection.rb', line 587

def store
	class_const.store
end

#temp_key ⇒ Object 



248
249
250
 
# File 'lib/redis_object/collection.rb', line 248

def temp_key
	"#{key}::zintersect_temp::#{RedisObject.new_id(4)}"
end