From a1cb93191c844ab4cd4cd2a127560c9867c5edf7 Mon Sep 17 00:00:00 2001 From: Noe Date: Tue, 12 Sep 2017 14:19:11 -0400 Subject: [PATCH] NIFI-3349 retry stale connections --- .../apache/nifi/processors/splunk/GetSplunk.java | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/nifi-nar-bundles/nifi-splunk-bundle/nifi-splunk-processors/src/main/java/org/apache/nifi/processors/splunk/GetSplunk.java b/nifi-nar-bundles/nifi-splunk-bundle/nifi-splunk-processors/src/main/java/org/apache/nifi/processors/splunk/GetSplunk.java index 956a6538930f..7e0092ba7b2d 100644 --- a/nifi-nar-bundles/nifi-splunk-bundle/nifi-splunk-processors/src/main/java/org/apache/nifi/processors/splunk/GetSplunk.java +++ b/nifi-nar-bundles/nifi-splunk-bundle/nifi-splunk-processors/src/main/java/org/apache/nifi/processors/splunk/GetSplunk.java @@ -458,7 +458,18 @@ public void onTrigger(ProcessContext context, ProcessSession session) throws Pro getLogger().debug("Using index_earliest of {} and index_latest of {}", new Object[]{earliestTime, latestTime}); } - final InputStream exportSearch = splunkService.export(query, exportArgs); + InputStream export; + try{ + export = splunkService.export(query, exportArgs); + //Catch Stale connection exception, reinitialize, and retry + } catch (com.splunk.HttpException e) { + getLogger().error("Splunk request status code:" + e.getStatus() + " Retrying the request."); + splunkService.logout(); + splunkService = createSplunkService(context); + export = splunkService.export(query, exportArgs); + } + + final InputStream exportSearch = export; FlowFile flowFile = session.create(); flowFile = session.write(flowFile, new OutputStreamCallback() {