generator.server.springboot.dbmigration.cassandra.execute-cql.sh Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jhlite Show documentation
JHipster Lite
There is a newer version: 1.23.0
#!/usr/bin/env bash

MIGRATION_KEYSPACE_NAME=jhipstermigration

unset scripts
declare -A scripts

function log {
  echo "[$(date)]: $*"
}

function logDebug {
  ((DEBUG_LOG)) && echo "[DEBUG][$(date)]: $*"
}

function exitWithError() {
  echo "ERROR :
        $*"
  exit 1
}

# load already executed scripts in the `scripts` global variable: dictionary[scriptName->checksum]
function loadExecutedScripts {
  local rows
  # after SELECT, removes blank lines, then column names, then the query summary at the last line
  mapfile -t rows < <(cqlsh -k $MIGRATION_KEYSPACE_NAME -e "select * from schema_version" "$CASSANDRA_CONTACT_POINT" | grep "\S" | tail -n +3 | sed '$d')
  for r in "${rows[@]}"; do
    local scriptName
    local checksum
    scriptName=$(echo "$r" | cut -d '|' -f 1 | sed s'/^[[:space:]]*//' | sed s'/[[:space:]]*$//')
    checksum=$(echo "$r" | cut -d '|' -f 2 | sed s'/^[[:space:]]*//' | sed s'/[[:space:]]*$//')
    scripts+=(["$scriptName"]="$checksum")
  done
}

function exists {
  if [ "$2" != in ]; then
    echo "Incorrect usage."
    echo "Correct usage: exists {key} in {array}"
    return 1
  fi

  eval '[ ${'"$3"'[$1]+exists} ]'
}

function checksumEquals {
  local checksum
  checksum=$(md5sum "$cqlFile" | cut -d ' ' -f 1)
  local foundChecksum=${scripts[${filename}]}

  if [[ "$checksum" == "$foundChecksum" ]]; then
    logDebug "checksum equals for $cqlFile, checksum=$checksum"
    return 0
  else
    logDebug "different checksum found for $cqlFile
        checksum=$checksum
        foundChecksum=$foundChecksum"
    return 1
  fi
}

function isExecuted {
  if exists "$filename" in scripts; then
    if checksumEquals "$cqlFile"; then
      return 0
    else
      exitWithError "$cqlFile has already been executed but has a different checksum logged in the schema_version table.
            scripts must not be changed after being executed.
            to resolve this issue you can:
            - revert the modified script to its initial state and create a new script
            OR
            - delete the script entry from the schema_version table
            "
    fi
  else
    return 1
  fi
}

function executeCqlScript {
  log "execute: $cqlFile"
  cqlsh -f "$cqlFile" "$CASSANDRA_CONTACT_POINT"

  # if execution failed
  if [ $? -ne 0 ]; then
    exitWithError "fail to apply script $filename
        stop applying database changes"
  fi
  logDebug "execution of $cqlFile succeeded"
}

function logExecutedScript {
  local duration=$1
  local checksum
  checksum=$(md5sum "$cqlFile" | cut -d ' ' -f 1)

  logDebug "save $cqlFile execution in schema_version table"
  local query="INSERT INTO schema_version (script_name, checksum, executed_by, executed_on, execution_time, status) VALUES ('$filename', '$checksum', '$USER', toTimestamp(now()), $duration, 'success');"
  cqlsh -k $MIGRATION_KEYSPACE_NAME -e "$query" "$CASSANDRA_CONTACT_POINT"
}

#usage checks
if [ -z "$1" ]; then
  echo "usage: ./execute-cql cqlFile.cql"
  exit 1
fi
if [ -z "$CASSANDRA_CONTACT_POINT" ]; then
  echo "CASSANDRA_CONTACT_POINT environment variable must be defined"
  exit 1
fi

cqlFile=$1
filename=$(basename "$1")

loadExecutedScripts
if isExecuted; then
  logDebug "skipping $cqlFile already executed"
else
  _start=$(date +"%s")
  executeCqlScript
  _end=$(date +"%s")
  duration=$((_end - _start))
  logExecutedScript "$duration"
  log "$cqlFile executed with success in $duration seconds"
fi