Repository: nifi Updated Branches: refs/heads/0.x 714a90bbd -> 5748e381a
NIFI-1913 Properly removing flowfile in HandleHttpRequest when the Context Map fails to register the request This closes #462 Project: http://git-wip-us.apache.org/repos/asf/nifi/repo Commit: http://git-wip-us.apache.org/repos/asf/nifi/commit/5748e381 Tree: http://git-wip-us.apache.org/repos/asf/nifi/tree/5748e381 Diff: http://git-wip-us.apache.org/repos/asf/nifi/diff/5748e381 Branch: refs/heads/0.x Commit: 5748e381a53791df33cff9a7a7e0ddda5d454751 Parents: 714a90b Author: jpercivall <[email protected]> Authored: Mon May 23 12:24:43 2016 -0400 Committer: jpercivall <[email protected]> Committed: Tue May 24 10:53:55 2016 -0400 ---------------------------------------------------------------------- .../processors/standard/HandleHttpRequest.java | 1 + .../standard/TestHandleHttpRequest.java | 80 +++++++++++++++++++- 2 files changed, 79 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/nifi/blob/5748e381/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/main/java/org/apache/nifi/processors/standard/HandleHttpRequest.java ---------------------------------------------------------------------- diff --git a/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/main/java/org/apache/nifi/processors/standard/HandleHttpRequest.java b/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/main/java/org/apache/nifi/processors/standard/HandleHttpRequest.java index b0e91b4..5919623 100644 --- a/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/main/java/org/apache/nifi/processors/standard/HandleHttpRequest.java +++ b/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/main/java/org/apache/nifi/processors/standard/HandleHttpRequest.java @@ -609,6 +609,7 @@ public class HandleHttpRequest extends AbstractProcessor { new Object[]{request.getRemoteAddr(), e}); } + session.remove(flowFile); return; } http://git-wip-us.apache.org/repos/asf/nifi/blob/5748e381/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestHandleHttpRequest.java ---------------------------------------------------------------------- diff --git a/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestHandleHttpRequest.java b/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestHandleHttpRequest.java index 11c47ce..c6bb337 100644 --- a/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestHandleHttpRequest.java +++ b/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestHandleHttpRequest.java @@ -101,14 +101,82 @@ public class TestHandleHttpRequest { } } + + @Test(timeout=10000) + public void testFailToRegister() throws InitializationException, MalformedURLException, IOException, InterruptedException { + final TestRunner runner = TestRunners.newTestRunner(HandleHttpRequest.class); + runner.setProperty(HandleHttpRequest.PORT, "0"); + + final MockHttpContextMap contextMap = new MockHttpContextMap(); + runner.addControllerService("http-context-map", contextMap); + runner.enableControllerService(contextMap); + runner.setProperty(HandleHttpRequest.HTTP_CONTEXT_MAP, "http-context-map"); + contextMap.setRegisterSuccessfully(false); + + // trigger processor to stop but not shutdown. + runner.run(1, false); + try { + final int[] responseCode = new int[1]; + responseCode[0] = 0; + final Thread httpThread = new Thread(new Runnable() { + @Override + public void run() { + HttpURLConnection connection = null; + try { + final int port = ((HandleHttpRequest) runner.getProcessor()).getPort(); + connection = (HttpURLConnection) new URL("http://localhost:"; + + port + "/my/path?query=true&value1=value1&value2=&value3&value4=apple=orange").openConnection(); + connection.setDoOutput(false); + connection.setRequestMethod("GET"); + connection.setRequestProperty("header1", "value1"); + connection.setRequestProperty("header2", ""); + connection.setRequestProperty("header3", "apple=orange"); + connection.setConnectTimeout(3000); + connection.setReadTimeout(3000); + + StreamUtils.copy(connection.getInputStream(), new NullOutputStream()); + } catch (final Throwable t) { + t.printStackTrace(); + if(connection != null ) { + try { + responseCode[0] = connection.getResponseCode(); + } catch (IOException e) { + responseCode[0] = -1; + } + } else { + responseCode[0] = -2; + } + } + } + }); + httpThread.start(); + + while (responseCode[0] == 0) { + // process the request. + runner.run(1, false, false); + } + + runner.assertTransferCount(HandleHttpRequest.REL_SUCCESS, 0); + assertEquals(503, responseCode[0]); + + } finally { + // shut down the server + runner.run(1, true); + } + } + private static class MockHttpContextMap extends AbstractControllerService implements HttpContextMap { + private boolean registerSuccessfully = true; + private final ConcurrentMap<String, HttpServletResponse> responseMap = new ConcurrentHashMap<>(); @Override public boolean register(final String identifier, final HttpServletRequest request, final HttpServletResponse response, final AsyncContext context) { - responseMap.put(identifier, response); - return true; + if(registerSuccessfully) { + responseMap.put(identifier, response); + } + return registerSuccessfully; } @Override @@ -124,5 +192,13 @@ public class TestHandleHttpRequest { public int size() { return responseMap.size(); } + + public boolean isRegisterSuccessfully() { + return registerSuccessfully; + } + + public void setRegisterSuccessfully(boolean registerSuccessfully) { + this.registerSuccessfully = registerSuccessfully; + } } }
