fluent-plugin-vertica 0.0.2

data/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2013 Erik Selin
+
+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,21 @@
+Fluentd output plugin for Vertica
+=================================
+
+Simple batched output plugin for getting events into vertica.
+
+Example config
+==============
+
+    <match vertica.public.test>
+      type vertica
+
+      database mydb
+      schema public
+      table test
+
+      username dbadmin
+      password mypass
+
+      host 127.0.0.1
+      port 5433
+    </match>

data/lib/fluent/plugin/out_vertica.rb

@@ -0,0 +1,136 @@
+module Fluent
+  class VerticaOutput < Fluent::BufferedOutput
+    Fluent::Plugin.register_output('vertica', self)
+
+    config_param :host,           :string,  :default => '127.0.0.1'
+    config_param :port,           :integer, :default => 5433
+    config_param :username,       :string,  :default => 'dbadmin'
+    config_param :password,       :string,  :default => nil
+    config_param :database,       :string,  :default => nil
+    config_param :schema,         :string,  :default => nil
+    config_param :table,          :string,  :default => nil
+    config_param :ssl,            :bool,    :default => false
+
+    def initialize
+      super
+
+      require 'vertica'
+      require 'csv'
+    end
+
+    def format(tag, time, record)
+      values = columns.map { |col| record[col] }
+      CSV.generate_line(values, { :col_sep => "\t" })
+    end
+
+    def write(chunk)
+      chunk.open do |file|
+
+        reset
+
+        temp_table = "temp_#{@table}"
+        perm_table = "#{@schema}.#{@table}"
+
+        vertica.query(<<-SQL)
+          CREATE LOCAL TEMPORARY TABLE #{temp_table}
+          ON COMMIT DELETE ROWS
+          AS SELECT * FROM #{perm_table} LIMIT 0
+        SQL
+
+        vertica.copy(<<-SQL) { |handle| handle.write(file.read) }
+          COPY #{temp_table} (#{columns.join(",")})
+          FROM STDIN DELIMITER E'\t'
+          RECORD TERMINATOR E'\n' NULL AS '__NULL__'
+          ENFORCELENGTH
+          NO COMMIT
+        SQL
+
+        if primary_keys.empty?
+          vertica.query(<<-SQL)
+            INSERT INTO #{perm_table}
+              (#{columns.join(",")})
+            SELECT
+              #{columns.join(",")}
+            FROM #{temp_table}
+          SQL
+        else
+          condition = primary_keys.map do |key|
+            "#{perm_table}.#{key} = #{temp_table}.#{key}"
+          end
+
+          unless empty_table?(perm_table)
+            vertica.query(<<-SQL)
+              DELETE FROM #{perm_table}
+              WHERE EXISTS (
+                SELECT 1
+                FROM #{temp_table}
+                WHERE #{condition.join(" AND ")}
+              )
+            SQL
+          end
+
+          vertica.query(<<-SQL)
+            INSERT INTO #{perm_table}
+              (#{columns.join(",")})
+            SELECT
+              #{columns.join(",")}
+            FROM (
+              SELECT
+                #{columns.join(",")},
+                row_number() OVER (partition by #{primary_keys.join(",")}) AS r
+              FROM #{temp_table}
+            ) AS temp
+            WHERE r = 1
+          SQL
+        end
+
+        vertica.query("COMMIT")
+      end
+    end
+
+    private
+
+    def reset
+      @vertica      = nil
+      @columns      = nil
+      @primary_keys = nil
+      @empty        = nil
+    end
+
+    def vertica
+      @vertica ||= Vertica.connect({
+        :host     => @host,
+        :user     => @username,
+        :password => @password,
+        :ssl      => @ssl,
+        :port     => @port,
+        :database => @database
+      })
+    end
+
+    def columns
+       @columns ||= vertica.query(<<-SQL).map { |column| column[:column_name] }
+           SELECT column_name
+             FROM columns
+            WHERE table_schema ='#{@schema}'
+              AND table_name='#{@table}'
+         ORDER BY ordinal_position
+       SQL
+    end
+
+    def primary_keys
+      @primary_keys ||= vertica.query(<<-SQL).map { |column| column[:column_name] }
+        SELECT column_name
+          FROM v_catalog.primary_keys
+         WHERE table_schema = '#{@schema}'
+           AND table_name = '#{@table}'
+      SQL
+    end
+
+    def empty_table?(table)
+      @empty ||= vertica.query(<<-SQL).first[:count] == 0
+        SELECT count(1) FROM #{table}
+      SQL
+    end
+  end
+end

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification
+name: fluent-plugin-vertica
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+  prerelease: 
+platform: ruby
+authors:
+- Erik Selin
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-09-09 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.10.35
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.10.35
+- !ruby/object:Gem::Dependency
+  name: vertica
+  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:
+- erik.selin@ifelsestudio.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- lib/fluent/plugin/out_vertica.rb
+homepage: http://tyro89.github.com/fluent-plugin-vertica
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.1
+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 output plugin for Vertica.
+test_files: []