entrypoint.sh

#!/bin/bash

mkdir -p /var/state/
chmod 666 -R /var/state/

# add the db_file_name
export DB_FILE="latest_record.db"
DB_FILE_PATH="$PWD/$DB_FILE"

export CONFIG_FILE="{collector_platform_name}-config.json"
CONFIG_FILE_PATH="$PWD/$CONFIG_FILE"

echo $AGENT_STATE_DIR
echo $DB_FILE_PATH

if [ -z "$AGENT_STATE_DIR" ] 
then
    # If there is no state directory configured, use the root directory to store the DB
    echo "No state configured, keeping DB at application root"
    # Create the file if it doesn't exist
    if [ ! -f $DB_FILE_PATH ]
    then
        touch $DB_FILE_PATH
    fi

else
    # If an agent state directory is configured, use it instead
    DB_FILE_PATH="$AGENT_STATE_DIR/$DB_FILE"
    CONFIG_FILE_PATH="$AGENT_STATE_DIR/$CONFIG_FILE"

    if [ -f $DB_FILE_PATH ]
    then
        # If the file exists already, do nothing
        echo "DB file already exists, ignoring..."
    else
        # Otherwise make an empty db file in the state dir.
        echo "Making empty DB file inside state directory $AGENT_STATE_DIR"
        touch $DB_FILE_PATH
    fi

fi

# Print the size of the DB file before migration
echo "Size of the DB file BEFORE MIGRATION:"
du -h $DB_FILE_PATH

# RUN DATABASE MIGRATION TO CONFIGURE DB FILE BEFORE STARTUP
echo "Creating DB schema"
alembic upgrade head

# Print the size of the DB file after migration
echo "Size of the DB file AFTER MIGRATION:"
du -h $DB_FILE_PATH

# Run a vacuum at startup
echo "Vacuuming DB $DB_FILE_PATH at startup..."
sqlite3 $DB_FILE_PATH "VACUUM"

# Print the size of the DB file after vacuum
echo "Size of the DB file AFTER VACUUM:"
du -h $DB_FILE_PATH

# Change it acccording to the collector
EXECUTION_ARGS=(--endDate "$END_DATE" --startDate "$START_DATE" --output-queue --input-queue --save-state)

# Include project root path
export PYTHONPATH=$(pwd)

echo "Starting running src/main.py"
if [ ${#EXECUTION_ARGS[@]} -eq 0 ]
then
    echo "No input arguments supplied. Please provide input parameters."
    python3 src/main.py --help
else
    # echo $EXECUTION
    echo "Starting script"
    python3 -u src/main.py "${EXECUTION_ARGS[@]}"
fi