Merge pull request #117 from fluent/dont-handle-database-placeholders-when-connection_string-specified

Don't handle database placholders when specifying connection_string parameter

Author: repeatedly

lib/fluent/plugin/out_mongo.rb

@@ -173,7 +173,8 @@ def multi_workers_ready?
 
     def write(chunk)
       collection_name = extract_placeholders(@collection, chunk)
-      database_name = extract_placeholders(@database, chunk)
+      # In connection_string case, we shouldn't handle extract_placeholers for @database.
+      database_name = extract_placeholders(@database, chunk) unless @connection_string
       operate(database_name, format_collection_name(collection_name), collect_records(chunk))
     end
 

test/plugin/test_out_mongo.rb

@@ -183,6 +183,27 @@ def test_write
     assert_equal(expected, actual_documents)
   end
 
+  def test_write_with_connection_string
+    d = create_driver(%[
+      @type mongo
+      connection_string mongodb://localhost:#{port}/#{database_name}
+      collection #{collection_name}
+      capped
+      capped_size 100
+    ])
+    assert_equal("mongodb://localhost:#{port}/#{database_name}", d.instance.connection_string)
+    assert_nil d.instance.database
+
+    d.run(default_tag: 'test') do
+      emit_documents(d)
+    end
+    actual_documents = get_documents
+    time = event_time("2011-01-02 13:14:15 UTC")
+    expected = [{'a' => 1, d.instance.inject_config.time_key => Time.at(time).localtime},
+                {'a' => 2, d.instance.inject_config.time_key => Time.at(time).localtime}]
+    assert_equal(expected, actual_documents)
+  end
+
   class WriteWithCollectionPlaceholder < self
     def setup
       @tag = 'custom'