fluent-plugin-df 0.0.2

data/.gitignore

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

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2013 tiwakawa
+
+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,44 @@
+# Fluent::Plugin::Df
+
+Df input plugin for Fluent event collector
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'fluent-plugin-df'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install fluent-plugin-df
+
+## Usage
+
+    <source>
+      type df
+      option -k
+      interval 3
+      tag_prefix df
+      target_mounts /
+    </source>
+
+  If you specify more than one `target_mounts`, separated by commas.
+
+## Output Format
+
+    df._dev_disk0s2: {"size":"487546976","used":"52533512","avail":"434757464","capacity":"11"}
+
+  Tag name is the concatenation of the name of the `tag_prefix` and file system.
+  `/` in the file system is replaced by an `_`
+
+## 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,10 @@
+require "bundler/gem_tasks"
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+task default: :test

data/fluent-plugin-df.gemspec

@@ -0,0 +1,23 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+Gem::Specification.new do |spec|
+  spec.name          = "fluent-plugin-df"
+  spec.version       = "0.0.2"
+  spec.authors       = ["tiwakawa"]
+  spec.email         = ["tiwakawa@aiming-inc.com"]
+  spec.description   = %q{input plugin to get disk free data}
+  spec.summary       = %q{input plugin to get disk free data}
+  spec.homepage      = ""
+  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_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_runtime_dependency "fluentd"
+end

data/lib/fluent/plugin/in_df.rb

@@ -0,0 +1,83 @@
+module Fluent
+  class DfInput < Fluent::Input
+    Fluent::Plugin.register_input('df', self)
+
+    config_param :option,        :string,  default: '-k'
+    config_param :interval,      :integer, default: 3
+    config_param :tag_prefix,    :string,  default: 'df'
+    config_param :target_mounts, :string,  default: '/'
+
+    def configure(conf)
+      super
+      @command = "df #{@option}"
+    end
+
+    def start
+      super
+      @loop = Coolio::Loop.new
+      @timer = DfInputTimerWatcher.new(@interval, true, &method(:watch))
+      @loop.attach(@timer)
+      @thread = Thread.new(&method(:run))
+    end
+
+    def shutdown
+      super
+      @loop.watchers.each { |w| w.detach }
+      @loop.stop
+      @thread.terminate
+      @thread.join
+    end
+
+    def run
+      @loop.run
+    rescue => e
+      $log.error "#{e.class.name} - #{e.message}"
+    end
+
+    private
+    def df
+      fss = `#{@command} 2> /dev/null`.split($/)
+      fss.shift # remove header
+      fss.map do |fs|
+        f = fs.split(/\s+/)
+        if arrayed_target_mounts.include?(f[5])
+          {
+            'fs'       => f[0].gsub(/\//, '_'),
+            'size'     => f[1],
+            'used'     => f[2],
+            'avail'    => f[3],
+            'capacity' => f[4].delete('%')
+          }
+        end
+      end.compact
+    end
+
+    def arrayed_target_mounts
+      @target_mounts.gsub(/'/, '').split(',').map(&:strip)
+    end
+
+    def tag_name(fs)
+      @tag_prefix.empty? ? fs : "#{@tag_prefix}.#{fs}"
+    end
+
+    def watch
+      df.each do |result|
+        fs = result.delete('fs')
+        Fluent::Engine.emit(tag_name(fs), Fluent::Engine.now, result)
+      end
+    end
+  end
+
+  class DfInputTimerWatcher < Coolio::TimerWatcher
+    def initialize(interval, repeat, &callback)
+      @callback = callback
+      super(interval, repeat)
+    end
+
+    def on_timer
+      @callback.call
+    rescue => e
+      $log.error "#{e.class.name} - #{e.message}"
+    end
+  end
+end

data/test/helper.rb

@@ -0,0 +1,19 @@
+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'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+
+require 'fluent/test'
+require 'fluent/plugin/in_df'
+
+class Test::Unit::TestCase
+end

data/test/plugin/test_in_df.rb

@@ -0,0 +1,29 @@
+require 'helper'
+
+class DfInputTest < Test::Unit::TestCase
+  def setup
+    Fluent::Test.setup
+  end
+
+  CONFIG = %[
+    option        -k
+    interval      3
+    tag_prefix    df
+    target_mounts /
+  ]
+
+  def create_driver(conf=CONFIG)
+    Fluent::Test::InputTestDriver.new(Fluent::DfInput).configure(conf)
+  end
+
+  def test_configure
+    d = create_driver
+    assert_equal "-k",      d.instance.option
+    assert_equal "df",      d.instance.tag_prefix
+    assert_equal 3,         d.instance.interval
+    assert_equal "/",       d.instance.target_mounts
+  end
+
+#  def test_emit
+#  end
+end

metadata

@@ -0,0 +1,105 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-df
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+  prerelease: 
+platform: ruby
+authors:
+- tiwakawa
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-03-22 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: input plugin to get disk free data
+email:
+- tiwakawa@aiming-inc.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- fluent-plugin-df.gemspec
+- lib/fluent/plugin/in_df.rb
+- test/helper.rb
+- test/plugin/test_in_df.rb
+homepage: ''
+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: input plugin to get disk free data
+test_files:
+- test/helper.rb
+- test/plugin/test_in_df.rb