run.sh

#!/bin/bash
#
# 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.
#

# hadoop classpath is used for object_store-hdfs to get hadoop client
export CLASSPATH=$CLASSPATH:`hadoop classpath --glob`

rotate_logs ()
{
    log=$1
    num=5

    if [ -f "$log" ]; then # rotate logs
        while [ $num -gt 1 ]; do
            prev=`expr $num - 1`
            [ -f "$log.$prev" ] && mv "$log.$prev" "$log.$num"
            num=$prev
        done
        mv "$log" "$log.$num";
    fi
}

# Check if DELTA_DIR is not set or is empty
if [ -z "$DELTA_DIR" ]; then
    echo "WARNING: DELTA_DIR environment variable is not set. Using /delta as default."
    DELTA_DIR="/delta"
fi

if [ -z "LOG_LEVEL" ]; then
    echo "WARNING: LOG_LEVEL environment variable is not set. Using INFO as default."
    LOG_LEVEL="INFO"
fi

if [ -z "PORT" ]; then
    echo "WARNING: PORT environment variable is not set. Using 50051 as default."
    PORT="50051"
fi

mkdir -p ./log

stdout_log="./log/stdout.log"
stderr_log="./log/stderr.log"

# Rotate logs
rotate_logs "$stdout_log"
rotate_logs "$stderr_log"

if [ -z "$MEM_LIMIT" ]; then
  echo "Starting datafusion-jdbc with DELTA_DIR=$DELTA_DIR, PORT=$PORT, LOG_LEVEL=$LOG_LEVEL"
  datafusion-jdbc -d $DELTA_DIR -p $PORT -l $LOG_LEVEL >> "$stdout_log" 2>> "$stderr_log"
else
  echo "Starting datafusion-jdbc with DELTA_DIR=$DELTA_DIR, PORT=$PORT, MEM_LIMIT=$MEM_LIMIT, LOG_LEVEL=$LOG_LEVEL"
  datafusion-jdbc -d $DELTA_DIR -p $PORT -m $MEM_LIMIT -l $LOG_LEVEL >> "$stdout_log" 2>> "$stderr_log"
fi