#!/bin/bash
#==============================================================================
# Copyright and license info is available in the LICENSE file included with
# the Server Deployment Package (SDP), and also available online:
# https://swarm.workshop.perforce.com/projects/perforce-software-sdp/view/main/LICENSE
#------------------------------------------------------------------------------

# This script is designed to rebuild an Edge server from a checkpoint of the master WHILE
# KEEPING THE EXISTING EDGE SPECIFIC DATA.
# You have to first copy the checkpoint from the master to the edge server before running this script.
# Then you run this script on the Edge server with the instance number and full path and filename
# of the master checkpoint as parameters.
#
# Run example:
#  ./recover_edge.sh 1 /depotdata/p4_1.ckp.9188.gz

export SDP_INSTANCE=${SDP_INSTANCE:-Undefined}
export SDP_INSTANCE=${1:-$SDP_INSTANCE}
if [[ $SDP_INSTANCE == Undefined ]]; then
   echo "Instance parameter not supplied."
   echo "You must supply the Perforce instance as a parameter to this script."
   exit 1
fi

. /p4/common/bin/p4_vars $SDP_INSTANCE
. /p4/common/bin/backup_functions.sh
LOGFILE=$LOGS/recover_edge.log

######### Start of Script ##########

check_vars
set_vars
ckp_running

if [[ "$2" == "" ]]; then
   echo You must pass in the full path and filename of the checkpoint you copied over from the master server.
   exit 2
fi

MASTERCKP=$2

log "Remove offline db"
rm -f /p4/${SDP_INSTANCE}/offline_db/db.* > $LOGFILE

# With -K we filter out the various Edge-specific tables which will be replaced with 
# current live versions.
EXCLUDED_TABLES=db.have,db.working,db.resolve,db.locks,db.revsh,db.workingx,db.resolvex

log "Recover checkpoint from master into offline_db skipping tables not used on the edge."
/p4/${SDP_INSTANCE}/bin/p4d_${SDP_INSTANCE} -r /p4/${SDP_INSTANCE}/offline_db/ -K $EXCLUDED_TABLES -jr -z $MASTERCKP >> $LOGFILE 2>&1

log "Stopping the edge server."
/p4/${SDP_INSTANCE}/bin/p4d_${SDP_INSTANCE}_init stop  >> $LOGFILE 2>&1

# With -k we filter and only checkpoint the specified tables from the current live Edge DB.
CKP_TABLES=$EXCLUDED_TABLES,db.view,db.label,db.revsx,db.revux

log "Creating a dump of the edge specific data from P4ROOT"
/p4/${SDP_INSTANCE}/bin/p4d_${SDP_INSTANCE} -r /p4/${SDP_INSTANCE}/root/ -k $CKP_TABLES -jd /p4/${SDP_INSTANCE}/checkpoints/edgedump  >> $LOGFILE 2>&1

log "Recover the edge dump into offline_db"
/p4/${SDP_INSTANCE}/bin/p4d_${SDP_INSTANCE} -r /p4/${SDP_INSTANCE}/offline_db -jr /p4/${SDP_INSTANCE}/checkpoints/edgedump  >> $LOGFILE 2>&1

log "Reset the replication state and clear the P4ROOT folder db files."
rm -f /p4/${SDP_INSTANCE}/root/db.*  >> $LOGFILE 2>&1
rm -f /p4/${SDP_INSTANCE}/root/state  >> $LOGFILE 2>&1
rm -f /p4/${SDP_INSTANCE}/root/rdb.lbr  >> $LOGFILE 2>&1
rm -f /p4/${SDP_INSTANCE}/logs/journal  >> $LOGFILE 2>&1

log "Move the rebuilt database to P4ROOT"
mv /p4/${SDP_INSTANCE}/offline_db/db.* /p4/${SDP_INSTANCE}/root/  >> $LOGFILE 2>&1

log "Start the edge server back up."
/p4/${SDP_INSTANCE}/bin/p4d_${SDP_INSTANCE}_init start  >> $LOGFILE 2>&1

log "Recreate the offline_db"
/p4/${SDP_INSTANCE}/bin/p4d_${SDP_INSTANCE} -r /p4/${SDP_INSTANCE}/offline_db/ -K $EXCLUDED_TABLES -jr -z $MASTERCKP  >> $LOGFILE 2>&1
/p4/${SDP_INSTANCE}/bin/p4d_${SDP_INSTANCE} -r /p4/${SDP_INSTANCE}/offline_db -jr /p4/${SDP_INSTANCE}/checkpoints/edgedump  >> $LOGFILE 2>&1

log "Create a new edge checkpoint from offline_db"
# Set P4ROOT to offline_db so that get_journalnum will get the journal number and set CHECKPOINTNUM from there.
export P4ROOT=/p4/${SDP_INSTANCE}/offline_db
get_journalnum
# Reset P4ROOT
export P4ROOT=/p4/${SDP_INSTANCE}/root

rm -f /p4/${SDP_INSTANCE}/checkpoints/${CHECKPOINTNUM}.gz >> $LOGFILE 2>&1
/p4/${SDP_INSTANCE}/bin/p4d_${SDP_INSTANCE} -r /p4/${SDP_INSTANCE}/offline_db -jd -z /p4/${SDP_INSTANCE}/checkpoints/${CHECKPOINTNUM}.gz  >> $LOGFILE 2>&1

ckp_complete
log "End $P4SERVER Recover Edge"
mail_log_file "$HOSTNAME $P4SERVER Recover Edge log."

