fluent-plugin-http-pull 0.5.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 56348b6b227f806a70821eaed3fafddcdc94ee6c
-  data.tar.gz: 34e8e11a24b5b2dac781f4556f0e534b08e403fc
+  metadata.gz: 4431d08247cac9ded691561df1f37e0ca301812c
+  data.tar.gz: 34533a363fe85ac8a9e8fe6f7732d25cebe0ea43
 SHA512:
-  metadata.gz: 0b5aeea895ccc1cfaf7923f8c21c9bdb54328677d384e66c508c289b0da9f8a7ab23778eb8097323e18822f493a7f1ad491544750d1230e2597910620d90e129
-  data.tar.gz: b0b8ced808f05bdb9c0804af5dad1c7b171c6f6e096d5eab29d840ad6305ec079276692eea1a45b875629ed26cb82650be4462c7912c8831eb96145b222fb9d5
+  metadata.gz: 3ab0a2aad719fb68cda02595b9b7df31c5d8f611534898aee88323c325358ba0915c7937deff0b317ca51ebea4d3fa535ebc929f08d8783cb4932022d514c55c
+  data.tar.gz: c8ed089e7978a0ae4dc7dea11ac71e5b5b819def2d2136cc9e43f1111c4ddf8cf5d694ebc0b2d9f22707bcecbf2b53677670794153f509bd59c8e19ee84b1d51

data/README.md

@@ -34,6 +34,8 @@ $ bundle
 
 ## Example
 
+You can found more examples in `test/plugin/test_in_http_pull.rb`
+
 ### Monitoring http status code only
 ```
 <source>
@@ -133,7 +135,17 @@ for more detail.
 
 ### status_only (bool) (optional, default: false)
 
-If status_only is true, body is not parsed.
+If `status_only` is true, body is not parsed.
+
+### http_method (enum) (optional, default: :get)
+
+The http request method for each requests. Avaliable options are listed below.
+
+* `get`
+* `post`
+* `delete`
+
+If `status_only` is true, `http_method` was override to `head`
 
 ### timeout (time) (optional, default: 10s)
 
@@ -151,6 +163,14 @@ The user for basic auth
 
 The password for basic auth
 
+### response_header (section) (optional, default: nil)
+
+The name of response header for capture.
+
+### request_header (section) (optional, default: nil)
+
+The name, value pair of custom reuqest header.
+
 ## In case of remote error
 
 ### Can receive response from remote

data/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.5.0"
+  spec.version = "0.6.0"
   spec.authors = ["filepang"]
   spec.email   = ["filepang@gmail.com"]
 

data/lib/fluent/plugin/in_http_pull.rb

@@ -20,7 +20,6 @@ module Fluent
   module Plugin
     class HttpPullInput < Fluent::Plugin::Input
       Fluent::Plugin.register_input("http_pull", self)
-
       helpers :timer, :parser, :compat_parameters
 
       def initialize
@@ -29,19 +28,28 @@ module Fluent
 
       desc 'The tag of the event.'
       config_param :tag, :string
+
       desc 'The url of monitoring target'
       config_param :url, :string
+
       desc 'The interval time between periodic request'
       config_param :interval, :time
+
       desc 'status_only'
       config_param :status_only, :bool, default: false
-      desc 'The timeout stime of each request'
+
+      desc 'The http method for each request'
+      config_param :http_method, :enum, list: [:get, :post, :delete], default: :get
+
+      desc 'The timeout second of each request'
       config_param :timeout, :time, default: 10
+
       desc 'The HTTP proxy URL to use for each requests'
       config_param :proxy, :string, default: nil
 
       desc 'user of basic auth'
       config_param :user, :string, default: nil
+
       desc 'password of basic auth'
       config_param :password, :string, default: nil
 
@@ -69,6 +77,8 @@ module Fluent
 
           [header.to_sym, value]
         end.to_h
+
+        @http_method = :head if @status_only
       end
 
       def start
@@ -81,7 +91,7 @@ module Fluent
         record = { "url" => @url }
 
         begin
-          request_options = { method: :get, url: @url, timeout: @timeout }
+          request_options = { method: @http_method, url: @url, timeout: @timeout }
 
           request_options[:proxy] = @proxy if @proxy
           request_options[:user] = @user if @user

data/test/helper/stub_server.rb

@@ -1,5 +1,17 @@
 require 'webrick'
 
+class DeleteService < WEBrick::HTTPServlet::AbstractServlet
+  def service(req, res)
+    if req.request_method != "DELETE"
+      res.status = 405
+    else
+      res.status = 200
+      res['Content-Type'] = 'application/json'
+      res.body = '{ "status": "OK" }'
+    end
+  end
+end
+
 class StubServer
   def initialize
     create_server
@@ -12,6 +24,9 @@ class StubServer
     @server.mount_proc '/redirect', &method(:redirect)
     @server.mount_proc '/protected', &method(:protected)
     @server.mount_proc '/custom_header', &method(:custom_header)
+
+    @server.mount_proc '/method_post', &method(:method_post)
+    @server.mount '/method_delete', DeleteService
   end
 
   def start
@@ -90,4 +105,15 @@ class StubServer
     res['Content-Type'] = 'application/json'
     res.body = '{ "status": "OK" }'
   end
+
+  def method_post(req, res)
+    if req.request_method != "POST"
+      res.status = 405
+    else
+      res.status = 200
+      res['Content-Type'] = 'application/json'
+      res.body = '{ "status": "OK" }'
+    end
+  end
 end
+

data/test/plugin/test_in_http_pull.rb

@@ -58,6 +58,13 @@ class HttpPullInputTest < Test::Unit::TestCase
 
       assert_equal(nil, d.instance.proxy)
     end
+
+    test 'http_method' do
+      d = create_driver TEST_DEFAULT_VALUE_CONFIG
+      assert_equal("test", d.instance.tag)
+
+      assert_equal(:get, d.instance.http_method)
+    end
   end
 
   sub_test_case "success case" do
@@ -556,6 +563,62 @@ class HttpPullInputTest < Test::Unit::TestCase
     end
   end
 
+  sub_test_case "request method" do
+    TEST_INTERVAL_3_POST = %[
+      tag test
+      url http://127.0.0.1:3939/method_post
+
+      interval 3s
+      format json
+      http_method post
+    ]
+
+    TEST_INTERVAL_3_DELETE = %[
+      tag test
+      url http://127.0.0.1:3939/method_delete
+
+      interval 3s
+      format json
+      http_method delete
+    ]
+
+    test 'interval 3 with :post' do
+      d = create_driver TEST_INTERVAL_3_POST
+      assert_equal("test", d.instance.tag)
+      assert_equal(3, d.instance.interval)
+
+      d.run(timeout: 8) do
+        sleep 7
+      end
+      assert_equal(2, d.events.size)
+
+      d.events.each do |tag, time, record|
+        assert_equal("test", tag)
+
+        assert_equal({"url"=>"http://127.0.0.1:3939/method_post","status"=>200, "message"=>{"status"=>"OK"}}, record)
+        assert(time.is_a?(Fluent::EventTime))
+      end
+    end
+
+    test 'interval 3 with :delete' do
+      d = create_driver TEST_INTERVAL_3_DELETE
+      assert_equal("test", d.instance.tag)
+      assert_equal(3, d.instance.interval)
+
+      d.run(timeout: 8) do
+        sleep 7
+      end
+      assert_equal(2, d.events.size)
+
+      d.events.each do |tag, time, record|
+        assert_equal("test", tag)
+
+        assert_equal({"url"=>"http://127.0.0.1:3939/method_delete","status"=>200, "message"=>{"status"=>"OK"}}, record)
+        assert(time.is_a?(Fluent::EventTime))
+      end
+    end
+  end
+
   private
 
   def create_driver(conf)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: fluent-plugin-http-pull
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors:
 - filepang
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-09-08 00:00:00.000000000 Z
+date: 2017-11-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -167,7 +167,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.13
+rubygems_version: 2.5.2
 signing_key: 
 specification_version: 4
 summary: fluent-plugin-http-pull