Merge pull request #137 from soutaro/master-worker

Spawn workers for performance

lib/steep.rb

@@ -10,6 +10,8 @@
 require "listen"
 require 'pry'
 require 'language_server-protocol'
+require "etc"
+require "open3"
 
 require "rbs"
 
@@ -70,6 +72,14 @@
 require "steep/type_inference/logic"
 require "steep/ast/types"
 
+require "steep/server/utils"
+require "steep/server/base_worker"
+require "steep/server/code_worker"
+require "steep/server/signature_worker"
+require "steep/server/worker_process"
+require "steep/server/interaction_worker"
+require "steep/server/master"
+
 require "steep/project"
 require "steep/project/file"
 require "steep/project/options"
@@ -89,6 +99,7 @@
 require "steep/drivers/print_project"
 require "steep/drivers/init"
 require "steep/drivers/vendor"
+require "steep/drivers/worker"
 
 if ENV["NO_COLOR"]
   Rainbow.enabled = false

lib/steep/cli.rb

@@ -34,7 +34,7 @@ def process_global_options
 
     def setup_command
       @command = argv.shift&.to_sym
-      if CLI.available_commands.include?(@command)
+      if CLI.available_commands.include?(@command) || @command == :worker
         true
       else
         stderr.puts "Unknown command: #{command}"
@@ -159,5 +159,20 @@ def process_version
       stdout.puts Steep::VERSION
       0
     end
+
+    def process_worker
+      Drivers::Worker.new(stdout: stdout, stderr: stderr, stdin: stdin).tap do |command|
+        OptionParser.new do |opts|
+          opts.banner = "Usage: steep worker [options] [dir]"
+          handle_logging_options opts
+
+          opts.on("--interaction") { command.worker_type = :interaction }
+          opts.on("--code") { command.worker_type = :code }
+          opts.on("--signature") { command.worker_type = :signature }
+          opts.on("--steepfile=PATH") {|path| command.steepfile = Pathname(path) }
+          opts.on("--name=NAME") {|name| command.worker_name = name }
+        end.parse!(argv)
+      end.run
+    end
   end
 end

lib/steep/drivers/annotations.rb

@@ -25,7 +25,7 @@ def run
 
         project.targets.each do |target|
           Steep.logger.tagged "target=#{target.name}" do
-            target.load_signatures do |_, subtyping, _|
+            target.load_signatures(validate: false) do |_, subtyping, _|
               case (status = target.status)
               when nil     # status set on error cases
                 target.source_files.each_value do |file|