Merge branch 'master' into spiper-guardduty_support

airbnb · Feb 14, 2018 · 6291359 · 6291359
2 parents 8e380b9 + 4c2d663
commit 6291359
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/stream_alert/rule_processor/firehose.py b/stream_alert/rule_processor/firehose.py
@@ -286,3 +286,7 @@ def send(self):
                 self._limit_record_size(record_batch)
                 for sized_batch in self._segment_records_by_size(record_batch):
                     self._firehose_request_helper(stream_name, sized_batch)
+
+        # explicitly close firehose client to resolve connection reset issue, suggested
+        # by AWS support team.
+        self._firehose_client._endpoint.http_session.close() #pylint: disable=protected-access