cfcm 0.0.3 → 0.0.4

data/.gitignore

@@ -9,6 +9,7 @@ pkg
 rdoc
 spec/reports
 test/tmp
+test/input
 test/version_tmp
 tmp
 

data/Gemfile

@@ -1,2 +1,3 @@
 gem 'vmc', "0.4.0.beta.36"
-gem 'eventmachine', "0.12.10"
+gem 'eventmachine', "0.12.10"
+gem 'rbvmomi', "1.5.1"

data/Gemfile.lock

@@ -1,6 +1,7 @@
 GEM
   specs:
     addressable (2.2.6)
+    builder (3.0.0)
     cfoundry (0.3.23)
       multi_json (~> 1.3.6)
       rest-client (~> 1.6.1)
@@ -14,8 +15,13 @@ GEM
     mime-types (1.19)
     mothership (0.0.14)
     multi_json (1.3.6)
+    nokogiri (1.5.5)
     rake (0.9.2.2)
     rb-readline (0.4.2)
+    rbvmomi (1.5.1)
+      builder
+      nokogiri (>= 1.4.1)
+      trollop
     rest-client (1.6.7)
       mime-types (>= 1.16)
     rspec (1.3.2)
@@ -25,6 +31,7 @@ GEM
       simplecov-html (~> 0.5.3)
     simplecov-html (0.5.3)
     terminal-table (1.4.5)
+    trollop (2.0)
     uuidtools (2.1.3)
     vmc (0.4.0.beta.36)
       addressable (~> 2.2.6)
@@ -56,4 +63,5 @@ PLATFORMS
 
 DEPENDENCIES
   eventmachine (= 0.12.10)
+  rbvmomi (= 1.5.1)
   vmc (= 0.4.0.beta.36)

data/README.md

@@ -1,4 +1,17 @@
 cloudfoundry-chaos-monkey
 =========================
 
-Chaos Monkey style service for Cloud Foundry
+Chaos Monkey style service for Cloud Foundry
+
+Using Cloud Foundry Chaos Monkey
+--------------------------------
+
+First off, grab the Cloud Foundry Chaos Monkey gem
+
+`gem install cfcm`
+
+and then unleash the monkey! In this example, we'll use CFCM on an application named `myapp`, keeping between 2 and 5 instances. We want CFCM to have a 30% chance of adding or removing an instance every 10 seconds. Adding and removeing instances is known as "soft mode", that is, this is equvilant to doing "vmc instane [app] -1". There is a "hard mode" planned that is a bit more chaotic.
+
+`cfcm soft [email] [password] myapp --min 2 --max 5 -p 30 -f 10`
+
+It's as easy as that!

data/lib/cfcm/cli.rb

@@ -6,11 +6,11 @@ module CFCM
     class Command < Thor
       
       desc "soft USERNAME PASSWORD APP", "Removes/Adds instances via VMC APIs"
-      method_option :target, :aliases => ["--target", "-t"], :type => :string, :desc => "Cloud Foundry API Target URL"
-      method_option :min, :alias => ["--min"], :type => :numeric, :desc => "Minimum number of instances"
-      method_option :max, :alias => ["--max"], :type => :numeric, :desc => "Maximum number of instances"
-      method_option :probability, :aliases => ["--probability", "-p"], :type => :numeric, :desc => "[1-100] Probablity for the Chaos Monkey to add/remove an instance"
-      method_option :frequency, :aliases => ["--frequency", "-f"], :type => :numeric, :desc => "Number of times to attempt to add/remove an instance"
+      method_option :target, :aliases => ["--target", "-t"], :type => :string, :desc => "Cloud Foundry API Target URL (Default: http://api.cloudfoundry.com)"
+      method_option :min, :alias => ["--min"], :type => :numeric, :desc => "Minimum number of instances (Default: 1)"
+      method_option :max, :alias => ["--max"], :type => :numeric, :desc => "Maximum number of instances (Default: 5)"
+      method_option :probability, :aliases => ["--probability", "-p"], :type => :numeric, :desc => "[1-100] Probablity for the Chaos Monkey to add/remove an instance (Default: 10)"
+      method_option :frequency, :aliases => ["--frequency", "-f"], :type => :numeric, :desc => "Number of seconds between attempts to add/remove an instance (Default: 10)"
       def soft(username, password, app, target = "http://api.cloudfoundry.com", min = 1, max = 5, probability = 10, frequency = 10)
         
         if options[:target]
@@ -50,6 +50,34 @@ module CFCM
         monkey = CFCM::Monkey::SoftMonkey.new(session, app, probability, min, max, frequency)
         monkey.start
       end     
+      
+      
+      desc "hard", "Removes/Adds DEAs through IaaS commands"
+      method_option :iaas, :alias => ["--iaas"], :type => :string, :desc => "IaaS to communicate with [VSPHERE]"
+      method_option :input, :aliases => ["--input", "-i"], :type => :string, :desc => "Path to file listing valid DEAs (See Documentation)"
+      method_option :config, :aliases => ["--config", "-c"], :type => :string, :desc => "Path to IaaS config YML file (See Documentation)"
+      method_option :probability, :aliases => ["--probability", "-p"], :type => :numeric, :desc => "[1-100] Probablity for the Chaos Monkey to add/remove an instance (Default: 10)"
+      method_option :frequency, :aliases => ["--frequency", "-f"], :type => :numeric, :desc => "Number of seconds between attempts to add/remove an instance (Default: 10)"
+      def hard(probability = 10, frequency = 10)
+        if (!options[:iaas] || !options[:input] || !options[:config])
+          CFCM::Monkey::HardMonkey.new.show_help
+        else
+          if options[:probability]
+            if options[:probability] < 1 || options[:probability] > 100
+              puts "Probability must be between 1 and 100"
+              return
+            else
+              probability = options[:probability]
+            end
+          end
+          if options[:frequency]
+            frequency = options[:frequency]
+          end
+          
+          monkey = CFCM::Monkey::HardMonkey.new(options[:iaas], options[:config], options[:input], probability, frequency)
+          monkey.start
+        end
+      end
     end
   end
 end

data/lib/cfcm/iaas/iaas.rb

@@ -0,0 +1,20 @@
+require 'rbvmomi'
+
+module CFCM
+  module IAAS
+    class IaaS
+        
+      def initalize(host, user, password, config)
+        @host = host
+        @user = user
+        @password = password
+        @config = config
+      end
+      
+      def power_off_vm(vm)
+        puts "Not Implemented"
+      end
+        
+    end
+  end
+end

data/lib/cfcm/iaas/vsphere.rb

@@ -0,0 +1,24 @@
+require 'rbvmomi'
+
+module CFCM
+  module IAAS
+    class Vsphere
+        
+      def initialize(host, user, password, config)
+        @host = host
+        @user = user
+        @password = password
+        @config = config
+      end
+      
+      def power_off_vm(vm_name)
+        opts = {:host=>@host, :user=>@user, :password=>@password, :insecure=>true}
+        vim = RbVmomi::VIM.connect opts
+        dc = vim.serviceInstance.content.rootFolder.traverse(config["datacenter"], RbVmomi::VIM::Datacenter) or abort "datacenter not found"
+        vm = dc.vmFolder.traverse(vm_name, VIM::VirtualMachine) or abort "VM not found"
+        vm.PowerOffVM_Task.wait_for_completion
+      end
+        
+    end
+  end
+end

data/lib/cfcm/monkey.rb

@@ -1,4 +1,7 @@
-require "eventmachine"
+require 'eventmachine'
+require 'yaml'
+require 'cfcm/iaas/iaas'
+require 'cfcm/iaas/vsphere'
 
 module CFCM
   module Monkey
@@ -50,5 +53,48 @@ module CFCM
         end
       end
     end
+    
+    
+    class HardMonkey
+      def show_help
+        puts "Description coming soon"
+      end
+      
+      def initialize(iaas, config_file, input_file, probability, frequency)
+        
+        @probability = probability
+        @frequency = frequency
+        
+        # Parse the config file
+        @config = YAML.load_file(config_file)
+        
+        # Parse the input file
+        input_fd = File.open(input_file, "rb")
+        @input = input_fd.read.split
+        
+        # Build the IaaS interface
+        @iaas_interface = nil          
+        case iaas.downcase
+        when "vsphere"
+          @iaas_interface = CFCM::IAAS::Vsphere.new(@config["host"], @config["user"], @config["password"], @config)
+        else
+          puts "Unknown IaaS -- #{iaas}"
+        end
+      end
+      
+      def start
+        # Start the Eventmachine loop, similar to above and shut down VMs at random
+        EventMachine.run do
+          EventMachine.add_periodic_timer(@frequency) do
+            # Determine if we should unleash the monkey
+            if (Random.rand(100) + 1) <= @probability
+              vm = @input.sample
+              puts "Sending Power Off to #{vm}"
+              @iaas_interface.power_off_vm(vm)
+            end
+          end
+        end
+      end
+    end
   end
 end

data/lib/cfcm/version.rb

@@ -1,3 +1,3 @@
 module CFCM
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cfcm
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-31 00:00:00.000000000 Z
+date: 2012-09-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -60,6 +60,8 @@ files:
 - cfcm.gemspec
 - lib/cfcm/cf.rb
 - lib/cfcm/cli.rb
+- lib/cfcm/iaas/iaas.rb
+- lib/cfcm/iaas/vsphere.rb
 - lib/cfcm/monkey.rb
 - lib/cfcm/version.rb
 homepage: https://github.com/BrianMMcClain/cloudfoundry-chaos-monkey