How to sum a stream of events

Problem:

you have data in a Kafka topic and want to calculate the sum of one or more fields.

Edit this page

Example use case:

Suppose you have a topic with events that represent ticket sales for movies. Each event contains the movie that the ticket was purchased for as well as its price. In this tutorial, we'll write a program that calculates the sum of all ticket sales per movie.

Code example:

Try it

1
Initialize the project

To get started, make a new directory anywhere you’d like for this project:

mkdir aggregate-sum && cd aggregate-sum

Then make the following directories to set up its structure:

mkdir src test

2
Get Confluent Platform

Next, create the following docker-compose.yml file to obtain Confluent Platform:

---
version: '2'

services:
  zookeeper:
    image: confluentinc/cp-zookeeper:5.3.0
    hostname: zookeeper
    container_name: zookeeper
    ports:
      - "2181:2181"
    environment:
      ZOOKEEPER_CLIENT_PORT: 2181
      ZOOKEEPER_TICK_TIME: 2000

  broker:
    image: confluentinc/cp-enterprise-kafka:5.3.0
    hostname: broker
    container_name: broker
    depends_on:
      - zookeeper
    ports:
      - "29092:29092"
    environment:
      KAFKA_BROKER_ID: 1
      KAFKA_ZOOKEEPER_CONNECT: 'zookeeper:2181'
      KAFKA_LISTENER_SECURITY_PROTOCOL_MAP: PLAINTEXT:PLAINTEXT,PLAINTEXT_HOST:PLAINTEXT
      KAFKA_ADVERTISED_LISTENERS: PLAINTEXT://broker:9092,PLAINTEXT_HOST://localhost:29092
      KAFKA_METRIC_REPORTERS: io.confluent.metrics.reporter.ConfluentMetricsReporter
      KAFKA_OFFSETS_TOPIC_REPLICATION_FACTOR: 1
      KAFKA_GROUP_INITIAL_REBALANCE_DELAY_MS: 0
      CONFLUENT_METRICS_REPORTER_BOOTSTRAP_SERVERS: broker:9092
      CONFLUENT_METRICS_REPORTER_ZOOKEEPER_CONNECT: zookeeper:2181
      CONFLUENT_METRICS_REPORTER_TOPIC_REPLICAS: 1
      CONFLUENT_METRICS_ENABLE: 'true'
      CONFLUENT_SUPPORT_CUSTOMER_ID: 'anonymous'

  schema-registry:
    image: confluentinc/cp-schema-registry:5.3.0
    hostname: schema-registry
    container_name: schema-registry
    depends_on:
      - zookeeper
      - broker
    ports:
      - "8081:8081"
    environment:
      SCHEMA_REGISTRY_HOST_NAME: schema-registry
      SCHEMA_REGISTRY_KAFKASTORE_CONNECTION_URL: 'zookeeper:2181'

  ksql-server:
    image: confluentinc/cp-ksql-server:5.3.0
    hostname: ksql-server
    container_name: ksql-server
    depends_on:
      - broker
      - schema-registry
    ports:
      - "8088:8088"
    environment:
      KSQL_CONFIG_DIR: "/etc/ksql"
      KSQL_LOG4J_OPTS: "-Dlog4j.configuration=file:/etc/ksql/log4j-rolling.properties"
      KSQL_BOOTSTRAP_SERVERS: "broker:9092"
      KSQL_HOST_NAME: ksql-server
      KSQL_APPLICATION_ID: "cp-all-in-one"
      KSQL_LISTENERS: "http://0.0.0.0:8088"
      KSQL_CACHE_MAX_BYTES_BUFFERING: 0
      KSQL_KSQL_SCHEMA_REGISTRY_URL: "http://schema-registry:8081"
      KSQL_PRODUCER_INTERCEPTOR_CLASSES: "io.confluent.monitoring.clients.interceptor.MonitoringProducerInterceptor"
      KSQL_CONSUMER_INTERCEPTOR_CLASSES: "io.confluent.monitoring.clients.interceptor.MonitoringConsumerInterceptor"

  ksql-cli:
    image: confluentinc/cp-ksql-cli:5.3.0
    container_name: ksql-cli
    depends_on:
      - broker
      - ksql-server
    entrypoint: /bin/sh
    tty: true
    volumes:
      - ./src:/opt/app/src
      - ./test:/opt/app/test

And launch it by running:

docker-compose up

3
Write the program interactively using the CLI

To begin developing interactively, open up the KSQL CLI:

docker exec -it ksql-cli ksql http://ksql-server:8088

First, you’ll need to create a Kafka topic and stream to represent the ticket sales. The statement below creates both at the same time. This stream contains the name of the movie and the price of the ticket to watch it. We model the price as an integer to make the example simple.

Another important characteristic of the data is the timestamp column, sale_ts. Every message in Kafka is timestamped, and unless you specify otherwise, KSQL will use that existing timestamp for any time-related processing. In this example, we’re telling it to use a field in the message for the timestamp. This is called the event time rather than the ingestion time.

CREATE STREAM MOVIE_TICKET_SALES (title VARCHAR, sale_ts VARCHAR, ticket_total_value INT)
    WITH (KAFKA_TOPIC='movie-ticket-sales',
          PARTITIONS=1,
          VALUE_FORMAT='avro',
          TIMESTAMP='sale_ts',
          TIMESTAMP_FORMAT='yyyy-MM-dd''T''HH:mm:ssX');

With the stream in place, we can now produce the following events to it:

INSERT INTO MOVIE_TICKET_SALES (title, sale_ts, ticket_total_value) VALUES ('Die Hard',         '2019-07-18T10:00:00Z', 12);
INSERT INTO MOVIE_TICKET_SALES (title, sale_ts, ticket_total_value) VALUES ('Die Hard',         '2019-07-18T10:01:00Z', 12);
INSERT INTO MOVIE_TICKET_SALES (title, sale_ts, ticket_total_value) VALUES ('The Godfather',    '2019-07-18T10:01:31Z', 12);
INSERT INTO MOVIE_TICKET_SALES (title, sale_ts, ticket_total_value) VALUES ('Die Hard',         '2019-07-18T10:01:36Z', 24);
INSERT INTO MOVIE_TICKET_SALES (title, sale_ts, ticket_total_value) VALUES ('The Godfather',    '2019-07-18T10:02:00Z', 18);
INSERT INTO MOVIE_TICKET_SALES (title, sale_ts, ticket_total_value) VALUES ('The Big Lebowski', '2019-07-18T11:03:21Z', 12);
INSERT INTO MOVIE_TICKET_SALES (title, sale_ts, ticket_total_value) VALUES ('The Big Lebowski', '2019-07-18T11:03:50Z', 12);
INSERT INTO MOVIE_TICKET_SALES (title, sale_ts, ticket_total_value) VALUES ('The Godfather',    '2019-07-18T11:40:00Z', 36);
INSERT INTO MOVIE_TICKET_SALES (title, sale_ts, ticket_total_value) VALUES ('The Godfather',    '2019-07-18T11:40:09Z', 18);

Before we continue, let’s make sure that each KSQL query we execute will begin its processing from the beginning of the stream:

SET 'auto.offset.reset' = 'earliest';

For the purposes of this example only, we are also going to configure KSQL to buffer the aggregates as it builds them. This makes the query feel like it responds more slowly, but means that you get just one row per movie from which it is simpler to understand the concept:

SET 'ksql.streams.cache.max.bytes.buffering' = '10000000';

Let’s calculate the total sales per movie using a SUM aggregation on the TICKET_TOTAL_VALUE field. This will block and continue to return results until it’s limit is reached or you tell it to stop.

SELECT TITLE,
       SUM(TICKET_TOTAL_VALUE) AS TOTAL_VALUE
FROM MOVIE_TICKET_SALES
GROUP BY TITLE
LIMIT 3;

This should yield the following output:

Die Hard | 48
The Big Lebowski | 24
The Godfather | 84
Limit Reached
Query terminated

Since the output looks right, the next step is to make the query persistent. We do this with the CREATE TABLE AS statement. This statement creates a stream processor that runs continuously, always consuming events from the source stream (MOVIE_TICKET_SALES) and creating and updating entries in the resulting table (MOVIE_REVENUE).

It should not escape your notice that we are turning a stream into a table. A table is always the result of using the GROUP BY clause on a stream. As we noted in the previous step, we are also computing an aggregate over the grouped values with SUM(TICKET_TOTAL_VALUE). This function createes a new column in the resulting table, which we give a readable name using the AS TOTAL_VALUE clause.

Issue the following to create the new table:

CREATE TABLE MOVIE_REVENUE AS
    SELECT TITLE,
           SUM(TICKET_TOTAL_VALUE) AS TOTAL_VALUE
    FROM MOVIE_TICKET_SALES
    GROUP BY TITLE;

To check that it’s working, print out the contents of the output stream’s underlying topic:

PRINT 'MOVIE_REVENUE' FROM BEGINNING LIMIT 3;

This should yield the following output:

Format:AVRO
7/18/19 10:01:36 AM UTC, Die Hard, {"TITLE": "Die Hard", "TOTAL_VALUE": 48}
7/18/19 11:03:50 AM UTC, The Big Lebowski, {"TITLE": "The Big Lebowski", "TOTAL_VALUE": 24}
7/18/19 11:40:09 AM UTC, The Godfather, {"TITLE": "The Godfather", "TOTAL_VALUE": 84}

4
Write your statements to a file

Now that you have a series of statements that’s doing the right thing, the last step is to put them into a file so that they can be used outside the CLI session. Create a file at src/statements.sql with the following content:

CREATE STREAM MOVIE_TICKET_SALES (title VARCHAR, sale_ts VARCHAR, ticket_total_value INT)
    WITH (KAFKA_TOPIC='movie-ticket-sales',
          PARTITIONS=1,
          VALUE_FORMAT='avro',
          TIMESTAMP='sale_ts',
          TIMESTAMP_FORMAT='yyyy-MM-dd''T''HH:mm:ssX');

CREATE TABLE MOVIE_REVENUE AS
    SELECT TITLE,
           SUM(TICKET_TOTAL_VALUE) AS TOTAL_VALUE
    FROM MOVIE_TICKET_SALES
    GROUP BY TITLE;

Test it

1
Create the test data

Create a file at test/input.json with the inputs for testing:

{
  "inputs": [
    {"topic": "movie-ticket-sales" ,"key": null, "value": {"TITLE": "Die Hard", "SALE_TS": "2019-07-18T10:00:00Z", "TICKET_TOTAL_VALUE": 12}},
    {"topic": "movie-ticket-sales" ,"key": null, "value": {"TITLE": "Die Hard", "SALE_TS": "2019-07-18T10:01:00Z", "TICKET_TOTAL_VALUE": 12}},
    {"topic": "movie-ticket-sales" ,"key": null, "value": {"TITLE": "The Godfather", "SALE_TS": "2019-07-18T10:01:31Z", "TICKET_TOTAL_VALUE": 12}},
    {"topic": "movie-ticket-sales" ,"key": null, "value": {"TITLE": "Die Hard", "SALE_TS": "2019-07-18T10:01:36Z", "TICKET_TOTAL_VALUE": 24}},
    {"topic": "movie-ticket-sales" ,"key": null, "value": {"TITLE": "The Godfather", "SALE_TS": "2019-07-18T10:02:00Z", "TICKET_TOTAL_VALUE": 18}},
    {"topic": "movie-ticket-sales" ,"key": null, "value": {"TITLE": "The Big Lebowski", "SALE_TS": "2019-07-18T11:03:21Z", "TICKET_TOTAL_VALUE": 12}},
    {"topic": "movie-ticket-sales" ,"key": null, "value": {"TITLE": "The Big Lebowski", "SALE_TS": "2019-07-18T11:03:50Z", "TICKET_TOTAL_VALUE": 12}},
    {"topic": "movie-ticket-sales" ,"key": null, "value": {"TITLE": "The Godfather", "SALE_TS": "2019-07-18T11:40:00Z", "TICKET_TOTAL_VALUE": 36}},
    {"topic": "movie-ticket-sales" ,"key": null, "value": {"TITLE": "The Godfather", "SALE_TS": "2019-07-18T11:40:09Z", "TICKET_TOTAL_VALUE": 18}}
  ]
}

Similarly, create a file at test/output.json with the expected outputs:

{
  "outputs": [
    {"topic": "MOVIE_REVENUE", "key": "Die Hard", "value": {"TITLE": "Die Hard", "TOTAL_VALUE": 12}, "timestamp": 1563444000000},
    {"topic": "MOVIE_REVENUE", "key": "Die Hard", "value": {"TITLE": "Die Hard", "TOTAL_VALUE": 24}, "timestamp": 1563444060000},
    {"topic": "MOVIE_REVENUE", "key": "The Godfather", "value": {"TITLE": "The Godfather", "TOTAL_VALUE": 12}, "timestamp": 1563444091000},
    {"topic": "MOVIE_REVENUE", "key": "Die Hard", "value": {"TITLE":"Die Hard", "TOTAL_VALUE": 48}, "timestamp": 1563444096000},
    {"topic": "MOVIE_REVENUE", "key": "The Godfather", "value": {"TITLE": "The Godfather","TOTAL_VALUE": 30}, "timestamp": 1563444120000},
    {"topic": "MOVIE_REVENUE", "key": "The Big Lebowski", "value": {"TITLE": "The Big Lebowski", "TOTAL_VALUE": 12}, "timestamp": 1563447801000},
    {"topic": "MOVIE_REVENUE", "key": "The Big Lebowski", "value": {"TITLE": "The Big Lebowski", "TOTAL_VALUE": 24}, "timestamp": 1563447830000},
    {"topic": "MOVIE_REVENUE", "key": "The Godfather", "value": {"TITLE": "The Godfather","TOTAL_VALUE": 66}, "timestamp": 1563450000000},
    {"topic": "MOVIE_REVENUE", "key": "The Godfather", "value": {"TITLE": "The Godfather","TOTAL_VALUE": 84}, "timestamp": 1563450009000}
  ]
}

2
Invoke the tests

Lastly, invoke the tests using the test runner and the statements file that you created earlier:

docker exec ksql-cli ksql-test-runner -i /opt/app/test/input.json -s /opt/app/src/statements.sql -o /opt/app/test/output.json

Which should pass:

	 >>> Test passed!

Take it to production

1
Send the statements to the REST API

Launch your statements into production by sending them to the REST API with the following command:

statements=$(< src/statements.sql) && \
    echo '{"ksql":"'$statements'", "streamsProperties": {}}' | \
        curl -X "POST" "http://localhost:8088/ksql" \
             -H "Content-Type: application/vnd.ksql.v1+json; charset=utf-8" \
             -d @- | \
        jq