health_status 0.0.1

data/.gitignore

@@ -0,0 +1,20 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+.*.swp
+vendor/bundle
+*.db

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in health_status.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2012 riywo
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,32 @@
+# HealthStatus
+
+API server to store and visualize applications' health status.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'health_status'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install health_status
+
+## Usage
+
+    $ health_status_server -d /path/to/sqlite.db
+
+    $ curl -v -d "status=1" localhost:5678/service1/application1/metric1
+    $ curl -v -d "status=2" localhost:5678/service1/application1/metric2
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,3 @@
+require "bundler/gem_tasks"
+
+require 'sinatra/activerecord/rake'

data/bin/health_status_server

@@ -0,0 +1,17 @@
+#!/usr/bin/env ruby
+$:.unshift File.expand_path("../../lib", __FILE__)
+begin
+  require 'vegas'
+rescue LoadError
+  require 'rubygems'
+  require 'vegas'
+end
+require "health_status/app"
+require "health_status/model"
+
+Vegas::Runner.new(HealthStatus::App, 'health_status_server') do |runner, opts, app|
+  opts.on("-d", "--database DATABASE_FILE", "path/to/sqlite.db") do |database|
+    runner.logger.info "Using database #{database}"
+    HealthStatus::App.database_path = database
+  end
+end

data/db/migrate/20121130030246_health_status_migration.rb

@@ -0,0 +1,22 @@
+class HealthStatusMigration < ActiveRecord::Migration
+  def up
+    create_table :applications do |t|
+      t.string     :name,             :null => false
+      t.integer    :status,           :null => false
+      t.datetime   :datetime,         :null => false
+      t.datetime   :saved_at,         :null => false
+    end
+    add_index :applications, [ :name ], :unique => true
+
+    create_table :half_hour_statuses do |t|
+      t.integer    :application_id,   :null => false
+      t.integer    :status,           :null => false
+      t.datetime   :datetime,         :null => false
+      t.datetime   :saved_at,         :null => false
+    end
+    add_index :half_hour_statuses, [:application_id]
+  end
+
+  def down
+  end
+end

data/db/migrate/20121204050629_remove_datetime.rb

@@ -0,0 +1,9 @@
+class RemoveDatetime < ActiveRecord::Migration
+  def up
+    remove_column(:applications, :datetime)
+  end
+
+  def down
+    add_column(:applications, :datetime, :datetime, { :null => false })
+  end
+end

data/db/migrate/20121204171946_add_service_metric.rb

@@ -0,0 +1,64 @@
+class AddServiceMetric < ActiveRecord::Migration
+  def up
+    rename_table :applications, :v1_applications
+    rename_table :half_hour_statuses, :v1_half_hour_statuses
+
+    create_table :services do |t|
+      t.string     :name,             :null => false
+      t.integer    :status,           :null => false
+      t.datetime   :saved_at,         :null => false
+    end
+    add_index :services, [ :name ], :unique => true
+
+    create_table :service_half_hour_statuses do |t|
+      t.integer    :service_id,       :null => false
+      t.integer    :status,           :null => false
+      t.datetime   :datetime,         :null => false
+      t.datetime   :saved_at,         :null => false
+    end
+    add_index :service_half_hour_statuses, [ :service_id ]
+
+    create_table :applications do |t|
+      t.integer    :service_id,       :null => false
+      t.string     :name,             :null => false
+      t.integer    :status,           :null => false
+      t.datetime   :saved_at,         :null => false
+    end
+    add_index :applications, [ :service_id, :name ], :unique => true
+
+    create_table :application_half_hour_statuses do |t|
+      t.integer    :application_id,   :null => false
+      t.integer    :status,           :null => false
+      t.datetime   :datetime,         :null => false
+      t.datetime   :saved_at,         :null => false
+    end
+    add_index :application_half_hour_statuses, [ :application_id ]
+
+    create_table :metrics do |t|
+      t.integer    :application_id,   :null => false
+      t.string     :name,             :null => false
+      t.integer    :status,           :null => false
+      t.datetime   :saved_at,         :null => false
+    end
+    add_index :metrics, [ :application_id, :name ], :unique => true
+
+    create_table :metric_half_hour_statuses do |t|
+      t.integer    :metric_id,        :null => false
+      t.integer    :status,           :null => false
+      t.datetime   :datetime,         :null => false
+      t.datetime   :saved_at,         :null => false
+    end
+    add_index :metric_half_hour_statuses, [ :metric_id ]
+  end
+
+  def down
+    drop_table :services
+    drop_table :service_half_hour_statuses
+    drop_table :applications
+    drop_table :application_half_hour_statuses
+    drop_table :metrics
+    drop_table :metric_half_hour_statuses
+    rename_table :v1_applications, :applications
+    rename_table :v1_half_hour_statuses, :half_hour_statuses
+  end
+end

data/health_status.gemspec

@@ -0,0 +1,31 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'health_status/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "health_status"
+  spec.version       = HealthStatus::VERSION
+  spec.authors       = ["Ryosuke IWANAGA"]
+  spec.email         = ["riywo.jp@gmail.com"]
+  spec.description   = %q{Health status API server}
+  spec.summary       = %q{API server to store and visualize applications' health status}
+  spec.homepage      = "https://github.com/riywo/health_status"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency 'sinatra'
+  spec.add_dependency 'sinatra-contrib'
+  spec.add_dependency 'activerecord'
+  spec.add_dependency 'sinatra-activerecord'
+  spec.add_dependency 'sqlite3'
+  spec.add_dependency 'vegas'
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency 'tapp'
+  spec.add_development_dependency 'rake'
+end

data/lib/health_status.rb

@@ -0,0 +1,5 @@
+require "health_status/version"
+
+module HealthStatus
+  # Your code goes here...
+end

data/lib/health_status/app.rb

@@ -0,0 +1,83 @@
+require "health_status"
+require "health_status/model"
+require "health_status/web"
+require "sinatra/base"
+require "sinatra/activerecord"
+require "sinatra/cookies"
+
+class HealthStatus::App < Sinatra::Base
+
+  class << self
+    attr_accessor :database_path
+  end
+
+  helpers  Sinatra::Cookies, ERB::Util
+  register Sinatra::ActiveRecordExtension
+
+  set :public_folder, File.expand_path("../../../public", __FILE__)
+  set :views,         File.expand_path("../../../views", __FILE__)
+
+  before do
+    settings.database = "sqlite3:///#{HealthStatus::App.database_path}"
+    ActiveRecord::Base.logger = nil
+    HealthStatus::Model::Migrate.migrate
+  end
+
+  get '/' do
+    @timezone = params["timezone"] || cookies[:timezone] || HealthStatus::Web.system_timezone
+    @zones  = HealthStatus::Web.timezones
+    erb :index
+  end
+
+  get '/api/v2/' do
+    HealthStatus::Model::Service.readonly.fetch_all_info().to_json
+  end
+
+  get '/api/v2/:service' do |service_name|
+    args = {}
+    args[:end_time] = Time.now.in_time_zone(params["timezone"])
+    args[:with_status] = true
+    service = HealthStatus::Model::Service.find_by_name(service_name)
+    service.fetch_info(args).to_json
+  end
+
+  get '/api/v2/:service/:application' do |service_name, application_name|
+    Time.zone = params["timezone"]
+    args = {}
+    args[:end_time] = Time.now.in_time_zone(params["timezone"])
+    args[:with_status] = true
+    application = HealthStatus::Model::Service.find_by_name(service_name).applications.find_by_name(application_name)
+    application.fetch_info(args).to_json
+  end
+
+  get '/api/v2/:service/:application/:metric' do |service_name, application_name, metric_name|
+    Time.zone = params["timezone"]
+    args = {}
+    args[:end_time] = Time.now.in_time_zone(params["timezone"])
+    args[:with_status] = true
+    metric = HealthStatus::Model::Service.find_by_name(service_name).applications.find_by_name(application_name).metrics.find_by_name(metric_name)
+    metric.fetch_info(args).to_json
+  end
+
+  post '/api/v2/:service/:application/:metric' do |service_name, application_name, metric_name|
+    raise unless params.has_key? "status"
+    HealthStatus::Model::Service.save_metric(service_name, application_name, metric_name, params["status"])
+    "OK"
+  end
+
+  delete '/api/v2/:service' do |service_name|
+    HealthStatus::Model::Service.find_by_name(service_name).destroy
+    "OK"
+  end
+
+  delete '/api/v2/:service/:application' do |service_name, application_name|
+    HealthStatus::Model::Service.find_by_name(service_name).applications.find_by_name(application_name).destroy
+    "OK"
+  end
+
+  delete '/api/v2/:service/:application/:metric' do |service_name, application_name, metric_name|
+    HealthStatus::Model::Service.find_by_name(service_name).applications.find_by_name(application_name).metrics.find_by_name(metric_name).destroy
+    "OK"
+  end
+
+end

data/lib/health_status/model.rb

@@ -0,0 +1,284 @@
+require "health_status"
+require 'sinatra/activerecord'
+require "sinatra/activerecord/rake"
+
+class HealthStatus::Model
+
+  module AggregateStatus
+    @@default_timezone = :utc
+    @@half_hour = 30 * 60
+    @@hour      = 60 * 60
+    @@day       = 24 * @@hour
+
+    def fetch_current_status
+      if saved_at < floor_half_hour(Time.now.utc)
+        nil
+      else
+        status
+      end
+    end
+
+    def fetch_saved_at(args)
+      end_time = args[:end_time]
+      end_time ||= Time.now
+      Time.at(saved_at.to_i).localtime(end_time.utc_offset)
+    end
+
+    def fetch_status_with_interval(interval, args = {})
+      validate_args = validate_fetch_status(interval, args)
+      interval_status = []
+      time = validate_args[:start_time]
+      while time <= validate_args[:end_time]
+        data = {
+          :datetime => time,
+          :status   => self_half_hour_statuses.where(:datetime => time..(time + interval - 1)).maximum(:status),
+        }
+        interval_status << data
+        time += interval
+      end
+      interval_status
+    end
+
+    def fetch_hourly_status(args = {})
+      fetch_status_with_interval(@@hour, args)
+    end
+
+    def fetch_daily_status(args = {})
+      fetch_status_with_interval(@@day, args)
+    end
+
+    def fetch_info(args = {})
+      depth = args[:depth].nil? ? 0 : args[:depth]
+      data = {
+        :id             => id,
+        :name           => name,
+        :current_status => fetch_current_status,
+        :saved_at       => fetch_saved_at(args),
+      }
+
+      if args[:with_status]
+        data[:hourly_status] = fetch_hourly_status(args)
+        data[:daily_status]  = fetch_daily_status(args)
+      end
+
+      if args[:with_id]
+        data[:service_id] = service.id if respond_to? :service
+        if respond_to? :application
+          data[:service_id] = application.service.id
+          data[:application_id] = application.id
+        end
+      end
+
+      if children.respond_to? :map and depth > 0
+        data[children_name] = children.map do |child|
+          args[:depth] = depth - 1
+          child.fetch_info(args)
+        end
+      end
+      data
+    end
+
+    def save_status(new_status)
+      children_status = children.maximum(:status)
+      if !children_status.nil? and children_status <= new_status.to_i
+        self.status = new_status.to_i
+      end
+      save!
+    end
+
+    private
+
+    def validate_fetch_status(interval, args)
+      end_time = args[:end_time]
+      end_time ||= Time.now
+      start_time = args[:start_time]
+      start_time ||= end_time - (interval == @@day ? 7 * @@day : @@day)
+
+      offset = start_time.utc_offset
+      raise if offset != end_time.utc_offset
+      raise if offset % @@half_hour != 0
+
+      start_time = floor(start_time, interval)
+      end_time   = floor(end_time, interval)
+      { :start_time => start_time, :end_time => end_time }
+    end
+
+    def floor_half_hour(datetime)
+      floor(datetime, @@half_hour)
+    end
+
+    def floor_hour(datetime)
+      floor(datetime, @@hour)
+    end
+
+    def floor_day(datetime)
+      floor(datetime, @@day)
+    end
+
+    def floor(datetime, seconds)
+      offset = datetime.utc_offset
+      Time.at(datetime.to_i - ((datetime.to_i + offset) % seconds).floor).localtime(offset)
+    end
+
+    def update_half_hour_status
+      half_hour = floor_half_hour(self.saved_at)
+      current = self.self_half_hour_statuses.find(:all, :conditions => { :datetime => half_hour } ).first
+      if current
+        current.status = status if current.status < status
+        current.saved_at = self.saved_at
+        current.save
+      else
+        self_half_hour_statuses.create(
+          :status   => self.status,
+          :datetime => half_hour,
+          :saved_at => self.saved_at,
+        )
+      end
+    end
+  end
+
+  class Service < ActiveRecord::Base
+    include AggregateStatus
+
+    has_many   :applications, :dependent => :destroy
+    has_many   :service_half_hour_statuses, :order => "datetime ASC", :dependent => :destroy
+
+    before_validation do self.saved_at = Time.now.utc end
+    after_save :update_half_hour_status
+    validates_inclusion_of :status, :in => 1..3
+
+    def self.fetch_all_info(args = {})
+      all.map do |service|
+        args[:depth] = 2
+        service.fetch_info(args)
+      end
+    end
+
+    def self.save_metric(service_name, application_name, metric_name, new_status)
+      ActiveRecord::Base.transaction do
+        service     = find_or_initialize_by_name(service_name)
+        application = service.applications.find_or_initialize_by_name(application_name)
+        metric      = application.metrics.find_or_initialize_by_name(metric_name)
+
+        if service.new_record?
+          service.status = new_status
+          application.status = new_status
+        elsif application.new_record?
+          application.status = new_status
+        end
+        metric.status = new_status
+        service.save!
+        application.save!
+        metric.save!
+
+        application.save_status(new_status)
+        service.save_status(new_status)
+      end
+    end
+
+    def self_half_hour_statuses
+      service_half_hour_statuses
+    end
+
+    def children_name
+      :applications
+    end
+
+    def children
+      applications
+    end
+  end
+
+  class Application < ActiveRecord::Base
+    include AggregateStatus
+
+    belongs_to :service
+    has_many   :metrics, :dependent => :destroy
+    has_many   :application_half_hour_statuses, :order => "datetime ASC", :dependent => :destroy
+
+    before_validation do self.saved_at = Time.now.utc end
+    after_save :update_half_hour_status
+    validates_inclusion_of :status, :in => 1..3
+
+    def self_half_hour_statuses
+      application_half_hour_statuses
+    end
+
+    def children_name
+      :metrics
+    end
+
+    def children
+      metrics
+    end
+  end
+
+  class Metric < ActiveRecord::Base
+    include AggregateStatus
+
+    belongs_to :application
+    has_many   :metric_half_hour_statuses, :order => "datetime ASC", :dependent => :destroy
+
+    before_validation do self.saved_at = Time.now.utc end
+    after_save :update_half_hour_status
+    validates_inclusion_of :status, :in => 1..3
+
+    def self_half_hour_statuses
+      metric_half_hour_statuses
+    end
+
+    def children
+      self
+    end
+
+  end
+
+  class ServiceHalfHourStatus < ActiveRecord::Base
+    @@default_timezone = :utc
+    belongs_to :service
+    validates_inclusion_of :status, :in => 1..3
+  end
+
+  class ApplicationHalfHourStatus < ActiveRecord::Base
+    @@default_timezone = :utc
+    belongs_to :application
+    validates_inclusion_of :status, :in => 1..3
+  end
+
+  class MetricHalfHourStatus < ActiveRecord::Base
+    @@default_timezone = :utc
+    belongs_to :metric
+    validates_inclusion_of :status, :in => 1..3
+  end
+
+  module Sort
+    def self.names_sort_by_status
+      all.sort do |a, b|
+        a_status = a.fetch_current_status
+        b_status = b.fetch_current_status
+        if !a_status and !b_status
+          a.name <=> b.name
+        elsif !a_status
+          1
+        elsif !b_status
+          -1
+        else
+          (b.fetch_current_status <=> a.fetch_current_status).nonzero? or
+          a.name <=> b.name
+        end
+      end.map { |app| app.name }
+    end
+  end
+
+  module Migrate
+    extend Sinatra::ActiveRecordTasks
+    extend self
+
+    private
+
+    def migrations_dir
+      File.expand_path("../../../db/migrate", __FILE__)
+    end
+  end
+
+end