riemann-tools 0.2.11 → 1.0.0

data/tools/riemann-riak/riak_status/riak_status.rb

@@ -0,0 +1,152 @@
+#!/usr/bin/env ruby
+Process.setproctitle($0)
+
+$LOAD_PATH.unshift File.expand_path("#{File.dirname(__FILE__)}/../vodpod-common/lib")
+require 'rubygems'
+require 'vodpod-common'
+require 'vodpod/alerts'
+require 'vodpod/starling'
+require 'net/http'
+require 'yajl/json_gem'
+
+class RiakStatus
+  PORT = 8098
+  PATH = '/stats'
+  INTERVAL = 10
+
+  FSM_LIMITS = {
+    :get => {
+      50 => 1000,
+      95 => 2000,
+      99 => 10000
+    },
+    :put => {
+       50 => 1000,
+       95 => 2000,
+       99 => 10000
+    }
+  }
+
+  def initialize(opts = {})
+    @host = opts[:host] || `hostname`.chomp
+    @port = opts[:port] || PORT
+    @path = opts[:path] || PATH
+  end
+
+  def alert(subservice, state, metric, description)
+    Vodpod.alert(
+      :service => "riak #{subservice}",
+      :state => state,
+      :metric => metric,
+      :description => description
+    )
+  end
+
+  def check_ring
+    str = `#{File.expand_path(File.dirname(__FILE__))}/ringready.erl riak@#{`hostname`}`.chomp
+    if str =~ /^TRUE/
+      alert 'ring', :ok, nil, str
+    else
+      alert 'ring', :warning, nil, str
+    end
+  end
+
+  def check_keys
+    keys = `#{File.expand_path(File.dirname(__FILE__))}/key_count.erl riak@#{`hostname`}`.chomp
+    if keys =~ /^\d+$/
+      alert 'keys', :ok, keys.to_i, keys
+    else
+      alert 'keys', :error, nil, keys
+    end
+  end
+
+  def check_disk
+    gb = `du -s /var/lib/riak/bitcask/`.split(/\s+/).first.to_i / (1024.0**2)
+    alert 'disk', :ok, gb, "#{gb} GB in bitcask"
+  end
+
+  # Returns the riak stat for the given fsm type and percentile.
+  def fsm_stat(type, percentile)
+    "node_#{type}_fsm_time_#{percentile == 50 ? 'median' : percentile}"
+  end
+
+  # Returns the alerts state for the given fsm.
+  def fsm_state(type, percentile, val)
+    limit = FSM_LIMITS[type][percentile]
+    case val
+    when 0 .. limit
+      :ok
+    when limit .. limit * 2
+      :warning
+    else
+      :critical
+    end
+  end
+
+  def check_stats
+    begin
+      res = Net::HTTP.start(@host, @port) do |http|
+        http.get('/stats')
+      end
+    rescue => e
+      Vodpod.alert(
+        :service => 'riak',
+        :state => :critical,
+        :description => "error fetching /stats: #{e.class}, #{e.message}"
+      )
+      return
+    end
+
+    if res.code.to_i == 200
+      stats = JSON.parse(res.body)
+    else
+      Vodpod.alert(
+        :service => 'riak',
+        :state => :critical,
+        :description => "stats returned HTTP #{res.code}:\n\n#{res.body}"
+      )
+      return
+    end
+
+    Vodpod.alert(
+      :service => 'riak',
+      :state => :ok
+    )
+
+    # Gets/puts/rr
+    [
+      'vnode_gets', 
+      'vnode_puts', 
+      'node_gets', 
+      'node_puts', 
+      'read_repairs' 
+    ].each do |s|
+      alert s, :ok, stats[s]/60.0, "#{stats[s]/60.0}/sec"
+    end
+
+    # FSMs
+    [:get, :put].each do |type|
+      [50, 95, 99].each do |percentile|
+        val = stats[fsm_stat(type, percentile)] || 0
+        val = 0 if val == 'undefined'
+        val /= 1000.0 # Convert us to ms
+        state = fsm_state(type, percentile, val) 
+        alert "#{type} #{percentile}", state, val, "#{val} ms"
+      end
+    end
+  end
+
+  def run
+    loop do
+#      check_keys
+      check_stats
+      check_ring
+      check_disk
+      sleep INTERVAL
+    end
+  end
+end
+
+if $0 == __FILE__
+  RiakStatus.new.run
+end

data/tools/riemann-riak/riak_status/ringready.erl

@@ -0,0 +1,9 @@
+#!/usr/bin/env escript
+Process.setproctitle($0)
+%%! -name riakstatuscheck -setcookie riak -hidden
+
+main([]) -> main(["riak@127.0.0.1"]);
+main([Node]) ->
+  io:format("~p\n", [
+    rpc:call(list_to_atom(Node), riak_kv_console, ringready, [[]])
+  ]).

metadata

@@ -1,59 +1,108 @@
 --- !ruby/object:Gem::Specification
 name: riemann-tools
 version: !ruby/object:Gem::Version
-  version: 0.2.11
+  version: 1.0.0
 platform: ruby
 authors:
 - Kyle Kingsbury
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-12-04 00:00:00.000000000 Z
+date: 2022-06-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: riemann-client
+  name: json
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.2
+        version: '1.8'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.2
+        version: '1.8'
 - !ruby/object:Gem::Dependency
-  name: trollop
+  name: optimist
   requirement: !ruby/object:Gem::Requirement
     requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.16.2
+        version: 3.0.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.16.2
+        version: 3.0.0
 - !ruby/object:Gem::Dependency
-  name: json
+  name: riemann-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: github_changelog_generator
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-  type: :runtime
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  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'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  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: Utilities which submit events to Riemann.
-email: aphyr@aphyr.com
+description: Collection of utilities which submit events to Riemann,
+email:
+- aphyr@aphyr.com
 executables:
 - riemann-apache-status
 - riemann-bench
@@ -71,14 +120,25 @@ executables:
 - riemann-net
 - riemann-nginx-status
 - riemann-ntp
+- riemann-portcheck
 - riemann-proc
 - riemann-varnish
 - riemann-zookeeper
 extensions: []
 extra_rdoc_files: []
 files:
+- ".docker/Dockerfile"
+- ".docker/publish.sh"
+- ".github/workflows/ci.yml"
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- CHANGELOG.md
+- Gemfile
+- ISSUE_TEMPLATE.md
 - LICENSE
 - README.markdown
+- Rakefile
 - bin/riemann-apache-status
 - bin/riemann-bench
 - bin/riemann-cloudant
@@ -95,14 +155,69 @@ files:
 - bin/riemann-net
 - bin/riemann-nginx-status
 - bin/riemann-ntp
+- bin/riemann-portcheck
 - bin/riemann-proc
 - bin/riemann-varnish
 - bin/riemann-zookeeper
 - lib/riemann/tools.rb
+- lib/riemann/tools/utils.rb
+- lib/riemann/tools/version.rb
+- riemann-tools.gemspec
+- tools/riemann-aws/LICENSE
+- tools/riemann-aws/README.md
+- tools/riemann-aws/Rakefile.rb
+- tools/riemann-aws/bin/riemann-aws-billing
+- tools/riemann-aws/bin/riemann-aws-rds-status
+- tools/riemann-aws/bin/riemann-aws-sqs-status
+- tools/riemann-aws/bin/riemann-aws-status
+- tools/riemann-aws/bin/riemann-elb-metrics
+- tools/riemann-aws/bin/riemann-s3-list
+- tools/riemann-aws/bin/riemann-s3-status
+- tools/riemann-chronos/LICENSE
+- tools/riemann-chronos/README.md
+- tools/riemann-chronos/Rakefile.rb
+- tools/riemann-chronos/bin/riemann-chronos
+- tools/riemann-docker/LICENSE
+- tools/riemann-docker/README.md
+- tools/riemann-docker/Rakefile.rb
+- tools/riemann-docker/bin/riemann-docker
+- tools/riemann-elasticsearch/LICENSE
+- tools/riemann-elasticsearch/README.md
+- tools/riemann-elasticsearch/Rakefile.rb
+- tools/riemann-elasticsearch/bin/riemann-elasticsearch
+- tools/riemann-marathon/LICENSE
+- tools/riemann-marathon/README.md
+- tools/riemann-marathon/Rakefile.rb
+- tools/riemann-marathon/bin/riemann-marathon
+- tools/riemann-mesos/LICENSE
+- tools/riemann-mesos/README.md
+- tools/riemann-mesos/Rakefile.rb
+- tools/riemann-mesos/bin/riemann-mesos
+- tools/riemann-munin/LICENSE
+- tools/riemann-munin/README.md
+- tools/riemann-munin/Rakefile.rb
+- tools/riemann-munin/bin/riemann-munin
+- tools/riemann-rabbitmq/LICENSE
+- tools/riemann-rabbitmq/README.md
+- tools/riemann-rabbitmq/Rakefile.rb
+- tools/riemann-rabbitmq/bin/riemann-rabbitmq
+- tools/riemann-riak/LICENSE
+- tools/riemann-riak/README.md
+- tools/riemann-riak/Rakefile.rb
+- tools/riemann-riak/bin/riemann-riak
+- tools/riemann-riak/bin/riemann-riak-keys
+- tools/riemann-riak/bin/riemann-riak-ring
+- tools/riemann-riak/riak_status/key_count.erl
+- tools/riemann-riak/riak_status/riak_status.rb
+- tools/riemann-riak/riak_status/ringready.erl
 homepage: https://github.com/aphyr/riemann-tools
 licenses:
 - MIT
-metadata: {}
+metadata:
+  allowed_push_host: https://rubygems.org/
+  homepage_uri: https://github.com/aphyr/riemann-tools
+  source_code_uri: https://github.com/aphyr/riemann-tools
+  changelog_uri: https://github.com/aphyr/riemann-tools
 post_install_message: 
 rdoc_options: []
 require_paths:
@@ -111,15 +226,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 1.8.7
+      version: 2.5.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: riemann-tools
-rubygems_version: 2.4.5
+rubygems_version: 3.2.5
 signing_key: 
 specification_version: 4
 summary: Utilities which submit events to Riemann.