Search in sources :

Example 1 with VALIDATED

use of io.confluent.examples.streams.avro.microservices.OrderState.VALIDATED in project kafka-streams-examples by confluentinc.

the class ValidationsAggregatorService method aggregateOrderValidations.

private KafkaStreams aggregateOrderValidations(String bootstrapServers, String stateDir) {
    // TODO put into a KTable to make dynamically configurable
    final int numberOfRules = 3;
    StreamsBuilder builder = new StreamsBuilder();
    KStream<String, OrderValidation> validations = builder.stream(ORDER_VALIDATIONS.name(), serdes1);
    KStream<String, Order> orders = builder.stream(ORDERS.name(), serdes2).filter((id, order) -> OrderState.CREATED.equals(order.getState()));
    // If all rules pass then validate the order
    validations.groupByKey(serdes3).windowedBy(SessionWindows.with(5 * MIN)).aggregate(() -> 0L, (id, result, total) -> PASS.equals(result.getValidationResult()) ? total + 1 : total, // include a merger as we're using session windows.
    (k, a, b) -> b == null ? a : b, Materialized.with(null, Serdes.Long())).toStream((windowedKey, total) -> windowedKey.key()).filter((k1, v) -> v != null).filter((k, total) -> total >= numberOfRules).join(orders, (id, order) -> newBuilder(order).setState(VALIDATED).build(), JoinWindows.of(5 * MIN), serdes4).to(ORDERS.name(), serdes5);
    // If any rule fails then fail the order
    validations.filter((id, rule) -> FAIL.equals(rule.getValidationResult())).join(orders, (id, order) -> newBuilder(order).setState(OrderState.FAILED).build(), JoinWindows.of(5 * MIN), serdes7).groupByKey(serdes6).reduce((order, v1) -> order).toStream().to(ORDERS.name(), Produced.with(ORDERS.keySerde(), ORDERS.valueSerde()));
    return new KafkaStreams(builder.build(), baseStreamsConfig(bootstrapServers, stateDir, ORDERS_SERVICE_APP_ID));
}
Also used : StreamsBuilder(org.apache.kafka.streams.StreamsBuilder) Order(io.confluent.examples.streams.avro.microservices.Order) MIN(io.confluent.examples.streams.microservices.util.MicroserviceUtils.MIN) Order.newBuilder(io.confluent.examples.streams.avro.microservices.Order.newBuilder) Produced(org.apache.kafka.streams.kstream.Produced) SessionWindows(org.apache.kafka.streams.kstream.SessionWindows) Serialized(org.apache.kafka.streams.kstream.Serialized) LoggerFactory(org.slf4j.LoggerFactory) KStream(org.apache.kafka.streams.kstream.KStream) Joined(org.apache.kafka.streams.kstream.Joined) JoinWindows(org.apache.kafka.streams.kstream.JoinWindows) Consumed(org.apache.kafka.streams.Consumed) VALIDATED(io.confluent.examples.streams.avro.microservices.OrderState.VALIDATED) Serdes(org.apache.kafka.common.serialization.Serdes) ORDER_VALIDATIONS(io.confluent.examples.streams.microservices.domain.Schemas.Topics.ORDER_VALIDATIONS) Order(io.confluent.examples.streams.avro.microservices.Order) MicroserviceUtils.parseArgsAndConfigure(io.confluent.examples.streams.microservices.util.MicroserviceUtils.parseArgsAndConfigure) OrderState(io.confluent.examples.streams.avro.microservices.OrderState) StreamsBuilder(org.apache.kafka.streams.StreamsBuilder) Logger(org.slf4j.Logger) ORDERS(io.confluent.examples.streams.microservices.domain.Schemas.Topics.ORDERS) MicroserviceUtils.addShutdownHookAndBlock(io.confluent.examples.streams.microservices.util.MicroserviceUtils.addShutdownHookAndBlock) FAIL(io.confluent.examples.streams.avro.microservices.OrderValidationResult.FAIL) PASS(io.confluent.examples.streams.avro.microservices.OrderValidationResult.PASS) MicroserviceUtils.baseStreamsConfig(io.confluent.examples.streams.microservices.util.MicroserviceUtils.baseStreamsConfig) OrderValidation(io.confluent.examples.streams.avro.microservices.OrderValidation) Materialized(org.apache.kafka.streams.kstream.Materialized) KafkaStreams(org.apache.kafka.streams.KafkaStreams) KafkaStreams(org.apache.kafka.streams.KafkaStreams) OrderValidation(io.confluent.examples.streams.avro.microservices.OrderValidation)

Example 2 with VALIDATED

use of io.confluent.examples.streams.avro.microservices.OrderState.VALIDATED in project kafka-streams-examples by confluentinc.

the class ValidationsAggregatorServiceTest method shouldAggregateRuleSuccesses.

@Test
public void shouldAggregateRuleSuccesses() throws Exception {
    // Given
    ordersService = new ValidationsAggregatorService();
    orders = asList(new Order(id(0L), 0L, CREATED, UNDERPANTS, 3, 5.00d), new Order(id(1L), 0L, CREATED, JUMPERS, 1, 75.00d));
    sendOrders(orders);
    ruleResults = asList(new OrderValidation(id(0L), OrderValidationType.FRAUD_CHECK, OrderValidationResult.PASS), new OrderValidation(id(0L), OrderValidationType.ORDER_DETAILS_CHECK, OrderValidationResult.PASS), new OrderValidation(id(0L), OrderValidationType.INVENTORY_CHECK, OrderValidationResult.PASS), new OrderValidation(id(1L), OrderValidationType.FRAUD_CHECK, OrderValidationResult.PASS), new OrderValidation(id(1L), OrderValidationType.ORDER_DETAILS_CHECK, OrderValidationResult.FAIL), new OrderValidation(id(1L), OrderValidationType.INVENTORY_CHECK, OrderValidationResult.PASS));
    sendOrderValuations(ruleResults);
    // When
    ordersService.start(CLUSTER.bootstrapServers());
    // Then
    List<KeyValue<String, Order>> finalOrders = MicroserviceTestUtils.readKeyValues(Topics.ORDERS, 4, CLUSTER.bootstrapServers());
    assertThat(finalOrders.size()).isEqualTo(4);
    // And the first order should have been validated but the second should have failed
    assertThat(finalOrders.stream().map(kv -> kv.value).collect(Collectors.toList())).contains(new Order(id(0L), 0L, VALIDATED, UNDERPANTS, 3, 5.00d), new Order(id(1L), 0L, FAILED, JUMPERS, 1, 75.00d));
}
Also used : Order(io.confluent.examples.streams.avro.microservices.Order) MicroserviceTestUtils(io.confluent.examples.streams.microservices.util.MicroserviceTestUtils) OrderValidationResult(io.confluent.examples.streams.avro.microservices.OrderValidationResult) OrderValidationType(io.confluent.examples.streams.avro.microservices.OrderValidationType) JUMPERS(io.confluent.examples.streams.avro.microservices.Product.JUMPERS) BeforeClass(org.junit.BeforeClass) Assertions.assertThat(org.assertj.core.api.Assertions.assertThat) KeyValue(org.apache.kafka.streams.KeyValue) OrderId.id(io.confluent.examples.streams.microservices.domain.beans.OrderId.id) FAILED(io.confluent.examples.streams.avro.microservices.OrderState.FAILED) Test(org.junit.Test) Collectors(java.util.stream.Collectors) UNDERPANTS(io.confluent.examples.streams.avro.microservices.Product.UNDERPANTS) Topics(io.confluent.examples.streams.microservices.domain.Schemas.Topics) Schemas(io.confluent.examples.streams.microservices.domain.Schemas) List(java.util.List) Arrays.asList(java.util.Arrays.asList) After(org.junit.After) VALIDATED(io.confluent.examples.streams.avro.microservices.OrderState.VALIDATED) OrderValidation(io.confluent.examples.streams.avro.microservices.OrderValidation) CREATED(io.confluent.examples.streams.avro.microservices.OrderState.CREATED) Order(io.confluent.examples.streams.avro.microservices.Order) KeyValue(org.apache.kafka.streams.KeyValue) OrderValidation(io.confluent.examples.streams.avro.microservices.OrderValidation) Test(org.junit.Test)

Aggregations

Order (io.confluent.examples.streams.avro.microservices.Order)2 VALIDATED (io.confluent.examples.streams.avro.microservices.OrderState.VALIDATED)2 OrderValidation (io.confluent.examples.streams.avro.microservices.OrderValidation)2 Order.newBuilder (io.confluent.examples.streams.avro.microservices.Order.newBuilder)1 OrderState (io.confluent.examples.streams.avro.microservices.OrderState)1 CREATED (io.confluent.examples.streams.avro.microservices.OrderState.CREATED)1 FAILED (io.confluent.examples.streams.avro.microservices.OrderState.FAILED)1 OrderValidationResult (io.confluent.examples.streams.avro.microservices.OrderValidationResult)1 FAIL (io.confluent.examples.streams.avro.microservices.OrderValidationResult.FAIL)1 PASS (io.confluent.examples.streams.avro.microservices.OrderValidationResult.PASS)1 OrderValidationType (io.confluent.examples.streams.avro.microservices.OrderValidationType)1 JUMPERS (io.confluent.examples.streams.avro.microservices.Product.JUMPERS)1 UNDERPANTS (io.confluent.examples.streams.avro.microservices.Product.UNDERPANTS)1 Schemas (io.confluent.examples.streams.microservices.domain.Schemas)1 Topics (io.confluent.examples.streams.microservices.domain.Schemas.Topics)1 ORDERS (io.confluent.examples.streams.microservices.domain.Schemas.Topics.ORDERS)1 ORDER_VALIDATIONS (io.confluent.examples.streams.microservices.domain.Schemas.Topics.ORDER_VALIDATIONS)1 OrderId.id (io.confluent.examples.streams.microservices.domain.beans.OrderId.id)1 MicroserviceTestUtils (io.confluent.examples.streams.microservices.util.MicroserviceTestUtils)1 MIN (io.confluent.examples.streams.microservices.util.MicroserviceUtils.MIN)1