#!/bin/bash
#Auther Sun Ying
##Copy left
##Version: Demo Version
##Basic Compare the datebase changes
#ER=`whoami`
#if [ $USER != "postgres" ];then
# echo -e "\033[31mWrong User,please use postgres user\033[0m"
# exit 1
#fi
trap "Clear_Temp;exit" SIGINT SIGQUIT SIGHUP
cmd_help(){
echo -e "\033[36mUsage: \033[33mdump2sql [smgr|ipoc] [Options]\033[0m"
echo -e "\033[32m -d|--dump {-b|--before|-f|--after} Generate the dumpfile\033[0m"
echo -e "\033[32m -c|--compare {d|delete} Compare the dump file by Change|Delete Method\033[0m"
exit 5
}
#check_dump(){
# l=1
# line2=0
# for i in `sed -n -e "/#Table/=" $table_file`
# do
# if [ $line2 -eq $l ];then
# c[$l]=$i
# break
# fi
# table_name=`sed -n "${i}p" $table_file|awk '{print $2}'`
# c[$l]=$i
# if [ $table_name == "$1" ];then
# line1=$i
# let line2=$l+1
# fi
#
# let l=$l+1
# done
# sed -n "$line1,${c[$line2]}p" $table_file
#}
Clear_Temp(){
rm $FullTable &>/dev/null
rm $FullTable1 &>/dev/null
}
generator() {
[ -e $TARGET ] && rm -f $TARGET
echo
echo -e "\033[33mGenrating the output files...Please wait...\033[0m"
j=0
for i in `cat $FullTable1`
do echo "#Table $i" >>$TARGET 2>/dev/null
export PGPASSWORD=$PGPASSWORD;psql -h $HOST -U $ACCOUNT $DATABASE -c "select * from $i">>$TARGET 2>/dev/null
echo >>$TARGET
echo >>$TARGET
echo >>$TARGET
printf "*"
done
}
get_all_tables(){
FullTable=`mktemp /tmp/full_table_XXXXXXX`
FullTable1=`mktemp /tmp/full_table_XXXXXX`
export PGPASSWORD=$PGPASSWORD;psql -h $HOST -U $ACCOUNT $DATABASE -c '\d' > $FullTable
cat $FullTable|awk -F "|" '{print $2}' > $FullTable1
}
get_information(){
read -p "Please input your Host:" HOST
read -p "Please input your User Account:" ACCOUNT
read -p "Please input the database your want to get:" DATABASE
read -s -p "Please input your pgsql User password:" PGPASSWORD
# HOST="localhost"
# ACCOUNT="postgres"
# DATABASE="avmgmt"
# PGPASSWORD="postgres"
}
smgr(){
HOST="localhost"
ACCOUNT="postgres"
DATABASE="avmgmt"
PGPASSWORD="postgres"
}
compare_change(){
file_before=/tmp/dump_before
file_after=/tmp/dump_after
m=1
for i in `sed -n -e "/#Table/=" $file_after`
do
a[$m]=$i
#echo ${a[$m]}
let m=m+1
done
echo -e "\033[33m-------------Compare change started, please wait-------------\033[0m"
n=1
for i in $(diff $file_before $file_after -c|egrep "\-\-\-.*\,.*\-\-\-"|awk -F "," '{print $2}'|awk '{print $1}')
do
let i=$i-3
b[$n]=$i
#echo ${b[$n]}
let n=n+1
done
for((i=1;i<n;++i))
do
if [[ ${b[$i]} -le ${a[$mid]} && ${b[$i]} -ge ${a[$mid_1]} ]];then
if [ $mark1 -eq "0" ];then
continue
else
sed -n "${b[$i]}p" $file_after|egrep "^\\(.*row.*\\)|^$|#Table.*" &>/dev/null
if [ $? -eq 0 ];then
continue
else
echo -e "\033[33m====Below line also changed in this Table===\033[0m"
echo -e "\033[32m`sed -n "${b[$i]}p" $file_after`\033[0m"
continue
fi
fi
fi
let max=$m-1
let min=1
let mid=$max+$min
let mid=$mid/2
let mid_1=$mid-1
let Num=${b[$i]}
until [[ $Num -le ${a[$mid]} && $Num -ge ${a[$mid_1]} ]];
do
if [ $Num -gt ${a[$mid]} ];then
let min=$mid
let mid=$min+$max
let mid=$mid/2
let mid_1=$mid-1
else
let max=$mid
let mid=$min+$max
let mid=$mid/2
let mid_1=$mid-1
fi
done
table_name=$(sed -n "${a[$mid_1]}p" $file_after)
if [ "$table_name" == "#Table log_store" -o "$table_name" == "#Table sched_completed_job_status_view" -o "$table_name" == "#Table sched_job_status" -o "$table_name" == "#Table sched_jobs" -o "$table_name" == "#Table sched_pending_job_status_view" -o "$table_name" == "#Table sched_qrtz_simple_triggers" -o "$table_name" == "#Table sched_qrtz_triggers" -o "$table_name" == "#Table timer" -o "$table_name" == "#Table sched_qrtz_fired_triggers" -o "$table_name" == "#Table sched_job_parameter" ];then
mark1=0
continue
fi
mark1=1
let lastline=${a[$mid]}-1
let Point_line=$Num
sed -n "${Point_line}p" $file_after|egrep "^\\(.*row.*\\)" &>/dev/null
[ $? -eq 0 ] && let Point_line=$Point_line-1
sed -n "${Point_line}p" $file_after|grep "#Table" &>/dev/null
if [ $? -ne 0 ];then
let Point_line_1=$Point_line-1
let Point_line_2=$Point_line+1
# sed -n "${Point_line}p" $file_after
sed -n "${a[$mid_1]},${Point_line_1}p" $file_after
echo -e "\033[32m`sed -n "${Point_line}p" $file_after`\t\t(Insertd|Modified)\033[0m"
sed -n "${Point_line_2},${lastline}p" $file_after|grep -v "^$"
else
echo -e "\033[33m`sed -n "${a[$mid_1]},${lastline}p" $file_after`\033[0m"
fi
done
}
compare_delete(){
file_before=/tmp/dump_before
file_after=/tmp/dump_after
m=1
for i in `sed -n -e "/#Table/=" $file_before`
do
a[$m]=$i
#echo ${a[$m]}
let m=m+1
done
echo -e "\033[33m-------------Compare delete started, please wait-------------\033[0m"
n=1
for i in $(diff $file_after $file_before -c|egrep "\-\-\-.*\,.*\-\-\-"|awk -F "," '{print $2}'|awk '{print $1}')
do
let i=$i-3
b[$n]=$i
#echo ${b[$n]}
let n=n+1
done
for((i=1;i<n;++i))
do
if [[ ${b[$i]} -le ${a[$mid]} && ${b[$i]} -ge ${a[$mid_1]} ]];then
if [ $mark1 -eq "0" ];then
continue
else
sed -n "${b[$i]}p" $file_after|egrep "^\\(.*row.*\\)|^$|#Table.*" &>/dev/null
if [ $? -eq 0 ];then
continue
else
echo -e "\033[33m====Below line also changed in this Table===\033[0m"
echo -e "\033[32m`sed -n "${b[$i]}p" $file_after`\033[0m"
continue
fi
fi
fi
let max=$m-1
let min=1
let mid=$max+$min
let mid=$mid/2
let mid_1=$mid-1
let Num=${b[$i]}
until [[ $Num -le ${a[$mid]} && $Num -ge ${a[$mid_1]} ]];
do
if [ $Num -gt ${a[$mid]} ];then
let min=$mid
let mid=$min+$max
let mid=$mid/2
let mid_1=$mid-1
else
let max=$mid
let mid=$min+$max
let mid=$mid/2
let mid_1=$mid-1
fi
done
table_name=$(sed -n "${a[$mid_1]}p" $file_before)
if [ "$table_name" == "#Table log_store" -o "$table_name" == "#Table sched_completed_job_status_view" -o "$table_name" == "#Table sched_job_status" -o "$table_name" == "#Table sched_jobs" -o "$table_name" == "#Table sched_pending_job_status_view" -o "$table_name" == "#Table sched_qrtz_simple_triggers" -o "$table_name" == "#Table sched_qrtz_triggers" -o "$table_name" == "#Table timer" -o "$table_name" == "#Table sched_qrtz_fired_triggers" -o "$table_name" == "#Table sched_job_parameter" ];then
mark1=0
continue
fi
mark1=1
let lastline=${a[$mid]}-1
let Point_line=$Num
# sed -n "${Point_line}p" $file_after|grep "#Table" &>/dev/null
# if [ $? -ne 0 ];then
sed -n "${Point_line}p" $file_before|egrep "^\\(.*row.*\\)" &>/dev/null
[ $? -eq 0 ] && let Point_line=$Point_line-1
let Point_line_1=$Point_line-1
let Point_line_2=$Point_line+1
# sed -n "${Point_line}p" $file_after
sed -n "${a[$mid_1]},${Point_line_1}p" $file_before
echo -e "\033[31m`sed -n "${Point_line}p" $file_before`\t\t(Deleted|Modified)\033[0m"
sed -n "${Point_line_2},${lastline}p" $file_before|grep -v "^$"
# else
# echo -e "\033[33m`sed -n "${a[$mid_1]},${lastline}p" $file_after`\033[0m"
# fi
done
}
#####main()#####
if [ $# -ne 0 ];then
if [ $1 == "SMGR" -o $1 == "smgr" ];then
smgr
shift
fi
if [ $1 == "IPOC" -o $1 == "ipoc" ];then
ipoc
shift
fi
if [ $1 == "-c" -o $1 == "--compare" ];then
if [ $# -lt 2 ];then
compare_change
exit 0
elif [ $2 == "d" -o $2 == "delete" ];then
compare_delete
exit 0
else
cmd_help
fi
fi
if [ $1 == "-d" -o $1 == "--dump" ];then
if [ $# -ge 2 ];then
if [ $2 == "b" -o $2 == "before" ];then
TARGET="/tmp/dump_before"
elif [ $2 == "f" -o $2 == "after" ];then
TARGET="/tmp/dump_after"
else
cmd_help
fi
elif [ ! -e /tmp/dump_before ];then
TARGET="/tmp/dump_before"
elif [[ -e /tmp/dump_before && ! -e /tmp/dump_after ]];then
TARGET="/tmp/dump_after"
elif [[ -e /tmp/dump_before && -e /tmp/dump_after ]];then
mv -f /tmp/dump_after /tmp/dump_before
TARGET="/tmp/dump_after"
else
TARGET="/tmp/dump_before"
fi
[ ! $HOST ] && get_information
get_all_tables
generator
echo
echo -e "You can find your output file under \033[33m$TARGET\033[0m"
exit 0
fi
# if [ $1 == "-k" -o $1 == "--check" ];then
# if [ $# -lt 2 ];then
# cmd_help
# fi
# if [ $# -ge 3 ];then
# if [ $2 == "b" -o $2 == "before" ];then
# table_file=/tmp/dump_before
# shift
# elif [ $2 == "f" -o $2 == "after" ];then
# table_file=/tmp/dump_after
# shift
# else
# cmd_help
# fi
# fi
# shift
# table_file=/tmp/dump_after
# check_dump $1
# fi
Clear_Temp
cmd_help
else
cmd_help
fi
