chore: import upstream snapshot with attribution

2026-07-13 12:32:25 +08:00
commit e014feafe1
2285 changed files with 1131979 additions and 0 deletions
@@ -0,0 +1,144 @@
+SERVER_REQS_INSTALLED="FALSE"
+SERVER_PID=""
+DEFAULT_DB=""
+PASSWORD=""
+USERNAME=""
+
+# wait_for_connection(<PORT>, <TIMEOUT IN MS>) attempts to connect to the sql-server at the specified
+# port on localhost, using the $SQL_USER (or 'postgres' if unspecified) as the user name, and trying once
+# per second until the millisecond timeout is reached. If a connection is successfully established,
+# this function returns 0. If a connection was not able to be established within the timeout period,
+# this function returns 1.
+wait_for_connection() {
+  port=$1
+  timeout=$2
+  end_time=$((SECONDS+($timeout/1000)))
+  USERNAME="${USERNAME:=postgres}"
+  PASSWORD="${PASSWORD:=password}"
+
+  nativevar PGPASSWORD "$PASSWORD" /w
+  
+  while [ $SECONDS -lt $end_time ]; do
+    run psql -U $USERNAME -h localhost -p $port -c "SELECT 1;" $DEFAULT_DB
+    if [ $status -eq 0 ]; then
+      echo "Connected successfully!"
+      return 0
+    else
+      echo "$output"
+    fi
+    sleep 1
+  done
+
+  echo "Failed to connect to database $DEFAULT_DB on port $port within $timeout ms."
+  return 1
+}
+
+start_sql_server() {
+    DEFAULT_DB="$1"
+    logFile=$2
+    USERNAME=$3
+    PASSWORD=$4
+
+    if [ -n "$DEFAULT_DB" ]; then
+        nativevar DOLTGRES_DB "$DEFAULT_DB" /w
+    fi
+
+    if [ -n "$PASSWORD" ]; then
+        nativevar PGPASSWORD "password" /w
+        nativevar DOLTGRES_PASSWORD "$PASSWORD" /w
+    else
+        nativevar PGPASSWORD "$PASSWORD" /w
+        PASSWORD="password"
+    fi
+
+    if [ -n "$USERNAME" ]; then
+        nativevar DOLTGRES_USER "$USERNAME" /w
+        if [ -z "$DEFAULT_DB" ]; then
+            DEFAULT_DB="$USERNAME"
+        fi
+    else
+        USERNAME="postgres"
+    fi
+
+    DEFAULT_DB="${DEFAULT_DB:=postgres}"
+
+    PORT=$( definePORT )
+    CONFIG=$( defineCONFIG $PORT )
+    echo "$CONFIG" > config.yaml
+    if [[ $logFile ]]
+    then
+        doltgres -data-dir=. -config=config.yaml> $logFile 2>&1 &
+    else
+        doltgres -data-dir=. -config=config.yaml &
+    fi
+    SERVER_PID=$!
+    wait_for_connection $PORT 7500
+}
+
+# like start_sql_server, but the second argument is a string with all arguments to doltgres. The
+# port argument is handled separately: if the variable $PORT is not defined and the --port argument
+# is not included in the argument list, a random port is chosen for $PORT and the argument --port is
+# appended to the argument list.
+start_sql_server_with_args() {
+    DEFAULT_DB=""
+    nativevar DEFAULT_DB "$DEFAULT_DB" /w
+    nativevar PGPASSWORD "password" /w
+
+    echo "running doltgres $@"
+    doltgres "$@" &
+    SERVER_PID=$!
+    wait_for_connection $PORT 7500
+}
+
+# stop_sql_server stops the SQL server. For cases where it's important
+# to wait for the process to exit after the kill signal (e.g. waiting
+# for an async replication push), pass 1.
+# kill the process if it's still running
+stop_sql_server() {
+    # Clean up any mysql.sock file in the default, global location
+    if [ -f "/tmp/mysql.sock" ]; then rm -f /tmp/mysql.sock; fi
+    if [ -f "/tmp/postgres.sock" ]; then rm -f /tmp/mysql.sock; fi
+    if [ -f "/tmp/dolt.$PORT.sock" ]; then rm -f /tmp/dolt.$PORT.sock; fi
+
+    wait=$1
+    if [ ! -z "$SERVER_PID" ]; then
+        # ignore failures of kill command in the case the server is already dead
+        run kill $SERVER_PID
+        if [ $wait ]; then
+            while ps -p $SERVER_PID > /dev/null; do
+                sleep .1;
+            done
+        fi;
+    fi
+    SERVER_PID=
+    PORT=
+}
+
+definePORT() {
+  for i in {0..99}
+  do
+    port=$((RANDOM % 4096 + 2048))
+    # nc (netcat) returns 0 when it _can_ connect to a port (therefore in use), 1 otherwise.
+    run nc -z localhost $port
+    if [ "$status" -eq 1 ]; then
+      echo $port
+      break
+    fi
+  done
+}
+
+defineCONFIG() {
+    PORT=$1
+    cat <<EOF
+    log_level: debug
+
+    behavior:
+      read_only: false
+      disable_client_multi_statements: false
+      dolt_transaction_commit: false
+
+    listener:
+      host: localhost
+      port: $PORT
+EOF
+}