fluent-plugin-mysqlslowquerylog 0.0.1

data/.gitignore

@@ -0,0 +1,11 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+# For TextMate, emacs, vim
+*.tmproj
+tmtags
+*~
+\#*
+.\#*
+*.swp

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,2 @@
+Copyright (c) 2012 Satoshi SUZUKI
+Apache License, Version 2.0

data/README.md

@@ -0,0 +1,37 @@
+# fluent-plugin-mysqlslowquerylog
+
+## Component
+
+## Synopsis
+
+## Configuration
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'fluent-plugin-mysqlslowquerylog'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install fluent-plugin-mysqlslowquerylog
+
+## 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
+
+## Copyright
+
+### Copyright
+Copyright (c) 2012- Satoshi SUZUKI (@studio3104)
+
+### License
+Apache License, Version 2.0

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-mysqlslowquerylog.gemspec

@@ -0,0 +1,21 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+
+Gem::Specification.new do |gem|
+  gem.name          = "fluent-plugin-mysqlslowquerylog"
+  gem.version       = "0.0.1"
+  gem.authors       = ["Satoshi SUZUKI"]
+  gem.email         = ["studio3104.com@gmail.com"]
+  gem.description   = %q{Fluentd plugin to concat MySQL slowquerylog.}
+  gem.summary       = %q{Fluentd plugin to concat MySQL slowquerylog.}
+  gem.homepage      = "https://github.com/studio3104/fluent-plugin-mysqlslowquerylog"
+
+  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 "fluentd"
+  gem.add_runtime_dependency "fluentd"
+end

data/lib/fluent/plugin/out_mysqlslowquerylog.rb

@@ -0,0 +1,96 @@
+class Fluent::MySQLSlowQueryLogOutput < Fluent::Output
+  Fluent::Plugin.register_output('mysqlslowquerylog', self)
+  include Fluent::HandleTagNameMixin
+
+  config_param :explain,    :bool,   :default => false
+  config_param :dbuser,     :string, :default => nil
+  config_param :dbpassword, :string, :default => nil
+
+  def configure(conf)
+    super
+    @slowlogs = Hash.new
+
+    if !@remove_tag_prefix && !@remove_tag_suffix && !@add_tag_prefix && !@add_tag_suffix
+      raise ConfigError, "out_slowquery: At least one of option, remove_tag_prefix, remove_tag_suffix, add_tag_prefix or add_tag_suffix is required to be set."
+    end
+  end
+
+  def start
+    super
+  end
+
+  def shutdown
+    super
+  end
+
+  def emit(tag, es, chain)
+    if !@slowlogs[:"#{tag}"]
+      @slowlogs[:"#{tag}"] = Array.new
+    end
+    es.each do |time, record|
+      concat_messages(tag, time, record)
+    end
+
+    chain.next
+  end
+
+  def concat_messages(tag, time, record)
+    record.each do |key, value|
+      @slowlogs[:"#{tag}"] << value
+      if value !~ /^set timestamp=\d+\;$/i && value !~ /^use /i && value.end_with?(';')
+        parse_message(tag, time)
+      end
+    end
+  end
+
+  def parse_message(tag, time)
+    record = {}
+    date   = nil
+
+    # Skip the message that is output when after flush-logs or restart mysqld.
+    # e.g.) /usr/sbin/mysqld, Version: 5.5.28-0ubuntu0.12.04.2-log ((Ubuntu)). started with:
+    message = @slowlogs[:"#{tag}"].shift
+    while !message.start_with?('#')
+      message = @slowlogs[:"#{tag}"].shift
+    end
+
+    if message.start_with?('# Time: ')
+      date    = Time.parse(message[8..-1].strip)
+      message = @slowlogs[:"#{tag}"].shift
+    end
+
+    message =~ /^#? User\@Host:\s+(\S+)\s+\@\s+(\S+).*/
+    record[:user] = $1
+    record[:host] = $2
+    message = @slowlogs[:"#{tag}"].shift
+
+    message =~ /^# Query_time: ([0-9.]+)\s+Lock_time: ([0-9.]+)\s+Rows_sent: ([0-9.]+)\s+Rows_examined: ([0-9.]+).*/
+    record[:query_time]    = $1.to_f
+    record[:lock_time]     = $2.to_f
+    record[:rows_sent]     = $3.to_i
+    record[:rows_examined] = $4.to_i
+
+    query = []
+    @slowlogs[:"#{tag}"].each do |m|
+      query << m.strip
+    end
+    record[:sql] = query.join(' ')
+
+    if date
+      flush_emit(tag, date.to_i, record)
+    else
+      flush_emit(tag, time, record)
+    end
+  end
+
+  def flush_emit(tag, time, record)
+    @slowlogs[:"#{tag}"].clear
+    _tag = tag.clone
+    filter_record(_tag, time, record)
+    if tag != _tag
+      Fluent::Engine.emit(_tag, time, record)
+    else
+      $log.warn "Can not emit message because the tag has not changed. Dropped record #{record}"
+    end
+  end
+end

data/test/helper.rb

@@ -0,0 +1,29 @@
+#!/usr/bin/env ruby
+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.has_key?('VERBOSE')
+  nulllogger = Object.new
+  nulllogger.instance_eval {|obj|
+    def method_missing(method, *args)
+      # pass
+    end
+  }
+  $log = nulllogger
+end
+
+require 'fluent/plugin/out_mysqlslowquerylog'
+
+class Test::Unit::TestCase
+end

data/test/plugin/test_out_mysqlslowquerylog.rb

@@ -0,0 +1,125 @@
+require 'helper'
+
+class MySQLSlowQueryLogOutputTest < Test::Unit::TestCase
+  def setup
+    Fluent::Test.setup
+  end
+
+  CONFIG = %[
+    add_tag_prefix concated.
+  ]
+
+  def create_driver(conf = CONFIG, tag='test')
+    Fluent::Test::OutputTestDriver.new(Fluent::MySQLSlowQueryLogOutput, tag).configure(conf)
+  end
+
+  def test_emit
+    d1 = create_driver
+    d2 = create_driver(CONFIG,'test2')
+    d3 = create_driver(CONFIG,'test3')
+    d4 = create_driver(CONFIG,'test4')
+
+    d1.run do
+      d1.emit('message' => '/usr/sbin/mysqld, Version: 5.5.28-0ubuntu0.12.04.2-log ((Ubuntu)). started with:')
+      d1.emit('message' => 'Tcp port: 3306  Unix socket: /var/run/mysqld/mysqld.sock')
+      d1.emit('message' => 'Time                 Id Command    Argument')
+      d1.emit('message' => '# Time: 130105 16:43:42')
+      d1.emit('message' => '# User@Host: debian-sys-maint[debian-sys-maint] @ localhost []')
+      d1.emit('message' => '# Query_time: 0.000167  Lock_time: 0.000057 Rows_sent: 1  Rows_examined: 7')
+      d1.emit('message' => 'SET timestamp=1357371822;')
+      d1.emit('message' => "SELECT count(*) FROM mysql.user WHERE user='root' and password='';")
+    end
+
+    d2.run do
+      d2.emit('message' => '# User@Host: debian-sys-maint[debian-sys-maint] @ localhost []')
+      d2.emit('message' => '# Query_time: 0.002998  Lock_time: 0.000078 Rows_sent: 31  Rows_examined: 81')
+      d2.emit('message' => 'SET timestamp=61357371822;')
+      d2.emit('message' => "select concat('select count(*) into @discard from `',")
+      d2.emit('message' => "                    TABLE_SCHEMA, '`.`', TABLE_NAME, '`')")
+      d2.emit('message' => "      from information_schema.TABLES where ENGINE='MyISAM';")
+
+      d2.emit('message' => "# Time: 130105 18:04:21")
+      d2.emit('message' => "# User@Host: root[root] @ localhost []")
+      d2.emit('message' => "# Query_time: 0.000398  Lock_time: 0.000117 Rows_sent: 7  Rows_examined: 7")
+      d2.emit('message' => "use mysql;")
+      d2.emit('message' => "SET timestamp=1357376661;")
+      d2.emit('message' => "select * from user;")
+    end
+
+    d3.run do
+      d3.emit('message' => "# User@Host: debian-sys-maint[debian-sys-maint] @ localhost []")
+      d3.emit('message' => "# Query_time: 0.014260  Lock_time: 0.000182 Rows_sent: 0  Rows_examined: 808")
+
+      d4.emit('message' => "# User@Host: debian-sys-maint[debian-sys-maint] @ localhost []")
+      d4.emit('message' => "# Query_time: 0.000262  Lock_time: 0.000200 Rows_sent: 0  Rows_examined: 0")
+      d4.emit('message' => "SET timestamp=1357371822;")
+      d4.emit('message' => "select count(*) into @discard from `information_schema`.`EVENTS`;")
+
+      d3.emit('message' => "SET timestamp=1357371822;")
+      d3.emit('message' => "select count(*) into @discard from `information_schema`.`COLUMNS`;")
+    end
+
+    assert_equal 1, d1.emits.size
+    assert_equal 2, d2.emits.size
+    assert_equal 2, d3.emits.size
+
+    assert_equal '2013-01-05 16:43:42 +0900', Time.at(d1.emits[0][1]).to_s
+    assert_equal '2013-01-05 18:04:21 +0900', Time.at(d2.emits[1][1]).to_s
+
+    assert_equal 'concated.test',  d1.emits[0][0]
+    assert_equal 'concated.test2', d2.emits[0][0]
+    assert_equal 'concated.test2', d2.emits[1][0]
+    assert_equal 'concated.test4', d3.emits[0][0]
+    assert_equal 'concated.test3', d3.emits[1][0]
+
+    assert_equal({
+      :user=>"debian-sys-maint[debian-sys-maint]",
+      :host=>"localhost",
+      :query_time=>0.000167,
+      :lock_time=>5.7e-05,
+      :rows_sent=>1,
+      :rows_examined=>7,
+      :sql=>"SET timestamp=1357371822; SELECT count(*) FROM mysql.user WHERE user='root' and password='';"
+    }, d1.emits[0][2])
+
+    assert_equal({
+      :user=>"debian-sys-maint[debian-sys-maint]",
+      :host=>"localhost",
+      :query_time=>0.002998,
+      :lock_time=>7.8e-05,
+      :rows_sent=>31,
+      :rows_examined=>81,
+      :sql=>"SET timestamp=61357371822; select concat('select count(*) into @discard from `', TABLE_SCHEMA, '`.`', TABLE_NAME, '`') from information_schema.TABLES where ENGINE='MyISAM';"
+    }, d2.emits[0][2])
+
+    assert_equal({
+      :user=>"root[root]",
+      :host=>"localhost",
+      :query_time=>0.000398,
+      :lock_time=>0.000117,
+      :rows_sent=>7,
+      :rows_examined=>7,
+      :sql=>"use mysql; SET timestamp=1357376661; select * from user;"
+    }, d2.emits[1][2])
+
+    assert_equal({
+      :user=>"debian-sys-maint[debian-sys-maint]",
+      :host=>"localhost",
+      :query_time=>0.000262,
+      :lock_time=>0.0002,
+      :rows_sent=>0,
+      :rows_examined=>0,
+      :sql=>"SET timestamp=1357371822; select count(*) into @discard from `information_schema`.`EVENTS`;"
+    }, d3.emits[0][2])
+
+    assert_equal({
+      :user=>"debian-sys-maint[debian-sys-maint]",
+      :host=>"localhost",
+      :query_time=>0.01426,
+      :lock_time=>0.000182,
+      :rows_sent=>0,
+      :rows_examined=>808,
+      :sql=>"SET timestamp=1357371822; select count(*) into @discard from `information_schema`.`COLUMNS`;"
+    }, d3.emits[1][2])
+  end
+end

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-mysqlslowquerylog
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Satoshi SUZUKI
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-01-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: fluentd
+  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: Fluentd plugin to concat MySQL slowquerylog.
+email:
+- studio3104.com@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- fluent-plugin-mysqlslowquerylog.gemspec
+- lib/fluent/plugin/out_mysqlslowquerylog.rb
+- test/helper.rb
+- test/plugin/test_out_mysqlslowquerylog.rb
+homepage: https://github.com/studio3104/fluent-plugin-mysqlslowquerylog
+licenses: []
+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: Fluentd plugin to concat MySQL slowquerylog.
+test_files:
+- test/helper.rb
+- test/plugin/test_out_mysqlslowquerylog.rb