Add grpc support

lib/puppet/functions/consul/validate_checks.rb

@@ -35,6 +35,10 @@ def validate_checks(obj)
             if (obj.key?("http") || obj.key?("tcp"))
                 raise Puppet::ParseError.new('http and tcp must not be defined for script checks')
             end
+            elsif (obj.key("grpc")
+            if (obj.key?("http") || obj.key?("tcp"))
+              raise Puppet::ParseError.new('http and tcp must not be defined for grpc checks')
+            end
             else
             raise Puppet::ParseError.new('One of ttl, script, tcp, or http must be defined.')
             end

manifests/check.pp

@@ -50,20 +50,24 @@
 #   Value in seconds before the http endpoint considers a failing healthcheck
 #   to be "HARD" down.
 #
+# [*grpc*]
+#    gRPC endpoint for the gRPC service healthcheck
+
 define consul::check (
-  $ensure     = present,
-  $http       = undef,
-  $id         = $title,
-  $interval   = undef,
-  $notes      = undef,
-  $script     = undef,
-  $args       = undef,
-  $service_id = undef,
-  $status     = undef,
-  $tcp        = undef,
-  $timeout    = undef,
-  $token      = undef,
-  $ttl        = undef,
+  $ensure           = present,
+  $http             = undef,
+  $id               = $title,
+  $interval         = undef,
+  $notes            = undef,
+  $script           = undef,
+  $args             = undef,
+  $service_id       = undef,
+  $status           = undef,
+  $tcp              = undef,
+  $timeout          = undef,
+  $token            = undef,
+  $ttl              = undef,
+  String[1] $grpc   = undef,
 ) {
   include consul
 
@@ -81,6 +85,7 @@
     'notes'      => $notes,
     'token'      => $token,
     'status'     => $status,
+    'grpc'       => $grpc
   }
 
   $check_hash = {

spec/defines/consul_check_spec.rb

@@ -190,6 +190,46 @@
             .with_content(/"interval" *: *"30s"/)
         }
       end
+      describe 'with grpc' do
+        let(:params) {{
+          'grpc' => 'localhost:4244'
+        }}
+        it {
+          should contain_file("/etc/consul/check_my_check.json") \
+            .with_content(/"id" *: *"my_check"/) \
+            .with_content(/"name" *: *"my_check"/) \
+            .with_content(/"check" *: *\{/) \
+            .with_content(/"grpc" *: *"localhost:4244"/)
+        }
+      end
+      describe 'with grpc with tls' do
+        let(:params) {{
+          'grpc' => 'localhost:4244',
+          'grpc_use_tls' => true
+        }}
+        it {
+          should contain_file("/etc/consul/check_my_check.json") \
+            .with_content(/"id" *: *"my_check"/) \
+            .with_content(/"name" *: *"my_check"/) \
+            .with_content(/"check" *: *\{/) \
+            .with_content(/"grpc" *: *"localhost:4244"/) \
+            .with_content(/"grpc_use_tls" *: *"true"/)
+        }
+      end
+      describe 'with grpc with interval' do
+        let(:params) {{
+          'grpc' => 'localhost:4244',
+          'interval' => '10s'
+        }}
+        it {
+          should contain_file("/etc/consul/check_my_check.json") \
+            .with_content(/"id" *: *"my_check"/) \
+            .with_content(/"name" *: *"my_check"/) \
+            .with_content(/"check" *: *\{/) \
+            .with_content(/"grpc" *: *"localhost:4244"/) \
+            .with_content(/"interval" *: *"10s"/)
+        }
+      end
       describe 'with script and service_id' do
         let(:params) {{
           'tcp'        => 'localhost:80',