rubix 0.0.6 → 0.0.7

data/VERSION

@@ -1 +1 @@
-0.0.6
+0.0.7

data/lib/rubix/models/item.rb

@@ -63,6 +63,10 @@ module Rubix
       self.application_ids = properties[:application_ids]
     end
 
+    def resource_name
+      "#{self.class.resource_name} #{self.key || self.id}"
+    end
+    
     def self.find_request options={}
       request('item.get', 'hostids' => [options[:host_id]], 'filter' => {'key_' => options[:key], 'id' => options[:id]}, "select_applications" => "refer", "output" => "extend")
     end

data/lib/rubix/sender.rb

@@ -28,51 +28,63 @@ module Rubix
     def initialize settings
       @settings = settings
       confirm_settings
-      self.host = Host.new(:name => settings['host'])
-      if settings['fast']
-        info("Forwarding for #{self.host.name}...") if settings['verbose']
+      if fast?
+        info("Forwarding for #{settings['host']}...") if settings['verbose']
       else
         initialize_hostgroups
         initialize_templates
         initialize_host
         initialize_applications
-        info("Forwarding for #{self.host.name}...") if settings['verbose'] && host.exists?
+        info("Forwarding for #{host.name}...") if settings['verbose']
       end
     end
 
-    def alive?
-      settings['fast'] || host.exists?
+    def fast?
+      settings['fast']
     end
 
+    def auto_vivify?
+      !fast?
+    end
+    
     def initialize_hostgroups
-      self.host_groups = settings['host_groups'].split(',').flatten.compact.map { |group_name | HostGroup.find_or_create_by_name(group_name.strip) }
+      self.host_groups = settings['host_groups'].split(',').flatten.compact.map { |group_name | HostGroup.find_or_create(:name => group_name.strip) }
     end
 
     def initialize_templates
-      self.templates = (settings['templates'] || '').split(',').flatten.compact.map { |template_name | Template.find_or_create_by_name(template_name.strip) }
+      self.templates = (settings['templates'] || '').split(',').flatten.compact.map { |template_name | Template.find(:name => template_name.strip) }.compact
     end
 
     def initialize_host
-      unless host.exists?
-        host.host_groups = host_groups
-        host.templates   = templates
-        host.create
-      end
-      # if settings['verbose']
-      #   puts "Forwarding data for Host '#{settings['host']}' (#{host_id}) from #{settings['pipe']} to #{settings['server']}"
-      #   puts "Creating Items in Application '#{settings['application']}' (#{application_id}) at #{settings['api_server']} as #{settings['username']}"
-      # end
+      self.host = (Host.find(:name => settings['host']) || Host.new(:name => settings['host']))
+      host.host_groups = host_groups
+      host.templates   = templates
+      host.save
     end
 
     def initialize_applications
-      self.applications = (settings['applications'] || '').split(',').flatten.compact.map { |app_name| Application.find_or_create_by_name_and_host(app_name, host) }
+      application_names = (settings['applications'] || '').split(',').flatten.compact
+      self.applications = []
+      application_names.each do |app_name|
+        app = Application.find(:name => app_name, :host_id => host.id)
+        if app
+          self.applications << app
+        else
+          app = Application.new(:name => app_name, :host_id => host.id)
+          if app.save
+            self.applications << app
+          else
+            warn("Could not create application '#{app_name}' for host #{host.name}")
+          end
+        end
+      end
     end
 
     def confirm_settings
       raise ConnectionError.new("Must specify a Zabbix server to send data to.")     unless settings['server']
       raise Error.new("Must specify the path to a local configuraiton file")         unless settings['configuration_file'] && File.file?(settings['configuration_file'])
       raise ConnectionError.new("Must specify the name of a host to send data for.") unless settings['host']
-      raise ValidationError.new("Must define at least one host group.")              if settings['host_groups'].nil? || settings['host_groups'].empty?      
+      raise ValidationError.new("Must define at least one host group.")              if auto_vivify? && (settings['host_groups'].nil? || settings['host_groups'].empty?)
     end
     
     #
@@ -80,7 +92,6 @@ module Rubix
     #
 
     def run
-      return unless alive?
       case
       when settings['pipe']
         process_pipe
@@ -196,8 +207,8 @@ module Rubix
     # Or when sending for another host:
     # 
     #   {
-    #     'hostname': 'shazaam',
-    #     'application': 'silly',
+    #     'host': 'shazaam',
+    #     'applications': 'silly',
     #     'data': [
     #       {'key': 'foo.bar.baz',      'value': 10},
     #       {'key': 'snap.crackle.pop', 'value': 8 }
@@ -226,7 +237,6 @@ module Rubix
         # present in the line.
         begin
           daughter_pipe = self.class.new(settings.stringify_keys.merge(json))
-          return unless daughter_pipe.alive?
         rescue Error => e
           error(e.message)
           return
@@ -257,9 +267,23 @@ module Rubix
     # If the +key+ doesn't exist for this local agent's host, it will be
     # added.
     def send key, value, timestamp
-      item = Item.new(:key => key, :host => host, :applications => applications, :value_type => Item.value_type_from_value(value))
-      unless settings['fast'] || item.exists?
-        return unless item.create
+      ensure_item_exists(key, value) unless fast?
+      command = "#{settings['sender']} --config #{settings['configuration_file']} --zabbix-server #{settings['server']} --host #{settings['host']} --key #{key} --value '#{value}'"
+      process_zabbix_sender_output(key, `#{command}`)
+
+      # command = "zabbix_sender --config #{configuration_file} --zabbix-server #{server} --input-file - --with-timestamps"
+      # open(command, 'w') do |zabbix_sender|
+      #   zabbix_sender.write([settings['host'], key, timestamp.to_i, value].map(&:to_s).join("\t"))
+      #   zabbix_sender.close_write
+      #   process_zabbix_sender_output(zabbix_sender.read)
+      # end
+    end
+
+    def ensure_item_exists key, value
+      item = Item.find(:key => key, :host_id => host.id)
+      unless item
+        Item.new(:key => key, :host_id => host.id, :applications => applications, :value_type => Item.value_type_from_value(value)).save
+        
         # There is a time lag of about 15-30 seconds between (successfully)
         # creating an item on the Zabbix server and having the Zabbix accept
         # new data for that item.
@@ -274,17 +298,8 @@ module Rubix
         # points you send to Zabbix, then don't worry about it.
         sleep settings['create_item_sleep']
       end
-      command = "#{settings['sender']} --config #{settings['configuration_file']} --zabbix-server #{settings['server']} --host #{settings['host']} --key #{key} --value '#{value}'"
-      process_zabbix_sender_output(key, `#{command}`)
-
-      # command = "zabbix_sender --config #{configuration_file} --zabbix-server #{server} --input-file - --with-timestamps"
-      # open(command, 'w') do |zabbix_sender|
-      #   zabbix_sender.write([settings['host'], key, timestamp.to_i, value].map(&:to_s).join("\t"))
-      #   zabbix_sender.close_write
-      #   process_zabbix_sender_output(zabbix_sender.read)
-      # end
     end
-
+    
     # Parse the +text+ output by +zabbix_sender+.
     def process_zabbix_sender_output key, text
       return unless settings['verbose']

data/spec/rubix/monitors/chef_monitor_spec.rb

@@ -1,5 +1,4 @@
 require 'spec_helper'
-require 'tempfile'
 
 describe Rubix::ChefMonitor do
 

data/spec/rubix/monitors/cluster_monitor_spec.rb

@@ -1,5 +1,4 @@
 require 'spec_helper'
-require 'tempfile'
 
 describe Rubix::ClusterMonitor do
 

data/spec/rubix/sender_spec.rb

@@ -3,6 +3,20 @@ require 'spec_helper'
 describe Rubix::Sender do
 
   before do
+    @config_file = Tempfile.new('sender', '/tmp')
+  end
+
+  describe "running in --fast mode" do
+
+    it "should not attempt to make any calls to the Zabbix API when writing values" do
+      Rubix.connection.should_not_receive(:request)
+      @sender = Rubix::Sender.new(mock_settings('configuration_file' => @config_file.path, 'host' => 'foohost', 'server' => 'fooserver', 'fast' => true, 'sender' => 'echo'))
+      @sender.process_line("foo.bar.baz	123\n")
+      @sender.process_line({:host => 'newhost', :host_groups => 'foobar,baz', :data => [{:key => 'foo.bar.baz', :value => 123}]}.to_json)
+    end
+  end
+
+  describe "running in auto-vivify mode" do
   end
   
 end

data/spec/spec_helper.rb

@@ -1,3 +1,4 @@
+require 'tempfile'
 require 'rspec'
 
 RUBIX_ROOT = File.expand_path(__FILE__, '../../lib')
@@ -12,6 +13,7 @@ RSpec.configure do |config|
   config.mock_with :rspec
   config.include Rubix::ResponseSpecs
   config.include Rubix::IntegrationHelper
+  config.include Rubix::ConfigliereHelper
 
   test_yml_path = File.expand_path('../test.yml', __FILE__)
   if File.exist?(test_yml_path)

data/spec/support/configliere_helper.rb

@@ -0,0 +1,12 @@
+module Rubix
+  module ConfigliereHelper
+    def mock_settings options={}, rest=[]
+      param = options
+      param.stub!(:rest).and_return(rest)
+      param.stub!(:stringify_keys).and_return(param) # FIXME...
+      param
+    end
+  end
+end
+
+

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: rubix
 version: !ruby/object:Gem::Version 
-  hash: 19
+  hash: 17
   prerelease: 
   segments: 
   - 0
   - 0
-  - 6
-  version: 0.0.6
+  - 7
+  version: 0.0.7
 platform: ruby
 authors: 
 - Dhruv Bansal
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-11-22 00:00:00 Z
+date: 2011-11-23 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: rspec
@@ -117,6 +117,7 @@ files:
 - spec/spec_helper.rb
 - spec/support/integration_helper.rb
 - spec/support/response_helper.rb
+- spec/support/configliere_helper.rb
 - LICENSE
 - README.rdoc
 - VERSION