Sha256: 6a514d67e7045273cb97e8b1288c9ef0e1e7eff9fc673704696455fff27c0dcc

Contents?: true

Size: 1.56 KB

Versions: 6

Compression:

Stored size: 1.56 KB

Contents

module Berkshelf::API
  class CacheBuilder
    require_relative 'cache_builder/worker'

    class WorkerSupervisor < Celluloid::SupervisionGroup; end

    include Berkshelf::API::GenericServer
    include Berkshelf::API::Logging
    include Berkshelf::API::Mixin::Services

    server_name :cache_builder
    finalizer :finalize_callback

    def initialize
      log.info "Cache Builder starting..."
      @worker_registry   = Celluloid::Registry.new
      @worker_supervisor = WorkerSupervisor.new(@worker_registry)
      @building          = false
      @build_interval   = Application.config.build_interval

      Application.config.endpoints.each_with_index do |endpoint, index|
        endpoint_options = (endpoint.options || {}).to_hash.deep_symbolize_keys
        @worker_supervisor.supervise(CacheBuilder::Worker[endpoint.type], endpoint_options.merge(priority: index))
      end
    end

    def build
      cache_manager.process_workers(workers)
    end

    # Issue a build command to all workers at the scheduled interval
    #
    # @param [Fixnum, Float] interval
    def build_loop(interval = @build_interval)
      return if @building

      loop do
        @building = true
        build
        sleep interval
      end
    end

    # Return the list of running workers
    #
    # @return [Array<CacheBuilder::Worker::Base>]
    def workers
      @worker_supervisor.actors
    end

    private

      def finalize_callback
        log.info "Cache Builder shutting down..."
        @worker_supervisor.terminate if @worker_supervisor && @worker_supervisor.alive?
      end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version Path
berkshelf-api-3.0.0 lib/berkshelf/api/cache_builder.rb
berkshelf-api-2.2.0 lib/berkshelf/api/cache_builder.rb
berkshelf-api-2.1.3 lib/berkshelf/api/cache_builder.rb
berkshelf-api-2.1.2 lib/berkshelf/api/cache_builder.rb
berkshelf-api-2.1.1 lib/berkshelf/api/cache_builder.rb
berkshelf-api-2.1.0 lib/berkshelf/api/cache_builder.rb