logstash-filter-empowclassifier 0.3.18 → 0.3.19

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bf5b905aefb124ab7b19acf66fc8353dc6184c18cff3283123757bea00a6c14b
-  data.tar.gz: 6444d99e54834955caf25a39db38b7e09b2646044ea8c0380eff36e7f1912223
+  metadata.gz: 9d8c5a0a9811c55399f68e4576b616806abc888f782178850e4dcc3f961c0ff7
+  data.tar.gz: d7014ddfca0ba50073de7c86336d660543b3a90293a886c0d66273ac317dae8c
 SHA512:
-  metadata.gz: e9104d6a2d640512b22841c1a69087a8daa7c11a4bda6cb79b15e85a2e9db5f3d24d9ec9d0c1d97f5214d041fb7839c5576e08c977b35eb3d90f7beefc480f04
-  data.tar.gz: '081f993e370beb1e944aa369bba155912354490bcc0ab82b956a9026656ebdba9b34cc620fe23ff6060fc2ca589c26427f395363c6d3f616995b577cc2190627'
+  metadata.gz: 69e9291c7a229244108c39a8cb8d6ae82f68180e9fb5b6a9fc9615c3adee44a4bb123c5bb475df2d7483fba7ef1ef34970b72f314482e1c10b76045a191b74e7
+  data.tar.gz: ce48de1456ad02ea41728bb7d74f176a5575f61facf5633572f739282dd4ab3aca98f783a3497a79f330c02042d6178a9240ac7009290319e4299168c2d47ebb

data/lib/logstash/filters/center-client.rb

@@ -185,17 +185,21 @@ module LogStash
           results = Hash.new
 
           responses.each_with_index do |response, i|
+            req = requests[i]
             res = nil
 
-            if response['responseStatus'] == 'SUCCESS'
+            status = response['responseStatus']
+
+            case status
+            when 'SUCCESS'
               res = LogStash::Filters::Empow::SuccessfulResponse.new(response)
+            when 'IN_PROGRESS'
+              res = LogStash::Filters::Empow::InProgressResponse.new(response)
             else
               failure_reason = response['failedReason']
               res = LogStash::Filters::Empow::FailureResponse.new(failure_reason)
             end
 
-            req = requests[i]
-
             results[req] = res
           end
 

data/lib/logstash/filters/classifier.rb

@@ -308,7 +308,9 @@ module LogStash; module Filters; module Empow;
 						@logger.error("encountered an error while trying to process result", :request => request, :error => e, :backtrace => e.backtrace)
 					end
 
-					@requests_queue.delete(request)
+					if res.is_final # in case of anti-malware, the result may change till the classification process is done
+						@requests_queue.delete(request)
+					end
 				end
 			end
 

data/logstash-filter-empowclassifier.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name          = 'logstash-filter-empowclassifier'
-  s.version       = '0.3.18'
+  s.version       = '0.3.19'
   s.licenses      = ['Apache-2.0']
   s.summary       = 'Logstash intent classification plugin client for accessing empow''s classifiction cloud'
   #s.description   = 'Write a longer description or delete this line.'

data/spec/filters/assaf_spec.rb

@@ -0,0 +1,39 @@
+require_relative '../spec_helper'
+require "logstash/filters/center-client"
+require "logstash/filters/response"
+require "logstash/filters/classification-request"
+
+# client = LogStash::Filters::Empow::ClassificationCenterClient.new('assaf', , , )
+
+describe LogStash::Filters::Empow::ClassificationCenterClient do
+
+  # before(:each) do
+  #   local_classifier = instance_double(LogStash::Filters::Empow::LocalClassifier)
+  #   allow(LogStash::Filters::Empow::LocalClassifier).to receive(:new).and_return(local_classifier)
+  # end
+
+  let(:url_base) { 'https://s0apxz9wik.execute-api.us-east-2.amazonaws.com' }
+  let(:username) { 'assaf' }
+  let(:password) { 'Empow2018!' }
+  let(:pool_id) { '8dljcvt4jfif762le0ald6j' }
+
+  describe "classification center api" do
+    it "test missing ids request" do
+  		client = described_class.new(username, password, pool_id, url_base)
+
+  		client.authenticate
+
+      term = {}
+      term[:malware_name] = 'assaf.clicker'
+      req1 = LogStash::Filters::Empow::ClassificationRequest.new('AM', 'lastline', term)
+
+  		results = client.classify([req1])
+
+      #p results
+
+      results.each do |res|
+        p "res: #{res}"
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: logstash-filter-empowclassifier
 version: !ruby/object:Gem::Version
-  version: 0.3.18
+  version: 0.3.19
 platform: ruby
 authors:
 - empow
@@ -206,6 +206,7 @@ files:
 - lib/logstash/filters/response.rb
 - lib/logstash/filters/utils.rb
 - logstash-filter-empowclassifier.gemspec
+- spec/filters/assaf_spec.rb
 - spec/filters/bulk-processor_spec.rb
 - spec/filters/center-client_spec.rb
 - spec/filters/classifier-cache_spec.rb
@@ -246,6 +247,7 @@ specification_version: 4
 summary: Logstash intent classification plugin client for accessing empows classifiction
   cloud
 test_files:
+- spec/filters/assaf_spec.rb
 - spec/filters/bulk-processor_spec.rb
 - spec/filters/center-client_spec.rb
 - spec/filters/classifier-cache_spec.rb