burden 0.1.0

data/lib/burden/storage/mongoid/run.rb

@@ -0,0 +1,18 @@
+module Burden
+  module Storage
+    module Mongoid
+      class Run
+        include ::Mongoid::Document
+
+        store_in collection: 'burden_runs'
+        field :name, type: String
+        field :success, type: Boolean
+        field :execution_time, type: Float
+        field :timestamp, type: Time
+
+        scope :summary, ->{ sort(timestamp: -1).group_by(&:name) }
+        scope :history, ->(task_name){ where(name: task_name).sort(timestamp: -1).group_by(&:name) }
+      end
+    end
+  end
+end

data/lib/burden/storage.rb

@@ -0,0 +1,21 @@
+module Burden
+  module Storage
+    def run
+      case Burden.config.storage
+      when :active_record, :activerecord
+        require 'burden/storage/active_record/run'
+        Burden::Storage::ActiveRecord::Run
+      when :mongoid
+        require 'burden/storage/mongoid/run'
+        Burden::Storage::Mongoid::Run
+      when :mongo_mapper, :mongomapper
+        require 'burden/storage/mongo_mapper/run'
+        Burden::Storage::MongoMapper::Run
+      else
+        raise Exception.new("Unknown storage: #{storage}")
+      end
+    end
+
+    extend self
+  end
+end

data/lib/burden/version.rb

@@ -0,0 +1,3 @@
+module Burden
+  VERSION = '0.1.0'
+end

data/lib/burden/wrapper.rb

@@ -0,0 +1,58 @@
+module Burden
+  class Wrapper
+    attr_reader :name, :block, :success, :exception, :execution_time, :timestamp
+
+    def initialize(name, block)
+      @name = name
+      @block = block
+      @success = true
+      @timestamp = Time.now.utc
+    end
+
+    def execute
+      return block.call if ignored?
+
+      result = measure_time do
+        intercept_exceptions do
+          block.call
+        end
+      end
+      save_statistics
+
+      unless success
+        Burden.config.trigger_failure_callback(name, execution_time, timestamp)
+        raise(exception)
+      end
+
+      Burden.config.trigger_success_callback(name, execution_time, timestamp)
+      result
+    end
+
+  private
+
+    def measure_time
+      t = Time.now
+      result = yield
+      @execution_time = Time.now.to_f - t.to_f
+      result
+    end
+
+    def intercept_exceptions
+      begin
+        yield
+      rescue => e
+        @success = false
+        @exception = e
+        nil
+      end
+    end
+
+    def ignored?
+      Burden.config.ignored_tasks.collect{ |r| (name =~ r) ? 1 : 0 }.map(&:to_i).inject(:+) != 0
+    end
+
+    def save_statistics
+      Statistics.new(name: name, success: success, execution_time: execution_time, timestamp: timestamp).save
+    end
+  end
+end

data/lib/burden.rb

@@ -0,0 +1,19 @@
+require 'bundler/setup'
+require 'rake'
+require File.expand_path('../rake/task', __FILE__)
+
+require 'burden/config'
+require 'burden/statistics'
+require 'burden/storage'
+require 'burden/version'
+require 'burden/wrapper'
+
+module Burden
+  include Config::Helper
+
+  def wrap(name, &block)
+    Wrapper.new(name, block).execute
+  end
+
+  extend self
+end

data/lib/rails/generators/burden/install/USAGE

@@ -0,0 +1,9 @@
+Description:
+    Generates Burden migration and initializer
+
+Example:
+    rails generate burden:install
+
+    This will create:
+        config/initializers/burden.rb
+        db/migrate/create_burden_runs.rb

data/lib/rails/generators/burden/install/install_generator.rb

@@ -0,0 +1,24 @@
+require 'rails/generators'
+require 'rails/generators/migration'
+
+module Burden
+  module Generators
+    class InstallGenerator < Rails::Generators::Base
+      include Rails::Generators::Migration
+
+      source_root File.expand_path('../templates', __FILE__)
+
+      def self.next_migration_number(path)
+        Time.now.utc.strftime('%Y%m%d%H%M%S')
+      end
+
+      def create_migration
+        migration_template 'burden_runs_migration.rb', 'db/migrate/create_burden_runs.rb'
+      end
+
+      def create_initializer
+        template 'burden_initializer.rb', 'config/initializers/burden.rb'
+      end
+    end
+  end
+end

data/lib/rails/generators/burden/install/templates/burden_initializer.rb

@@ -0,0 +1,4 @@
+Burden.configure do |c|
+  c.storage = :active_record
+  c.ignored_tasks = [/environment/]
+end

data/lib/rails/generators/burden/install/templates/burden_runs_migration.rb

@@ -0,0 +1,14 @@
+class CreateBurdenRuns < ActiveRecord::Migration
+  def self.up
+    create_table :burden_runs do |t|
+      t.string :name
+      t.boolean :success
+      t.float :execution_time
+      t.datetime :timestamp
+    end
+  end
+
+  def self.down
+    drop_table :burden_runs
+  end
+end

data/lib/rake/task.rb

@@ -0,0 +1,9 @@
+module Rake
+  class Task
+    alias_method :original_execute, :execute
+
+    def execute(args)
+      Burden.wrap(name){ original_execute(args) }
+    end
+  end
+end

data/spec/burden/wrapper_spec.rb

@@ -0,0 +1,12 @@
+require 'spec_helper'
+
+describe Burden::Wrapper do
+  before :all do
+    Rake::Task.define_task('dummy_task'){}
+    Rake::Task['dummy_task'].invoke
+  end
+
+  it 'should work' do
+    'YAY!'
+  end
+end

data/spec/passing_spec.rb

@@ -0,0 +1,7 @@
+require 'spec_helper'
+
+describe 'nothing' do
+  it 'goes wrong' do
+    true.should be_true
+  end
+end

data/spec/rake/task_spec.rb

@@ -0,0 +1,14 @@
+require 'spec_helper'
+require 'ostruct'
+
+describe Rake::Task do
+  describe '#execute' do
+    let(:dummy_task){ Rake::Task.new('dummy', OpenStruct.new) }
+
+    context 'for dummy task' do
+      subject{ dummy_task }
+      its(:methods){ should include :execute }
+      its(:methods){ should include :original_execute }
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,7 @@
+require 'burden'
+
+RSpec.configure do |config|
+  config.formatter = :progress
+  config.color_enabled = true
+  config.tty = true
+end

metadata

@@ -0,0 +1,231 @@
+--- !ruby/object:Gem::Specification
+name: burden
+version: !ruby/object:Gem::Version
+  prerelease: 
+  version: 0.1.0
+platform: ruby
+authors:
+- Gregory Eremin
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-06-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  type: :runtime
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  type: :development
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  type: :development
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  type: :development
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rails
+  type: :development
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '3.1'
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '3.1'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  type: :development
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '2.0'
+description: Rake tasks manager and statistics collector
+email:
+- magnolia_fan@me.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- burden.gemspec
+- burden_web/.gitignore
+- burden_web/Gemfile
+- burden_web/LICENSE
+- burden_web/README.md
+- burden_web/Rakefile
+- burden_web/app/assets/javascripts/burden_web/application.js
+- burden_web/app/assets/stylesheets/burden_web/application.css
+- burden_web/app/assets/stylesheets/burden_web/foundation.min.css
+- burden_web/app/assets/stylesheets/burden_web/normalize.css
+- burden_web/app/controllers/burden_web/application_controller.rb
+- burden_web/app/controllers/burden_web/runs_controller.rb
+- burden_web/app/controllers/burden_web/tasks_controller.rb
+- burden_web/app/helpers/burden_web/application_helper.rb
+- burden_web/app/views/burden_web/runs/index.html.erb
+- burden_web/app/views/burden_web/tasks/index.html.erb
+- burden_web/app/views/layouts/burden_web/application.html.erb
+- burden_web/burden_web.gemspec
+- burden_web/config/routes.rb
+- burden_web/lib/burden_web.rb
+- burden_web/lib/burden_web/engine.rb
+- burden_web/lib/burden_web/version.rb
+- burden_web/script/rails
+- burden_web/spec/dummy/Rakefile
+- burden_web/spec/dummy/app/assets/javascripts/application.js
+- burden_web/spec/dummy/app/assets/stylesheets/application.css
+- burden_web/spec/dummy/app/controllers/application_controller.rb
+- burden_web/spec/dummy/app/helpers/application_helper.rb
+- burden_web/spec/dummy/app/mailers/.gitkeep
+- burden_web/spec/dummy/app/models/.gitkeep
+- burden_web/spec/dummy/app/views/layouts/application.html.erb
+- burden_web/spec/dummy/config.ru
+- burden_web/spec/dummy/config/application.rb
+- burden_web/spec/dummy/config/boot.rb
+- burden_web/spec/dummy/config/database.yml
+- burden_web/spec/dummy/config/environment.rb
+- burden_web/spec/dummy/config/environments/development.rb
+- burden_web/spec/dummy/config/environments/production.rb
+- burden_web/spec/dummy/config/environments/test.rb
+- burden_web/spec/dummy/config/initializers/backtrace_silencers.rb
+- burden_web/spec/dummy/config/initializers/burden.rb
+- burden_web/spec/dummy/config/initializers/inflections.rb
+- burden_web/spec/dummy/config/initializers/mime_types.rb
+- burden_web/spec/dummy/config/initializers/secret_token.rb
+- burden_web/spec/dummy/config/initializers/session_store.rb
+- burden_web/spec/dummy/config/initializers/wrap_parameters.rb
+- burden_web/spec/dummy/config/locales/en.yml
+- burden_web/spec/dummy/config/routes.rb
+- burden_web/spec/dummy/db/migrate/20130607082646_create_burden_runs.rb
+- burden_web/spec/dummy/db/schema.rb
+- burden_web/spec/dummy/lib/assets/.gitkeep
+- burden_web/spec/dummy/lib/tasks/bad_failure.rake
+- burden_web/spec/dummy/lib/tasks/good_bar.rake
+- burden_web/spec/dummy/lib/tasks/good_foo.rake
+- burden_web/spec/dummy/lib/tasks/unpredictable.rake
+- burden_web/spec/dummy/log/.gitkeep
+- burden_web/spec/dummy/public/404.html
+- burden_web/spec/dummy/public/422.html
+- burden_web/spec/dummy/public/500.html
+- burden_web/spec/dummy/public/favicon.ico
+- burden_web/spec/dummy/script/rails
+- lib/burden.rb
+- lib/burden/config.rb
+- lib/burden/statistics.rb
+- lib/burden/storage.rb
+- lib/burden/storage/abstract/run.rb
+- lib/burden/storage/active_record/run.rb
+- lib/burden/storage/mongo_mapper/run.rb
+- lib/burden/storage/mongoid/run.rb
+- lib/burden/version.rb
+- lib/burden/wrapper.rb
+- lib/rails/generators/burden/install/USAGE
+- lib/rails/generators/burden/install/install_generator.rb
+- lib/rails/generators/burden/install/templates/burden_initializer.rb
+- lib/rails/generators/burden/install/templates/burden_runs_migration.rb
+- lib/rake/task.rb
+- spec/burden/wrapper_spec.rb
+- spec/passing_spec.rb
+- spec/rake/task_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/magnolia-fan/burden
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Rake tasks manager and statistics collector
+test_files:
+- spec/burden/wrapper_spec.rb
+- spec/passing_spec.rb
+- spec/rake/task_spec.rb
+- spec/spec_helper.rb