Class: Disbatch::Node

Inherits:
Object
  • Object
show all
Defined in:
lib/disbatch/node.rb

Overview

Represesents a Disbatch node

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(id) ⇒ Node

Create a node object

Parameters:

  • id (String) 


11
12
13
14
 
# File 'lib/disbatch/node.rb', line 11

def initialize(id)
	@registered = false
	@id   = id
end

Instance Attribute Details

#id ⇒ Object

Returns the value of attribute id.



4
5
6
 
# File 'lib/disbatch/node.rb', line 4

def id
  @id
end

Class Method Details

.create(id = Disbatch.node_id) ⇒ Object

Create a new node

Parameters:

  • id (String) (defaults to: Disbatch.node_id) 


36
37
38
39
40
41
42
43
44
45
 
# File 'lib/disbatch/node.rb', line 36

def self.create(id = Disbatch.node_id)
	Mongo.try do
		Disbatch.db[:nodes].insert({
			:_id => id,
			:ctime => Time.now
		})
	end

	new(id)
end

.get(id = Disbatch.node_id) ⇒ Object

Get an existing node

Parameters:

  • id (String) (defaults to: Disbatch.node_id) 


19
20
21
22
23
24
 
# File 'lib/disbatch/node.rb', line 19

def self.get(id = Disbatch.node_id)
	Mongo.try do
		raise Disbatch::NoNodeError unless Disbatch.db[:nodes].find_one({:_id => id})
		new(id)
	end
end

.get_all ⇒ Object

Get all existing nodes



27
28
29
30
31
 
# File 'lib/disbatch/node.rb', line 27

def self.get_all
	Mongo.try do
		Disbatch.db[:nodes].find.map { |doc| new(doc['_id']) }
	end
end

Instance Method Details

#==(node) ⇒ Object

Check equality with another node object

Parameters:



109
110
111
 
# File 'lib/disbatch/node.rb', line 109

def ==(node)
	@id == node.id
end

#register(force = false) ⇒ Object

Register node

Raises:



48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
 
# File 'lib/disbatch/node.rb', line 48

def register(force = false)
	raise Disbatch::AlreadyRegisteredNodeError if @registered

	begin
		doc = Mongo.try do
			Disbatch.db[:nodes].find_and_modify({
				:query  => force ? { :_id => @id } : { :_id => @id, :pid => nil },
				:update => { :$set => {
					:pid          => Process.pid,
					:version      => Disbatch::VERSION,
					:spec_version => Disbatch::SPEC_VERSION
				} }
			})
		end
	rescue
		doc = Mongo.try do
			Disbatch.db[:nodes].find({ :_id => @id })
		end

		if doc.count == 0
			raise Disbatch::NoNodeError
		else
			raise Disbatch::RegisteredNodeError
		end
	end

	@registered = true
	self
end

#registered? ⇒ Boolean Also known as: open?

Check if process is registered as this node.

Returns:

  • (Boolean) 


102
103
104
 
# File 'lib/disbatch/node.rb', line 102

def registered?
	@registered
end

#release ⇒ Object

Release ownership of node

Raises:



79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
 
# File 'lib/disbatch/node.rb', line 79

def release
	raise Disbatch::NotRegisteredNodeError unless @registered

	doc = Mongo.try do
		Disbatch.db[:nodes].find_and_modify({
			:query  => { :_id => @id, :pid => Process.pid },
			:update => { :$set => {
				:pid   => nil,
				:mtime => Time.now,
				:atime => Time.now
			} }
		})
	end

	if doc.nil? 
		false
	else
		@registered = false
		true
	end
end