[tiktok] Make Kafka endpoint configurable.

Leonid Ryzhyk · ryzhyk · commit 3f05e47e3885 · 2024-08-19T13:53:43.000-07:00
Allow specifying custom Kafka broker address in the tiktok benchmark.

Signed-off-by: Leonid Ryzhyk &lt;leonid@feldera.com&gt;
diff --git a/benchmark/feldera-sql/benchmarks/tiktok/README.md b/benchmark/feldera-sql/benchmarks/tiktok/README.md
@@ -3,10 +3,12 @@
 
 1. Ensure that you have Feldera is running.
 2. Ensure that you have Kafka running.
-3. Set the environment variable `MAX_EVENTS` to the number of events to generate. 
+3. Set the environment variable `MAX_EVENTS` to the number of events to generate.
    `MAX_EVENTS=100000000`
-4. Run `generate.bash`. (`source generate.bash`)
-5. Run the `feldera/benchmark/feldera-sql/run.py` script as follows:
+4. Set the environment variable `BOOTSTRAP_SERVERS` to point to the Kafka port
+   `BOOTSTRAP_SERVERS=localhost:9092`
+5. Run `generate.bash`. (`source generate.bash`)
+6. Run the `feldera/benchmark/feldera-sql/run.py` script as follows:
 
 ```sh
 cd feldera/benchmark/feldera-sql
diff --git a/benchmark/feldera-sql/benchmarks/tiktok/generate.bash b/benchmark/feldera-sql/benchmarks/tiktok/generate.bash
@@ -2,5 +2,5 @@ BASEDIR="$(pwd)" # get the current directory
 cd "$(dirname "${BASH_SOURCE[0]}")" # cd to the directory this script is in
 TIKTOK_GEN="$(realpath ../../../../demo/project_demo12-HopsworksTikTokRecSys/tiktok-gen)"
 cd $TIKTOK_GEN # cd to the tiktok generator directory
-cargo run --release -- --historical --delete-topic-if-exists -I ${MAX_EVENTS:-50000000}
+cargo run --release -- --historical --delete-topic-if-exists -I ${MAX_EVENTS:-50000000} -B ${BOOTSTRAP_SERVERS:-localhost:9092}
 cd $BASEDIR
