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.



206
207
208
209
 
# File 'lib/redis_object/collection.rb', line 206

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

Class Method Details

.class_const_for(name) ⇒ Object 



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

def class_const_for(name)
	Object.const_get(name.to_s.classify.to_sym) rescue RedisObject
end

.load(name, owner) ⇒ Object 



580
581
582
583
584
 
# File 'lib/redis_object/collection.rb', line 580

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 



556
557
558
559
560
 
# File 'lib/redis_object/collection.rb', line 556

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

#[](k) ⇒ Object 



272
273
274
 
# File 'lib/redis_object/collection.rb', line 272

def [](k)
	find k
end

#class_const ⇒ Object 



566
567
568
 
# File 'lib/redis_object/collection.rb', line 566

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

#cleanup! ⇒ Object 



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

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 



552
553
554
 
# File 'lib/redis_object/collection.rb', line 552

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

#convert_regex_to_lua(reg) ⇒ Object 



471
472
473
 
# File 'lib/redis_object/collection.rb', line 471

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

#delete(obj) ⇒ Object 



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

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

#each ⇒ Object 



501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
 
# File 'lib/redis_object/collection.rb', line 501

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 



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

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 



493
494
495
 
# File 'lib/redis_object/collection.rb', line 493

def first
	class_const.find_by_key(super)
end

#index_key(idx) ⇒ Object 



253
254
255
 
# File 'lib/redis_object/collection.rb', line 253

def index_key(idx)
	class_const.index_key(idx)
end

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



219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
 
# File 'lib/redis_object/collection.rb', line 219

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 



458
459
460
461
462
463
464
465
466
467
468
469
 
# File 'lib/redis_object/collection.rb', line 458

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 



257
258
259
 
# File 'lib/redis_object/collection.rb', line 257

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

#key ⇒ Object 



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

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

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



244
245
246
247
248
249
250
251
 
# File 'lib/redis_object/collection.rb', line 244

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

#last ⇒ Object 



497
498
499
 
# File 'lib/redis_object/collection.rb', line 497

def last
	class_const.find_by_key(super)
end

#latest ⇒ Object 



215
216
217
 
# File 'lib/redis_object/collection.rb', line 215

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

#map(&block) ⇒ Object 



531
532
533
 
# File 'lib/redis_object/collection.rb', line 531

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

#match(pkt, use_or = false) ⇒ Object 



429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
 
# File 'lib/redis_object/collection.rb', line 429

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 



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

def objects
	each.to_a
end

#push(obj) ⇒ Object 



562
563
564
 
# File 'lib/redis_object/collection.rb', line 562

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



485
486
487
 
# File 'lib/redis_object/collection.rb', line 485

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

#remove! ⇒ Object 



211
212
213
 
# File 'lib/redis_object/collection.rb', line 211

def remove!
	@owner.remove_collection! @name
end

#select(&block) ⇒ Object 



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

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

#store ⇒ Object 



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

def store
	class_const.store
end

#temp_key ⇒ Object 



237
238
239
 
# File 'lib/redis_object/collection.rb', line 237

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