Contents

module PgObjects
  ##
  # Manages process to create objects
  #
  # Usage:
  #
  #   Manager.new.load_files(:before).create_objects
  #
  # or
  #
  #   Manager.new.load_files(:after).create_objects
  class Manager
    attr_reader :objects, :config, :log

    def initialize
      raise UnsupportedAdapterError if ActiveRecord::Base.connection.adapter_name != 'PostgreSQL'

      @objects = []
      @config = PgObjects.config
      @log = Logger.new(config.silent)
    end

    ##
    # event: +:before+ or +:after+
    #
    # used to reference configuration settings +before_path+ and +after_path+
    def load_files(event)
      dir = config.send "#{event}_path"
      Dir[File.join(dir, '**', "*.{#{config.extensions.join(',')}}")].each do |path|
        @objects << PgObjects::DbObject.new(path)
      end

      self
    end

    def create_objects
      @objects.each { |obj| create_object obj }
    end

    private

    def create_object(obj)
      return if obj.status == :done
      raise CyclicDependencyError, obj.name if obj.status == :processing

      obj.status = :processing

      create_dependencies(obj.dependencies)

      log.write("creating #{obj.name}")
      execute_query(obj)

      obj.status = :done
    end

    def execute_query(obj)
      exec_method = obj.multistatement? ? :execute : :exec_query
      ActiveRecord::Base.connection.send exec_method, obj.sql_query
    end

    def create_dependencies(dependencies)
      dependencies.each { |dep_name| create_object find_object(dep_name) }
    end

    def find_object(dep_name)
      result = @objects.select { |obj| [obj.name, obj.full_name, obj.object_name].compact.include? dep_name }

      raise AmbiguousDependencyError, dep_name if result.size > 1
      raise DependencyNotExistError, dep_name if result.empty?

      result[0]
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
pg_objects-0.4.2	lib/pg_objects/manager.rb
pg_objects-0.3.4	lib/pg_objects/manager.rb