Class: Chef::Provider::Service::Upstart

Inherits:
Simple show all
Defined in:
lib/chef/provider/service/upstart.rb

Constant Summary collapse

UPSTART_STATE_FORMAT =
/\w+ \(?(\w+)\)?[\/ ](\w+)/

Constants included from Mixin::ShellOut

Mixin::ShellOut::DEPRECATED_OPTIONS

Instance Attribute Summary

Attributes inherited from Chef::Provider

#action, #current_resource, #new_resource, #run_context

Instance Method Summary collapse

Methods inherited from Simple

#shared_resource_requirements, #whyrun_supported?

Methods included from Mixin::ShellOut

#run_command_compatible_options, #shell_out, #shell_out!

Methods inherited from Chef::Provider::Service

#action_disable, #action_enable, #action_reload, #action_restart, #action_start, #action_stop, #load_new_resource_state, #shared_resource_requirements, #whyrun_supported?

Methods included from Mixin::Command

#chdir_or_tmpdir, #handle_command_failures, #output_of_command, #run_command, #run_command_with_systems_locale

Methods included from Mixin::Command::Windows

#popen4

Methods included from Mixin::Command::Unix

#popen4

Methods inherited from Chef::Provider

#action_nothing, #cleanup_after_converge, #cookbook_name, #events, #node, #process_resource_requirements, #requirements, #resource_collection, #run_action, #set_updated_status, #whyrun_mode?, #whyrun_supported?

Methods included from DSL::Recipe

#method_missing

Methods included from Mixin::ConvertToClassName

#convert_to_class_name, #convert_to_snake_case, #filename_to_qualified_string, #snake_case_basename

Constructor Details

#initialize(new_resource, run_context) ⇒ Upstart

Upstart does more than start or stop a service, creating multiple ‘states’ [1] that a service can be in. In chef, when we ask a service to start, we expect it to have started before performing the next step since we have top down dependencies. Which is to say we may follow witha resource next that requires that service to be running. According to [2] we can trust that sending a ‘goal’ such as start will not return until that ‘goal’ is reached, or some error has occured.

1

upstart.ubuntu.com/wiki/JobStates

2

www.netsplit.com/2008/04/27/upstart-05-events/

Raises:

  • (ArgumentError)


39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# File 'lib/chef/provider/service/upstart.rb', line 39

def initialize(new_resource, run_context)
  # TODO: re-evaluate if this is needed after integrating cookbook fix
  raise ArgumentError, "run_context cannot be nil" unless run_context
  super
  
  run_context.node
  
  @job = @new_resource.service_name
  
  if @new_resource.parameters
    @new_resource.parameters.each do |key, value|
      @job << " #{key}=#{value}"
    end
  end
  
  platform, version = Chef::Platform.find_platform_and_version(run_context.node)
  if platform == "ubuntu" && (8.04..9.04).include?(version.to_f)
    @upstart_job_dir = "/etc/event.d"
    @upstart_conf_suffix = ""
  else
    @upstart_job_dir = "/etc/init"
    @upstart_conf_suffix = ".conf"
  end

  @command_success = true # new_resource.status_command= false, means upstart used 
  @config_file_found = true 
  @upstart_command_success = true
end

Dynamic Method Handling

This class handles dynamic methods through the method_missing method in the class Chef::DSL::Recipe

Instance Method Details

#define_resource_requirementsObject



68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
# File 'lib/chef/provider/service/upstart.rb', line 68

def define_resource_requirements
  # Do not call super, only call shared requirements
  shared_resource_requirements
  requirements.assert(:all_actions) do |a|
    if !@command_success
      whyrun_msg = @new_resource.status_command ? "Provided status command #{@new_resource.status_command} failed." : 
        "Could not determine upstart state for service"
    end
    a.assertion { @command_success }
    # no failure here, just document the assumptions made. 
    a.whyrun "#{whyrun_msg} Assuming service installed and not running." 
  end

  requirements.assert(:all_actions) do |a| 
    a.assertion  { @config_file_found } 
    # no failure here, just document the assumptions made. 
    a.whyrun "Could not find #{@upstart_job_dir}/#{@new_resource.service_name}#{@upstart_conf_suffix}. Assuming service is disabled." 
  end
end

#disable_serviceObject



206
207
208
209
210
211
# File 'lib/chef/provider/service/upstart.rb', line 206

def disable_service
  Chef::Log.debug("#{@new_resource} upstart lacks inherent support for disabling services, editing job config file")
  conf = Chef::Util::FileEdit.new("#{@upstart_job_dir}/#{@new_resource.service_name}#{@upstart_conf_suffix}")
  conf.search_file_replace(/^start on/, "#start on")
  conf.write_file
end

#enable_serviceObject



199
200
201
202
203
204
# File 'lib/chef/provider/service/upstart.rb', line 199

def enable_service
  Chef::Log.debug("#{@new_resource} upstart lacks inherent support for enabling services, editing job config file")
  conf = Chef::Util::FileEdit.new("#{@upstart_job_dir}/#{@new_resource.service_name}#{@upstart_conf_suffix}")
  conf.search_file_replace(/^#start on/, "start on")
  conf.write_file
end

#load_current_resourceObject



88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
# File 'lib/chef/provider/service/upstart.rb', line 88

def load_current_resource
  @current_resource = Chef::Resource::Service.new(@new_resource.name)
  @current_resource.service_name(@new_resource.service_name)

  # Get running/stopped state
  # We do not support searching for a service via ps when using upstart since status is a native
  # upstart function. We will however support status_command in case someone wants to do something special.
  if @new_resource.status_command
    Chef::Log.debug("#{@new_resource} you have specified a status command, running..")

    begin
      if run_command_with_systems_locale(:command => @new_resource.status_command) == 0
        @current_resource.running true
      end
    rescue Chef::Exceptions::Exec
      @command_success = false
      @current_resource.running false
      nil
    end
  else
    begin
      if upstart_state == "running"
        @current_resource.running true
      else
        @current_resource.running false
      end
    rescue Chef::Exceptions::Exec
      @command_success = false
      @current_resource.running false
      nil
    end
  end
  # Get enabled/disabled state by reading job configuration file
  if ::File.exists?("#{@upstart_job_dir}/#{@new_resource.service_name}#{@upstart_conf_suffix}")
    Chef::Log.debug("#{@new_resource} found #{@upstart_job_dir}/#{@new_resource.service_name}#{@upstart_conf_suffix}")
    ::File.open("#{@upstart_job_dir}/#{@new_resource.service_name}#{@upstart_conf_suffix}",'r') do |file|
      while line = file.gets
        case line
        when /^start on/
          Chef::Log.debug("#{@new_resource} enabled: #{line.chomp}")
          @current_resource.enabled true
          break
        when /^#start on/
          Chef::Log.debug("#{@new_resource} disabled: #{line.chomp}")
          @current_resource.enabled false
          break
        end
      end
    end
  else
    @config_file_found = false
    Chef::Log.debug("#{@new_resource} did not find #{@upstart_job_dir}/#{@new_resource.service_name}#{@upstart_conf_suffix}")
    @current_resource.enabled false
  end

  @current_resource
end

#reload_serviceObject



188
189
190
191
192
193
194
195
# File 'lib/chef/provider/service/upstart.rb', line 188

def reload_service
  if @new_resource.reload_command
    super
  else
    # upstart >= 0.6.3-4 supports reload (HUP)
    run_command_with_systems_locale(:command => "/sbin/reload #{@job}")
  end
end

#restart_serviceObject



174
175
176
177
178
179
180
181
182
183
184
185
186
# File 'lib/chef/provider/service/upstart.rb', line 174

def restart_service
  if @new_resource.restart_command
    super
  # Upstart always provides restart functionality so we don't need to mimic it with stop/sleep/start.
  # Older versions of upstart would fail on restart if the service was currently stopped, check for that. LP:430883
  else @new_resource.supports[:restart]
    if @current_resource.running
      run_command_with_systems_locale(:command => "/sbin/restart #{@job}")
    else
      start_service
    end
  end
end

#start_serviceObject



146
147
148
149
150
151
152
153
154
155
156
157
158
# File 'lib/chef/provider/service/upstart.rb', line 146

def start_service
  # Calling start on a service that is already started will return 1
  # Our 'goal' when we call start is to ensure the service is started
  if @current_resource.running
    Chef::Log.debug("#{@new_resource} already running, not starting")
  else
    if @new_resource.start_command
      super
    else
      run_command_with_systems_locale(:command => "/sbin/start #{@job}")
    end
  end
end

#stop_serviceObject



160
161
162
163
164
165
166
167
168
169
170
171
172
# File 'lib/chef/provider/service/upstart.rb', line 160

def stop_service
  # Calling stop on a service that is already stopped will return 1
  # Our 'goal' when we call stop is to ensure the service is stopped
  unless @current_resource.running
    Chef::Log.debug("#{@new_resource} not running, not stopping")
  else
    if @new_resource.stop_command
      super
    else
      run_command_with_systems_locale(:command => "/sbin/stop #{@job}")
    end
  end
end

#upstart_stateObject



213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
# File 'lib/chef/provider/service/upstart.rb', line 213

def upstart_state
  command = "/sbin/status #{@job}"
  status = popen4(command) do |pid, stdin, stdout, stderr|
    stdout.each_line do |line|
      # rsyslog stop/waiting
      # service goal/state
      # OR
      # rsyslog (stop) waiting
      # service (goal) state
      line =~ UPSTART_STATE_FORMAT
      data = Regexp.last_match
      return data[2]
    end
  end
end