wsdirector-cli 0.5.0 → 1.0.0

data/lib/wsdirector/scenario_reader.rb

@@ -1,115 +0,0 @@
-# frozen_string_literal: true
-
-require "erb"
-require "json"
-require "wsdirector/ext/deep_dup"
-
-module WSDirector
-  # Read and parse different scenarios
-  class ScenarioReader
-    using WSDirector::Ext::DeepDup
-
-    class << self
-      include WSDirector::Utils
-
-      def parse(scenario)
-        contents =
-          if File.file?(scenario)
-            parse_file(scenario)
-          else
-            [JSON.parse(scenario)]
-          end.flatten
-
-        if contents.first.key?("client")
-          contents = transform_with_loop(contents, multiple: true)
-          parse_multiple_scenarios(contents)
-        else
-          contents = transform_with_loop(contents)
-          {"total" => 1, "clients" => [parse_simple_scenario(contents)]}
-        end
-      end
-
-      private
-
-      JSON_FILE_FORMAT = /.+.(json)\z/.freeze
-      private_constant :JSON_FILE_FORMAT
-
-      def parse_file(file)
-        if file.match?(JSON_FILE_FORMAT)
-          JSON.parse(File.read(file))
-        else
-          ::YAML.load(ERB.new(File.read(file)).result) # rubocop:disable Security/YAMLLoad
-        end
-      end
-
-      def handle_steps(steps)
-        steps.flat_map.with_index do |step, id|
-          if step.is_a?(Hash)
-            type, data = step.to_a.first
-
-            data["sample"] = [1, parse_multiplier(data["sample"])].max if data["sample"]
-
-            multiplier = parse_multiplier(data.delete("multiplier") || "1")
-            Array.new(multiplier) { {"type" => type, "id" => id}.merge(data) }
-          else
-            {"type" => step, "id" => id}
-          end
-        end
-      end
-
-      def parse_simple_scenario(
-        steps,
-        multiplier: 1, name: "default", ignore: nil, protocol: "base"
-      )
-        {
-          "multiplier" => multiplier,
-          "steps" => handle_steps(steps),
-          "name" => name,
-          "ignore" => ignore,
-          "protocol" => protocol
-        }
-      end
-
-      def parse_multiple_scenarios(definitions)
-        total_count = 0
-        clients = definitions.map.with_index do |client, i|
-          _, client = client.to_a.first
-          multiplier = parse_multiplier(client.delete("multiplier") || "1")
-          name = client.delete("name") || (i + 1).to_s
-          total_count += multiplier
-          ignore = parse_ingore(client.fetch("ignore", nil))
-          parse_simple_scenario(
-            client.fetch("actions", []),
-            multiplier: multiplier,
-            name: name,
-            ignore: ignore,
-            protocol: client.fetch("protocol", "base")
-          )
-        end
-        {"total" => total_count, "clients" => clients}
-      end
-
-      def transform_with_loop(contents, multiple: false)
-        contents.flat_map do |content|
-          loop_data = content.dig("client", "loop") || content.dig("loop")
-          next content unless loop_data
-
-          loop_multiplier = parse_multiplier(loop_data["multiplier"] || "1")
-
-          if multiple
-            content["client"]["actions"] = (loop_data["actions"] * loop_multiplier).map(&:deep_dup)
-            content
-          else
-            loop_data["actions"] * loop_multiplier
-          end
-        end
-      end
-
-      def parse_ingore(str)
-        return unless str
-
-        Array(str)
-      end
-    end
-  end
-end

data/lib/wsdirector/task.rb

@@ -1,47 +0,0 @@
-# frozen_string_literal: true
-
-require "wsdirector/client"
-require "wsdirector/protocols"
-
-module WSDirector
-  # Single client operator
-  class Task
-    attr_reader :global_holder, :client
-
-    def initialize(config, global_holder:, result:)
-      @ignore = config.fetch("ignore")
-      @steps = config.fetch("steps")
-      @global_holder = global_holder
-      @result = result
-
-      protocol_class = Protocols.get(config.fetch("protocol", "base"))
-      @protocol = protocol_class.new(self)
-    end
-
-    def run
-      connect!
-
-      steps.each(&protocol)
-
-      result.succeed
-    rescue Error => e
-      result.failed(e.message)
-    end
-
-    def sampled?(step)
-      return true unless step["sample"]
-
-      id, max = step["id"], step["sample"]
-
-      result.track_sample(id, max)
-    end
-
-    private
-
-    attr_reader :steps, :result, :protocol
-
-    def connect!
-      protocol.init_client(ignore: @ignore)
-    end
-  end
-end

data/lib/wsdirector/utils.rb

@@ -1,15 +0,0 @@
-# frozen_string_literal: true
-
-module WSDirector
-  module Utils # :nodoc:
-    MULTIPLIER_FORMAT = /^[-+*\/\\\d ]+$/
-
-    def parse_multiplier(str)
-      prepared = str.to_s.gsub(":scale", WSDirector.config.scale.to_s)
-      raise WSDirector::Error, "Unknown multiplier format: #{str}" unless
-        MULTIPLIER_FORMAT.match?(prepared)
-
-      eval(prepared) # rubocop:disable Security/Eval
-    end
-  end
-end

data/lib/wsdirector/version.rb

@@ -1,5 +0,0 @@
-# frozen_string_literal: true
-
-module WSDirector
-  VERSION = "0.5.0"
-end

data/lib/wsdirector-cli.rb

@@ -1,3 +0,0 @@
-# frozen_string_literal: true
-
-require "wsdirector/cli"

data/lib/wsdirector.rb

@@ -1,23 +0,0 @@
-# frozen_string_literal: true
-
-require "concurrent"
-require "yaml"
-require "json"
-
-require "wsdirector/version"
-require "wsdirector/configuration"
-require "wsdirector/utils"
-
-# Command line tool for testing websocket servers using scenarios.
-module WSDirector
-  class Error < StandardError
-  end
-
-  class << self
-    def config
-      @config ||= Configuration.new
-    end
-  end
-end
-
-require "wsdirector/cli"