michaelmior / NoSE

require 'ostruct'

require 'json'

module NoSE

  module CLI

    # Add a command to run the advisor for a given workload

    class NoSECLI < Thor

      desc 'search NAME', 'run the workload NAME'

      long_desc <<-LONGDESC

        `nose search` is the primary command to run the NoSE advisor. It will

        load the given workload, enumerate indexes for each statement, and

        construct and solve an ILP to produce statement execution plans.

      LONGDESC

      shared_option :mix

      shared_option :format

      shared_option :output

      option :max_space, type: :numeric, default: Float::INFINITY,

                         aliases: '-s',

                         desc: 'maximum space allocated to indexes'

      option :enumerated, type: :boolean, default: false, aliases: '-e',

                          desc: 'whether enumerated indexes should be output'

      option :read_only, type: :boolean, default: false,

                         desc: 'whether to ignore update statements'

      option :objective, type: :string, default: 'cost',

                         enum: %w(cost space indexes),

                         desc: 'the objective function to use in the ILP'

        # Get the workload and cost model

        workload = Workload.load name

          unless options[:mix] == 'default' && workload.mix != :default

        workload.remove_updates if options[:read_only]

        cost_model = get_class_from_config options, 'cost', :cost_model

        # Execute the advisor

        objective = Search::Objective.const_get options[:objective].upcase

        result = search_result workload, cost_model, options[:max_space],

                               objective

        output_search_result result, options unless result.nil?

      end

      private

      # Output results from the search procedure

      # @return [void]

  end

end