msg-qworker 0.1.0

Sign up to get free protection for your applications and to get access to all the features.
data/LICENSE ADDED
@@ -0,0 +1,279 @@
1
+ GNU GENERAL PUBLIC LICENSE
2
+ Version 2, June 1991
3
+
4
+ Copyright (C) 1989, 1991 Free Software Foundation, Inc.,
5
+ 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
6
+ Everyone is permitted to copy and distribute verbatim copies
7
+ of this license document, but changing it is not allowed.
8
+
9
+ Preamble
10
+
11
+ The licenses for most software are designed to take away your
12
+ freedom to share and change it. By contrast, the GNU General Public
13
+ License is intended to guarantee your freedom to share and change free
14
+ software--to make sure the software is free for all its users. This
15
+ General Public License applies to most of the Free Software
16
+ Foundation's software and to any other program whose authors commit to
17
+ using it. (Some other Free Software Foundation software is covered by
18
+ the GNU Lesser General Public License instead.) You can apply it to
19
+ your programs, too.
20
+
21
+ When we speak of free software, we are referring to freedom, not
22
+ price. Our General Public Licenses are designed to make sure that you
23
+ have the freedom to distribute copies of free software (and charge for
24
+ this service if you wish), that you receive source code or can get it
25
+ if you want it, that you can change the software or use pieces of it
26
+ in new free programs; and that you know you can do these things.
27
+
28
+ To protect your rights, we need to make restrictions that forbid
29
+ anyone to deny you these rights or to ask you to surrender the rights.
30
+ These restrictions translate to certain responsibilities for you if you
31
+ distribute copies of the software, or if you modify it.
32
+
33
+ For example, if you distribute copies of such a program, whether
34
+ gratis or for a fee, you must give the recipients all the rights that
35
+ you have. You must make sure that they, too, receive or can get the
36
+ source code. And you must show them these terms so they know their
37
+ rights.
38
+
39
+ We protect your rights with two steps: (1) copyright the software, and
40
+ (2) offer you this license which gives you legal permission to copy,
41
+ distribute and/or modify the software.
42
+
43
+ Also, for each author's protection and ours, we want to make certain
44
+ that everyone understands that there is no warranty for this free
45
+ software. If the software is modified by someone else and passed on, we
46
+ want its recipients to know that what they have is not the original, so
47
+ that any problems introduced by others will not reflect on the original
48
+ authors' reputations.
49
+
50
+ Finally, any free program is threatened constantly by software
51
+ patents. We wish to avoid the danger that redistributors of a free
52
+ program will individually obtain patent licenses, in effect making the
53
+ program proprietary. To prevent this, we have made it clear that any
54
+ patent must be licensed for everyone's free use or not licensed at all.
55
+
56
+ The precise terms and conditions for copying, distribution and
57
+ modification follow.
58
+
59
+ GNU GENERAL PUBLIC LICENSE
60
+ TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION
61
+
62
+ 0. This License applies to any program or other work which contains
63
+ a notice placed by the copyright holder saying it may be distributed
64
+ under the terms of this General Public License. The "Program", below,
65
+ refers to any such program or work, and a "work based on the Program"
66
+ means either the Program or any derivative work under copyright law:
67
+ that is to say, a work containing the Program or a portion of it,
68
+ either verbatim or with modifications and/or translated into another
69
+ language. (Hereinafter, translation is included without limitation in
70
+ the term "modification".) Each licensee is addressed as "you".
71
+
72
+ Activities other than copying, distribution and modification are not
73
+ covered by this License; they are outside its scope. The act of
74
+ running the Program is not restricted, and the output from the Program
75
+ is covered only if its contents constitute a work based on the
76
+ Program (independent of having been made by running the Program).
77
+ Whether that is true depends on what the Program does.
78
+
79
+ 1. You may copy and distribute verbatim copies of the Program's
80
+ source code as you receive it, in any medium, provided that you
81
+ conspicuously and appropriately publish on each copy an appropriate
82
+ copyright notice and disclaimer of warranty; keep intact all the
83
+ notices that refer to this License and to the absence of any warranty;
84
+ and give any other recipients of the Program a copy of this License
85
+ along with the Program.
86
+
87
+ You may charge a fee for the physical act of transferring a copy, and
88
+ you may at your option offer warranty protection in exchange for a fee.
89
+
90
+ 2. You may modify your copy or copies of the Program or any portion
91
+ of it, thus forming a work based on the Program, and copy and
92
+ distribute such modifications or work under the terms of Section 1
93
+ above, provided that you also meet all of these conditions:
94
+
95
+ a) You must cause the modified files to carry prominent notices
96
+ stating that you changed the files and the date of any change.
97
+
98
+ b) You must cause any work that you distribute or publish, that in
99
+ whole or in part contains or is derived from the Program or any
100
+ part thereof, to be licensed as a whole at no charge to all third
101
+ parties under the terms of this License.
102
+
103
+ c) If the modified program normally reads commands interactively
104
+ when run, you must cause it, when started running for such
105
+ interactive use in the most ordinary way, to print or display an
106
+ announcement including an appropriate copyright notice and a
107
+ notice that there is no warranty (or else, saying that you provide
108
+ a warranty) and that users may redistribute the program under
109
+ these conditions, and telling the user how to view a copy of this
110
+ License. (Exception: if the Program itself is interactive but
111
+ does not normally print such an announcement, your work based on
112
+ the Program is not required to print an announcement.)
113
+
114
+ These requirements apply to the modified work as a whole. If
115
+ identifiable sections of that work are not derived from the Program,
116
+ and can be reasonably considered independent and separate works in
117
+ themselves, then this License, and its terms, do not apply to those
118
+ sections when you distribute them as separate works. But when you
119
+ distribute the same sections as part of a whole which is a work based
120
+ on the Program, the distribution of the whole must be on the terms of
121
+ this License, whose permissions for other licensees extend to the
122
+ entire whole, and thus to each and every part regardless of who wrote it.
123
+
124
+ Thus, it is not the intent of this section to claim rights or contest
125
+ your rights to work written entirely by you; rather, the intent is to
126
+ exercise the right to control the distribution of derivative or
127
+ collective works based on the Program.
128
+
129
+ In addition, mere aggregation of another work not based on the Program
130
+ with the Program (or with a work based on the Program) on a volume of
131
+ a storage or distribution medium does not bring the other work under
132
+ the scope of this License.
133
+
134
+ 3. You may copy and distribute the Program (or a work based on it,
135
+ under Section 2) in object code or executable form under the terms of
136
+ Sections 1 and 2 above provided that you also do one of the following:
137
+
138
+ a) Accompany it with the complete corresponding machine-readable
139
+ source code, which must be distributed under the terms of Sections
140
+ 1 and 2 above on a medium customarily used for software interchange; or,
141
+
142
+ b) Accompany it with a written offer, valid for at least three
143
+ years, to give any third party, for a charge no more than your
144
+ cost of physically performing source distribution, a complete
145
+ machine-readable copy of the corresponding source code, to be
146
+ distributed under the terms of Sections 1 and 2 above on a medium
147
+ customarily used for software interchange; or,
148
+
149
+ c) Accompany it with the information you received as to the offer
150
+ to distribute corresponding source code. (This alternative is
151
+ allowed only for noncommercial distribution and only if you
152
+ received the program in object code or executable form with such
153
+ an offer, in accord with Subsection b above.)
154
+
155
+ The source code for a work means the preferred form of the work for
156
+ making modifications to it. For an executable work, complete source
157
+ code means all the source code for all modules it contains, plus any
158
+ associated interface definition files, plus the scripts used to
159
+ control compilation and installation of the executable. However, as a
160
+ special exception, the source code distributed need not include
161
+ anything that is normally distributed (in either source or binary
162
+ form) with the major components (compiler, kernel, and so on) of the
163
+ operating system on which the executable runs, unless that component
164
+ itself accompanies the executable.
165
+
166
+ If distribution of executable or object code is made by offering
167
+ access to copy from a designated place, then offering equivalent
168
+ access to copy the source code from the same place counts as
169
+ distribution of the source code, even though third parties are not
170
+ compelled to copy the source along with the object code.
171
+
172
+ 4. You may not copy, modify, sublicense, or distribute the Program
173
+ except as expressly provided under this License. Any attempt
174
+ otherwise to copy, modify, sublicense or distribute the Program is
175
+ void, and will automatically terminate your rights under this License.
176
+ However, parties who have received copies, or rights, from you under
177
+ this License will not have their licenses terminated so long as such
178
+ parties remain in full compliance.
179
+
180
+ 5. You are not required to accept this License, since you have not
181
+ signed it. However, nothing else grants you permission to modify or
182
+ distribute the Program or its derivative works. These actions are
183
+ prohibited by law if you do not accept this License. Therefore, by
184
+ modifying or distributing the Program (or any work based on the
185
+ Program), you indicate your acceptance of this License to do so, and
186
+ all its terms and conditions for copying, distributing or modifying
187
+ the Program or works based on it.
188
+
189
+ 6. Each time you redistribute the Program (or any work based on the
190
+ Program), the recipient automatically receives a license from the
191
+ original licensor to copy, distribute or modify the Program subject to
192
+ these terms and conditions. You may not impose any further
193
+ restrictions on the recipients' exercise of the rights granted herein.
194
+ You are not responsible for enforcing compliance by third parties to
195
+ this License.
196
+
197
+ 7. If, as a consequence of a court judgment or allegation of patent
198
+ infringement or for any other reason (not limited to patent issues),
199
+ conditions are imposed on you (whether by court order, agreement or
200
+ otherwise) that contradict the conditions of this License, they do not
201
+ excuse you from the conditions of this License. If you cannot
202
+ distribute so as to satisfy simultaneously your obligations under this
203
+ License and any other pertinent obligations, then as a consequence you
204
+ may not distribute the Program at all. For example, if a patent
205
+ license would not permit royalty-free redistribution of the Program by
206
+ all those who receive copies directly or indirectly through you, then
207
+ the only way you could satisfy both it and this License would be to
208
+ refrain entirely from distribution of the Program.
209
+
210
+ If any portion of this section is held invalid or unenforceable under
211
+ any particular circumstance, the balance of the section is intended to
212
+ apply and the section as a whole is intended to apply in other
213
+ circumstances.
214
+
215
+ It is not the purpose of this section to induce you to infringe any
216
+ patents or other property right claims or to contest validity of any
217
+ such claims; this section has the sole purpose of protecting the
218
+ integrity of the free software distribution system, which is
219
+ implemented by public license practices. Many people have made
220
+ generous contributions to the wide range of software distributed
221
+ through that system in reliance on consistent application of that
222
+ system; it is up to the author/donor to decide if he or she is willing
223
+ to distribute software through any other system and a licensee cannot
224
+ impose that choice.
225
+
226
+ This section is intended to make thoroughly clear what is believed to
227
+ be a consequence of the rest of this License.
228
+
229
+ 8. If the distribution and/or use of the Program is restricted in
230
+ certain countries either by patents or by copyrighted interfaces, the
231
+ original copyright holder who places the Program under this License
232
+ may add an explicit geographical distribution limitation excluding
233
+ those countries, so that distribution is permitted only in or among
234
+ countries not thus excluded. In such case, this License incorporates
235
+ the limitation as if written in the body of this License.
236
+
237
+ 9. The Free Software Foundation may publish revised and/or new versions
238
+ of the General Public License from time to time. Such new versions will
239
+ be similar in spirit to the present version, but may differ in detail to
240
+ address new problems or concerns.
241
+
242
+ Each version is given a distinguishing version number. If the Program
243
+ specifies a version number of this License which applies to it and "any
244
+ later version", you have the option of following the terms and conditions
245
+ either of that version or of any later version published by the Free
246
+ Software Foundation. If the Program does not specify a version number of
247
+ this License, you may choose any version ever published by the Free Software
248
+ Foundation.
249
+
250
+ 10. If you wish to incorporate parts of the Program into other free
251
+ programs whose distribution conditions are different, write to the author
252
+ to ask for permission. For software which is copyrighted by the Free
253
+ Software Foundation, write to the Free Software Foundation; we sometimes
254
+ make exceptions for this. Our decision will be guided by the two goals
255
+ of preserving the free status of all derivatives of our free software and
256
+ of promoting the sharing and reuse of software generally.
257
+
258
+ NO WARRANTY
259
+
260
+ 11. BECAUSE THE PROGRAM IS LICENSED FREE OF CHARGE, THERE IS NO WARRANTY
261
+ FOR THE PROGRAM, TO THE EXTENT PERMITTED BY APPLICABLE LAW. EXCEPT WHEN
262
+ OTHERWISE STATED IN WRITING THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES
263
+ PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED
264
+ OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
265
+ MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE ENTIRE RISK AS
266
+ TO THE QUALITY AND PERFORMANCE OF THE PROGRAM IS WITH YOU. SHOULD THE
267
+ PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF ALL NECESSARY SERVICING,
268
+ REPAIR OR CORRECTION.
269
+
270
+ 12. IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
271
+ WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MAY MODIFY AND/OR
272
+ REDISTRIBUTE THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES,
273
+ INCLUDING ANY GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING
274
+ OUT OF THE USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED
275
+ TO LOSS OF DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY
276
+ YOU OR THIRD PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER
277
+ PROGRAMS), EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE
278
+ POSSIBILITY OF SUCH DAMAGES.
279
+
data/README ADDED
File without changes
data/Rakefile ADDED
@@ -0,0 +1,59 @@
1
+ require 'rubygems'
2
+ require 'rake'
3
+
4
+ begin
5
+ require 'jeweler'
6
+ Jeweler::Tasks.new do |gem|
7
+ gem.name = "msg-qworker"
8
+ gem.summary = %Q{A reliable beanstalk queue processor in Ruby }
9
+ gem.description = %Q{This gem provides a queue worker infrastructure for a beanstalk queue. Jobs are defined within the framework.}
10
+ gem.email = "md@gonium.net"
11
+ gem.homepage = "https://github.com/mysmartgrid/msg-qworker"
12
+ gem.authors = ["Mathias Dalheimer"]
13
+ gem.bindir = 'bin'
14
+ gem.executables = ["qworker.rb", "producer.rb"]
15
+ gem.default_executable = 'qworker.rb'
16
+ gem.files = FileList["[A-Z]*", "{lib,etc,test}/**/*"]
17
+
18
+ gem.add_dependency('uuid', '~> 2.3.2')
19
+ gem.add_dependency('beanstalk-client', '~> 1.1.0')
20
+ #gem.add_development_dependency "thoughtbot-shoulda", ">= 0"
21
+ # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
22
+ end
23
+ Jeweler::GemcutterTasks.new
24
+ rescue LoadError
25
+ puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
26
+ end
27
+
28
+ require 'rake/testtask'
29
+ Rake::TestTask.new(:test) do |test|
30
+ test.libs << 'lib' << 'test'
31
+ test.pattern = 'test/**/test_*.rb'
32
+ test.verbose = true
33
+ end
34
+
35
+ begin
36
+ require 'rcov/rcovtask'
37
+ Rcov::RcovTask.new do |test|
38
+ test.libs << 'test'
39
+ test.pattern = 'test/**/test_*.rb'
40
+ test.verbose = true
41
+ end
42
+ rescue LoadError
43
+ task :rcov do
44
+ abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
45
+ end
46
+ end
47
+
48
+ task :test => :check_dependencies
49
+ task :default => :test
50
+
51
+ require 'rake/rdoctask'
52
+ Rake::RDocTask.new do |rdoc|
53
+ version = File.exist?('VERSION') ? File.read('VERSION') : ""
54
+
55
+ rdoc.rdoc_dir = 'rdoc'
56
+ rdoc.title = "flukso4r #{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.0
data/bin/producer.rb ADDED
@@ -0,0 +1,149 @@
1
+ #!/usr/bin/env ruby
2
+ ###
3
+ ##
4
+ # qworker: A beanstalk ruby library for solid job processing
5
+ # Copyright (C) 2011 Mathias Dalheimer (md@gonium.net)
6
+ #
7
+ # This program is free software; you can redistribute it and/or modify
8
+ # it under the terms of the GNU General Public License as published by
9
+ # the Free Software Foundation; either version 2 of the License, or
10
+ # any later version.
11
+ #
12
+ # This program is distributed in the hope that it will be useful,
13
+ # but WITHOUT ANY WARRANTY; without even the implied warranty of
14
+ # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15
+ # GNU General Public License for more details.
16
+ #
17
+ # You should have received a copy of the GNU General Public License along
18
+ # with this program; if not, write to the Free Software Foundation, Inc.,
19
+ # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20
+ ##
21
+
22
+ # Read the qworker location
23
+ libpath=File.join(File.dirname(__FILE__), '..', 'lib')
24
+ $:.unshift << libpath
25
+ #puts "Using libraty path #{$:.join(":")}"
26
+
27
+ require 'rubygems'
28
+ require 'json'
29
+ require 'pp'
30
+ require 'beanstalk-client' # gem install beanstalk-client
31
+ require 'uuid' # gem install uuid
32
+ require 'optparse'
33
+ require 'ostruct'
34
+
35
+
36
+ ###
37
+ ## Commandline parser
38
+ #
39
+ class Optparser
40
+ CODES = %w[iso-2022-jp shift_jis euc-jp utf8 binary]
41
+ CODE_ALIASES = { "jis" => "iso-2022-jp", "sjis" => "shift_jis" }
42
+ #
43
+ # Return a structure describing the options.
44
+ #
45
+ def self.parse(args)
46
+ # The options specified on the command line will be collected in *options*.
47
+ # We set default values here.
48
+ options = OpenStruct.new
49
+ options.inplace = false
50
+ options.encoding = "utf8"
51
+ options.verbose = false
52
+ opts = OptionParser.new do |opts|
53
+ opts.banner = "Usage: #{$0} [options]"
54
+ opts.separator ""
55
+ opts.separator "Specific options:"
56
+ opts.on("-c", "--config FILE", "The file where the configuration lives.") do |file|
57
+ options.config_file = file
58
+ end
59
+ opts.on("-a", "--algorithm TYPE ", "The algorithm to execute, 'drunken_sailor' or 'sleep'.") do |type|
60
+ options.alg_type = type
61
+ end
62
+ # Boolean switch.
63
+ # Boolean switch.
64
+ opts.on("-v", "--verbose", "Run verbosely") do |v|
65
+ options.verbose = v
66
+ end
67
+ opts.on_tail("-h", "--help", "Show this message") do
68
+ puts opts
69
+ exit
70
+ end
71
+ end
72
+ opts.parse!(args)
73
+ options
74
+ end
75
+ end
76
+
77
+ ###
78
+ ## Script startup
79
+ #
80
+ options = Optparser.parse(ARGV)
81
+ $verbose = options.verbose
82
+ if (options.alg_type != "drunken_sailor" and options.alg_type != "sleep")
83
+ puts "Please provide a valid algorithm type... (-h for details)."
84
+ exit(-1);
85
+ end
86
+ if options.config_file == nil
87
+ puts "Please provide a configuration file... (-h for details)."
88
+ exit(-2);
89
+ end
90
+ if not File.exists?(options.config_file)
91
+ puts " Configuration file #{options.config_file} not found - no database configuration available!"
92
+ exit(-3);
93
+ else
94
+ $CONFIG=YAML.load_file(options.config_file);
95
+ puts "Using this configuration:" if $verbose
96
+ puts $CONFIG.to_yaml if $verbose
97
+ beanstalk_server=$CONFIG['BEANSTALK_SERVER']
98
+ beanstalk_port=$CONFIG['BEANSTALK_PORT']
99
+ puts "---" if $verbose
100
+ end
101
+
102
+ puts "Starting producer, connecting to #{beanstalk_server}:#{beanstalk_port}" if $verbose
103
+
104
+ beanstalk=nil;
105
+ uuidgen=UUID.new();
106
+
107
+ # Attempt to connect to beanstalkd. Do not loop and try to reconnect
108
+ # here! If the server is not reachable this can be a permanent error.
109
+ # The monit infrastructure will report this and automatically restart
110
+ # the worker. Just make sure to print a useful log message and exit
111
+ # with an error code.
112
+ # The Drupal website would do this differently: A pool is maintained by
113
+ # the website. A PHP module for this is pheanstalk:
114
+ # https://github.com/pda/pheanstalk/
115
+
116
+ begin
117
+ beanstalk = Beanstalk::Pool.new(["#{beanstalk_server}:#{beanstalk_port}"])
118
+ rescue
119
+ puts "Cannot connect to beanstalk server. Exiting."
120
+ exit 1
121
+ end
122
+
123
+ # Create a work package. This MUST be formatted as a JSON array, with
124
+ # two parts: A hash that contains the metadata (UUID and jobtype)
125
+ # and another hash containing the payload. The UUID will be used to
126
+ # publish the results. The algorithm identifies the worker binary to
127
+ # run.
128
+ # The UUID MUST be a 32 digit hexadecimal number such as
129
+ # 137753108f59012eaa2c549a20077664
130
+ uuid=uuidgen.generate(format=:compact);
131
+ payload=[
132
+ {'uuid' => uuid, 'type' => options.alg_type},
133
+ {'foo' => 'bar', '23' => '42'}
134
+ ]
135
+ json_payload=JSON.generate(payload);
136
+ puts "generated work package: #{json_payload}"
137
+
138
+ # Now, put the work package in the default tube.
139
+ begin
140
+ beanstalk.put(
141
+ json_payload, # Content of the job.
142
+ pri=1337, # Priority of this job. <1024 is considered urgent.
143
+ delay=0, # Should the job be delayed for X seconds?
144
+ ttr=3 # Seconds until the job will be re-queued.
145
+ )
146
+ rescue
147
+ puts "Cannot put work package in the tube."
148
+ end
149
+
data/bin/qworker.rb ADDED
@@ -0,0 +1,204 @@
1
+ #!/usr/bin/env ruby
2
+
3
+ ###
4
+ ##
5
+ # qworker: A beanstalk ruby library for solid job processing
6
+ # Copyright (C) 2011 Mathias Dalheimer (md@gonium.net)
7
+ #
8
+ # This program is free software; you can redistribute it and/or modify
9
+ # it under the terms of the GNU General Public License as published by
10
+ # the Free Software Foundation; either version 2 of the License, or
11
+ # any later version.
12
+ #
13
+ # This program is distributed in the hope that it will be useful,
14
+ # but WITHOUT ANY WARRANTY; without even the implied warranty of
15
+ # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16
+ # GNU General Public License for more details.
17
+ #
18
+ # You should have received a copy of the GNU General Public License along
19
+ # with this program; if not, write to the Free Software Foundation, Inc.,
20
+ # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
21
+ ##
22
+
23
+ # Read the qworker location
24
+ libpath=File.join(File.dirname(__FILE__), '..', 'lib')
25
+ $:.unshift << libpath
26
+ #puts "Using libraty path #{$:.join(":")}"
27
+
28
+
29
+ require 'rubygems'
30
+ require 'json'
31
+ require 'beanstalk-client' # gem install beanstalk-client
32
+ require 'timeout'
33
+ require 'optparse'
34
+ require 'ostruct'
35
+
36
+ ###
37
+ ## Commandline parser
38
+ #
39
+ class Optparser
40
+ CODES = %w[iso-2022-jp shift_jis euc-jp utf8 binary]
41
+ CODE_ALIASES = { "jis" => "iso-2022-jp", "sjis" => "shift_jis" }
42
+ #
43
+ # Return a structure describing the options.
44
+ #
45
+ def self.parse(args)
46
+ # The options specified on the command line will be collected in *options*.
47
+ # We set default values here.
48
+ options = OpenStruct.new
49
+ options.inplace = false
50
+ options.encoding = "utf8"
51
+ options.verbose = false
52
+ opts = OptionParser.new do |opts|
53
+ opts.banner = "Usage: #{$0} [options]"
54
+ opts.separator ""
55
+ opts.separator "Specific options:"
56
+ opts.on("-c", "--config FILE", "The file where the configuration lives.") do |file|
57
+ options.config_file = file
58
+ end
59
+ # Boolean switch.
60
+ opts.on("-v", "--verbose", "Run verbosely") do |v|
61
+ options.verbose = v
62
+ end
63
+ opts.on_tail("-h", "--help", "Show this message") do
64
+ puts opts
65
+ exit
66
+ end
67
+ end
68
+ opts.parse!(args)
69
+ options
70
+ end
71
+ end
72
+
73
+ ###
74
+ ## Script startup
75
+ #
76
+ options = Optparser.parse(ARGV)
77
+ $verbose = options.verbose
78
+ if options.config_file == nil
79
+ puts "Please provide a configuration file... (-h for details)."
80
+ exit(-2);
81
+ end
82
+ if not File.exists?(options.config_file)
83
+ puts " Configuration file #{options.config_file} not found!"
84
+ exit(-3);
85
+ else
86
+ $CONFIG=YAML.load_file(options.config_file);
87
+ puts "Using this configuration:" if $verbose
88
+ puts $CONFIG.to_yaml if $verbose
89
+ beanstalk_server=$CONFIG['BEANSTALK_SERVER']
90
+ beanstalk_port=$CONFIG['BEANSTALK_PORT']
91
+ $GRACE_TIME=$CONFIG['GRACE_TIME']
92
+ result_basedir=$CONFIG['RESULT_BASEDIR']
93
+ puts "---" if $verbose
94
+ end
95
+
96
+ if not File.directory?(result_basedir)
97
+ puts "FATAL: Result directory does not exist, exiting."
98
+ exit(-4)
99
+ end
100
+
101
+ puts "Starting producer, connecting to #{beanstalk_server}:#{beanstalk_port}" if $verbose
102
+ beanstalk=nil;
103
+ # Attempt to connect to beanstalkd. Do not loop and try to reconnect
104
+ # here! If the server is not reachable this can be a permanent error.
105
+ # The monit infrastructure will report this and automatically restart
106
+ # the worker. Just make sure to print a useful log message and exit
107
+ # with an error code.
108
+ begin
109
+ beanstalk = Beanstalk::Pool.new(["#{beanstalk_server}:#{beanstalk_port}"])
110
+ rescue
111
+ puts "FATAL: Cannot connect to beanstalk server. Exiting."
112
+ exit 1
113
+ end
114
+
115
+ loop do
116
+ job = beanstalk.reserve
117
+ begin
118
+ payload_json=job.body
119
+ payload=JSON.parse(payload_json)
120
+ header=payload[0]
121
+ job_uuid=header['uuid']
122
+ # calculate the result directory of the job. An HTTP server exports
123
+ # this directory to external clients.
124
+ job_result_path="#{result_basedir}/#{job_uuid[0..1]}/#{job_uuid}"
125
+ job_type=header['type']
126
+ body=payload[1]
127
+ puts
128
+ puts "### Dequeued job #{job_uuid}, algorithm #{job_type}"
129
+ puts " + result directory: #{job_result_path}"
130
+ puts " + input data: #{body}"
131
+ # after a timeout, beanstalk puts the job back into the queue.
132
+ puts " + time left for processing: #{job.time_left()}"
133
+ puts " + job queue priority: #{job.pri()}"
134
+
135
+ # Be careful! Only delete the job if the algorithm was really
136
+ # successful. This means that if you start external processes, you
137
+ # MUST evaluate the return code. If the external process failed,
138
+ # release the job so that it is still in the queue (and other
139
+ # consumers can process it)
140
+ pid=0;
141
+ first_dir="#{result_basedir}/#{job_uuid[0..1]}"
142
+ Dir.mkdir(first_dir) unless File.directory?(first_dir)
143
+ second_dir="#{result_basedir}/#{job_uuid[0..1]}/#{job_uuid}"
144
+ Dir.mkdir(second_dir) unless File.directory?(second_dir)
145
+ #TODO: Pass the result path to the job
146
+ begin
147
+ timeout(job.time_left() + $GRACE_TIME) {
148
+ pid = Process.fork
149
+ if pid.nil? then
150
+ # In child: Execute the algorithm. Decide which program to
151
+ # execute here.
152
+ case job_type
153
+ when "drunken_sailor":
154
+ sailor_bin = File.join(File.dirname(__FILE__), "drunken_sailor.rb")
155
+ exec "ruby", sailor_bin, "-s", "#{job.time_left()}", "-v"
156
+ when "sleep":
157
+ exec "sleep", "#{job.time_left()}"
158
+ else
159
+ puts "Unknown job type: #{job_type} - ignoring job."
160
+ end
161
+ else
162
+ # In parent
163
+ Process.wait(pid)
164
+ end
165
+ }
166
+ puts "Algorithm terminated within expected time constraints."
167
+ exitstatus=$?.exitstatus;
168
+ puts "Algorithm terminated with exitstatus #{exitstatus}"
169
+ if (exitstatus == 0)
170
+ puts "Job completed, removing from queue."
171
+ job.delete
172
+ else
173
+ puts "Job failed, leaving job in queue."
174
+ begin
175
+ job.release
176
+ rescue Beanstalk::NotFoundError => e
177
+ puts "Job not found, cannot release it. Ignoring."
178
+ end
179
+ end
180
+ rescue Timeout::Error
181
+ puts "Algorithm did not terminate within expected timeframe."
182
+ puts "Attempting to kill algorithm PID #{pid}"
183
+ Process.kill("KILL", pid)
184
+ Process.wait(pid)
185
+ begin
186
+ job.release
187
+ rescue Beanstalk::NotFoundError => e
188
+ puts "Job not found, cannot release it. Ignoring."
189
+ end
190
+ end
191
+ rescue Exception => e
192
+ puts "Unexpected problem during job execution: #{e} - Releasing job."
193
+ begin
194
+ job.release
195
+ rescue Beanstalk::NotFoundError => e
196
+ puts "Job not found, cannot release it. Ignoring."
197
+ end
198
+ end
199
+ end
200
+
201
+ # The consumer should never leave the loop. If this happens, an error
202
+ # occured, so just leave gracefully.
203
+ puts "Worker node left main loop - exiting."
204
+ exit 2
data/etc/flukso4rrc ADDED
@@ -0,0 +1,6 @@
1
+ DB_FILE: "/tmp/demo_database.sqlite3"
2
+ DB_TABLE_NAME: "UTCReadings"
3
+ ACCESS_TOKEN: "3f..."
4
+ SENSOR_ID: "0e...."
5
+ BASE_URL: "https://api.flukso.net/sensor"
6
+
data/etc/qproducerrc ADDED
@@ -0,0 +1,2 @@
1
+ BEANSTALK_SERVER: "localhost"
2
+ BEANSTALK_PORT: "11300"
data/etc/qworkerrc ADDED
@@ -0,0 +1,4 @@
1
+ BEANSTALK_SERVER: "localhost"
2
+ BEANSTALK_PORT: "11300"
3
+ RESULT_BASEDIR: "/var/tmp/qworker"
4
+ GRACE_TIME: 2
metadata ADDED
@@ -0,0 +1,108 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: msg-qworker
3
+ version: !ruby/object:Gem::Version
4
+ hash: 27
5
+ prerelease: false
6
+ segments:
7
+ - 0
8
+ - 1
9
+ - 0
10
+ version: 0.1.0
11
+ platform: ruby
12
+ authors:
13
+ - Mathias Dalheimer
14
+ autorequire:
15
+ bindir: bin
16
+ cert_chain: []
17
+
18
+ date: 2011-07-26 00:00:00 +02:00
19
+ default_executable: qworker.rb
20
+ dependencies:
21
+ - !ruby/object:Gem::Dependency
22
+ name: uuid
23
+ prerelease: false
24
+ requirement: &id001 !ruby/object:Gem::Requirement
25
+ none: false
26
+ requirements:
27
+ - - ~>
28
+ - !ruby/object:Gem::Version
29
+ hash: 7
30
+ segments:
31
+ - 2
32
+ - 3
33
+ - 2
34
+ version: 2.3.2
35
+ type: :runtime
36
+ version_requirements: *id001
37
+ - !ruby/object:Gem::Dependency
38
+ name: beanstalk-client
39
+ prerelease: false
40
+ requirement: &id002 !ruby/object:Gem::Requirement
41
+ none: false
42
+ requirements:
43
+ - - ~>
44
+ - !ruby/object:Gem::Version
45
+ hash: 19
46
+ segments:
47
+ - 1
48
+ - 1
49
+ - 0
50
+ version: 1.1.0
51
+ type: :runtime
52
+ version_requirements: *id002
53
+ description: This gem provides a queue worker infrastructure for a beanstalk queue. Jobs are defined within the framework.
54
+ email: md@gonium.net
55
+ executables:
56
+ - qworker.rb
57
+ - producer.rb
58
+ extensions: []
59
+
60
+ extra_rdoc_files:
61
+ - LICENSE
62
+ - README
63
+ files:
64
+ - LICENSE
65
+ - README
66
+ - Rakefile
67
+ - VERSION
68
+ - etc/flukso4rrc
69
+ - etc/qproducerrc
70
+ - etc/qworkerrc
71
+ - bin/qworker.rb
72
+ - bin/producer.rb
73
+ has_rdoc: true
74
+ homepage: https://github.com/mysmartgrid/msg-qworker
75
+ licenses: []
76
+
77
+ post_install_message:
78
+ rdoc_options:
79
+ - --charset=UTF-8
80
+ require_paths:
81
+ - lib
82
+ required_ruby_version: !ruby/object:Gem::Requirement
83
+ none: false
84
+ requirements:
85
+ - - ">="
86
+ - !ruby/object:Gem::Version
87
+ hash: 3
88
+ segments:
89
+ - 0
90
+ version: "0"
91
+ required_rubygems_version: !ruby/object:Gem::Requirement
92
+ none: false
93
+ requirements:
94
+ - - ">="
95
+ - !ruby/object:Gem::Version
96
+ hash: 3
97
+ segments:
98
+ - 0
99
+ version: "0"
100
+ requirements: []
101
+
102
+ rubyforge_project:
103
+ rubygems_version: 1.3.7
104
+ signing_key:
105
+ specification_version: 3
106
+ summary: A reliable beanstalk queue processor in Ruby
107
+ test_files: []
108
+