Source zookeeper-env.sh when running ZooKeeper

recipes/service.rb

@@ -18,5 +18,6 @@
 zookeeper_service 'zookeeper' do
   service_style node['zookeeper']['service_style']
   install_dir   "#{node['zookeeper']['install_dir']}/zookeeper"
+  conf_dir      node['zookeeper']['conf_dir']
   username      node['zookeeper']['user']
 end

resources/service.rb

@@ -25,13 +25,15 @@
              -> (service_style) { %w(runit upstart systemd exhibitor).include? service_style },
          }
 property :install_dir,         default: '/opt/zookeeper'
+property :conf_dir,            default: '/opt/zookeeper/conf'
 property :username,            default: 'zookeeper'
 property :service_actions,     default: [:enable, :start]
 property :template_cookbook,   default: 'zookeeper'
 property :restart_on_reconfig, default: false
 
 action :create do
   executable_path = "#{new_resource.install_dir}/bin/zkServer.sh"
+  env_path = "#{new_resource.conf_dir}/zookeeper-env.sh"
 
   case new_resource.service_style
   when 'runit'
@@ -43,6 +45,7 @@
       owner          new_resource.username
       group          new_resource.username
       options(
+        zk_env:   env_path,
         exec:     executable_path,
         username: new_resource.username
       )
@@ -54,6 +57,7 @@
       source 'zookeeper.upstart.erb'
       mode   '0644'
       variables(
+        zk_env:   env_path,
         exec:     executable_path,
         username: new_resource.username
       )
@@ -71,6 +75,7 @@
       source 'zookeeper.systemd.erb'
       mode   '0644'
       variables(
+        zk_env:   env_path,
         exec:     executable_path,
         username: new_resource.username
       )
@@ -89,7 +94,6 @@
       supports status: true, restart: true, nothing: true
       action   new_resource.service_actions
     end
-
   when 'exhibitor'
     Chef::Log.info 'Assuming Exhibitor will start up Zookeeper.'
   end

templates/default/sv-zookeeper-run.erb

@@ -1,3 +1,4 @@
 #!/bin/sh
 
+source <%= @options[:zk_env] %>
 exec chpst -u<%= @options[:username] %> <%= @options[:exec] %> start-foreground

templates/default/zookeeper.systemd.erb

@@ -9,6 +9,7 @@ After=network.target
 User=<%= @username %>
 Group=<%= @username %>
 SyslogIdentifier=zookeeper
+EnvironmentFile=<%= @zk_env %>
 Restart=on-failure
 ExecStart=<%= @exec %> start-foreground
 LimitNOFILE=8192

templates/default/zookeeper.upstart.erb

@@ -16,5 +16,7 @@ setuid <%= @username %>
 setgid <%= @username %>
 
 script
+    set -a
+    source <%= @zk_env %>
     exec <%= @exec %> start-foreground
 end script