fluent-plugin-mysql-replicator 0.0.1

data/.gitignore

@@ -0,0 +1,5 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+vendor/*

data/.travis.yml

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

data/Gemfile

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

data/LICENSE

@@ -0,0 +1,14 @@
+Copyright (c) 2013- Kentaro Yoshida
+
+   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,70 @@
+# fluent-plugin-mysql-replicator [![Build Status](https://travis-ci.org/y-ken/fluent-plugin-mysql-replicator.png?branch=master)](https://travis-ci.org/y-ken/fluent-plugin-mysql-replicator)
+
+## Overview
+
+Fluentd input plugin to track insert/update/delete event from MySQL database server.
+
+## Installation
+
+`````
+### native gem
+gem install fluent-plugin-mysql-replicator
+
+### td-agent gem
+/usr/lib64/fluent/ruby/bin/fluent-gem install fluent-plugin-mysql-replicator
+`````
+
+## Tutorial
+
+#### configuration
+
+`````
+<source>
+  type mysql_replicator
+  host localhost
+  username your_mysql_user
+  password your_mysql_password
+  database myweb
+  interval 5s
+  tag replicator
+  query SELECT id, text from search_test
+</source>
+
+<match replicator.*>
+  type stdout
+</match>
+`````
+
+#### sample query
+
+`````
+$ mysql -e "create database myweb"
+$ mysql myweb -e "create table search_test(id int auto_increment, text text, PRIMARY KEY (id))"
+$ sleep 10
+$ mysql myweb -e "insert into search_test(text) values('aaa')"
+$ sleep 10
+$ mysql myweb -e "update search_test set text='bbb' where text = 'aaa'"
+$ sleep 10
+$ mysql myweb -e "delete from search_test where text='bbb'"
+`````
+
+#### result
+
+`````
+$ tail -f /var/log/td-agent/td-agent.log
+2013-11-25 18:22:25 +0900 replicator.insert: {"id":"1","text":"aaa"}
+2013-11-25 18:22:35 +0900 replicator.update: {"id":"1","text":"bbb"}
+2013-11-25 18:22:45 +0900 replicator.delete: {"id":"1"}
+`````
+
+## TODO
+
+Pull requests are very welcome!!
+
+## Copyright
+
+Copyright © 2013- Kentaro Yoshida ([@yoshi_ken](https://twitter.com/yoshi_ken))
+
+## License
+
+Apache License, Version 2.0

data/Rakefile

@@ -0,0 +1,9 @@
+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-mysql-replicator.gemspec

@@ -0,0 +1,18 @@
+# -*- encoding: utf-8 -*-
+Gem::Specification.new do |s|
+  s.name        = "fluent-plugin-mysql-replicator"
+  s.version     = "0.0.1"
+  s.authors     = ["Kentaro Yoshida"]
+  s.email       = ["y.ken.studio@gmail.com"]
+  s.homepage    = "https://github.com/y-ken/fluent-plugin-mysql-replicator"
+  s.summary     = %q{Fluentd input plugin to track insert/update/delete event from MySQL database server.}
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_development_dependency "rake"
+  s.add_runtime_dependency "fluentd"
+  s.add_runtime_dependency "mysql2"
+end

data/lib/fluent/plugin/in_mysql_replicator.rb

@@ -0,0 +1,109 @@
+module Fluent
+  class MysqlReplicatorInput < Fluent::Input
+    Plugin.register_input('mysql_replicator', self)
+
+    def initialize
+      require 'mysql2'
+      require 'digest/sha1'
+      super
+    end
+
+    config_param :host, :string, :default => 'localhost'
+    config_param :port, :integer, :default => 3306
+    config_param :username, :string, :default => 'root'
+    config_param :password, :string, :default => nil
+    config_param :database, :string, :default => nil
+    config_param :encoding, :string, :default => 'utf8'
+    config_param :interval, :string, :default => '1m'
+    config_param :tag, :string
+    config_param :query, :string
+    config_param :primary_key, :string, :default => 'id'
+
+    def configure(conf)
+      super
+      @interval = Config.time_value(@interval)
+      $log.info "adding mysql_replicator job: [#{@query}] interval: #{@interval}sec"
+    end
+
+    def start
+      @thread = Thread.new(&method(:run))
+    end
+
+    def shutdown
+      Thread.kill(@thread)
+    end
+
+    def run
+      begin
+        poll
+      rescue StandardError => e
+        $log.error "error: #{e.message}"
+        $log.error e.backtrace.join("\n")
+      end
+    end
+
+    def poll
+      table_hash = Hash.new
+      ids = Array.new
+      loop do
+        previous_ids = ids
+        current_ids = Array.new
+        query(@query).each do |row|
+          current_ids << row[@primary_key]
+          current_hash = Digest::SHA1.hexdigest(row.flatten.join)
+          if !table_hash.include?(row[@primary_key])
+            emit_record(:insert, row)
+          elsif table_hash[row[@primary_key]] != current_hash
+            emit_record(:update, row)
+          end
+          table_hash[row[@primary_key]] = current_hash
+        end
+        ids = current_ids
+        deleted_ids = previous_ids - current_ids
+        if deleted_ids.count > 0
+          hash_delete_by_list(table_hash, deleted_ids)
+          deleted_ids.each {|id| emit_record(:delete, {@primary_key => id})}
+        end
+        sleep @interval        
+      end
+    end
+
+    def hash_delete_by_list (hash, deleted_keys)
+      deleted_keys.each{|k| hash.delete(k)}
+    end
+
+    def emit_record(type, record)
+      tag = "#{@tag}.#{type.to_s}"
+      Engine.emit(tag, Engine.now, record)
+    end
+
+    def query(query)
+      @mysql ||= get_connection
+      begin
+        return @mysql.query(query, :cast => false, :cache_rows => false)
+      rescue Exception => e
+        $log.warn "mysql_replicator: #{e}"
+        sleep @interval
+        retry
+      end
+    end
+
+    def get_connection
+      begin
+        return Mysql2::Client.new({
+          :host => @host,
+          :port => @port,
+          :username => @username,
+          :password => @password,
+          :database => @database,
+          :encoding => @encoding,
+          :reconnect => true
+        })
+      rescue Exception => e
+        $log.warn "mysql_replicator: #{e}"
+        sleep @interval
+        retry
+      end
+    end
+  end
+end

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

data/test/plugin/test_in_mysql_replicator.rb

@@ -0,0 +1,38 @@
+require 'helper'
+
+class MysqlReplicatorInputTest < Test::Unit::TestCase
+  def setup
+    Fluent::Test.setup
+  end
+
+  CONFIG = %[
+    host            localhost
+    port            3306
+    interval        30
+    tag             input.mysql
+    query           SELECT id, text from search_text
+    record_hostname yes
+  ]
+
+  def create_driver(conf=CONFIG,tag='test')
+    Fluent::Test::OutputTestDriver.new(Fluent::MysqlReplicatorInput, tag).configure(conf)
+  end
+
+  def test_configure
+    assert_raise(Fluent::ConfigError) {
+      d = create_driver('')
+    }
+    d = create_driver %[
+      host            localhost
+      port            3306
+      interval        30
+      tag             input.mysql
+      query           SELECT id, text from search_text
+    ]
+    d.instance.inspect
+    assert_equal 'localhost', d.instance.host
+    assert_equal 3306, d.instance.port
+    assert_equal 30, d.instance.interval
+    assert_equal 'input.mysql', d.instance.tag
+  end
+end

metadata

@@ -0,0 +1,106 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-mysql-replicator
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Kentaro Yoshida
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-11-26 00:00:00.000000000 Z
+dependencies:
+- !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'
+- !ruby/object:Gem::Dependency
+  name: mysql2
+  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: 
+email:
+- y.ken.studio@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .travis.yml
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- fluent-plugin-mysql-replicator.gemspec
+- lib/fluent/plugin/in_mysql_replicator.rb
+- test/helper.rb
+- test/plugin/test_in_mysql_replicator.rb
+homepage: https://github.com/y-ken/fluent-plugin-mysql-replicator
+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 input plugin to track insert/update/delete event from MySQL database
+  server.
+test_files:
+- test/helper.rb
+- test/plugin/test_in_mysql_replicator.rb