fluent-plugin-pgjson 0.0.1 → 0.0.2

data/fluent-plugin-pgjson.gemspec

@@ -3,9 +3,9 @@ $:.push File.expand_path("../lib", __FILE__)
 
 Gem::Specification.new do |s|
   s.name        = "fluent-plugin-pgjson"
-  s.version     = "0.0.1"
+  s.version     = "0.0.2"
   s.authors     = ["OKUNO Akihiro"]
-  s.email       = ["okuno.akihiro@gmail.com"]
+  s.email       = ["choplin.choplin@gmail.com"]
   s.homepage    = "https://github.com/choplin/fluent-plugin-pgjson"
   s.summary     = %q{}
   s.description = %q{}

data/lib/fluent/plugin/out_pgjson.rb

@@ -13,10 +13,10 @@ class PgJsonOutput < Fluent::BufferedOutput
   config_param :tag_col    , :string  , :default => 'tag'
   config_param :record_col , :string  , :default => 'record'
 
-
   def initialize
     super
     require 'pg'
+    @conn = nil
   end
 
   def configure(conf)
@@ -24,14 +24,10 @@ class PgJsonOutput < Fluent::BufferedOutput
     @stmt_name = 'insert'
   end
 
-  def start
-    super
-    init_connection
-  end
-
   def shutdown
     super
-    if !@conn.nil? and !@conn.finished?
+
+    if ! @conn.nil? and ! @conn.finished?
       @conn.close()
     end
   end
@@ -42,30 +38,40 @@ class PgJsonOutput < Fluent::BufferedOutput
 
   def write(chunk)
     begin
+      init_connection
       sql = build_sql(chunk)
       @conn.exec(sql)
     rescue
-      #TODO
+      begin
+        @conn.close()
+      rescue
+      end
+
+      @conn = nil 
       raise
     end
   end
 
   private
   def init_connection
-    begin
-      @conn = PGconn.new(:dbname => @database, :host => @host, :port => @port, :user => @user, :password => @password)
-      @conn.setnonblocking(true)
-    rescue
-      raise Fluent::ConfigError, "failed to connect PostgreSQL Server at #{@host}:#{@port}"
-    end
-  end
+    if @conn.nil?
+      $log.debug "Connecting to PostgreSQL server #{@host}:#{@port}, database #{@database}..."
 
-  def prepare_statement
-    begin
-      @conn.prepare(@stmt_name, sql)
-    rescue
-      #TODO
-      raise
+      begin
+        @conn = PGconn.new(:dbname => @database, :host => @host, :port => @port, :user => @user, :password => @password)
+        @conn.setnonblocking(true)
+      rescue
+        if ! conn.nil?
+          begin
+            @conn.close()
+          rescue
+          end
+
+          @conn - nil
+        end
+
+        raise
+      end
     end
   end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: fluent-plugin-pgjson
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-07-29 00:00:00.000000000 Z
+date: 2013-07-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: fluentd
@@ -45,7 +45,7 @@ dependencies:
         version: '0'
 description: ''
 email:
-- okuno.akihiro@gmail.com
+- choplin.choplin@gmail.com
 executables: []
 extensions: []
 extra_rdoc_files: []