ruby-cli-daemon 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: f0a57291fb717787ecf32c800114d81c9935386917e3e49c7515df597e83296f
+  data.tar.gz: b9495d51c4ad1311dc183efde2d78ef3b7c9156c4f57fb49e191539fa3defb80
+SHA512:
+  metadata.gz: b4246d548789712085fa31fc1ec2452776fd36aa02d9b9b458e294db1677c9cae662f575b1b0b4620622c3ca0fbccb2b02c163087da186da1cd5b18d34f8a310
+  data.tar.gz: 1f55ac0797a42592da5e3b92b2dc98e8a6ebb70d5dbc5f47ec007581361b6c218715862ea4a3d7a6f4f5b660218f1f87c146a254392d13255bc04fabc1e2b9d8

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright (C) 2013 Michael Grosser <michael@grosser.it>
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/bin/ruby-sli-daemon.sh

@@ -0,0 +1,68 @@
+#!/usr/bin/env sh
+set -e # add "x" to debug
+
+lib=$(dirname $(realpath $0))/../lib
+
+case "$1" in
+stop)
+  # Not ruby-cli-daemon, so it does not kill my current editor in that folder
+  exec pkill -f ruby_cli_daemon
+  ;;
+-v|--version)
+  exec ruby -r$lib/ruby_cli_daemon/version.rb -e "puts RubyCliDaemon::VERSION"
+  ;;
+-h|--help)
+  echo "Usage:"
+  echo "  ruby-cli-daemon <ruby-executable> [arg]*"
+  echo "    Start or use background worker to execute command"
+  echo "    For example: ruby-cli-daemon rake --version"
+  echo ""
+  echo "  ruby-cli-daemon stop"
+  echo "    Kill all spawned processes"
+  echo ""
+  echo "Options:"
+  echo "  -v / --version     Show version"
+  echo "  -h / --help        Show this help"
+  exit
+  ;;
+esac
+
+executable=$1
+shift
+
+# matching the `stop` pattern so everything can be killed quickly
+socket=${TMPDIR}ruby_cli_daemon/$(basename $PWD)/${executable}
+log=${TMPDIR}ruby_cli_daemon.log
+
+# spawn new daemon if none exists
+if [[ ! -e $socket ]]; then
+  # absolute executable so a single gem install is enough for all rubies
+  nohup ruby -r$lib/ruby_cli_daemon.rb -rbundler/setup -e RubyCliDaemon.start\ \"$socket\",\ \"$executable\" 0<&- &>$log &
+  while [ ! -e $socket ]; do
+    sleep 0.1
+    kill -0 $(jobs -p) || (echo "Failed to start worker, check $log" && false) # fail fast when worker failed
+  done
+fi
+
+# prepare output so we can start tailing
+status="${socket}.status"
+stdout="${socket}.out"
+stderr="${socket}.err"
+rm -f $status $stdout $stderr # clear previous
+touch $stdout $stderr
+
+# send the command to the daemon
+echo $@ | nc -U $socket
+
+# stream output
+tail -f $stdout &
+tail -f $stderr >&2 &
+
+# wait for command to finish, tight loop so we don't lose time
+while [ ! -f $status ]; do sleep 0.02; done
+
+# kill log streamers, they should be done (but not the spawned worker)
+for job in `jobs -p | tail -n2`; do kill $job; done
+
+# replay exit status
+exit "$(cat $status)"

data/lib/ruby_cli_daemon.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+require "fileutils"
+require "socket"
+require "shellwords"
+
+module RubyCliDaemon
+  TIMEOUT = 60 * 60
+
+  class << self
+    def install(path)
+      File.unlink(path) if File.exist?(path)
+      File.symlink(File.expand_path("../bin/ruby-sli-daemon.sh", __dir__), path)
+      FileUtils.chmod("+x", path)
+    end
+
+    def start(socket, executable)
+      server = create_socket(socket)
+      path = preload_gem(executable)
+
+      loop do
+        return unless (command = wait_for_command(server))
+
+        # execute the command in a fork
+        capture :STDOUT, "#{socket}.out" do
+          capture :STDERR, "#{socket}.err" do
+            _, status = Process.wait2(fork do
+              ARGV.replace(command)
+              load path
+            end)
+
+            # send back response
+            File.write("#{socket}.status", status.exitstatus)
+          end
+        end
+      end
+    ensure
+      # signal that this program is done so ruby-sli-daemon.sh restarts it
+      File.unlink socket if File.exist?(socket)
+    end
+
+    private
+
+    # preload the libraries we'll need to speed up execution
+    def preload_gem(executable)
+      require executable
+      path = Gem.bin_path(executable, executable)
+      GC.start # https://bugs.ruby-lang.org/issues/15878
+      path
+    end
+
+    def wait_for_command(server)
+      return unless IO.select([server], nil, nil, TIMEOUT)
+      connection = server.accept
+      command = connection.gets.shellsplit
+      connection.close
+      command
+    end
+
+    def create_socket(socket)
+      File.unlink socket if File.exist?(socket)
+      FileUtils.mkdir_p(File.dirname(socket))
+      UNIXServer.new(socket)
+    end
+
+    # StringIO does not work with rubies `system` call that `sh` uses under the hood, so using Tempfile + reopen
+    # https://grosser.it/2018/11/23/ruby-capture-stdout-without-stdout/
+    def capture(stream, path)
+      const = Object.const_get(stream)
+      old_stream = const.dup
+      const.flush # not sure if that's necessary
+      const.reopen(path)
+      yield
+    ensure
+      const.flush # not sure if that's necessary
+      const.reopen(old_stream)
+    end
+  end
+end

data/lib/ruby_cli_daemon/version.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+module RubyCliDaemon
+  VERSION = "0.1.0"
+end

metadata

@@ -0,0 +1,47 @@
+--- !ruby/object:Gem::Specification
+name: ruby-cli-daemon
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Michael Grosser
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-05-28 00:00:00.000000000 Z
+dependencies: []
+description: 
+email: michael@grosser.it
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- MIT-LICENSE
+- bin/ruby-sli-daemon.sh
+- lib/ruby_cli_daemon.rb
+- lib/ruby_cli_daemon/version.rb
+homepage: https://github.com/grosser/ruby-cli-daemon
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.3.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.7.6
+signing_key: 
+specification_version: 4
+summary: Preforking daemon that makes all ruby binaries faster
+test_files: []