hadouken 0.1.4.pre

data/lib/hadouken/task.rb

@@ -0,0 +1,58 @@
+module Hadouken::Task
+  class Base
+    attr_reader :plan
+    attr_reader :group_name
+
+    def initialize(opts)
+      @group_name = opts[:group] 
+      @plan       = opts[:plan ]
+    end
+
+    def group?
+      !! @group_name
+    end
+
+    def group
+      @plan.group(@group_name)
+    end
+
+    def self.create!(instance, opts)
+      return case instance
+        # autovivify
+        when String then Hadouken::Task::Command.new  instance, opts
+        when Array  then Hadouken::Task::Command.new  instance, opts
+        when Proc   then Hadouken::Task::Callback.new instance, opts
+
+        # go with it
+        when Hadouken::Strategy::Base then instance
+        when Hadouken::Task::Base     then instance
+
+        # no chance
+        else raise ArgumentError
+      end
+    end
+  end
+
+  class Command < Base
+    def initialize(command, opts)
+      @command = command
+      super(opts)
+    end
+  
+    # TODO: sanitize command so it has a chance of working
+    def command
+      @command
+    end
+  end  
+
+  class Callback < Base
+    def initialize(fn, opts)
+      @proc = fn
+      super opts
+    end
+  
+    def call(opts)
+      @proc.call(*opts)
+    end
+  end
+end

data/lib/hadouken/tasks.rb

@@ -0,0 +1,17 @@
+class Hadouken::Tasks
+  include Enumerable
+
+  def initialize
+    @tasks = []
+  end
+
+  def each
+    @tasks.each do |task|
+      yield task
+    end
+  end
+
+  def add(task, opts={})
+    @tasks << Hadouken::Task::Base.create!(task, opts)
+  end
+end

data/test/helper.rb

@@ -0,0 +1,21 @@
+require 'rubygems'
+require 'bundler'
+
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+require 'shoulda'
+require 'mocha'
+require 'pp'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'hadouken'
+
+class Test::Unit::TestCase
+end

data/test/test_hadouken.rb

@@ -0,0 +1,118 @@
+require 'helper'
+
+class TestHadouken < Test::Unit::TestCase
+
+  context "yeah" do
+    setup do 
+      Net::SSH::Multi::Session.any_instance.stubs(:exec).returns(true)
+      Net::SSH::Multi::Subsession.any_instance.stubs(:exec).returns(true)
+    end
+
+    should "create a group" do 
+      g = Hadouken::Group.new(:name => :pony, :range => (1..5), :pattern => "pony-%05d")
+      assert_equal g.name,    :pony
+      assert_equal g.range,   (1..5)
+      assert_equal g.pattern, "pony-%05d"
+      assert_equal g.size,    5
+      assert_equal g.hosts,   g.range.map{|i| g.pattern % i}
+    end
+
+    should "create a collection of groups" do
+      plan = Hadouken::Plan.new
+      plan.verbose = true
+      plan.add_group Hadouken::Group.new(:name => :cat, :range => (1..3), :pattern => "cat-%05d")
+      plan.add_group Hadouken::Group.new(:name => :dog, :range => (1..5), :pattern => "dog-%05d")
+    
+      assert_equal 8, plan.groups.hosts.size
+      assert_equal plan.groups.fetch(:cat), plan.groups[:cat]
+      assert_equal 'cat-00003', plan.groups.fetch(:cat).hosts.last
+      assert_equal 'dog-00005', plan.groups.fetch(:dog).hosts.last
+    end
+
+    should "use by-host strategy" do
+      plan = Hadouken::Plan.new
+      plan.verbose = true
+      plan.add_group Hadouken::Group.new(:name => :cat, :range => (1..3), :pattern => "cat-%05d")
+      plan.add_group Hadouken::Group.new(:name => :dog, :range => (1..3), :pattern => "dog-%05d")
+
+      cats  = (1..3).map{|i| "cat-%05d" % i}
+      dogs  = (1..3).map{|i| "dog-%05d" % i}
+      crazy = [ cats + dogs ]
+
+      servers = Hadouken::Strategy::ByHost.new(plan).host_strategy
+      assert_equal crazy, servers
+    end
+
+    should "use by-host strategy with two at a time" do
+      plan = Hadouken::Plan.new
+      plan.verbose = true
+      plan.add_group Hadouken::Group.new(:name => :cat, :range => (1..3), :pattern => "cat-%05d")
+      plan.add_group Hadouken::Group.new(:name => :dog, :range => (1..3), :pattern => "dog-%05d")
+
+      cats  = (1..3).map{|i| "cat-%05d" % i}
+      dogs  = (1..3).map{|i| "dog-%05d" % i}
+      crazy = []
+      (cats + dogs).each_slice(2) do |s|
+        crazy << s
+      end
+
+      servers = Hadouken::Strategy::ByHost.new(plan, :max_hosts => 2).host_strategy
+      assert_equal crazy, servers
+    end
+
+
+    should "use by-group strategy with two at a time" do
+      plan = Hadouken::Plan.new
+      plan.verbose = true
+      plan.add_group Hadouken::Group.new(:name => :cat, :range => (1..3), :pattern => "cat-%05d")
+      plan.add_group Hadouken::Group.new(:name => :dog, :range => (1..3), :pattern => "dog-%05d")
+
+      cats = (1..3).map{|i| "cat-%05d" % i}
+      dogs = (1..3).map{|i| "dog-%05d" % i}
+
+      servers = Hadouken::Strategy::ByGroup.new(plan, :max_hosts => 2).host_strategy
+      assert_equal cats[0..1], servers[0]
+      assert_equal [cats[2]],  servers[1]
+      assert_equal dogs[0..1], servers[2]
+      assert_equal [dogs[2]],  servers[3]
+    end
+
+    
+    should "use by-group-parallel strategy" do
+      plan = Hadouken::Plan.new
+      plan.verbose = true
+      plan.name = :pet_store
+      plan.add_group Hadouken::Group.new(:name => :cat, :range => (1..3), :pattern => "cat-%05d")
+      plan.add_group Hadouken::Group.new(:name => :dog, :range => (1..3), :pattern => "dog-%05d")
+      plan.add_group Hadouken::Group.new(:name => :hog, :range => (1..3), :pattern => "hog-%05d")
+
+      cats  = (1..3).map{|i| "cat-%05d" % i}
+      dogs  = (1..3).map{|i| "dog-%05d" % i}
+      hogs  = (1..3).map{|i| "hog-%05d" % i}
+
+      crazy  = []
+      crazy << [ cats[0], dogs[0], hogs[0] ]
+      crazy << [ cats[1], dogs[1], hogs[1] ]
+      crazy << [ cats[2], dogs[2], hogs[2] ]
+
+      plan.tasks.add Hadouken::Strategy::ByHost.new(plan)
+      plan.tasks.add Hadouken::Task::Command.new("aaa", :plan => plan)
+      plan.tasks.add "bbb", :plan => plan
+      plan.tasks.add Proc.new{|host|
+      }, :plan => plan
+
+      plan.tasks.add Hadouken::Strategy::ByGroupParallel.new(plan, :max_hosts => 4, :traversal => :depth)
+      plan.tasks.add Hadouken::Task::Command.new("doggie", :plan => plan, :group_name => :dog)
+      plan.tasks.add Hadouken::Task::Command.new("kitty",  :plan => plan, :group_name => :cat)
+      plan.tasks.add Hadouken::Task::Command.new("pig",    :plan => plan, :group_name => :hog)
+      plan.tasks.add Proc.new{|host|
+        puts "**************** #{host} ******************"
+      }, :plan => plan
+
+      Hadouken::Executor.run!(plan)
+    end
+
+
+
+  end
+end

metadata

@@ -0,0 +1,164 @@
+--- !ruby/object:Gem::Specification
+name: hadouken
+version: !ruby/object:Gem::Version
+  version: 0.1.4.pre
+  prerelease: 6
+platform: ruby
+authors:
+- Matt Knopp
+- Chris Gray
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-01-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: &70333320089840 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70333320089840
+- !ruby/object:Gem::Dependency
+  name: net-ssh-multi
+  requirement: &70333320089100 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70333320089100
+- !ruby/object:Gem::Dependency
+  name: yajl-ruby
+  requirement: &70333320088360 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70333320088360
+- !ruby/object:Gem::Dependency
+  name: shoulda
+  requirement: &70333320087540 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70333320087540
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: &70333320086600 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70333320086600
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: &70333320085440 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :development
+  prerelease: false
+  version_requirements: *70333320085440
+- !ruby/object:Gem::Dependency
+  name: jeweler
+  requirement: &70333320084320 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.6.4
+  type: :development
+  prerelease: false
+  version_requirements: *70333320084320
+- !ruby/object:Gem::Dependency
+  name: rcov
+  requirement: &70333320098940 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70333320098940
+description: run commands over ssh in a way that makes sense for deploying artifacts
+email:
+- mknopp@yammer-inc.com, cgray@yammer-inc.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- LICENSE.txt
+- README.md
+files:
+- .document
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- VERSION
+- hadouken.gemspec
+- lib/hadouken.rb
+- lib/hadouken/executor.rb
+- lib/hadouken/ext/net_ssh_multi_session_actions.rb
+- lib/hadouken/group.rb
+- lib/hadouken/groups.rb
+- lib/hadouken/host.rb
+- lib/hadouken/plan.rb
+- lib/hadouken/runner.rb
+- lib/hadouken/strategy/base.rb
+- lib/hadouken/strategy/by_group.rb
+- lib/hadouken/strategy/by_group_parallel.rb
+- lib/hadouken/strategy/by_host.rb
+- lib/hadouken/task.rb
+- lib/hadouken/tasks.rb
+- test/helper.rb
+- test/test_hadouken.rb
+homepage: http://github.com/mhat/hadouken
+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'
+      segments:
+      - 0
+      hash: -2045406218975922941
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>'
+    - !ruby/object:Gem::Version
+      version: 1.3.1
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: run commands over ssh in a way that makes sense for deploying artifacts
+test_files: []