protected void handleError(String sensorType, byte[] originalMessage, Tuple tuple, Throwable ex, OutputCollector collector) { MetronError error = new MetronError() .withErrorType(Constants.ErrorType.PARSER_ERROR) .withThrowable(ex) .withSensorType(Collections.singleton(sensorType)) .addRawMessage(originalMessage); ErrorUtils.handleError(collector, error); }
public void error(String sensorType, Throwable e, Iterable<Tuple> tuples, MessageGetStrategy messageGetStrategy) { LOG.error(format("Failing %d tuple(s); sensorType=%s", Iterables.size(tuples), sensorType), e); tuples.forEach(t -> { MetronError error = new MetronError() .withSensorType(Collections.singleton(sensorType)) .withErrorType(Constants.ErrorType.INDEXING_ERROR) .withThrowable(e) .addRawMessage(messageGetStrategy.get(t)); collector.emit(Constants.ERROR_STREAM, new Values(error.getJSONObject())); }); if (handleCommit) { commit(tuples); } // there is only one error to report for all of the failed tuples collector.reportError(e); }
protected void handleError(String key, JSONObject rawMessage, String subGroup, JSONObject enrichedMessage, Exception e) { LOG.error("[Metron] Unable to enrich message: {}", rawMessage, e); if (key != null) { collector.emit(enrichmentType, new Values(key, enrichedMessage, subGroup)); } MetronError error = new MetronError() .withErrorType(Constants.ErrorType.ENRICHMENT_ERROR) .withThrowable(e) .addRawMessage(rawMessage); ErrorUtils.handleError(collector, error); }
@Test public void getJSONObjectShouldIncludeErrorFields() { JSONObject message = new JSONObject(); message.put("field1", "value1"); message.put("field2", "value2"); MetronError error = new MetronError().addRawMessage(message).withErrorFields(Sets.newHashSet("field1", "field2")); JSONObject errorJSON = error.getJSONObject(); assertEquals(Sets.newHashSet("field1", "field2"), Sets.newHashSet(((String) errorJSON.get(Constants.ErrorFields.ERROR_FIELDS.getName())).split(","))); assertEquals("04a2629c39e098c3944be85f35c75876598f2b44b8e5e3f52c59fa1ac182817c", errorJSON.get(Constants.ErrorFields.ERROR_HASH.getName())); } }
.withThrowable(throwable) .withSensorType(Collections.singleton(sensorType)) .addRawMessage(rawMessage.getMessage()))); .withThrowable(entry.getValue()) .withSensorType(Collections.singleton(sensorType)) .addRawMessage(entry.getKey())).collect(Collectors.toList()));
.withMessage("Joining problem: " + message) .withThrowable(e) .addRawMessage(message); ErrorUtils.handleError(collector, error); collector.ack(tuple);
.withErrorType(Constants.ErrorType.PARSER_INVALID) .withSensorType(Collections.singleton(sensorType)) .addRawMessage(message); Set<String> errorFields = failedValidators == null ? null : failedValidators.stream() .flatMap(fieldValidator -> fieldValidator.getInput().stream())
@Test public void testNonBatchErrorPathErrorInWrite() throws Exception { ParserConfigurations configurations = getConfigurations(1); String sensorType = "test"; Tuple t = mock(Tuple.class); when(t.toString()).thenReturn("tuple"); when(t.getValueByField(eq("message"))).thenReturn(new JSONObject()); WriterBolt bolt = new WriterBolt(new WriterHandler(writer), configurations, sensorType); bolt.prepare(new HashMap(), topologyContext, outputCollector); doThrow(new Exception("write error")).when(writer).write(any(), any(), any(), any()); verify(writer, times(1)).init(); bolt.execute(t); verify(outputCollector, times(1)).ack(t); verify(writer, times(1)).write(eq(sensorType), any(), any(), any()); verify(outputCollector, times(1)).reportError(any()); verify(outputCollector, times(0)).fail(any()); MetronError error = new MetronError() .withErrorType(Constants.ErrorType.DEFAULT_ERROR) .withThrowable(new IllegalStateException("Unhandled bulk errors in response: {java.lang.Exception: write error=[tuple]}")) .withSensorType(Collections.singleton(sensorType)) .addRawMessage(new JSONObject()); verify(outputCollector, times(1)).emit(eq(Constants.ERROR_STREAM), argThat(new MetronErrorJSONMatcher(error.getJSONObject()))); }
@SuppressWarnings("unchecked") @Test public void testExecuteShouldReportError() throws ExecutionException { joinBolt.withMaxCacheSize(100); joinBolt.withMaxTimeRetain(10000); joinBolt.prepare(new HashMap(), topologyContext, outputCollector); when(tuple.getValueByField("key")).thenReturn(key); when(tuple.getValueByField("message")).thenReturn(new JSONObject()); joinBolt.cache = mock(LoadingCache.class); when(joinBolt.cache.get(any())).thenThrow(new RuntimeException(new Exception("join exception"))); joinBolt.execute(tuple); RuntimeException expectedExecutionException = new RuntimeException(new Exception("join exception")); MetronError error = new MetronError() .withErrorType(Constants.ErrorType.ENRICHMENT_ERROR) .withMessage("Joining problem: {}") .withThrowable(expectedExecutionException) .addRawMessage(new JSONObject()); verify(outputCollector, times(1)).emit(eq(Constants.ERROR_STREAM), argThat(new MetronErrorJSONMatcher(error.getJSONObject()))); verify(outputCollector, times(1)).reportError(any(ExecutionException.class)); verify(outputCollector, times(1)).ack(eq(tuple)); verifyNoMoreInteractions(outputCollector); } }
@Test public void shouldExecuteWithMasterThrowable() { parserRunner = spy(parserRunner); RawMessage rawMessage = new RawMessage("raw_message".getBytes(), new HashMap<>()); Throwable masterThrowable = mock(Throwable.class); MessageParserResult<JSONObject> messageParserResult = new DefaultMessageParserResult<>(masterThrowable); when(broParser.parseOptionalResult(rawMessage.getMessage())).thenReturn(Optional.of(messageParserResult)); parserRunner.setSensorToParserComponentMap(new HashMap<String, ParserComponent>() {{ put("bro", new ParserComponent(broParser, stellarFilter)); }}); ParserRunnerResults<JSONObject> parserRunnerResults = parserRunner.execute("bro", rawMessage, parserConfigurations); verify(parserRunner, times(0)) .processMessage(any(), any(), any(), any(), any()); MetronError expectedError = new MetronError() .withErrorType(Constants.ErrorType.PARSER_ERROR) .withThrowable(masterThrowable) .withSensorType(Collections.singleton("bro")) .addRawMessage(rawMessage.getMessage()); Assert.assertEquals(1, parserRunnerResults.getErrors().size()); Assert.assertTrue(parserRunnerResults.getErrors().contains(expectedError)); }
.withErrorType(Constants.ErrorType.PARSER_INVALID) .withSensorType(Collections.singleton("bro")) .addRawMessage(inputMessage) .withErrorFields(new HashSet<>(Arrays.asList("ip_src_addr", "ip_dst_addr")));
.withMessage(t.getValue().getMessage()) .withThrowable(t.getValue()) .addRawMessage(t.getKey()); ErrorUtils.handleError(collector, error); .withMessage(e.getMessage()) .withThrowable(e) .addRawMessage(message); ErrorUtils.handleError(collector, error);
@Override public void execute(Tuple tuple) { JSONObject message = null; try { message = (JSONObject) messageGetStrategy.get(tuple); handler.write(sensorType, tuple, message, configuration, messageGetStrategy); if(!handler.handleAck()) { collector.ack(tuple); } } catch (Throwable e) { MetronError error = new MetronError() .withErrorType(errorType) .withThrowable(e) .withSensorType(Collections.singleton(sensorType)) .addRawMessage(message); ErrorUtils.handleError(collector, error); collector.ack(tuple); } }
@Test public void getJSONObjectShouldIncludeRawMessages() { JSONObject message1 = new JSONObject(); JSONObject message2 = new JSONObject(); message1.put("value", "message1"); message2.put("value", "message2"); MetronError error = new MetronError().withRawMessages(Arrays.asList(message1, message2)); JSONObject errorJSON = error.getJSONObject(); assertEquals("{\"value\":\"message1\"}", errorJSON.get(Constants.ErrorFields.RAW_MESSAGE.getName() + "_0")); assertEquals("{\"value\":\"message2\"}", errorJSON.get(Constants.ErrorFields.RAW_MESSAGE.getName() + "_1")); error = new MetronError().addRawMessage("raw message".getBytes()); errorJSON = error.getJSONObject(); assertEquals("raw message", errorJSON.get(Constants.ErrorFields.RAW_MESSAGE.getName())); // It's unclear if we need a rawMessageBytes field so commenting out for now //assertEquals(Bytes.asList("raw message".getBytes()), errorJSON.get(Constants.ErrorFields.RAW_MESSAGE_BYTES.getName())); assertEquals("3b02cb29676bc448c69da1ec5eef7c89f4d6dc6a5a7ce0296ea25b207eea36be", errorJSON.get(Constants.ErrorFields.ERROR_HASH.getName())); error = new MetronError().addRawMessage(message1); errorJSON = error.getJSONObject(); assertEquals("{\"value\":\"message1\"}", errorJSON.get(Constants.ErrorFields.RAW_MESSAGE.getName())); assertEquals("e8aaf87c8494d345aac2d612ffd94fcf0b98c975fe6c4b991e2f8280a3a0bd10", errorJSON.get(Constants.ErrorFields.ERROR_HASH.getName())); }
@Test public void shouldReturnMetronErrorOnInvalidMessage() { JSONObject inputMessage = new JSONObject(); inputMessage.put("guid", "guid"); RawMessage rawMessage = new RawMessage("raw_message".getBytes(), new HashMap<>()); JSONObject expectedOutput = new JSONObject(); expectedOutput.put("guid", "guid"); expectedOutput.put("source.type", "bro"); MetronError expectedMetronError = new MetronError() .withErrorType(Constants.ErrorType.PARSER_INVALID) .withSensorType(Collections.singleton("bro")) .addRawMessage(inputMessage); when(stellarFilter.emit(expectedOutput, parserRunner.getStellarContext())).thenReturn(true); when(broParser.validate(expectedOutput)).thenReturn(false); parserRunner.setSensorToParserComponentMap(new HashMap<String, ParserComponent>() {{ put("bro", new ParserComponent(broParser, stellarFilter)); }}); Optional<ParserRunnerImpl.ProcessResult> processResult = parserRunner.processMessage("bro", inputMessage, rawMessage, broParser, parserConfigurations); Assert.assertTrue(processResult.isPresent()); Assert.assertTrue(processResult.get().isError()); Assert.assertEquals(expectedMetronError, processResult.get().getError()); }
@Override public ParserRunnerResults<JSONObject> execute(String sensorType, RawMessage rawMessage, ParserConfigurations parserConfigurations) { DefaultParserRunnerResults parserRunnerResults = new DefaultParserRunnerResults(); this.rawMessage = rawMessage; if (!isInvalid) { parserRunnerResults.addMessage(message); } else { MetronError error = new MetronError() .withErrorType(Constants.ErrorType.PARSER_INVALID) .withSensorType(Collections.singleton(sensorType)) .addRawMessage(message); parserRunnerResults.addError(error); } return parserRunnerResults; }
.withThrowable(new IllegalStateException("parserRunner.execute failed")) .withSensorType(Collections.singleton("yaf")) .addRawMessage("originalMessage".getBytes(StandardCharsets.UTF_8));
.withThrowable(new IllegalStateException("write failed")) .withSensorType(Collections.singleton("yaf")) .addRawMessage("originalMessage".getBytes(StandardCharsets.UTF_8));
.withThrowable(throwable1) .withSensorType(Collections.singleton("bro")) .addRawMessage(rawMessage1); MetronError expectedParseError2 = new MetronError() .withErrorType(Constants.ErrorType.PARSER_ERROR) .withThrowable(throwable2) .withSensorType(Collections.singleton("bro")) .addRawMessage(rawMessage2);
.withErrorType(Constants.ErrorType.PARSER_INVALID) .withSensorType(Collections.singleton("yaf")) .addRawMessage(message);