Feature/http check

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## 2.0.3
+
+- 2017-10-06 Lars Remes [[email protected]][1] Support HTTP service checks
+
 ## 2.0.2
 
 - 2017-08-23 Trevor Wood [[email protected]][1] Revert the change to single values

README.md

@@ -180,6 +180,14 @@ services = Diplomat::Service.get_all({ :dc => 'My_Datacenter' })
 # => #<OpenStruct consul=[], foo=[], bar=[]>
 ```
 
+#### Checks
+
+Register a check:
+```ruby
+headers = { "Authorization" => [ "Basic ZGlwbG9tYXQ6cGFzc3dvcmQ=" ] }
+Diplomat::Check.register_http('health-check-1', 'Health check', 'Node level HTTP health check', "http://#{addr}:#{port}/health_check", 'GET', headers, '10s')
+```
+
 ### Datacenters
 
 Getting a list of datacenters is quite simple and gives you the option to extract all services out of

lib/diplomat/check.rb

@@ -1,7 +1,7 @@
 module Diplomat
   # Methods for interacting with the Consul check API endpoint
   class Check < Diplomat::RestClient
-    @access_methods = %i[checks register_script register_ttl
+    @access_methods = %i[checks register_http register_script register_ttl
                          deregister pass warn fail]
 
     # Get registered checks
@@ -11,7 +11,27 @@ def checks
       JSON.parse(ret.body)
     end
 
-    # Register a check
+    # Register a HTTP check
+    # @param check_id [String] the unique id of the check
+    # @param name [String] the name
+    # @param notes [String] notes about the check
+    # @param script [String] command to be run for check
+    # @param interval [String] frequency (with units) of the check execution
+    # @param ttl [String] time (with units) to mark a check down
+    # @return [Integer] Status code
+    #
+    def register_http(check_id, name, notes, url, method, header, interval)
+      ret = @conn.put do |req|
+        req.url '/v1/agent/check/register'
+        req.body = JSON.generate(
+          'ID' => check_id, 'Name' => name, 'Notes' => notes, 'HTTP' => url, 'Method' => method,
+          'Header' => header, 'Interval' => interval
+        )
+      end
+      ret.status == 200
+    end
+
+    # Register a script check
     # @param check_id [String] the unique id of the check
     # @param name [String] the name
     # @param notes [String] notes about the check

spec/check_spec.rb

@@ -32,6 +32,12 @@
       expect(check.register_script('foobar-1', 'Foobar', 'Foobar test', '/script/test', '10s')).to eq(true)
     end
 
+    it 'register_http' do
+      faraday.stub(:put).and_return(OpenStruct.new(body: '', status: 200))
+      check = Diplomat::Check.new(faraday)
+      expect(check.register_http('foobar-1', 'Foobar', 'Foobar test', 'localhost', 'GET', {}, '10s')).to eq(true)
+    end
+
     it 'register_ttl' do
       faraday.stub(:put).and_return(OpenStruct.new(body: '', status: 200))
       check = Diplomat::Check.new(faraday)