module Aggkit::Runner
Public Instance Methods
die(message)
click to toggle source
# File lib/aggkit/runner.rb, line 25 def die(message) error(message) exit 1 end
ensure_env_defined!(key, env = ENV)
click to toggle source
# File lib/aggkit/runner.rb, line 72 def ensure_env_defined!(key, env = ENV) env.key?(key) || die("Environment variable #{key} must present!") env[key] end
envsubst(*paths)
click to toggle source
# File lib/aggkit/runner.rb, line 77 def envsubst(*paths) paths = paths.flatten.map{|c| c.to_s.strip }.reject(&:empty?) paths.each do |path| Dir.glob("#{path}/**/*.in") do |templ| output = templ.sub(/\.in$/, '') cmd = "cat '#{templ}' | envsubst > '#{output}'" system(cmd) || die("envsubst failed: #{cmd}") end end end
envsubst_file(templ, output = nil)
click to toggle source
# File lib/aggkit/runner.rb, line 88 def envsubst_file(templ, output = nil) output ||= templ.sub(/\.in$/, '') die('filename must ends with .in or output must be provided') if output.strip == templ.strip cmd = "cat '#{templ}' | envsubst > '#{output}'" system(cmd) || die("envsubst failed: #{cmd}") end
error(message)
click to toggle source
# File lib/aggkit/runner.rb, line 21 def error(message) STDERR.puts "Error: #{message}" end
execute(cmd)
click to toggle source
# File lib/aggkit/runner.rb, line 48 def execute(cmd) puts "Executing: #{cmd}" io = IO.popen(cmd) output = io.read begin io.close rescue StandardError nil end @last_result = $? output end
execute!(cmd, error = nil)
click to toggle source
# File lib/aggkit/runner.rb, line 61 def execute!(cmd, error = nil) output = execute(cmd) if @last_result @last_result.success? || die(error || "Can't execute: #{cmd}") else die("Already collected: #{cmd}") end output end
init_service(service, options = nil)
click to toggle source
# File lib/aggkit/runner.rb, line 9 def init_service(service, options = nil) STDOUT.sync = true STDERR.sync = true puts "Starting #{service}..." puts "Options: #{options.inspect}" if options trap('EXIT') do puts "Stopping #{service}..." end end
load_envs(string, env = ENV)
click to toggle source
# File lib/aggkit/runner.rb, line 30 def load_envs(string, env = ENV) envs = Dotenv::Parser.new(string).call envs.each_pair do |k, v| env[k] = v end envs end
load_envs_from_consul(consul, service)
click to toggle source
# File lib/aggkit/runner.rb, line 38 def load_envs_from_consul(consul, service) consul_http = if consul['http'] consul else "http://#{consul}:8500" end envs = execute!("aggconsul --consul=#{consul_http} --env services/env/#{service} --pristine -d", "Can't load envs") load_envs(envs) end