多租户更改

2026-08-11 15:59:57 +08:00
parent 6c800de67d
commit c2076721fc
453 changed files with 102349 additions and 47 deletions
@@ -0,0 +1,80 @@
+#!/bin/bash
+
+# Start/stop test cluster on localhost. This is NOT a production script. Use it for reference only.
+
+# Names of cluster nodes
+ALL_NODE_NAMES=( one two three )
+# Port where the first node will listen for client connections over http
+HTTP_BASE_PORT=6060
+# Port where the first node will listen for gRPC intra-cluster connections.
+GRPC_BASE_PORT=16060
+
+USAGE="Usage: $0 [ --config <path_to_tinode.conf> ] {start|stop}"
+
+# Your server binary may have a different name and location.
+SERVER='./server'
+
+if [ "$#" -lt "1" ]; then
+  echo $USAGE
+  exit 1
+fi
+
+while [[ $# -gt 0 ]]; do
+  key="$1"
+  shift
+  echo "$key"
+  case "$key" in
+    -c|--config)
+      config=$1
+      shift # value
+      ;;
+    -s|--static_data)
+      static_data=$1
+      shift # value
+      ;;
+    start)
+      if [ ! -z "$config" ] ; then
+        TINODE_CONF=$config
+      else
+        TINODE_CONF="tinode.conf"
+      fi
+      if [ ! -z "${static_data+x}" ] ; then
+        STATIC_DATA_DIR=$static_data
+      else
+        STATIC_DATA_DIR="static"
+      fi
+
+      echo "HTTP ports 6060-6062, gRPC ports 16060-16062, config ${config}"
+
+      HTTP_PORT=$HTTP_BASE_PORT
+      GRPC_PORT=$GRPC_BASE_PORT
+      for NODE_NAME in "${ALL_NODE_NAMES[@]}"
+      do
+        # Start the node
+        $SERVER -config=${TINODE_CONF} -cluster_self=${NODE_NAME} -listen=:${HTTP_PORT} -grpc_listen=:${GRPC_PORT} -static_data=${STATIC_DATA_DIR} -log_flags=stdFlags,shortfile &
+        # Save PID of the node to a temp file.
+        # /var/tmp/ does not requre root access.
+        echo $!> "/var/tmp/tinode-${NODE_NAME}.pid"
+        # Increment ports for the next node.
+        HTTP_PORT=$((HTTP_PORT+1))
+        GRPC_PORT=$((GRPC_PORT+1))
+      done
+      exit 0
+      ;;
+    stop)
+      echo 'Stopping cluster'
+
+      for NODE_NAME in "${ALL_NODE_NAMES[@]}"
+      do
+        # Read PIDs of running nodes from temp files and kill them.
+        kill `cat /var/tmp/tinode-${NODE_NAME}.pid`
+        # Clean up: delete temp files.
+        rm "/var/tmp/tinode-${NODE_NAME}.pid"
+      done
+      exit 0
+      ;;
+    *)
+      echo $USAGE
+      exit 1
+  esac
+done