fluent-plugin-mobile-carrier 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 211678f21ba539d46ac9c1ac103d04f0030ef9a0
+  data.tar.gz: d0f28df60642ae53bba779798a6ba1e53092a18e
+SHA512:
+  metadata.gz: 9c08cc8c43d7be6abc8ad4ca7cc65cb00296abc0d9a118900d9d89e72970808e817c6120b7df0dfa3e0f6db9cc85d0e2de8d703a925c730b1f5b8ba178b5d5bf
+  data.tar.gz: b4457d79c197f4bcd2327d2c9522b3108bbeac8247e1373d936f69897a4e342c40450d96ad33649763fcd3d1989c0ab28a589eacb45bb604af512f1edcb5f21b

data/.gitignore

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

data/.travis.yml

@@ -0,0 +1,4 @@
+language: ruby
+rvm:
+  - 1.9.3
+  - 2.0.0

data/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gemspec

data/LICENSE.txt

@@ -0,0 +1,13 @@
+Copyright (c) HARUYAMA Seigo
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

data/README.md

@@ -0,0 +1,25 @@
+# fluent-plugin-mobile-carrier
+
+## MobileCarrierOutput
+
+'fluent-plugin-mobile-carrier' is a Fluentd plugin to judge mobile carriers from IP address.
+'fluent-plugin-mobile-carrier' needs external config yaml file, which includes Carrier-IPAddress settings.
+
+
+## Configuration
+
+To add mobile_carrier result into matched messages:
+
+    <match input.**>
+      type mobile_carrier
+      key_name ip_address
+      config_yaml config.yaml
+      remove_prefix input
+      add_prefix merged
+    </match>
+
+## Copyright
+
+* Copyright (c) HARUYAMA Seigo
+* License
+  * Apache License, Version 2.0

data/Rakefile

@@ -0,0 +1,11 @@
+#!/usr/bin/env rake
+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-mobile-carrier.gemspec

@@ -0,0 +1,20 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |gem|
+  gem.name          = 'fluent-plugin-mobile-carrier'
+  gem.version       = '0.0.1'
+  gem.authors       = ['HARUYAMA Seigo']
+  gem.email         = ['haruyama@unixuser.org']
+  gem.description   = %q{judge mobile carrier by ip address.}
+  gem.summary       = %q{Fluentd plugin to judge mobile carrier by ip address.}
+  gem.homepage      = 'http://github.com/haruyama/fluent-plugin-mobile-carrier'
+  gem.license       = 'APLv2'
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ['lib']
+
+  gem.add_development_dependency 'rake'
+  gem.add_runtime_dependency 'fluentd'
+end

data/lib/fluent/plugin/.rubocop.yml

@@ -0,0 +1,12 @@
+LineLength:
+  Enabled: false
+
+CyclomaticComplexity:
+  Enabled: false
+
+MethodLength:
+  CountComments: false # count full line comments?
+  Max: 50
+
+GlobalVars:
+  Enabled: false

data/lib/fluent/plugin/out_mobile_carrier.rb

@@ -0,0 +1,87 @@
+
+# mobile carrieroutput
+class Fluent::MobileCarrierOutput < Fluent::Output
+  Fluent::Plugin.register_output('mobile_carrier', self)
+
+  config_param :config_yaml,   :string
+  config_param :remove_prefix, :string, default: nil
+  config_param :add_prefix,    :string, default: nil
+
+  config_param :key_name,               :string
+  config_param :unknown_carrier,        :string, default: 'UNKNOWN'
+  config_param :out_key_mobile_carrier, :string, default: 'mobile_carrier'
+
+  def initialize
+    super
+    require 'ipaddr'
+    require 'yaml'
+  end
+
+  def configure(conf)
+    super
+
+    raise 'config_yaml is not set!' unless (@config_yaml)
+    @config = prepare_config(YAML.load_file(@config_yaml))
+
+    if !@tag && !@remove_prefix && !@add_prefix
+      fail Fluent::ConfigError, 'missing both of remove_prefix and add_prefix'
+    end
+    if @tag && (@remove_prefix || @add_prefix)
+      fail Fluent::ConfigError, 'both of tag and remove_prefix/add_prefix must not be specified'
+    end
+    if @remove_prefix
+      @removed_prefix_string = @remove_prefix + '.'
+      @removed_length = @removed_prefix_string.length
+    end
+    @added_prefix_string = @add_prefix + '.' if @add_prefix
+  end
+
+  def tag_mangle(tag)
+    if @tag
+      @tag
+    else
+      if @remove_prefix &&
+          ( (tag.start_with?(@removed_prefix_string) && tag.length > @removed_length) || tag == @remove_prefix)
+        tag = tag[@removed_length..-1]
+      end
+      if @add_prefix
+        tag = if tag && tag.length > 0
+                @added_prefix_string + tag
+              else
+                @add_prefix
+              end
+      end
+      tag
+    end
+  end
+
+  def prepare_config(raw_config)
+    config = {}
+    raw_config.each { |carrier, ip_address_list|
+      config[carrier] = ip_address_list.map { |ip_address|
+        IPAddr.new(ip_address)
+      }
+    }
+    config
+  end
+
+  def judge_mobile_carrier(ip_address, config)
+    config.each { |carrier, ipaddr_list|
+      if ipaddr_list.any? { |ipaddr| ipaddr.include? ip_address }
+        return carrier
+      end
+    }
+    @unknown_carrier
+  end
+
+  def emit(tag, es, chain)
+    tag = tag_mangle(tag)
+    es.each do |time, record|
+      record.merge!(
+        @out_key_mobile_carrier => judge_mobile_carrier(record[@key_name], @config)
+      )
+      Fluent::Engine.emit(tag, time, record)
+    end
+    chain.next
+  end
+end

data/test/data/config1.yaml

@@ -0,0 +1,7 @@
+---
+D:
+- 192.168.0.1/24
+- 192.168.2.1/24
+E:
+- 192.168.1.1/24
+- 192.168.3.1/24

data/test/helper.rb

@@ -0,0 +1,28 @@
+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'
+unless ENV.key?('VERBOSE')
+  nulllogger = Object.new
+  nulllogger.instance_eval do |obj|
+    def method_missing(method, *args)
+      # pass
+    end
+  end
+  $log = nulllogger
+end
+
+require 'fluent/plugin/out_mobile_carrier'
+
+class Test::Unit::TestCase
+end

data/test/plugin/.rubocop.yml

@@ -0,0 +1,25 @@
+LineLength:
+  Enabled: false
+
+MethodLength:
+  Enabled: true
+  CountComments: false # count full line comments?
+  Max: 50
+
+FavorSprintf:
+  Enabled: false
+
+MultilineBlocks:
+  Enabled: false
+
+AvoidPerlBackrefs:
+  Enabled: false
+
+PercentLiterals:
+  Enabled: false
+
+BraceAfterPercent:
+  Enabled: false
+
+ClassLength:
+  Enabled: false

data/test/plugin/test_out_mobile_carrier.rb

@@ -0,0 +1,98 @@
+require 'helper'
+
+# MobileCarrierOutput test
+class Fluent::MobileCarrierOutputTest < Test::Unit::TestCase
+  # through & merge
+  CONFIG1 = %[
+type mobile_carrier
+remove_prefix test
+add_prefix merged
+config_yaml test/data/config1.yaml
+key_name ip_address
+]
+
+  CONFIG2 = %[
+type mobile_carrier
+remove_prefix test
+add_prefix merged
+config_yaml test/data/config1.yaml
+key_name ip
+unknown_carrier unknown
+out_key_mobile_carrier mc
+]
+
+  def create_driver(conf = CONFIG1, tag = 'test')
+    Fluent::Test::OutputTestDriver.new(Fluent::MobileCarrierOutput, tag).configure(conf)
+  end
+
+  def test_configure
+    # through & merge
+    d = create_driver CONFIG1
+    assert_equal 'ip_address', d.instance.key_name
+    assert_equal 'test',    d.instance.remove_prefix
+    assert_equal 'merged',  d.instance.add_prefix
+
+    assert_equal 'UNKNOWN',        d.instance.unknown_carrier
+    assert_equal 'mobile_carrier', d.instance.out_key_mobile_carrier
+
+    # filter & merge
+    d = create_driver CONFIG2
+    assert_equal 'ip',     d.instance.key_name
+    assert_equal 'test',   d.instance.remove_prefix
+    assert_equal 'merged', d.instance.add_prefix
+
+    assert_equal 'unknown', d.instance.unknown_carrier
+    assert_equal 'mc',      d.instance.out_key_mobile_carrier
+  end
+
+  def test_tag_mangle
+    p = create_driver(CONFIG1).instance
+    assert_equal 'merged.data', p.tag_mangle('data')
+    assert_equal 'merged.data', p.tag_mangle('test.data')
+    assert_equal 'merged.test.data', p.tag_mangle('test.test.data')
+    assert_equal 'merged', p.tag_mangle('test')
+  end
+
+  def test_emit1
+    d = create_driver(CONFIG1, 'test.message')
+    time = Time.parse('2012-07-20 16:40:30').to_i
+    d.run do
+      d.emit({ 'value' => 0 }, time)
+      d.emit({ 'value' => 1, 'ip_address' => '192.168.0.5' }, time)
+      d.emit({ 'value' => 2, 'ip_address' => '192.168.1.10' }, time)
+      d.emit({ 'value' => 3, 'ip_address' => '192.168.2.100' }, time)
+      d.emit({ 'value' => 4, 'ip_address' => '192.168.3.200' }, time)
+      d.emit({ 'value' => 5, 'ip_address' => '192.168.4.1' }, time)
+    end
+
+    emits = d.emits
+    assert_equal 6,                emits.size
+    assert_equal 'merged.message', emits[0][0]
+    assert_equal time,             emits[0][1]
+
+    m = emits[0][2]
+    assert_equal 0,         m['value']
+    assert_equal 'UNKNOWN', m['mobile_carrier']
+
+    m = emits[1][2]
+    assert_equal 1,   m['value']
+    assert_equal 'D', m['mobile_carrier']
+
+    m = emits[2][2]
+    assert_equal 2,   m['value']
+    assert_equal 'E', m['mobile_carrier']
+
+    m = emits[3][2]
+    assert_equal 3,   m['value']
+    assert_equal 'D', m['mobile_carrier']
+
+    m = emits[4][2]
+    assert_equal 4,   m['value']
+    assert_equal 'E', m['mobile_carrier']
+
+    m = emits[5][2]
+    assert_equal 5,         m['value']
+    assert_equal 'UNKNOWN', m['mobile_carrier']
+  end
+
+end

metadata

@@ -0,0 +1,89 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-mobile-carrier
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- HARUYAMA Seigo
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-01-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: judge mobile carrier by ip address.
+email:
+- haruyama@unixuser.org
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- fluent-plugin-mobile-carrier.gemspec
+- lib/fluent/plugin/.rubocop.yml
+- lib/fluent/plugin/out_mobile_carrier.rb
+- test/data/config1.yaml
+- test/helper.rb
+- test/plugin/.rubocop.yml
+- test/plugin/test_out_mobile_carrier.rb
+homepage: http://github.com/haruyama/fluent-plugin-mobile-carrier
+licenses:
+- APLv2
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.0
+signing_key: 
+specification_version: 4
+summary: Fluentd plugin to judge mobile carrier by ip address.
+test_files:
+- test/data/config1.yaml
+- test/helper.rb
+- test/plugin/.rubocop.yml
+- test/plugin/test_out_mobile_carrier.rb