fluent-plugin-timescaledb 1.0.0 → 1.0.1

data/lib/fluent/plugin/out_timescaledb.rb

@@ -13,17 +13,21 @@ module Fluent
         super
 
         @conn = PG.connect(@db_conn_string)
-        @conn.prepare('insert_record', "INSERT INTO \"#{@db_table_name}\" (time, tag, record) VALUES ($1, $2, $3::jsonb)")
       end
 
       def close
-        @conn.close if @conn
+        @conn.close if @conn and !@conn.finished?
       end
 
       def write(chunk)
+        reconnect_if_connection_bad!
+
+        values = []
         chunk.msgpack_each do | tag, time, record |
-          @conn.exec_prepared('insert_record', [Time.at(time.to_f), tag, record.to_json])
+          values << "('#{@conn.escape_string(format_time(time))}','#{@conn.escape_string(tag)}','#{@conn.escape_string(record.to_json)}'::jsonb)"
         end
+
+        @conn.exec("INSERT INTO #{@conn.escape_identifier(@db_table_name)} (time, tag, record) VALUES #{values.join(',')}")
       end
 
       def format(tag, time, record)
@@ -33,6 +37,20 @@ module Fluent
       def formatted_to_msgpack_binary
         true
       end
+
+      private
+
+      TIME_FORMAT = "%Y-%m-%d %H:%M:%S.%N".freeze
+
+      def format_time(time)
+        Time.at(time.to_f).utc.strftime(TIME_FORMAT)
+      end
+
+      def reconnect_if_connection_bad!
+        if @conn.status == PG::CONNECTION_BAD
+          @conn.reset
+        end
+      end
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: fluent-plugin-timescaledb
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-04-14 00:00:00.000000000 Z
+date: 2019-04-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: pg