Merge pull request #36 from glennsarti/speedup-server-startup

Speedup server startup

Author: jpogran

server/README.md

@@ -76,6 +76,7 @@ Usage: puppet-languageserver.rb [options]
     -t, --timeout=TIMEOUT            Stop the language server if a client does not connection within TIMEOUT seconds.  A value of zero will not timeout.  Default is 10 seconds
     -d, --no-preload                 Do not preload Puppet information when the language server starts.  Default is to preload
         --debug=DEBUG                Output debug information.  Either specify a filename or 'STDOUT'.  Default is no debug output
+    -s, --slow-start                 Delay starting the TCP Server until Puppet initialisation has completed.  Default is to start fast
     -h, --help                       Prints this help
 ```
 

server/lib/puppet-languageserver.rb

@@ -26,7 +26,8 @@ def self.parse(options)
         :stop_on_client_exit => true,
         :connection_timeout => 10,
         :preload_puppet => true,
-        :debug => nil
+        :debug => nil,
+        :fast_start_tcpserver => true
       }
 
       opt_parser = OptionParser.new do |opts|
@@ -56,6 +57,10 @@ def self.parse(options)
           args[:debug] = debug
         end
 
+        opts.on('-s', '--slow-start', 'Delay starting the TCP Server until Puppet initialisation has completed.  Default is to start fast') do |_misc|
+          args[:fast_start_tcpserver] = false
+        end
+
         opts.on('-h', '--help', 'Prints this help') do
           puts opts
           exit
@@ -98,6 +103,19 @@ def self.init_puppet(options)
     log_message(:info, "Using Puppet v#{Puppet.version}")
 
     log_message(:info, 'Initializing settings...')
+    if options[:fast_start_tcpserver]
+      Thread.new do
+        init_puppet_worker(options)
+      end
+    else
+      init_puppet_worker(options)
+    end
+
+    true
+  end
+
+  def self.init_puppet_worker(options)
+    log_message('information', 'Initializing settings...')
     Puppet.initialize_settings
 
     log_message(:info, 'Creating puppet function environment...')
@@ -117,8 +135,6 @@ def self.init_puppet(options)
     else
       log_message(:info, 'Skipping preloading Puppet')
     end
-
-    true
   end
 
   def self.rpc_server(options)