Class: TaxGenerator::Processor
- Inherits:
-
Object
- Object
- TaxGenerator::Processor
- Includes:
- Celluloid, Celluloid::Logger, Celluloid::Notifications, ApplicationHelper
- Defined in:
- lib/tax_generator/classes/processor.rb
Overview
class used to process xml files and create html files
Instance Attribute Summary collapse
-
#condition ⇒ Celluloid::Condition
The supervision group that supervises workers.
-
#job_to_worker ⇒ Hash
Each key from the list is the job id, and the value is the worker that will handle the job.
-
#jobs ⇒ Hash
Each key from the job list is the job id, and the value is the job itself.
-
#options ⇒ Hash
The options that can determine the input and output files and folders.
-
#taxonomy ⇒ TaxGenerator::TaxonomyTree
The taxonomy tree that holds the nodes from the taxonomy xml document.
-
#worker_supervisor ⇒ Celluloid::SupervisionGroup
The supervision group that supervises workers.
-
#worker_to_job ⇒ Hash
Each key from the list is the workers mailbox address, and the value is the job being handled by the worker.
-
#workers ⇒ Celluloid::Actor
The actors that will work on the jobs.
Instance Method Summary collapse
-
#all_workers_finished? ⇒ Boolean
checks if all workers finished and returns true or false.
-
#delegate_job(*jobs) ⇒ void
registers all the jobs, and then delegates them to workers.
-
#destinations_file_name ⇒ String
returns the destinations filename from the option list otherwise the default filename.
-
#destinations_file_path ⇒ String
returns the full path to the destinations file.
-
#fetch_file_jobs ⇒ Array<Hash>
parses the destinations xml document, gets each destination and adds a new job for that destination in the job list and then returns it.
-
#generate_files ⇒ void
fetches the jobs for file generation, then delegates the jobs to workers and waits untill workers finish.
-
#initialize(options = {}) ⇒ void
constructor
receives a list of options that are used to determine the input files and output and input folders.
-
#input_folder ⇒ String
returns the input folder from the options list otherwise the default path.
-
#output_folder ⇒ String
returns the output folder path from the option list otherwise the default path.
-
#prepare_output_dirs ⇒ void
cleans the output folder and re-creates it and the static folder.
-
#register_jobs(*jobs) ⇒ void
registers all the jobs so that the managers can have access to them at any time.
-
#register_worker_for_job(job, worker) ⇒ void
registers the worker so that the current actor has access to it at any given time and starts the worker.
-
#static_output_dir ⇒ String
returns the full path to the static folder.
-
#taxonomy_file_name ⇒ String
returns the taxonomy filename from the option list otherwise the default filename.
-
#taxonomy_file_path ⇒ String
returns the full path to the taxonomy file.
-
#wait_jobs_termination ⇒ void
retrieves the information about the node from the tree and generates for each destination a new File.
-
#work ⇒ void
generates the taxonomy tree , prints it and generates the files.
-
#worker_died(worker, reason) ⇒ void
logs the message about working being dead if a worker crashes.
Methods included from ApplicationHelper
create_directories, elements_with_content, erb_template, execute_with_rescue, format_error, log_error, log_message, nokogiri_xml, rescue_interrupt, root, set_celluloid_exception_handling
Constructor Details
#initialize(options = {}) ⇒ void
receives a list of options that are used to determine the input files and output and input folders
45 46 47 48 49 50 51 52 53 54 |
# File 'lib/tax_generator/classes/processor.rb', line 45 def initialize( = {}) Celluloid.boot @options = .is_a?(Hash) ? .symbolize_keys : {} @worker_supervisor = Celluloid::SupervisionGroup.run! @workers = @worker_supervisor.pool(TaxGenerator::FileCreator, as: :workers, size: 50) Actor.current.link @workers @jobs = {} @job_to_worker = {} @worker_to_job = {} end |
Instance Attribute Details
#condition ⇒ Celluloid::Condition
Returns the supervision group that supervises workers.
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 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 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 |
# File 'lib/tax_generator/classes/processor.rb', line 22 class Processor include Celluloid include Celluloid::Logger include Celluloid::Notifications include TaxGenerator::ApplicationHelper attr_reader :options, :worker_supervisor, :workers, :taxonomy, :jobs, :job_to_worker, :worker_to_job, :condition trap_exit :worker_died # receives a list of options that are used to determine the input files and output and input folders # # @param [Hash] options the options that can determine the input and output files and folders # @option options [String] :input_dir The input directory # @option options [String]:output_dir The output directory # @option options [String] :taxonomy_file_name The taxonomy file name # @option options [String] :destinations_file_name The destinations file name # # @see #work # # @return [void] # # @api public def initialize( = {}) Celluloid.boot @options = .is_a?(Hash) ? .symbolize_keys : {} @worker_supervisor = Celluloid::SupervisionGroup.run! @workers = @worker_supervisor.pool(TaxGenerator::FileCreator, as: :workers, size: 50) Actor.current.link @workers @jobs = {} @job_to_worker = {} @worker_to_job = {} end # returns the input folder from the options list # otherwise the default path # # @return [String] # # @api public def input_folder @options.fetch(:input_dir, "#{root}/data/input") end # returns the taxonomy filename from the option list # otherwise the default filename # # @return [String] # # @api public def taxonomy_file_name @options.fetch(:taxonomy_filename, 'taxonomy.xml') end # returns the destinations filename from the option list # otherwise the default filename # # @return [String] # # @api public def destinations_file_name @options.fetch(:destinations_filename, 'destinations.xml') end # returns the output folder path from the option list # otherwise the default path # # @return [String] # # @api public def output_folder @options.fetch(:output_dir, "#{root}/data/output") end # returns the full path to the taxonomy file # # @return [String] # # @api public def taxonomy_file_path File.join(input_folder, taxonomy_file_name) end # returns the full path to the destinations file # # @return [String] # # @api public def destinations_file_path File.join(input_folder, destinations_file_name) end # returns the full path to the static folder # # @return [String] # # @api public def static_output_dir File.join(output_folder, 'static') end # cleans the output folder and re-creates it and the static folder # # @return [void] # # @api public def prepare_output_dirs FileUtils.rm_rf Dir["#{output_folder}/**/*"] create_directories(output_folder, static_output_dir) FileUtils.cp_r(Dir["#{File.join(root, 'templates', 'static')}/*"], static_output_dir) end # checks if all workers finished and returns true or false # # @return [Boolean] # # @api public def all_workers_finished? @jobs.all? { |_job_id, job| job['status'] == 'finished' } end # registers all the jobs so that the managers can have access to them at any time # # @param [Array] jobs the jobs that will be registered # # @return [void] # # @api public def register_jobs(*jobs) jobs.pmap do |job| job = job.stringify_keys @jobs[job['atlas_id']] = job end end # registers all the jobs, and then delegates them to workers # @see #register_jobs # @see TaxGenerator::FileCreator#work # # @param [Array] jobs the jobs that will be delegated to the workers # # @return [void] # # @api public def delegate_job(*jobs) # jobs need to be added into the manager before starting task to avoid adding new key while iterating register_jobs(*jobs) current_actor = Actor.current @jobs.pmap do |_job_id, job| @workers.async.work(job, current_actor) if @workers.alive? end end # parses the destinations xml document, gets each destination and adds a new job for that # destination in the job list and then returns it # @see #nokogiri_xml # # @return [Array<Hash>] # # @api public def fetch_file_jobs jobs = [{ atlas_id: 0, taxonomy: @taxonomy, destination: nil, output_folder: output_folder }] nokogiri_xml(destinations_file_path).xpath('//destination').pmap do |destination| atlas_id = destination.attributes['atlas_id'] jobs << { atlas_id: atlas_id.value, taxonomy: @taxonomy, destination: destination, output_folder: output_folder } end jobs end # fetches the jobs for file generation, then delegates the jobs to workers and waits untill workers finish # @see #fetch_file_jobs # @see #delegate_job # @see #wait_jobs_termination # # @return [void] # # @api public def generate_files @condition = Celluloid::Condition.new jobs = fetch_file_jobs delegate_job(*jobs) wait_jobs_termination end # retrieves the information about the node from the tree and generates for each destination a new File # @see #create_file # # @param [TaxGenerator::TaxonomyTree] taxonomy the taxonomy tree that will be used for fetching node information # # @return [void] # # @api public def wait_jobs_termination result = @condition.wait return unless result.present? terminate end # registers the worker so that the current actor has access to it at any given time and starts the worker # @see TaxGenerator::FileCreator#start_work # # @param [Hash] job the job that the worker will work # @param [TaxGenerator::FileCreator] worker the worker that will create the file # # @return [void] # # @api public def register_worker_for_job(job, worker) @job_to_worker[job['atlas_id']] = worker @worker_to_job[worker.mailbox.address] = job ("worker #{worker.job_id} registed into manager") Actor.current.link worker worker.async.start_work end # generates the taxonomy tree , prints it and generates the files # @see TaxGenerator::TaxonomyTree#new # @see Tree::TreeNode#print_tree # @see #generate_files # # @return [void] # # @api public def work prepare_output_dirs if File.directory?(input_folder) && File.file?(taxonomy_file_path) && File.file?(destinations_file_path) @taxonomy = TaxGenerator::TaxonomyTree.new(taxonomy_file_path) @taxonomy.print_tree generate_files else ('Please provide valid options', log_method: 'fatal') end end # logs the message about working being dead if a worker crashes # @param [TaxGenerator::FileCreator] worker the worker that died # @param [String] reason the reason for which the worker died # # @return [void] # # @api public def worker_died(worker, reason) mailbox_address = worker.mailbox.address job = @worker_to_job.delete(mailbox_address) return if reason.blank? || job.blank? ("worker job #{job['atlas_id']} with mailbox #{mailbox_address.inspect} died for reason: #{log_error(reason)}", log_method: 'fatal') end end |
#job_to_worker ⇒ Hash
Returns each key from the list is the job id, and the value is the worker that will handle the job.
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 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 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 |
# File 'lib/tax_generator/classes/processor.rb', line 22 class Processor include Celluloid include Celluloid::Logger include Celluloid::Notifications include TaxGenerator::ApplicationHelper attr_reader :options, :worker_supervisor, :workers, :taxonomy, :jobs, :job_to_worker, :worker_to_job, :condition trap_exit :worker_died # receives a list of options that are used to determine the input files and output and input folders # # @param [Hash] options the options that can determine the input and output files and folders # @option options [String] :input_dir The input directory # @option options [String]:output_dir The output directory # @option options [String] :taxonomy_file_name The taxonomy file name # @option options [String] :destinations_file_name The destinations file name # # @see #work # # @return [void] # # @api public def initialize( = {}) Celluloid.boot @options = .is_a?(Hash) ? .symbolize_keys : {} @worker_supervisor = Celluloid::SupervisionGroup.run! @workers = @worker_supervisor.pool(TaxGenerator::FileCreator, as: :workers, size: 50) Actor.current.link @workers @jobs = {} @job_to_worker = {} @worker_to_job = {} end # returns the input folder from the options list # otherwise the default path # # @return [String] # # @api public def input_folder @options.fetch(:input_dir, "#{root}/data/input") end # returns the taxonomy filename from the option list # otherwise the default filename # # @return [String] # # @api public def taxonomy_file_name @options.fetch(:taxonomy_filename, 'taxonomy.xml') end # returns the destinations filename from the option list # otherwise the default filename # # @return [String] # # @api public def destinations_file_name @options.fetch(:destinations_filename, 'destinations.xml') end # returns the output folder path from the option list # otherwise the default path # # @return [String] # # @api public def output_folder @options.fetch(:output_dir, "#{root}/data/output") end # returns the full path to the taxonomy file # # @return [String] # # @api public def taxonomy_file_path File.join(input_folder, taxonomy_file_name) end # returns the full path to the destinations file # # @return [String] # # @api public def destinations_file_path File.join(input_folder, destinations_file_name) end # returns the full path to the static folder # # @return [String] # # @api public def static_output_dir File.join(output_folder, 'static') end # cleans the output folder and re-creates it and the static folder # # @return [void] # # @api public def prepare_output_dirs FileUtils.rm_rf Dir["#{output_folder}/**/*"] create_directories(output_folder, static_output_dir) FileUtils.cp_r(Dir["#{File.join(root, 'templates', 'static')}/*"], static_output_dir) end # checks if all workers finished and returns true or false # # @return [Boolean] # # @api public def all_workers_finished? @jobs.all? { |_job_id, job| job['status'] == 'finished' } end # registers all the jobs so that the managers can have access to them at any time # # @param [Array] jobs the jobs that will be registered # # @return [void] # # @api public def register_jobs(*jobs) jobs.pmap do |job| job = job.stringify_keys @jobs[job['atlas_id']] = job end end # registers all the jobs, and then delegates them to workers # @see #register_jobs # @see TaxGenerator::FileCreator#work # # @param [Array] jobs the jobs that will be delegated to the workers # # @return [void] # # @api public def delegate_job(*jobs) # jobs need to be added into the manager before starting task to avoid adding new key while iterating register_jobs(*jobs) current_actor = Actor.current @jobs.pmap do |_job_id, job| @workers.async.work(job, current_actor) if @workers.alive? end end # parses the destinations xml document, gets each destination and adds a new job for that # destination in the job list and then returns it # @see #nokogiri_xml # # @return [Array<Hash>] # # @api public def fetch_file_jobs jobs = [{ atlas_id: 0, taxonomy: @taxonomy, destination: nil, output_folder: output_folder }] nokogiri_xml(destinations_file_path).xpath('//destination').pmap do |destination| atlas_id = destination.attributes['atlas_id'] jobs << { atlas_id: atlas_id.value, taxonomy: @taxonomy, destination: destination, output_folder: output_folder } end jobs end # fetches the jobs for file generation, then delegates the jobs to workers and waits untill workers finish # @see #fetch_file_jobs # @see #delegate_job # @see #wait_jobs_termination # # @return [void] # # @api public def generate_files @condition = Celluloid::Condition.new jobs = fetch_file_jobs delegate_job(*jobs) wait_jobs_termination end # retrieves the information about the node from the tree and generates for each destination a new File # @see #create_file # # @param [TaxGenerator::TaxonomyTree] taxonomy the taxonomy tree that will be used for fetching node information # # @return [void] # # @api public def wait_jobs_termination result = @condition.wait return unless result.present? terminate end # registers the worker so that the current actor has access to it at any given time and starts the worker # @see TaxGenerator::FileCreator#start_work # # @param [Hash] job the job that the worker will work # @param [TaxGenerator::FileCreator] worker the worker that will create the file # # @return [void] # # @api public def register_worker_for_job(job, worker) @job_to_worker[job['atlas_id']] = worker @worker_to_job[worker.mailbox.address] = job ("worker #{worker.job_id} registed into manager") Actor.current.link worker worker.async.start_work end # generates the taxonomy tree , prints it and generates the files # @see TaxGenerator::TaxonomyTree#new # @see Tree::TreeNode#print_tree # @see #generate_files # # @return [void] # # @api public def work prepare_output_dirs if File.directory?(input_folder) && File.file?(taxonomy_file_path) && File.file?(destinations_file_path) @taxonomy = TaxGenerator::TaxonomyTree.new(taxonomy_file_path) @taxonomy.print_tree generate_files else ('Please provide valid options', log_method: 'fatal') end end # logs the message about working being dead if a worker crashes # @param [TaxGenerator::FileCreator] worker the worker that died # @param [String] reason the reason for which the worker died # # @return [void] # # @api public def worker_died(worker, reason) mailbox_address = worker.mailbox.address job = @worker_to_job.delete(mailbox_address) return if reason.blank? || job.blank? ("worker job #{job['atlas_id']} with mailbox #{mailbox_address.inspect} died for reason: #{log_error(reason)}", log_method: 'fatal') end end |
#jobs ⇒ Hash
Returns each key from the job list is the job id, and the value is the job itself.
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 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 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 |
# File 'lib/tax_generator/classes/processor.rb', line 22 class Processor include Celluloid include Celluloid::Logger include Celluloid::Notifications include TaxGenerator::ApplicationHelper attr_reader :options, :worker_supervisor, :workers, :taxonomy, :jobs, :job_to_worker, :worker_to_job, :condition trap_exit :worker_died # receives a list of options that are used to determine the input files and output and input folders # # @param [Hash] options the options that can determine the input and output files and folders # @option options [String] :input_dir The input directory # @option options [String]:output_dir The output directory # @option options [String] :taxonomy_file_name The taxonomy file name # @option options [String] :destinations_file_name The destinations file name # # @see #work # # @return [void] # # @api public def initialize( = {}) Celluloid.boot @options = .is_a?(Hash) ? .symbolize_keys : {} @worker_supervisor = Celluloid::SupervisionGroup.run! @workers = @worker_supervisor.pool(TaxGenerator::FileCreator, as: :workers, size: 50) Actor.current.link @workers @jobs = {} @job_to_worker = {} @worker_to_job = {} end # returns the input folder from the options list # otherwise the default path # # @return [String] # # @api public def input_folder @options.fetch(:input_dir, "#{root}/data/input") end # returns the taxonomy filename from the option list # otherwise the default filename # # @return [String] # # @api public def taxonomy_file_name @options.fetch(:taxonomy_filename, 'taxonomy.xml') end # returns the destinations filename from the option list # otherwise the default filename # # @return [String] # # @api public def destinations_file_name @options.fetch(:destinations_filename, 'destinations.xml') end # returns the output folder path from the option list # otherwise the default path # # @return [String] # # @api public def output_folder @options.fetch(:output_dir, "#{root}/data/output") end # returns the full path to the taxonomy file # # @return [String] # # @api public def taxonomy_file_path File.join(input_folder, taxonomy_file_name) end # returns the full path to the destinations file # # @return [String] # # @api public def destinations_file_path File.join(input_folder, destinations_file_name) end # returns the full path to the static folder # # @return [String] # # @api public def static_output_dir File.join(output_folder, 'static') end # cleans the output folder and re-creates it and the static folder # # @return [void] # # @api public def prepare_output_dirs FileUtils.rm_rf Dir["#{output_folder}/**/*"] create_directories(output_folder, static_output_dir) FileUtils.cp_r(Dir["#{File.join(root, 'templates', 'static')}/*"], static_output_dir) end # checks if all workers finished and returns true or false # # @return [Boolean] # # @api public def all_workers_finished? @jobs.all? { |_job_id, job| job['status'] == 'finished' } end # registers all the jobs so that the managers can have access to them at any time # # @param [Array] jobs the jobs that will be registered # # @return [void] # # @api public def register_jobs(*jobs) jobs.pmap do |job| job = job.stringify_keys @jobs[job['atlas_id']] = job end end # registers all the jobs, and then delegates them to workers # @see #register_jobs # @see TaxGenerator::FileCreator#work # # @param [Array] jobs the jobs that will be delegated to the workers # # @return [void] # # @api public def delegate_job(*jobs) # jobs need to be added into the manager before starting task to avoid adding new key while iterating register_jobs(*jobs) current_actor = Actor.current @jobs.pmap do |_job_id, job| @workers.async.work(job, current_actor) if @workers.alive? end end # parses the destinations xml document, gets each destination and adds a new job for that # destination in the job list and then returns it # @see #nokogiri_xml # # @return [Array<Hash>] # # @api public def fetch_file_jobs jobs = [{ atlas_id: 0, taxonomy: @taxonomy, destination: nil, output_folder: output_folder }] nokogiri_xml(destinations_file_path).xpath('//destination').pmap do |destination| atlas_id = destination.attributes['atlas_id'] jobs << { atlas_id: atlas_id.value, taxonomy: @taxonomy, destination: destination, output_folder: output_folder } end jobs end # fetches the jobs for file generation, then delegates the jobs to workers and waits untill workers finish # @see #fetch_file_jobs # @see #delegate_job # @see #wait_jobs_termination # # @return [void] # # @api public def generate_files @condition = Celluloid::Condition.new jobs = fetch_file_jobs delegate_job(*jobs) wait_jobs_termination end # retrieves the information about the node from the tree and generates for each destination a new File # @see #create_file # # @param [TaxGenerator::TaxonomyTree] taxonomy the taxonomy tree that will be used for fetching node information # # @return [void] # # @api public def wait_jobs_termination result = @condition.wait return unless result.present? terminate end # registers the worker so that the current actor has access to it at any given time and starts the worker # @see TaxGenerator::FileCreator#start_work # # @param [Hash] job the job that the worker will work # @param [TaxGenerator::FileCreator] worker the worker that will create the file # # @return [void] # # @api public def register_worker_for_job(job, worker) @job_to_worker[job['atlas_id']] = worker @worker_to_job[worker.mailbox.address] = job ("worker #{worker.job_id} registed into manager") Actor.current.link worker worker.async.start_work end # generates the taxonomy tree , prints it and generates the files # @see TaxGenerator::TaxonomyTree#new # @see Tree::TreeNode#print_tree # @see #generate_files # # @return [void] # # @api public def work prepare_output_dirs if File.directory?(input_folder) && File.file?(taxonomy_file_path) && File.file?(destinations_file_path) @taxonomy = TaxGenerator::TaxonomyTree.new(taxonomy_file_path) @taxonomy.print_tree generate_files else ('Please provide valid options', log_method: 'fatal') end end # logs the message about working being dead if a worker crashes # @param [TaxGenerator::FileCreator] worker the worker that died # @param [String] reason the reason for which the worker died # # @return [void] # # @api public def worker_died(worker, reason) mailbox_address = worker.mailbox.address job = @worker_to_job.delete(mailbox_address) return if reason.blank? || job.blank? ("worker job #{job['atlas_id']} with mailbox #{mailbox_address.inspect} died for reason: #{log_error(reason)}", log_method: 'fatal') end end |
#options ⇒ Hash
Returns the options that can determine the input and output files and folders.
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 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 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 |
# File 'lib/tax_generator/classes/processor.rb', line 22 class Processor include Celluloid include Celluloid::Logger include Celluloid::Notifications include TaxGenerator::ApplicationHelper attr_reader :options, :worker_supervisor, :workers, :taxonomy, :jobs, :job_to_worker, :worker_to_job, :condition trap_exit :worker_died # receives a list of options that are used to determine the input files and output and input folders # # @param [Hash] options the options that can determine the input and output files and folders # @option options [String] :input_dir The input directory # @option options [String]:output_dir The output directory # @option options [String] :taxonomy_file_name The taxonomy file name # @option options [String] :destinations_file_name The destinations file name # # @see #work # # @return [void] # # @api public def initialize( = {}) Celluloid.boot @options = .is_a?(Hash) ? .symbolize_keys : {} @worker_supervisor = Celluloid::SupervisionGroup.run! @workers = @worker_supervisor.pool(TaxGenerator::FileCreator, as: :workers, size: 50) Actor.current.link @workers @jobs = {} @job_to_worker = {} @worker_to_job = {} end # returns the input folder from the options list # otherwise the default path # # @return [String] # # @api public def input_folder @options.fetch(:input_dir, "#{root}/data/input") end # returns the taxonomy filename from the option list # otherwise the default filename # # @return [String] # # @api public def taxonomy_file_name @options.fetch(:taxonomy_filename, 'taxonomy.xml') end # returns the destinations filename from the option list # otherwise the default filename # # @return [String] # # @api public def destinations_file_name @options.fetch(:destinations_filename, 'destinations.xml') end # returns the output folder path from the option list # otherwise the default path # # @return [String] # # @api public def output_folder @options.fetch(:output_dir, "#{root}/data/output") end # returns the full path to the taxonomy file # # @return [String] # # @api public def taxonomy_file_path File.join(input_folder, taxonomy_file_name) end # returns the full path to the destinations file # # @return [String] # # @api public def destinations_file_path File.join(input_folder, destinations_file_name) end # returns the full path to the static folder # # @return [String] # # @api public def static_output_dir File.join(output_folder, 'static') end # cleans the output folder and re-creates it and the static folder # # @return [void] # # @api public def prepare_output_dirs FileUtils.rm_rf Dir["#{output_folder}/**/*"] create_directories(output_folder, static_output_dir) FileUtils.cp_r(Dir["#{File.join(root, 'templates', 'static')}/*"], static_output_dir) end # checks if all workers finished and returns true or false # # @return [Boolean] # # @api public def all_workers_finished? @jobs.all? { |_job_id, job| job['status'] == 'finished' } end # registers all the jobs so that the managers can have access to them at any time # # @param [Array] jobs the jobs that will be registered # # @return [void] # # @api public def register_jobs(*jobs) jobs.pmap do |job| job = job.stringify_keys @jobs[job['atlas_id']] = job end end # registers all the jobs, and then delegates them to workers # @see #register_jobs # @see TaxGenerator::FileCreator#work # # @param [Array] jobs the jobs that will be delegated to the workers # # @return [void] # # @api public def delegate_job(*jobs) # jobs need to be added into the manager before starting task to avoid adding new key while iterating register_jobs(*jobs) current_actor = Actor.current @jobs.pmap do |_job_id, job| @workers.async.work(job, current_actor) if @workers.alive? end end # parses the destinations xml document, gets each destination and adds a new job for that # destination in the job list and then returns it # @see #nokogiri_xml # # @return [Array<Hash>] # # @api public def fetch_file_jobs jobs = [{ atlas_id: 0, taxonomy: @taxonomy, destination: nil, output_folder: output_folder }] nokogiri_xml(destinations_file_path).xpath('//destination').pmap do |destination| atlas_id = destination.attributes['atlas_id'] jobs << { atlas_id: atlas_id.value, taxonomy: @taxonomy, destination: destination, output_folder: output_folder } end jobs end # fetches the jobs for file generation, then delegates the jobs to workers and waits untill workers finish # @see #fetch_file_jobs # @see #delegate_job # @see #wait_jobs_termination # # @return [void] # # @api public def generate_files @condition = Celluloid::Condition.new jobs = fetch_file_jobs delegate_job(*jobs) wait_jobs_termination end # retrieves the information about the node from the tree and generates for each destination a new File # @see #create_file # # @param [TaxGenerator::TaxonomyTree] taxonomy the taxonomy tree that will be used for fetching node information # # @return [void] # # @api public def wait_jobs_termination result = @condition.wait return unless result.present? terminate end # registers the worker so that the current actor has access to it at any given time and starts the worker # @see TaxGenerator::FileCreator#start_work # # @param [Hash] job the job that the worker will work # @param [TaxGenerator::FileCreator] worker the worker that will create the file # # @return [void] # # @api public def register_worker_for_job(job, worker) @job_to_worker[job['atlas_id']] = worker @worker_to_job[worker.mailbox.address] = job ("worker #{worker.job_id} registed into manager") Actor.current.link worker worker.async.start_work end # generates the taxonomy tree , prints it and generates the files # @see TaxGenerator::TaxonomyTree#new # @see Tree::TreeNode#print_tree # @see #generate_files # # @return [void] # # @api public def work prepare_output_dirs if File.directory?(input_folder) && File.file?(taxonomy_file_path) && File.file?(destinations_file_path) @taxonomy = TaxGenerator::TaxonomyTree.new(taxonomy_file_path) @taxonomy.print_tree generate_files else ('Please provide valid options', log_method: 'fatal') end end # logs the message about working being dead if a worker crashes # @param [TaxGenerator::FileCreator] worker the worker that died # @param [String] reason the reason for which the worker died # # @return [void] # # @api public def worker_died(worker, reason) mailbox_address = worker.mailbox.address job = @worker_to_job.delete(mailbox_address) return if reason.blank? || job.blank? ("worker job #{job['atlas_id']} with mailbox #{mailbox_address.inspect} died for reason: #{log_error(reason)}", log_method: 'fatal') end end |
#taxonomy ⇒ TaxGenerator::TaxonomyTree
Returns the taxonomy tree that holds the nodes from the taxonomy xml document.
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 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 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 |
# File 'lib/tax_generator/classes/processor.rb', line 22 class Processor include Celluloid include Celluloid::Logger include Celluloid::Notifications include TaxGenerator::ApplicationHelper attr_reader :options, :worker_supervisor, :workers, :taxonomy, :jobs, :job_to_worker, :worker_to_job, :condition trap_exit :worker_died # receives a list of options that are used to determine the input files and output and input folders # # @param [Hash] options the options that can determine the input and output files and folders # @option options [String] :input_dir The input directory # @option options [String]:output_dir The output directory # @option options [String] :taxonomy_file_name The taxonomy file name # @option options [String] :destinations_file_name The destinations file name # # @see #work # # @return [void] # # @api public def initialize( = {}) Celluloid.boot @options = .is_a?(Hash) ? .symbolize_keys : {} @worker_supervisor = Celluloid::SupervisionGroup.run! @workers = @worker_supervisor.pool(TaxGenerator::FileCreator, as: :workers, size: 50) Actor.current.link @workers @jobs = {} @job_to_worker = {} @worker_to_job = {} end # returns the input folder from the options list # otherwise the default path # # @return [String] # # @api public def input_folder @options.fetch(:input_dir, "#{root}/data/input") end # returns the taxonomy filename from the option list # otherwise the default filename # # @return [String] # # @api public def taxonomy_file_name @options.fetch(:taxonomy_filename, 'taxonomy.xml') end # returns the destinations filename from the option list # otherwise the default filename # # @return [String] # # @api public def destinations_file_name @options.fetch(:destinations_filename, 'destinations.xml') end # returns the output folder path from the option list # otherwise the default path # # @return [String] # # @api public def output_folder @options.fetch(:output_dir, "#{root}/data/output") end # returns the full path to the taxonomy file # # @return [String] # # @api public def taxonomy_file_path File.join(input_folder, taxonomy_file_name) end # returns the full path to the destinations file # # @return [String] # # @api public def destinations_file_path File.join(input_folder, destinations_file_name) end # returns the full path to the static folder # # @return [String] # # @api public def static_output_dir File.join(output_folder, 'static') end # cleans the output folder and re-creates it and the static folder # # @return [void] # # @api public def prepare_output_dirs FileUtils.rm_rf Dir["#{output_folder}/**/*"] create_directories(output_folder, static_output_dir) FileUtils.cp_r(Dir["#{File.join(root, 'templates', 'static')}/*"], static_output_dir) end # checks if all workers finished and returns true or false # # @return [Boolean] # # @api public def all_workers_finished? @jobs.all? { |_job_id, job| job['status'] == 'finished' } end # registers all the jobs so that the managers can have access to them at any time # # @param [Array] jobs the jobs that will be registered # # @return [void] # # @api public def register_jobs(*jobs) jobs.pmap do |job| job = job.stringify_keys @jobs[job['atlas_id']] = job end end # registers all the jobs, and then delegates them to workers # @see #register_jobs # @see TaxGenerator::FileCreator#work # # @param [Array] jobs the jobs that will be delegated to the workers # # @return [void] # # @api public def delegate_job(*jobs) # jobs need to be added into the manager before starting task to avoid adding new key while iterating register_jobs(*jobs) current_actor = Actor.current @jobs.pmap do |_job_id, job| @workers.async.work(job, current_actor) if @workers.alive? end end # parses the destinations xml document, gets each destination and adds a new job for that # destination in the job list and then returns it # @see #nokogiri_xml # # @return [Array<Hash>] # # @api public def fetch_file_jobs jobs = [{ atlas_id: 0, taxonomy: @taxonomy, destination: nil, output_folder: output_folder }] nokogiri_xml(destinations_file_path).xpath('//destination').pmap do |destination| atlas_id = destination.attributes['atlas_id'] jobs << { atlas_id: atlas_id.value, taxonomy: @taxonomy, destination: destination, output_folder: output_folder } end jobs end # fetches the jobs for file generation, then delegates the jobs to workers and waits untill workers finish # @see #fetch_file_jobs # @see #delegate_job # @see #wait_jobs_termination # # @return [void] # # @api public def generate_files @condition = Celluloid::Condition.new jobs = fetch_file_jobs delegate_job(*jobs) wait_jobs_termination end # retrieves the information about the node from the tree and generates for each destination a new File # @see #create_file # # @param [TaxGenerator::TaxonomyTree] taxonomy the taxonomy tree that will be used for fetching node information # # @return [void] # # @api public def wait_jobs_termination result = @condition.wait return unless result.present? terminate end # registers the worker so that the current actor has access to it at any given time and starts the worker # @see TaxGenerator::FileCreator#start_work # # @param [Hash] job the job that the worker will work # @param [TaxGenerator::FileCreator] worker the worker that will create the file # # @return [void] # # @api public def register_worker_for_job(job, worker) @job_to_worker[job['atlas_id']] = worker @worker_to_job[worker.mailbox.address] = job ("worker #{worker.job_id} registed into manager") Actor.current.link worker worker.async.start_work end # generates the taxonomy tree , prints it and generates the files # @see TaxGenerator::TaxonomyTree#new # @see Tree::TreeNode#print_tree # @see #generate_files # # @return [void] # # @api public def work prepare_output_dirs if File.directory?(input_folder) && File.file?(taxonomy_file_path) && File.file?(destinations_file_path) @taxonomy = TaxGenerator::TaxonomyTree.new(taxonomy_file_path) @taxonomy.print_tree generate_files else ('Please provide valid options', log_method: 'fatal') end end # logs the message about working being dead if a worker crashes # @param [TaxGenerator::FileCreator] worker the worker that died # @param [String] reason the reason for which the worker died # # @return [void] # # @api public def worker_died(worker, reason) mailbox_address = worker.mailbox.address job = @worker_to_job.delete(mailbox_address) return if reason.blank? || job.blank? ("worker job #{job['atlas_id']} with mailbox #{mailbox_address.inspect} died for reason: #{log_error(reason)}", log_method: 'fatal') end end |
#worker_supervisor ⇒ Celluloid::SupervisionGroup
Returns the supervision group that supervises workers.
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 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 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 |
# File 'lib/tax_generator/classes/processor.rb', line 22 class Processor include Celluloid include Celluloid::Logger include Celluloid::Notifications include TaxGenerator::ApplicationHelper attr_reader :options, :worker_supervisor, :workers, :taxonomy, :jobs, :job_to_worker, :worker_to_job, :condition trap_exit :worker_died # receives a list of options that are used to determine the input files and output and input folders # # @param [Hash] options the options that can determine the input and output files and folders # @option options [String] :input_dir The input directory # @option options [String]:output_dir The output directory # @option options [String] :taxonomy_file_name The taxonomy file name # @option options [String] :destinations_file_name The destinations file name # # @see #work # # @return [void] # # @api public def initialize( = {}) Celluloid.boot @options = .is_a?(Hash) ? .symbolize_keys : {} @worker_supervisor = Celluloid::SupervisionGroup.run! @workers = @worker_supervisor.pool(TaxGenerator::FileCreator, as: :workers, size: 50) Actor.current.link @workers @jobs = {} @job_to_worker = {} @worker_to_job = {} end # returns the input folder from the options list # otherwise the default path # # @return [String] # # @api public def input_folder @options.fetch(:input_dir, "#{root}/data/input") end # returns the taxonomy filename from the option list # otherwise the default filename # # @return [String] # # @api public def taxonomy_file_name @options.fetch(:taxonomy_filename, 'taxonomy.xml') end # returns the destinations filename from the option list # otherwise the default filename # # @return [String] # # @api public def destinations_file_name @options.fetch(:destinations_filename, 'destinations.xml') end # returns the output folder path from the option list # otherwise the default path # # @return [String] # # @api public def output_folder @options.fetch(:output_dir, "#{root}/data/output") end # returns the full path to the taxonomy file # # @return [String] # # @api public def taxonomy_file_path File.join(input_folder, taxonomy_file_name) end # returns the full path to the destinations file # # @return [String] # # @api public def destinations_file_path File.join(input_folder, destinations_file_name) end # returns the full path to the static folder # # @return [String] # # @api public def static_output_dir File.join(output_folder, 'static') end # cleans the output folder and re-creates it and the static folder # # @return [void] # # @api public def prepare_output_dirs FileUtils.rm_rf Dir["#{output_folder}/**/*"] create_directories(output_folder, static_output_dir) FileUtils.cp_r(Dir["#{File.join(root, 'templates', 'static')}/*"], static_output_dir) end # checks if all workers finished and returns true or false # # @return [Boolean] # # @api public def all_workers_finished? @jobs.all? { |_job_id, job| job['status'] == 'finished' } end # registers all the jobs so that the managers can have access to them at any time # # @param [Array] jobs the jobs that will be registered # # @return [void] # # @api public def register_jobs(*jobs) jobs.pmap do |job| job = job.stringify_keys @jobs[job['atlas_id']] = job end end # registers all the jobs, and then delegates them to workers # @see #register_jobs # @see TaxGenerator::FileCreator#work # # @param [Array] jobs the jobs that will be delegated to the workers # # @return [void] # # @api public def delegate_job(*jobs) # jobs need to be added into the manager before starting task to avoid adding new key while iterating register_jobs(*jobs) current_actor = Actor.current @jobs.pmap do |_job_id, job| @workers.async.work(job, current_actor) if @workers.alive? end end # parses the destinations xml document, gets each destination and adds a new job for that # destination in the job list and then returns it # @see #nokogiri_xml # # @return [Array<Hash>] # # @api public def fetch_file_jobs jobs = [{ atlas_id: 0, taxonomy: @taxonomy, destination: nil, output_folder: output_folder }] nokogiri_xml(destinations_file_path).xpath('//destination').pmap do |destination| atlas_id = destination.attributes['atlas_id'] jobs << { atlas_id: atlas_id.value, taxonomy: @taxonomy, destination: destination, output_folder: output_folder } end jobs end # fetches the jobs for file generation, then delegates the jobs to workers and waits untill workers finish # @see #fetch_file_jobs # @see #delegate_job # @see #wait_jobs_termination # # @return [void] # # @api public def generate_files @condition = Celluloid::Condition.new jobs = fetch_file_jobs delegate_job(*jobs) wait_jobs_termination end # retrieves the information about the node from the tree and generates for each destination a new File # @see #create_file # # @param [TaxGenerator::TaxonomyTree] taxonomy the taxonomy tree that will be used for fetching node information # # @return [void] # # @api public def wait_jobs_termination result = @condition.wait return unless result.present? terminate end # registers the worker so that the current actor has access to it at any given time and starts the worker # @see TaxGenerator::FileCreator#start_work # # @param [Hash] job the job that the worker will work # @param [TaxGenerator::FileCreator] worker the worker that will create the file # # @return [void] # # @api public def register_worker_for_job(job, worker) @job_to_worker[job['atlas_id']] = worker @worker_to_job[worker.mailbox.address] = job ("worker #{worker.job_id} registed into manager") Actor.current.link worker worker.async.start_work end # generates the taxonomy tree , prints it and generates the files # @see TaxGenerator::TaxonomyTree#new # @see Tree::TreeNode#print_tree # @see #generate_files # # @return [void] # # @api public def work prepare_output_dirs if File.directory?(input_folder) && File.file?(taxonomy_file_path) && File.file?(destinations_file_path) @taxonomy = TaxGenerator::TaxonomyTree.new(taxonomy_file_path) @taxonomy.print_tree generate_files else ('Please provide valid options', log_method: 'fatal') end end # logs the message about working being dead if a worker crashes # @param [TaxGenerator::FileCreator] worker the worker that died # @param [String] reason the reason for which the worker died # # @return [void] # # @api public def worker_died(worker, reason) mailbox_address = worker.mailbox.address job = @worker_to_job.delete(mailbox_address) return if reason.blank? || job.blank? ("worker job #{job['atlas_id']} with mailbox #{mailbox_address.inspect} died for reason: #{log_error(reason)}", log_method: 'fatal') end end |
#worker_to_job ⇒ Hash
Returns each key from the list is the workers mailbox address, and the value is the job being handled by the worker.
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 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 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 |
# File 'lib/tax_generator/classes/processor.rb', line 22 class Processor include Celluloid include Celluloid::Logger include Celluloid::Notifications include TaxGenerator::ApplicationHelper attr_reader :options, :worker_supervisor, :workers, :taxonomy, :jobs, :job_to_worker, :worker_to_job, :condition trap_exit :worker_died # receives a list of options that are used to determine the input files and output and input folders # # @param [Hash] options the options that can determine the input and output files and folders # @option options [String] :input_dir The input directory # @option options [String]:output_dir The output directory # @option options [String] :taxonomy_file_name The taxonomy file name # @option options [String] :destinations_file_name The destinations file name # # @see #work # # @return [void] # # @api public def initialize( = {}) Celluloid.boot @options = .is_a?(Hash) ? .symbolize_keys : {} @worker_supervisor = Celluloid::SupervisionGroup.run! @workers = @worker_supervisor.pool(TaxGenerator::FileCreator, as: :workers, size: 50) Actor.current.link @workers @jobs = {} @job_to_worker = {} @worker_to_job = {} end # returns the input folder from the options list # otherwise the default path # # @return [String] # # @api public def input_folder @options.fetch(:input_dir, "#{root}/data/input") end # returns the taxonomy filename from the option list # otherwise the default filename # # @return [String] # # @api public def taxonomy_file_name @options.fetch(:taxonomy_filename, 'taxonomy.xml') end # returns the destinations filename from the option list # otherwise the default filename # # @return [String] # # @api public def destinations_file_name @options.fetch(:destinations_filename, 'destinations.xml') end # returns the output folder path from the option list # otherwise the default path # # @return [String] # # @api public def output_folder @options.fetch(:output_dir, "#{root}/data/output") end # returns the full path to the taxonomy file # # @return [String] # # @api public def taxonomy_file_path File.join(input_folder, taxonomy_file_name) end # returns the full path to the destinations file # # @return [String] # # @api public def destinations_file_path File.join(input_folder, destinations_file_name) end # returns the full path to the static folder # # @return [String] # # @api public def static_output_dir File.join(output_folder, 'static') end # cleans the output folder and re-creates it and the static folder # # @return [void] # # @api public def prepare_output_dirs FileUtils.rm_rf Dir["#{output_folder}/**/*"] create_directories(output_folder, static_output_dir) FileUtils.cp_r(Dir["#{File.join(root, 'templates', 'static')}/*"], static_output_dir) end # checks if all workers finished and returns true or false # # @return [Boolean] # # @api public def all_workers_finished? @jobs.all? { |_job_id, job| job['status'] == 'finished' } end # registers all the jobs so that the managers can have access to them at any time # # @param [Array] jobs the jobs that will be registered # # @return [void] # # @api public def register_jobs(*jobs) jobs.pmap do |job| job = job.stringify_keys @jobs[job['atlas_id']] = job end end # registers all the jobs, and then delegates them to workers # @see #register_jobs # @see TaxGenerator::FileCreator#work # # @param [Array] jobs the jobs that will be delegated to the workers # # @return [void] # # @api public def delegate_job(*jobs) # jobs need to be added into the manager before starting task to avoid adding new key while iterating register_jobs(*jobs) current_actor = Actor.current @jobs.pmap do |_job_id, job| @workers.async.work(job, current_actor) if @workers.alive? end end # parses the destinations xml document, gets each destination and adds a new job for that # destination in the job list and then returns it # @see #nokogiri_xml # # @return [Array<Hash>] # # @api public def fetch_file_jobs jobs = [{ atlas_id: 0, taxonomy: @taxonomy, destination: nil, output_folder: output_folder }] nokogiri_xml(destinations_file_path).xpath('//destination').pmap do |destination| atlas_id = destination.attributes['atlas_id'] jobs << { atlas_id: atlas_id.value, taxonomy: @taxonomy, destination: destination, output_folder: output_folder } end jobs end # fetches the jobs for file generation, then delegates the jobs to workers and waits untill workers finish # @see #fetch_file_jobs # @see #delegate_job # @see #wait_jobs_termination # # @return [void] # # @api public def generate_files @condition = Celluloid::Condition.new jobs = fetch_file_jobs delegate_job(*jobs) wait_jobs_termination end # retrieves the information about the node from the tree and generates for each destination a new File # @see #create_file # # @param [TaxGenerator::TaxonomyTree] taxonomy the taxonomy tree that will be used for fetching node information # # @return [void] # # @api public def wait_jobs_termination result = @condition.wait return unless result.present? terminate end # registers the worker so that the current actor has access to it at any given time and starts the worker # @see TaxGenerator::FileCreator#start_work # # @param [Hash] job the job that the worker will work # @param [TaxGenerator::FileCreator] worker the worker that will create the file # # @return [void] # # @api public def register_worker_for_job(job, worker) @job_to_worker[job['atlas_id']] = worker @worker_to_job[worker.mailbox.address] = job ("worker #{worker.job_id} registed into manager") Actor.current.link worker worker.async.start_work end # generates the taxonomy tree , prints it and generates the files # @see TaxGenerator::TaxonomyTree#new # @see Tree::TreeNode#print_tree # @see #generate_files # # @return [void] # # @api public def work prepare_output_dirs if File.directory?(input_folder) && File.file?(taxonomy_file_path) && File.file?(destinations_file_path) @taxonomy = TaxGenerator::TaxonomyTree.new(taxonomy_file_path) @taxonomy.print_tree generate_files else ('Please provide valid options', log_method: 'fatal') end end # logs the message about working being dead if a worker crashes # @param [TaxGenerator::FileCreator] worker the worker that died # @param [String] reason the reason for which the worker died # # @return [void] # # @api public def worker_died(worker, reason) mailbox_address = worker.mailbox.address job = @worker_to_job.delete(mailbox_address) return if reason.blank? || job.blank? ("worker job #{job['atlas_id']} with mailbox #{mailbox_address.inspect} died for reason: #{log_error(reason)}", log_method: 'fatal') end end |
#workers ⇒ Celluloid::Actor
Returns the actors that will work on the jobs.
22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 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 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 |
# File 'lib/tax_generator/classes/processor.rb', line 22 class Processor include Celluloid include Celluloid::Logger include Celluloid::Notifications include TaxGenerator::ApplicationHelper attr_reader :options, :worker_supervisor, :workers, :taxonomy, :jobs, :job_to_worker, :worker_to_job, :condition trap_exit :worker_died # receives a list of options that are used to determine the input files and output and input folders # # @param [Hash] options the options that can determine the input and output files and folders # @option options [String] :input_dir The input directory # @option options [String]:output_dir The output directory # @option options [String] :taxonomy_file_name The taxonomy file name # @option options [String] :destinations_file_name The destinations file name # # @see #work # # @return [void] # # @api public def initialize( = {}) Celluloid.boot @options = .is_a?(Hash) ? .symbolize_keys : {} @worker_supervisor = Celluloid::SupervisionGroup.run! @workers = @worker_supervisor.pool(TaxGenerator::FileCreator, as: :workers, size: 50) Actor.current.link @workers @jobs = {} @job_to_worker = {} @worker_to_job = {} end # returns the input folder from the options list # otherwise the default path # # @return [String] # # @api public def input_folder @options.fetch(:input_dir, "#{root}/data/input") end # returns the taxonomy filename from the option list # otherwise the default filename # # @return [String] # # @api public def taxonomy_file_name @options.fetch(:taxonomy_filename, 'taxonomy.xml') end # returns the destinations filename from the option list # otherwise the default filename # # @return [String] # # @api public def destinations_file_name @options.fetch(:destinations_filename, 'destinations.xml') end # returns the output folder path from the option list # otherwise the default path # # @return [String] # # @api public def output_folder @options.fetch(:output_dir, "#{root}/data/output") end # returns the full path to the taxonomy file # # @return [String] # # @api public def taxonomy_file_path File.join(input_folder, taxonomy_file_name) end # returns the full path to the destinations file # # @return [String] # # @api public def destinations_file_path File.join(input_folder, destinations_file_name) end # returns the full path to the static folder # # @return [String] # # @api public def static_output_dir File.join(output_folder, 'static') end # cleans the output folder and re-creates it and the static folder # # @return [void] # # @api public def prepare_output_dirs FileUtils.rm_rf Dir["#{output_folder}/**/*"] create_directories(output_folder, static_output_dir) FileUtils.cp_r(Dir["#{File.join(root, 'templates', 'static')}/*"], static_output_dir) end # checks if all workers finished and returns true or false # # @return [Boolean] # # @api public def all_workers_finished? @jobs.all? { |_job_id, job| job['status'] == 'finished' } end # registers all the jobs so that the managers can have access to them at any time # # @param [Array] jobs the jobs that will be registered # # @return [void] # # @api public def register_jobs(*jobs) jobs.pmap do |job| job = job.stringify_keys @jobs[job['atlas_id']] = job end end # registers all the jobs, and then delegates them to workers # @see #register_jobs # @see TaxGenerator::FileCreator#work # # @param [Array] jobs the jobs that will be delegated to the workers # # @return [void] # # @api public def delegate_job(*jobs) # jobs need to be added into the manager before starting task to avoid adding new key while iterating register_jobs(*jobs) current_actor = Actor.current @jobs.pmap do |_job_id, job| @workers.async.work(job, current_actor) if @workers.alive? end end # parses the destinations xml document, gets each destination and adds a new job for that # destination in the job list and then returns it # @see #nokogiri_xml # # @return [Array<Hash>] # # @api public def fetch_file_jobs jobs = [{ atlas_id: 0, taxonomy: @taxonomy, destination: nil, output_folder: output_folder }] nokogiri_xml(destinations_file_path).xpath('//destination').pmap do |destination| atlas_id = destination.attributes['atlas_id'] jobs << { atlas_id: atlas_id.value, taxonomy: @taxonomy, destination: destination, output_folder: output_folder } end jobs end # fetches the jobs for file generation, then delegates the jobs to workers and waits untill workers finish # @see #fetch_file_jobs # @see #delegate_job # @see #wait_jobs_termination # # @return [void] # # @api public def generate_files @condition = Celluloid::Condition.new jobs = fetch_file_jobs delegate_job(*jobs) wait_jobs_termination end # retrieves the information about the node from the tree and generates for each destination a new File # @see #create_file # # @param [TaxGenerator::TaxonomyTree] taxonomy the taxonomy tree that will be used for fetching node information # # @return [void] # # @api public def wait_jobs_termination result = @condition.wait return unless result.present? terminate end # registers the worker so that the current actor has access to it at any given time and starts the worker # @see TaxGenerator::FileCreator#start_work # # @param [Hash] job the job that the worker will work # @param [TaxGenerator::FileCreator] worker the worker that will create the file # # @return [void] # # @api public def register_worker_for_job(job, worker) @job_to_worker[job['atlas_id']] = worker @worker_to_job[worker.mailbox.address] = job ("worker #{worker.job_id} registed into manager") Actor.current.link worker worker.async.start_work end # generates the taxonomy tree , prints it and generates the files # @see TaxGenerator::TaxonomyTree#new # @see Tree::TreeNode#print_tree # @see #generate_files # # @return [void] # # @api public def work prepare_output_dirs if File.directory?(input_folder) && File.file?(taxonomy_file_path) && File.file?(destinations_file_path) @taxonomy = TaxGenerator::TaxonomyTree.new(taxonomy_file_path) @taxonomy.print_tree generate_files else ('Please provide valid options', log_method: 'fatal') end end # logs the message about working being dead if a worker crashes # @param [TaxGenerator::FileCreator] worker the worker that died # @param [String] reason the reason for which the worker died # # @return [void] # # @api public def worker_died(worker, reason) mailbox_address = worker.mailbox.address job = @worker_to_job.delete(mailbox_address) return if reason.blank? || job.blank? ("worker job #{job['atlas_id']} with mailbox #{mailbox_address.inspect} died for reason: #{log_error(reason)}", log_method: 'fatal') end end |
Instance Method Details
#all_workers_finished? ⇒ Boolean
checks if all workers finished and returns true or false
139 140 141 |
# File 'lib/tax_generator/classes/processor.rb', line 139 def all_workers_finished? @jobs.all? { |_job_id, job| job['status'] == 'finished' } end |
#delegate_job(*jobs) ⇒ void
This method returns an undefined value.
registers all the jobs, and then delegates them to workers
166 167 168 169 170 171 172 173 |
# File 'lib/tax_generator/classes/processor.rb', line 166 def delegate_job(*jobs) # jobs need to be added into the manager before starting task to avoid adding new key while iterating register_jobs(*jobs) current_actor = Actor.current @jobs.pmap do |_job_id, job| @workers.async.work(job, current_actor) if @workers.alive? end end |
#destinations_file_name ⇒ String
returns the destinations filename from the option list otherwise the default filename
82 83 84 |
# File 'lib/tax_generator/classes/processor.rb', line 82 def destinations_file_name @options.fetch(:destinations_filename, 'destinations.xml') end |
#destinations_file_path ⇒ String
returns the full path to the destinations file
110 111 112 |
# File 'lib/tax_generator/classes/processor.rb', line 110 def destinations_file_path File.join(input_folder, destinations_file_name) end |
#fetch_file_jobs ⇒ Array<Hash>
parses the destinations xml document, gets each destination and adds a new job for that destination in the job list and then returns it
182 183 184 185 186 187 188 189 |
# File 'lib/tax_generator/classes/processor.rb', line 182 def fetch_file_jobs jobs = [{ atlas_id: 0, taxonomy: @taxonomy, destination: nil, output_folder: output_folder }] nokogiri_xml(destinations_file_path).xpath('//destination').pmap do |destination| atlas_id = destination.attributes['atlas_id'] jobs << { atlas_id: atlas_id.value, taxonomy: @taxonomy, destination: destination, output_folder: output_folder } end jobs end |
#generate_files ⇒ void
This method returns an undefined value.
fetches the jobs for file generation, then delegates the jobs to workers and waits untill workers finish
199 200 201 202 203 204 |
# File 'lib/tax_generator/classes/processor.rb', line 199 def generate_files @condition = Celluloid::Condition.new jobs = fetch_file_jobs delegate_job(*jobs) wait_jobs_termination end |
#input_folder ⇒ String
returns the input folder from the options list otherwise the default path
62 63 64 |
# File 'lib/tax_generator/classes/processor.rb', line 62 def input_folder @options.fetch(:input_dir, "#{root}/data/input") end |
#output_folder ⇒ String
returns the output folder path from the option list otherwise the default path
92 93 94 |
# File 'lib/tax_generator/classes/processor.rb', line 92 def output_folder @options.fetch(:output_dir, "#{root}/data/output") end |
#prepare_output_dirs ⇒ void
This method returns an undefined value.
cleans the output folder and re-creates it and the static folder
128 129 130 131 132 |
# File 'lib/tax_generator/classes/processor.rb', line 128 def prepare_output_dirs FileUtils.rm_rf Dir["#{output_folder}/**/*"] create_directories(output_folder, static_output_dir) FileUtils.cp_r(Dir["#{File.join(root, 'templates', 'static')}/*"], static_output_dir) end |
#register_jobs(*jobs) ⇒ void
This method returns an undefined value.
registers all the jobs so that the managers can have access to them at any time
150 151 152 153 154 155 |
# File 'lib/tax_generator/classes/processor.rb', line 150 def register_jobs(*jobs) jobs.pmap do |job| job = job.stringify_keys @jobs[job['atlas_id']] = job end end |
#register_worker_for_job(job, worker) ⇒ void
This method returns an undefined value.
registers the worker so that the current actor has access to it at any given time and starts the worker
229 230 231 232 233 234 235 |
# File 'lib/tax_generator/classes/processor.rb', line 229 def register_worker_for_job(job, worker) @job_to_worker[job['atlas_id']] = worker @worker_to_job[worker.mailbox.address] = job ("worker #{worker.job_id} registed into manager") Actor.current.link worker worker.async.start_work end |
#static_output_dir ⇒ String
returns the full path to the static folder
119 120 121 |
# File 'lib/tax_generator/classes/processor.rb', line 119 def static_output_dir File.join(output_folder, 'static') end |
#taxonomy_file_name ⇒ String
returns the taxonomy filename from the option list otherwise the default filename
72 73 74 |
# File 'lib/tax_generator/classes/processor.rb', line 72 def taxonomy_file_name @options.fetch(:taxonomy_filename, 'taxonomy.xml') end |
#taxonomy_file_path ⇒ String
returns the full path to the taxonomy file
101 102 103 |
# File 'lib/tax_generator/classes/processor.rb', line 101 def taxonomy_file_path File.join(input_folder, taxonomy_file_name) end |
#wait_jobs_termination ⇒ void
This method returns an undefined value.
retrieves the information about the node from the tree and generates for each destination a new File
214 215 216 217 218 |
# File 'lib/tax_generator/classes/processor.rb', line 214 def wait_jobs_termination result = @condition.wait return unless result.present? terminate end |
#work ⇒ void
This method returns an undefined value.
generates the taxonomy tree , prints it and generates the files
245 246 247 248 249 250 251 252 253 254 |
# File 'lib/tax_generator/classes/processor.rb', line 245 def work prepare_output_dirs if File.directory?(input_folder) && File.file?(taxonomy_file_path) && File.file?(destinations_file_path) @taxonomy = TaxGenerator::TaxonomyTree.new(taxonomy_file_path) @taxonomy.print_tree generate_files else ('Please provide valid options', log_method: 'fatal') end end |
#worker_died(worker, reason) ⇒ void
This method returns an undefined value.
logs the message about working being dead if a worker crashes
263 264 265 266 267 268 |
# File 'lib/tax_generator/classes/processor.rb', line 263 def worker_died(worker, reason) mailbox_address = worker.mailbox.address job = @worker_to_job.delete(mailbox_address) return if reason.blank? || job.blank? ("worker job #{job['atlas_id']} with mailbox #{mailbox_address.inspect} died for reason: #{log_error(reason)}", log_method: 'fatal') end |