Add init_command parameter

README.md

@@ -25,6 +25,7 @@ sslca|path to ca cert(default: nil)
 sslcapath|path to ca certs(default: nil)
 sslcipher|ssl cipher(default: nil)
 sslverify|verify server certificate(default: nil)
+init_command|SQL string executed after the connection is established(default: nil)
 column_names|bulk insert column (require)
 key_names|value key names, ${time} is placeholder Time.at(time).strftime("%Y-%m-%d %H:%M:%S") (default : column_names)
 json_key_names|Key names which store data as json, comma separator.

lib/fluent/plugin/out_mysql.rb

@@ -15,6 +15,7 @@ class Fluent::MysqlOutput < Fluent::BufferedOutput
   config_param :sslcapath, :string, :default => nil
   config_param :sslcipher, :string, :default => nil
   config_param :sslverify, :bool, :default => nil
+  config_param :init_command, :string, :default => nil
 
   config_param :key_names, :string, :default => nil # nil allowed for json format
   config_param :sql, :string, :default => nil
@@ -112,6 +113,7 @@ def client
         :sslcapath => @sslcapath,
         :sslcipher => @sslcipher,
         :sslverify => @sslverify,
+        :init_command => @init_command,
         :flags => Mysql2::Client::MULTI_STATEMENTS,
       })
   end

lib/fluent/plugin/out_mysql_bulk.rb

@@ -29,6 +29,8 @@ class MysqlBulkOutput < Output
                          desc: "SSL cipher."
     config_param :sslverify, :bool, default: nil,
                              desc: "SSL Verify Server Certificate."
+    config_param :init_command, :string, default: nil,
+                 desc: "SQL string executed after the connection is established."
 
     config_param :column_names, :string,
                  desc: "Bulk insert column."
@@ -149,6 +151,7 @@ def client(database)
           sslcapath: @sslcapath,
           sslcipher: @sslcipher,
           sslverify: @sslverify,
+          init_command: @init_command,
           flags: Mysql2::Client::MULTI_STATEMENTS
         )
     end