lock-server 0.1.6

Sign up to get free protection for your applications and to get access to all the features.
data/.document ADDED
@@ -0,0 +1,5 @@
1
+ lib/**/*.rb
2
+ bin/*
3
+ -
4
+ features/**/*.feature
5
+ LICENSE.txt
data/Gemfile ADDED
@@ -0,0 +1,14 @@
1
+ source "http://rubygems.org"
2
+
3
+ gem "eventmachine", ">=0"
4
+
5
+
6
+ # Add dependencies to develop your gem here.
7
+ # Include everything needed to run rake, tests, features, etc.
8
+ group :development do
9
+ gem "shoulda", ">= 0"
10
+ gem "rdoc", "~> 3.12"
11
+ gem "bundler", "~> 1.0.0"
12
+ gem "jeweler", "~> 1.8.3"
13
+ gem "simplecov", ">= 0"
14
+ end
data/Gemfile.lock ADDED
@@ -0,0 +1,35 @@
1
+ GEM
2
+ remote: http://rubygems.org/
3
+ specs:
4
+ eventmachine (0.12.10)
5
+ git (1.2.5)
6
+ jeweler (1.8.3)
7
+ bundler (~> 1.0)
8
+ git (>= 1.2.5)
9
+ rake
10
+ rdoc
11
+ json (1.6.6)
12
+ multi_json (1.3.5)
13
+ rake (0.9.2.2)
14
+ rdoc (3.12)
15
+ json (~> 1.4)
16
+ shoulda (3.0.1)
17
+ shoulda-context (~> 1.0.0)
18
+ shoulda-matchers (~> 1.0.0)
19
+ shoulda-context (1.0.0)
20
+ shoulda-matchers (1.0.0)
21
+ simplecov (0.6.1)
22
+ multi_json (~> 1.0)
23
+ simplecov-html (~> 0.5.3)
24
+ simplecov-html (0.5.3)
25
+
26
+ PLATFORMS
27
+ ruby
28
+
29
+ DEPENDENCIES
30
+ bundler (~> 1.0.0)
31
+ eventmachine
32
+ jeweler (~> 1.8.3)
33
+ rdoc (~> 3.12)
34
+ shoulda
35
+ simplecov
data/LICENSE.txt ADDED
@@ -0,0 +1,20 @@
1
+ Copyright (c) 2012 V_M
2
+
3
+ Permission is hereby granted, free of charge, to any person obtaining
4
+ a copy of this software and associated documentation files (the
5
+ "Software"), to deal in the Software without restriction, including
6
+ without limitation the rights to use, copy, modify, merge, publish,
7
+ distribute, sublicense, and/or sell copies of the Software, and to
8
+ permit persons to whom the Software is furnished to do so, subject to
9
+ the following conditions:
10
+
11
+ The above copyright notice and this permission notice shall be
12
+ included in all copies or substantial portions of the Software.
13
+
14
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
15
+ EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
16
+ MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
17
+ NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
18
+ LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
19
+ OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
20
+ WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
data/README.rdoc ADDED
@@ -0,0 +1,19 @@
1
+ = lock-server
2
+
3
+ Description goes here.
4
+
5
+ == Contributing to lock-server
6
+
7
+ * Check out the latest master to make sure the feature hasn't been implemented or the bug hasn't been fixed yet.
8
+ * Check out the issue tracker to make sure someone already hasn't requested it and/or contributed it.
9
+ * Fork the project.
10
+ * Start a feature/bugfix branch.
11
+ * Commit and push until you are happy with your contribution.
12
+ * Make sure to add tests for it. This is important so I don't break it in a future version unintentionally.
13
+ * Please try not to mess with the Rakefile, version, or history. If you want to have your own version, or is otherwise necessary, that is fine, but please isolate to its own commit so I can cherry-pick around it.
14
+
15
+ == Copyright
16
+
17
+ Copyright (c) 2012 V_M. See LICENSE.txt for
18
+ further details.
19
+
data/Rakefile ADDED
@@ -0,0 +1,59 @@
1
+ # encoding: utf-8
2
+
3
+ require 'rubygems'
4
+ require 'bundler'
5
+ begin
6
+ Bundler.setup(:default, :development)
7
+ rescue Bundler::BundlerError => e
8
+ $stderr.puts e.message
9
+ $stderr.puts "Run `bundle install` to install missing gems"
10
+ exit e.status_code
11
+ end
12
+ require 'rake'
13
+
14
+ require 'jeweler'
15
+ Jeweler::Tasks.new do |gem|
16
+ # gem is a Gem::Specification... see http://docs.rubygems.org/read/chapter/20 for more options
17
+ gem.name = "lock-server"
18
+ gem.homepage = "http://github.com/blimp666/lock-server"
19
+ gem.license = "MIT"
20
+ gem.summary = %Q{Lock-server}
21
+ gem.description = %Q{General purpose lock-server}
22
+ gem.email = "nobody@nowhere.com"
23
+ gem.authors = ["V_M"]
24
+ # dependencies defined in Gemfile
25
+ gem.files.include 'lib/lock-server.rb'
26
+ gem.files.include 'lib/with_lock.rb'
27
+ gem.files.include 'lock-server-example.conf'
28
+ gem.executables << 'lock_server'
29
+ end
30
+
31
+ # Это чтобы на rubygems засылал
32
+ Jeweler::RubygemsDotOrgTasks.new
33
+
34
+ require 'rake/testtask'
35
+ Rake::TestTask.new(:test) do |test|
36
+ test.libs << 'lib' << 'test'
37
+ test.pattern = 'test/**/test_*.rb'
38
+ test.verbose = true
39
+ end
40
+
41
+ # require 'rcov/rcovtask'
42
+ # Rcov::RcovTask.new do |test|
43
+ # test.libs << 'test'
44
+ # test.pattern = 'test/**/test_*.rb'
45
+ # test.verbose = true
46
+ # test.rcov_opts << '--exclude "gems/*"'
47
+ # end
48
+
49
+ task :default => :test
50
+
51
+ require 'rdoc/task'
52
+ Rake::RDocTask.new do |rdoc|
53
+ version = File.exist?('VERSION') ? File.read('VERSION') : ""
54
+
55
+ rdoc.rdoc_dir = 'rdoc'
56
+ rdoc.title = "lock-server #{version}"
57
+ rdoc.rdoc_files.include('README*')
58
+ rdoc.rdoc_files.include('lib/**/*.rb')
59
+ end
data/VERSION ADDED
@@ -0,0 +1 @@
1
+ 0.1.6
data/bin/lock_server ADDED
@@ -0,0 +1,11 @@
1
+ #!/usr/bin/env ruby
2
+ require 'rubygems'
3
+ require 'daemons'
4
+ require 'lock-server'
5
+
6
+ file_name = ARGV[1] || './lock-server.conf'
7
+ config = eval(File.open("#{file_name}").read)
8
+
9
+ Daemons.run_proc(config[:name], config[:daemons_options]) do
10
+ LockServer.start_server(config[:server_socket])
11
+ end
@@ -0,0 +1,160 @@
1
+ require 'rubygems'
2
+ require 'eventmachine'
3
+ require 'socket'
4
+
5
+ module LockServer
6
+
7
+ CONNECTION_TIMEOUT = 600 # 10 minutes in seconds
8
+
9
+ # Simple resource locking protocol implementation, suitable for EventMachine.
10
+ module ResourceLockServer
11
+
12
+ # Returns a queue for specified resource
13
+ #
14
+ # ==== Parameters
15
+ # resource<String>::
16
+ # A name of resource which query should be returned
17
+ def queue_for(resource)
18
+ p @@resource_queues
19
+ @@resource_queues[resource] ||= []
20
+ end
21
+
22
+ # Returns a mutex for specified resource
23
+ #
24
+ # ==== Parameters
25
+ # resource<String>::
26
+ # A name of resource which mutex should be returned
27
+ def mutex_for(resource)
28
+ @@mutexes[resource] ||= Mutex.new
29
+ end
30
+
31
+ # Hash of resource queues. Keys are resource names, values are arrays of connections, trying to acquire resource.
32
+ @@resource_queues = { }
33
+ # Hash of mutexes used for syncronization of queue access. Keys are resource names, values are Mutex instances.
34
+ @@mutexes = { }
35
+
36
+ attr_reader :peer_ip
37
+ attr_reader :created_at, :updated_at
38
+
39
+ def post_init
40
+ @resource = nil
41
+ @peer_ip = Socket.unpack_sockaddr_in(self.get_peername)[1]
42
+ set_comm_inactivity_timeout(LockServer::CONNECTION_TIMEOUT)
43
+ end
44
+
45
+ def send_line(string)
46
+ send_data(string+"\n")
47
+ end
48
+
49
+ # A collback of EventMachine, being called when data is received. Used for processing of lock requests.
50
+ # To aquire lock for needed resource name application should send message 'lock resource_name', where resource_name is a
51
+ # name of needed resource.
52
+ # An answer will be either 'ok resource_name' (meaning lock have been obtained and application could proceed) or 'wait resource_name'
53
+ # (there's already lock on this resource, and application should wait).
54
+ # To unlock data, 'unlock' should be sent. This will remove current connection from a queue and a connection will be closed.
55
+ # Sending any other message will result in 'err: unknown message' and connection issued incorrect command will be closed.
56
+ #
57
+ # ==== Parameters
58
+ # data<String>::
59
+ # Data fetched from EventMachine
60
+ def receive_data(data)
61
+ @updated_at = Time.now
62
+ data.each_line do |line|
63
+ command, resource = line.strip.split(' ')
64
+ case command
65
+
66
+ when 'lock'
67
+ obtain_lock(resource)
68
+
69
+ when 'unlock'
70
+ send_line("released #{@resource}")
71
+ close_connection_after_writing
72
+
73
+ when "status"
74
+ write_server_status
75
+ close_connection_after_writing
76
+
77
+ else
78
+ send_line("err: unknown message")
79
+ close_connection_after_writing
80
+ end
81
+ end
82
+ end
83
+
84
+ def write_server_status
85
+ @@resource_queues.keys.each{ |k|
86
+ next if queue_for(k).empty?
87
+ send_line("Resource: #{k} (#{queue_for(k).size} connections waiting)")
88
+ send_line(queue_for(k).collect(&:peer_ip).join(", "))
89
+ }
90
+ end
91
+
92
+ # Creates a lock for requested resource, acquires mutex to protect resource queue manipulations and
93
+ # either notifies remote side about successful resource aquisition or tells it to wait.
94
+ # First connection in a queue is owning a resource, otherwise a connection will be put on hold for resource aquisition.
95
+ #
96
+ # ==== Parameters
97
+ # resource<String>::
98
+ # A name of resource to lock.
99
+ def obtain_lock(resource)
100
+ @resource = resource
101
+ queue = queue_for(@resource)
102
+ return if queue.include?(self)
103
+ mutex_for(@resource).synchronize do
104
+ queue << self
105
+ if queue.size == 1
106
+ allow_action
107
+ else
108
+ wait_for_lock_release
109
+ end
110
+ end
111
+ end
112
+
113
+ def interrupt_timed_out_connection
114
+ return unless Time.now-@updated_at
115
+ end
116
+
117
+ # Releases lock of previously specified resource.
118
+ # Calling this method will remove current connection from @resource aquisitors queue. Also, if current connection is owning a resource (is first in aquisitors queue),
119
+ # it will notify next connection that resource is free now.
120
+ #
121
+ # ==== Parameters
122
+ # none
123
+ def release_lock
124
+ queue = queue_for(@resource)
125
+ mutex_for(@resource).synchronize do
126
+ if queue.first == self
127
+ queue.shift
128
+ queue.first.allow_action unless queue.empty?
129
+ else
130
+ queue.delete self
131
+ end
132
+ end
133
+ end
134
+
135
+ # Sends a notification that requested resource is free now.
136
+ def allow_action
137
+ send_line("ok #{@resource}")
138
+ end
139
+
140
+ # Sends a notification that requested resource is locked and an application should wait for lock.
141
+ def wait_for_lock_release
142
+ send_line("wait #{@resource}")
143
+ end
144
+
145
+ def unbind
146
+ release_lock
147
+ end
148
+
149
+
150
+ end
151
+
152
+ def self.start_server(opts = { })
153
+ opts[:port] ||= 12312
154
+ opts[:host] ||= 'localhost'
155
+ EventMachine::run {
156
+ EventMachine::start_server opts[:host], opts[:port], ResourceLockServer
157
+ }
158
+ end
159
+ end
160
+
data/lib/with_lock.rb ADDED
@@ -0,0 +1,35 @@
1
+ require 'socket'
2
+
3
+ module WithLock
4
+
5
+ class LockTimeOut < Exception
6
+ end
7
+
8
+ # Request a lock at remote server, waits for aquisition, and yields a given block.
9
+ # Opens a connection to lock server and performs a communication with it, obtaining lock.
10
+ #
11
+ # ==== Parameters
12
+ # resource<Object>::
13
+ # A resource to aquire. Will be converted to string with to_str method and sent as a resource name to lock server.
14
+ def with_lock(resource, timeout_period = 20, &block)
15
+ # yield; return
16
+
17
+ resource = resource.to_s
18
+ if resource.empty?
19
+ yield; return;
20
+ end
21
+ sock = TCPSocket.new($lock_server_config[:host], $lock_server_config[:port])
22
+ sock.puts "lock #{resource}\n"
23
+ while 1 do
24
+ msg = sock.gets.to_s
25
+ break if msg.strip == "ok #{resource}"
26
+ sleep(0.1)
27
+ end
28
+ Timeout::timeout(timeout_period) { yield }
29
+ ensure
30
+ if sock
31
+ sock.puts("unlock #{resource}\n")
32
+ sock.close
33
+ end
34
+ end
35
+ end
@@ -0,0 +1,12 @@
1
+ {
2
+ name: "lock_server_blimp",
3
+ daemons_options: {
4
+ dir_mode: :normal,
5
+ dir: File.dirname(__FILE__)+'/log',
6
+ multiple: false,
7
+ backtrace: true,
8
+ monitor: true},
9
+ server_socket: {
10
+ host: '127.0.0.1',
11
+ port: 12370}
12
+ }
@@ -0,0 +1,71 @@
1
+ # Generated by jeweler
2
+ # DO NOT EDIT THIS FILE DIRECTLY
3
+ # Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
4
+ # -*- encoding: utf-8 -*-
5
+
6
+ Gem::Specification.new do |s|
7
+ s.name = "lock-server"
8
+ s.version = "0.1.6"
9
+
10
+ s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
11
+ s.authors = ["V_M"]
12
+ s.date = "2012-06-18"
13
+ s.description = "General purpose lock-server"
14
+ s.email = "nobody@nowhere.com"
15
+ s.executables = ["lock_server", "lock_server"]
16
+ s.extra_rdoc_files = [
17
+ "LICENSE.txt",
18
+ "README.rdoc"
19
+ ]
20
+ s.files = [
21
+ ".document",
22
+ "Gemfile",
23
+ "Gemfile.lock",
24
+ "LICENSE.txt",
25
+ "README.rdoc",
26
+ "Rakefile",
27
+ "VERSION",
28
+ "bin/lock_server",
29
+ "lib/lock-server.rb",
30
+ "lib/with_lock.rb",
31
+ "lock-server-example.conf",
32
+ "lock-server.gemspec",
33
+ "log/lock_server_blimp.pid",
34
+ "log/lock_server_blimp_monitor.pid",
35
+ "test/helper.rb",
36
+ "test/test_lock-server.rb"
37
+ ]
38
+ s.homepage = "http://github.com/blimp666/lock-server"
39
+ s.licenses = ["MIT"]
40
+ s.require_paths = ["lib"]
41
+ s.rubygems_version = "1.8.16"
42
+ s.summary = "Lock-server"
43
+
44
+ if s.respond_to? :specification_version then
45
+ s.specification_version = 3
46
+
47
+ if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
48
+ s.add_runtime_dependency(%q<eventmachine>, [">= 0"])
49
+ s.add_development_dependency(%q<shoulda>, [">= 0"])
50
+ s.add_development_dependency(%q<rdoc>, ["~> 3.12"])
51
+ s.add_development_dependency(%q<bundler>, ["~> 1.0.0"])
52
+ s.add_development_dependency(%q<jeweler>, ["~> 1.8.3"])
53
+ s.add_development_dependency(%q<simplecov>, [">= 0"])
54
+ else
55
+ s.add_dependency(%q<eventmachine>, [">= 0"])
56
+ s.add_dependency(%q<shoulda>, [">= 0"])
57
+ s.add_dependency(%q<rdoc>, ["~> 3.12"])
58
+ s.add_dependency(%q<bundler>, ["~> 1.0.0"])
59
+ s.add_dependency(%q<jeweler>, ["~> 1.8.3"])
60
+ s.add_dependency(%q<simplecov>, [">= 0"])
61
+ end
62
+ else
63
+ s.add_dependency(%q<eventmachine>, [">= 0"])
64
+ s.add_dependency(%q<shoulda>, [">= 0"])
65
+ s.add_dependency(%q<rdoc>, ["~> 3.12"])
66
+ s.add_dependency(%q<bundler>, ["~> 1.0.0"])
67
+ s.add_dependency(%q<jeweler>, ["~> 1.8.3"])
68
+ s.add_dependency(%q<simplecov>, [">= 0"])
69
+ end
70
+ end
71
+
@@ -0,0 +1 @@
1
+ 8482
@@ -0,0 +1 @@
1
+ 8486
data/test/helper.rb ADDED
@@ -0,0 +1,18 @@
1
+ require 'rubygems'
2
+ require 'bundler'
3
+ begin
4
+ Bundler.setup(:default, :development)
5
+ rescue Bundler::BundlerError => e
6
+ $stderr.puts e.message
7
+ $stderr.puts "Run `bundle install` to install missing gems"
8
+ exit e.status_code
9
+ end
10
+ require 'test/unit'
11
+ require 'shoulda'
12
+
13
+ $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
14
+ $LOAD_PATH.unshift(File.dirname(__FILE__))
15
+ require 'lock-server'
16
+
17
+ class Test::Unit::TestCase
18
+ end
@@ -0,0 +1,7 @@
1
+ require 'helper'
2
+
3
+ class TestLockServer < Test::Unit::TestCase
4
+ should "probably rename this file and start testing for real" do
5
+ flunk "hey buddy, you should probably rename this file and start testing for real"
6
+ end
7
+ end
metadata ADDED
@@ -0,0 +1,133 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: lock-server
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.1.6
5
+ prerelease:
6
+ platform: ruby
7
+ authors:
8
+ - V_M
9
+ autorequire:
10
+ bindir: bin
11
+ cert_chain: []
12
+ date: 2012-06-18 00:00:00.000000000 Z
13
+ dependencies:
14
+ - !ruby/object:Gem::Dependency
15
+ name: eventmachine
16
+ requirement: &85014590 !ruby/object:Gem::Requirement
17
+ none: false
18
+ requirements:
19
+ - - ! '>='
20
+ - !ruby/object:Gem::Version
21
+ version: '0'
22
+ type: :runtime
23
+ prerelease: false
24
+ version_requirements: *85014590
25
+ - !ruby/object:Gem::Dependency
26
+ name: shoulda
27
+ requirement: &85014320 !ruby/object:Gem::Requirement
28
+ none: false
29
+ requirements:
30
+ - - ! '>='
31
+ - !ruby/object:Gem::Version
32
+ version: '0'
33
+ type: :development
34
+ prerelease: false
35
+ version_requirements: *85014320
36
+ - !ruby/object:Gem::Dependency
37
+ name: rdoc
38
+ requirement: &85014040 !ruby/object:Gem::Requirement
39
+ none: false
40
+ requirements:
41
+ - - ~>
42
+ - !ruby/object:Gem::Version
43
+ version: '3.12'
44
+ type: :development
45
+ prerelease: false
46
+ version_requirements: *85014040
47
+ - !ruby/object:Gem::Dependency
48
+ name: bundler
49
+ requirement: &85030030 !ruby/object:Gem::Requirement
50
+ none: false
51
+ requirements:
52
+ - - ~>
53
+ - !ruby/object:Gem::Version
54
+ version: 1.0.0
55
+ type: :development
56
+ prerelease: false
57
+ version_requirements: *85030030
58
+ - !ruby/object:Gem::Dependency
59
+ name: jeweler
60
+ requirement: &85029420 !ruby/object:Gem::Requirement
61
+ none: false
62
+ requirements:
63
+ - - ~>
64
+ - !ruby/object:Gem::Version
65
+ version: 1.8.3
66
+ type: :development
67
+ prerelease: false
68
+ version_requirements: *85029420
69
+ - !ruby/object:Gem::Dependency
70
+ name: simplecov
71
+ requirement: &85028860 !ruby/object:Gem::Requirement
72
+ none: false
73
+ requirements:
74
+ - - ! '>='
75
+ - !ruby/object:Gem::Version
76
+ version: '0'
77
+ type: :development
78
+ prerelease: false
79
+ version_requirements: *85028860
80
+ description: General purpose lock-server
81
+ email: nobody@nowhere.com
82
+ executables:
83
+ - lock_server
84
+ extensions: []
85
+ extra_rdoc_files:
86
+ - LICENSE.txt
87
+ - README.rdoc
88
+ files:
89
+ - .document
90
+ - Gemfile
91
+ - Gemfile.lock
92
+ - LICENSE.txt
93
+ - README.rdoc
94
+ - Rakefile
95
+ - VERSION
96
+ - bin/lock_server
97
+ - lib/lock-server.rb
98
+ - lib/with_lock.rb
99
+ - lock-server-example.conf
100
+ - lock-server.gemspec
101
+ - log/lock_server_blimp.pid
102
+ - log/lock_server_blimp_monitor.pid
103
+ - test/helper.rb
104
+ - test/test_lock-server.rb
105
+ homepage: http://github.com/blimp666/lock-server
106
+ licenses:
107
+ - MIT
108
+ post_install_message:
109
+ rdoc_options: []
110
+ require_paths:
111
+ - lib
112
+ required_ruby_version: !ruby/object:Gem::Requirement
113
+ none: false
114
+ requirements:
115
+ - - ! '>='
116
+ - !ruby/object:Gem::Version
117
+ version: '0'
118
+ segments:
119
+ - 0
120
+ hash: 206688659
121
+ required_rubygems_version: !ruby/object:Gem::Requirement
122
+ none: false
123
+ requirements:
124
+ - - ! '>='
125
+ - !ruby/object:Gem::Version
126
+ version: '0'
127
+ requirements: []
128
+ rubyforge_project:
129
+ rubygems_version: 1.8.16
130
+ signing_key:
131
+ specification_version: 3
132
+ summary: Lock-server
133
+ test_files: []