Merge pull request #13 from HatsuneMiku3939/feature/add-user-agent-option

add user agent option

Author: HatsuneMiku3939

README.md

@@ -120,6 +120,10 @@ The tag of the event.
 
 The url of remote server.
 
+### agent (string) (optional, default: fluent-plugin-http-pull)
+
+The user agent string of request.
+
 ### interval (time) (required)
 
 The interval time between periodic request.

fluent-plugin-http-pull.gemspec

@@ -3,7 +3,7 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |spec|
   spec.name    = "fluent-plugin-http-pull"
-  spec.version = "0.6.1"
+  spec.version = "0.7.0"
   spec.authors = ["filepang"]
   spec.email   = ["[email protected]"]
 

lib/fluent/plugin/in_http_pull.rb

@@ -35,6 +35,9 @@ def initialize
       desc 'The interval time between periodic request'
       config_param :interval, :time
 
+      desc 'The user agent string of request'
+      config_param :agent, :string, default: "fluent-plugin-http-pull"
+
       desc 'status_only'
       config_param :status_only, :bool, default: false
 
@@ -72,7 +75,8 @@ def configure(conf)
 
         @parser = parser_create unless @status_only
         @_request_headers = {
-          "Content-Type" => "application/x-www-form-urlencoded"
+          "Content-Type" => "application/x-www-form-urlencoded",
+          "User-Agent" => @agent
         }.merge(@request_headers.map do |section|
           header = section["header"]
           value = section["value"]

test/plugin/test_in_http_pull.rb

@@ -54,6 +54,13 @@ class HttpPullInputTestDefaultOptions < Test::Unit::TestCase
 
       assert_equal(:get, d.instance.http_method)
     end
+
+    test 'agent' do
+      d = create_driver TEST_DEFAULT_VALUE_CONFIG
+      assert_equal("test", d.instance.tag)
+
+      assert_equal("fluent-plugin-http-pull", d.instance.agent)
+    end
   end
 
   private