From d8ddf673a6f3be7a133e12c46291e59699ffa2d2 Mon Sep 17 00:00:00 2001
From: cjcchen <ickchenjunchao@gmail.com>
Date: Sun, 8 Dec 2024 12:35:41 +0800
Subject: [PATCH] Use custom Ports (#165)

* add port reset
---
 README.md                                     | 16 ++++
 service/tools/config/generate_config.sh       | 92 +++++++++++++++++++
 .../tools/kv/server_tools/generate_config.sh  | 34 +++++++
 3 files changed, 142 insertions(+)
 create mode 100755 service/tools/config/generate_config.sh
 create mode 100755 service/tools/kv/server_tools/generate_config.sh

diff --git a/README.md b/README.md
index 6d3fea22e..c2f30236c 100644
--- a/README.md
+++ b/README.md
@@ -368,3 +368,19 @@ We also provide access to a [deployment script](https://github.com/resilientdb/r
    ```
 
    Verify the functionality of the service by performing set and get operations provided above [functions](README.md#functions).
+
+
+## Custom Ports ##
+When starting the service locally, current services are running on 10000 port-base with 5 services where the server config is located [here](https://github.com/apache/incubator-resilientdb/blob/master/service/tools/config/server/server.config)
+
+If you want to change the setting,  you need to generate the certificates.
+
+Go the the workspace where the resilientdb repo is localted.
+
+Change the setting parameters here and run the script:
+  ./service/tools/kv/server_tools/generate_config.sh
+
+Then re-run the start script:
+  ./service/tools/kv/server_tools/start_kv_service.sh
+
+
diff --git a/service/tools/config/generate_config.sh b/service/tools/config/generate_config.sh
new file mode 100755
index 000000000..8d493c55e
--- /dev/null
+++ b/service/tools/config/generate_config.sh
@@ -0,0 +1,92 @@
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+#
+base_path=$1; shift
+key_path=$1; shift
+output_cert_path=$1; shift
+output_path=$1; shift
+admin_key_path=$1; shift
+client_num=$1; shift
+base_port=$1; shift
+
+iplist=$@
+
+echo "generage certificates"
+
+echo "base path:"$base_path
+echo "key path:"$key_path
+echo "output cert path:"$output_cert_path
+echo "output path:"$output_path
+echo "admin_key_path:"$admin_key_path
+
+cd ${output_path}
+
+ADMIN_PRIVATE_KEY=${admin_key_path}/admin.key.pri
+ADMIN_PUBLIC_KEY=${admin_key_path}/admin.key.pub
+
+CERT_TOOLS_BIN=${base_path}/bazel-bin/tools/certificate_tools
+CONFIG_TOOLS_BIN=${base_path}/bazel-bin/tools/generate_region_config
+
+USERNAME=ubuntu
+BASE_PORT=${base_port}
+CLIENT_NUM=${client_num}
+
+
+echo "" > client.config
+echo "" > server.config
+
+bazel build //tools:certificate_tools
+bazel build //tools:generate_region_config
+
+idx=1
+tot=0
+for _ in ${iplist[@]};
+do
+  tot=$(($tot+1))
+done
+
+echo $PWD
+echo "node num:"$tot
+echo "base port:"${BASE_PORT}
+echo "client num:" ${CLIENT_NUM}
+
+for ip in ${iplist[@]};
+do
+  port=$((${BASE_PORT}+${idx}))
+  public_key=${key_path}/node${idx}.key.pub 
+  echo "get ip:"${ip}
+
+  # create public key
+  # create server config
+  # create the public key and certificate
+  if [ $(($idx+$CLIENT_NUM)) -gt $tot ] ; then
+    $CERT_TOOLS_BIN ${output_cert_path} ${ADMIN_PRIVATE_KEY} ${ADMIN_PUBLIC_KEY} ${public_key} ${idx} ${ip} ${port} client
+    echo "${idx} ${ip} ${port}" >> client.config
+  else
+    $CERT_TOOLS_BIN ${output_cert_path} ${ADMIN_PRIVATE_KEY} ${ADMIN_PUBLIC_KEY} ${public_key} ${idx} ${ip} ${port} replica
+    echo "${idx} ${ip} ${port}" >> server.config
+  fi
+
+  idx=$(($idx+1))
+done
+
+python3 ${CONFIG_TOOLS_BIN} ./server.config ./server.config.json 
+mv server.config.json ${output_path}/server/server.config
+mv client.config ${output_path}/interface/service.config
+echo "config done:" ${output_path}/server/server.config
+
diff --git a/service/tools/kv/server_tools/generate_config.sh b/service/tools/kv/server_tools/generate_config.sh
new file mode 100755
index 000000000..5f4944345
--- /dev/null
+++ b/service/tools/kv/server_tools/generate_config.sh
@@ -0,0 +1,34 @@
+#
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+#
+
+iplist=(
+127.0.0.1
+127.0.0.1
+127.0.0.1
+127.0.0.1
+127.0.0.1
+)
+
+WORKSPACE=$PWD
+CERT_PATH=$PWD/service/tools/data/cert/
+CONFIG_PATH=$PWD/service/tools/config/
+PORT_BASE=20000
+CLIENT_NUM=1
+
+./service/tools/config/generate_config.sh ${WORKSPACE} ${CERT_PATH} ${CERT_PATH} ${CONFIG_PATH} ${CERT_PATH} ${CLIENT_NUM} ${PORT_BASE} ${iplist[@]}