wehub-resource-sync
145 lines
3.9 KiB
Bash
145 lines
3.9 KiB
Bash
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
|
|
}
|