kubernetes_health_checker 0.0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 7d91a2df9bf41a797648a65e6c74775cb34ab6e5
+  data.tar.gz: 16e5ea1cf7275cb6137a32998fe8f18f59885980
+SHA512:
+  metadata.gz: 9c611e07d3c5cc1df5f645b753f84ac8d8a077009ac9dcd7b447949938459f8ee233f817f22a842ea8d77cac1cd327e22d8fab4e0a688720edb74422ff63c9a9
+  data.tar.gz: 5fac09e8356088c55a52039d2a5937833143e30cf9cc7c8f1ae19ae8c7df3674a06b8a2d36e5b5b1603d5f23a8f6081583d645ddf640799a7ecd67519c3d9bd9

data/bin/kubernetes_health_checker

@@ -0,0 +1,86 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'thor'
+require 'byebug'
+require 'json'
+
+module KubernetesHealthChecker
+  class Runner < Thor
+    default_task :start
+
+    desc 'start', 'Runs Kubernetes Health Checker'
+    method_option :channel,          aliases: '-c', type: :string,  desc: 'Specify the slack channel for alerting'
+    method_option :alert_threshold,  aliases: '-a', type: :numeric, desc: 'Specify the threshold of restarts before alerting'
+    method_option :namespace,        aliases: '-n', type: :string,  desc: 'Specify the namespace in which the pods to health check reside'
+    method_option :url,              aliases: '-u', type: :string,  desc: 'Specify the slack url you would like the alert POSTed to'
+    method_option :test,             aliases: '-t', type: :boolean, desc: 'Specify if running the gem in test mode'
+
+    ALERT_STATUSES = ['CrashLoopBackOff'].freeze
+
+    def start
+      opts = options.inject({}){|memo,(k,v)| memo[k.to_sym] = v; memo}
+      @alert_threshold = opts[:alert_threshold] || 5
+      @slack_channel = opts[:channel] || '@rohan'
+      @namespace = opts[:namespace] || 'default'
+      @test = opts[:test] || false
+
+      raise 'must provide a url to POST alert to' if !opts[:url]
+      @url = opts[:url]
+
+      if @test
+        path = File.expand_path('../mock_output.txt', __FILE__)
+        output = File.read(path)
+      else
+        output = `kubectl get pods --namespace #{options[:namespace]}`
+      end
+
+      message = ''
+      rows = output.split("\n")
+      rows.each_with_index do |row, index|
+        next if index == 0
+        row = row.split(" ")
+        response = check_pod_state(pod_name: row[0], status: row[2], restarts: row[3].to_i)
+
+        message += response[:message] if response[:should_alert]
+      end
+
+      send_alert(message) if !message.empty?
+    end
+
+    desc 'version', 'Prints version'
+    def version
+      say "KubernetesHealthChecker 0.0.0"
+    end
+
+    no_commands do
+      def send_alert(message)
+        payload = {
+          channel: @slack_channel,
+          username: 'kubernetes_health_check_bot',
+          text: message,
+          icon_emoji: ":face_with_thermometer:"
+        }
+
+        `curl -X POST --data-urlencode 'payload=#{JSON.generate(payload)}' #{@url}`
+      end
+
+      def check_pod_state(pod_name:, status:, restarts:)
+        should_alert = false
+        text = nil
+
+        if ALERT_STATUSES.include?(status)
+          should_alert = true
+          text = "Pod *#{pod_name}* is in status: *#{status}*.\n"
+        elsif restarts > @alert_threshold
+          should_alert = true
+          text = "Pod *#{pod_name}* has restarted *#{restarts}* times and has status: *#{status}*.\n"
+        end
+
+        {should_alert: should_alert, message: text}
+      end
+    end
+  end
+end
+
+KubernetesHealthChecker::Runner.start

metadata

@@ -0,0 +1,74 @@
+--- !ruby/object:Gem::Specification
+name: kubernetes_health_checker
+version: !ruby/object:Gem::Version
+  version: 0.0.0.1
+platform: ruby
+authors:
+- Rohan Sahai
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-08-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Simple script to send slack notifications for pods that are crashing/restarting
+  too often
+email: rohan@sah.ai
+executables:
+- kubernetes_health_checker
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/kubernetes_health_checker
+homepage: http://rubygems.org/gems/kubernetes_health_checker
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.12
+signing_key: 
+specification_version: 4
+summary: Get notifications for unhealthy kubernetes pods
+test_files: []