From 61044aa7f01aa548a01d0f943ef1d782ea819a6f Mon Sep 17 00:00:00 2001
From: John Spray <john@neon.tech>
Date: Fri, 22 Dec 2023 11:53:20 +0000
Subject: [PATCH] DNM demo script

---
 demo_sharding.sh | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/demo_sharding.sh b/demo_sharding.sh
index aff2090d16..189953d09e 100644
--- a/demo_sharding.sh
+++ b/demo_sharding.sh
@@ -1,9 +1,11 @@
 
 
 export RUST_LOG=DEBUG
-SHARDS=4
+INITIAL_SHARDS=4
+FINAL_SHARDS=4
+STRIPE_SIZE=1024
 SCALE=10
-ARGS=
+ARGS=--release
 
 set -e
 set -x
@@ -12,13 +14,16 @@ set +e
 cargo neon $ARGS stop ; killall -9 storage_broker ; killall -9 safekeeper ; killall -9 pageserver ; killall -9 postgres ; killall -9 attachment_service ; rm -rf .neon
 set -e
 
-cargo build $ARGS --package=pageserver
-cargo build $ARGS --package=control_plane
+cargo build $ARGS --features=testing
 
-cargo neon $ARGS init --num-pageservers=$SHARDS && RUST_LOG=debug cargo neon $ARGS start && cargo neon $ARGS tenant create --shard-count=$SHARDS --tenant-id=1f359dd625e519a1a4e8d7509690f6fc --timeline-id=3d34095be52fec4c44a92e774c573b57 --set-default
+cargo neon $ARGS init --num-pageservers=$FINAL_SHARDS && RUST_LOG=debug cargo neon $ARGS start && cargo neon $ARGS tenant create --shard-count=$INITIAL_SHARDS --shard-stripe-size=$STRIPE_SIZE  --tenant-id=1f359dd625e519a1a4e8d7509690f6fc --timeline-id=3d34095be52fec4c44a92e774c573b57 --set-default
 
-cargo neon $ARGS endpoint create && cargo neon endpoint start ep-main
+cargo neon $ARGS endpoint create && cargo neon $ARGS endpoint start ep-main
 
 pgbench postgres -i -h 127.0.0.1 -p 55432 -U cloud_admin -s $SCALE
 
-du -sh .neon/local_fs_remote_storage/pageserver/tenants/1f359dd625e519a1a4e8d7509690f6fc*
+cargo neon $ARGS tenant status
+
+cargo neon $ARGS tenant shard-split --shard-count=$FINAL_SHARDS
+
+cargo neon $ARGS tenant status