@Override public ReadinessState process(ComponentRunner runner) { KafkaComponent kc = runner.getComponent("kafka", KafkaComponent.class); outputMessages.addAll(readMessagesFromKafka(kc, outputKafkaTopic)); return calcReadiness(inputMessages.size(), outputMessages.size()); }
@Override public ReadinessState process(ComponentRunner runner) { KafkaComponent kafkaComponent = runner.getComponent("kafka", KafkaComponent.class); try { docs = readDocsFromDisk(hdfsDir); } catch (IOException e) { throw new IllegalStateException("Unable to retrieve indexed documents.", e); } if (docs.size() < inputMessages.size()) { errors = kafkaComponent.readMessages(ERROR_TOPIC); if(errors.size() > 0 && errors.size() + docs.size() == inputMessages.size()){ return ReadinessState.READY; } return ReadinessState.NOT_READY; } else { return ReadinessState.READY; } }
@Override public ReadinessState process(ComponentRunner runner) { ElasticSearchComponent elasticSearchComponent = runner.getComponent("search", ElasticSearchComponent.class); KafkaComponent kafkaComponent = runner.getComponent("kafka", KafkaComponent.class); if (elasticSearchComponent.hasIndex(index)) { try { docs = elasticSearchComponent.getAllIndexedDocs(index, testSensorType + "_doc"); } catch (IOException e) { throw new IllegalStateException("Unable to retrieve indexed documents.", e); } if (docs.size() < inputMessages.size() ) { errors = kafkaComponent.readMessages(ERROR_TOPIC); if(errors.size() > 0 && errors.size() + docs.size() == inputMessages.size()){ return ReadinessState.READY; } return ReadinessState.NOT_READY; } else { return ReadinessState.READY; } } else { return ReadinessState.NOT_READY; } }