shell脚本例子

#!/bin/ksh
#ident "%W%"
function sumRows {
cp $1 $1.tmp
num=`grep rows $1.tmp | awk '{print $2}'`
if [ -z "$num" ] ; then
num=0
fi

num=`echo "$num+$2" | bc`

grep -v rows $1.tmp > $1
echo rows $num >> $1
rm $1.tmp
}

function startMarxJob {
if [ "$isTrackStatus" != "true" ] ; then
return 0;
fi
while getopts j:p:c:m:d: args
do
case $args in
j) JOB_NAME="$OPTARG";;
p) PID="$OPTARG";;
c) COMMENT="$OPTARG";;
m) allowMultiExec="$OPTARG";;
d) date="$OPTARG";;
esac
done

SQL_CMD="sp_marx_job_start @jobname='$JOB_NAME', @pid=$PID, @jobdate=$date, @allowMultiExecution='${allowMultiExec}', @comment='$COMMENT' , @id_return=@jobid out"
echo exec $SQL_CMD

jobTmpFile=${GCDR_TMPDIR}/.$$.sql.result.tmp
$ISQL $SQL <<EOD > $jobTmpFile
declare @jobid int
exec $SQL_CMD
go
EOD

if [ $? != 0 ] ; then
echo failed to exec sp_marx_job_start;
return -1;
fi

jobid=`grep Job_ID $jobTmpFile | awk '{print $2}'`

rm $jobTmpFile
if [ -n "$jobid" ] ; then
echo Job_ID $jobid;
echo Job_ID $jobid > ${GCDR_TMPDIR}/.${PID}.status;
return 0;
else
echo failed to start job;
return 99;
fi
}

function endMarxJob {
if [ "$isTrackStatus" != "true" ] ; then
return 0;
fi
while getopts F:S:c:x: args
do
case $args in
F) FILE="$OPTARG";;
S) STATUS="$OPTARG";;
c) COMMENT="$OPTARG";;
x) EXIT_STATUS="$OPTARG";;
esac
done

if [ ! -f $FILE ] ; then
echo "$FILE" does not exists!;
return 1;
fi

#grep job id and row processed
JOB_ID=`grep Job_ID $FILE | awk '{print $2}'`
ROWS_NUM=`grep rows $FILE | awk '{print $2}'`

if [ -z $ROWS_NUM ] ; then
ROWS_NUM=0;
fi

if [ finish = "$STATUS" ] ; then
SQL_CMD="sp_marx_job_finish $JOB_ID, $EXIT_STATUS, $ROWS_NUM, '$COMMENT'";
else
SQL_CMD="sp_marx_job_error $JOB_ID, $EXIT_STATUS, $ROWS_NUM, '$COMMENT'";
fi

echo exec $SQL_CMD

jobTmpFile=${GCDR_TMPDIR}/.$$.sql.result.tmp
$ISQL $SQL <<EOD > $jobTmpFile
exec $SQL_CMD
go
EOD


if [ $? != 0 ] ; then
echo failed to end job due to DB error.;
return 1;
fi

hasError=`grep -i ERROR $jobTmpFile`

rm $jobTmpFile
if [ -z "$hasError" ] ; then
echo Job_ID $JOB_ID has been marked as $STATUS successfully.;
return 0;
else
echo Job_ID $JOB_ID has not been marked as $STATUS successfully due to $hasError;
return 1;
fi
}

function startMarxTask {
if [ "$isTrackStatus" != "true" ] ; then
return 0;
fi
while getopts n:t:c:d:F: args
do
case $args in
n) TASK_NAME="$OPTARG";;
t) TYPE="$OPTARG";;
c) COMMENT="$OPTARG";;
F) FILE="$OPTARG";;
esac
done

if [ ! -f $FILE ] ; then
echo job status file does not exist. ;
return 99;
fi

#grep job id and row processed
JOB_ID=`grep Job_ID $FILE | awk '{print $2}'`
SQL_CMD="sp_marx_task_start $JOB_ID, '$TASK_NAME', '$TYPE', '$COMMENT' , @taskid out"
echo $SQL_CMD
jobTmpFile=${GCDR_TMPDIR}/.$$.sql.result.tmp
$ISQL $SQL <<EOD > $jobTmpFile
declare @taskid int
exec $SQL_CMD
go
EOD

if [ $? != 0 ] ; then
echo failed to exec sp_marx_task_start due to DB error;
return 1;
fi

taskid=`grep Task_ID $jobTmpFile | awk '{print $2}'`

rm $jobTmpFile
if [ -n "$taskid" ] ; then
grep -v Task_ID $FILE > $FILE.tmp;
echo Task_ID $taskid >> $FILE.tmp;
rm $FILE;
mv $FILE.tmp $FILE;
return 0;
else
echo failed to get task ID;
return 99;
fi
}


function endMarxTask {
if [ "$isTrackStatus" != "true" ] ; then
return 0;
fi
while getopts x:s:f:c:r:S:F: args
do
case $args in
x) EXIT_STATUS="$OPTARG";;
s) SUCCESS_NUM="$OPTARG";;
f) FAILURE_NUM="$OPTARG";;
c) COMMENT="$OPTARG";;
r) PROCESS_NUM="$OPTARG";;
S) STATUS="$OPTARG";;
F) FILE="$OPTARG";;
esac
done

if [ ! -f $FILE ] ; then
echo job status file does not exist. ;
return 99;
fi

TASK_ID=`grep Task_ID $FILE | awk '{print $2}'`

if [ finish = "$STATUS" ] ; then
SQL_CMD="sp_marx_task_finish $TASK_ID , $EXIT_STATUS , $SUCCESS_NUM , $FAILURE_NUM , '$COMMENT' , @taskid out";
else
SQL_CMD="sp_marx_task_error $TASK_ID , $EXIT_STATUS , $SUCCESS_NUM , $FAILURE_NUM , '$COMMENT' , @taskid out";
fi
echo $SQL_CMD

jobTmpFile=${GCDR_TMPDIR}/.$$.sql.result.tmp

$ISQL $SQL <<EOD > $jobTmpFile
declare @taskid varchar(80)
exec $SQL_CMD
go
EOD


if [ $? != 0 ] ; then
echo failed to end task due to DB error;
return 1;
fi

hasError=`grep -i ERROR $jobTmpFile`

rm $jobTmpFile
if [ -z "$hasError" ] ; then
#add the rows num
sumRows $FILE $PROCESS_NUM;
echo TASK_ID $TASK_ID has been marked as $STATUS successfully.;
echo $PROCESS_NUM rows been processed.;
return 0;
else
echo TASK_ID $TASK_ID has not been marked as $STATUS successfully due to $hasError;
return 1;
fi
}

function insertLogName {
if [ "$isTrackStatus" != "true" ] ; then
return 0;
fi
while getopts l:F: args
do
case $args in
l) LOGNAME="$OPTARG";;
F) FILE="$OPTARG";;
esac
done

if [ ! -f $FILE ] ; then
echo job status file does not exist. ;
return 99;
fi

JOB_ID=`grep Job_ID $FILE | awk '{print $2}'`

SQL_CMD="insert into marx_job_log values ( $JOB_ID , '${LOGNAME}' )"

echo $SQL_CMD

jobTmpFile=${GCDR_TMPDIR}/sql.result.tmp

$ISQL $SQL <<EOD > $jobTmpFile
declare @taskid varchar(80)
$SQL_CMD
go
EOD


if [ $? != 0 ] ; then
echo failed to record $LOGNAME to JOB_ID $JOB_ID;
return 1;
fi

hasError=`grep -i ERROR $jobTmpFile`

rm $jobTmpFile
if [ -z "$hasError" ] ; then
echo record $LOGNAME to JOB_ID $JOB_ID successfully.;
return 0;
else
echo failed to record $LOGNAME to JOB_ID $JOB_ID due to $hasError;
return 1;
fi
}


function insertMarxJob {
if [ "$isTrackStatus" != "true" ] ; then
return 0;
fi
PROCESS_NUM=0
COMMENT=""
while getopts j:p:x:r:S:c:d:l: args
do
case $args in
j) JOB_NAME="$OPTARG";;
p) PID="$OPTARG";;
x) exit_status="$OPTARG";;
r) PROCESS_NUM="$OPTARG";;
S) status="$OPTARG";;
c) COMMENT="$OPTARG";;
d) date="$OPTARG";;
l) logname="$OPTARG";;
esac
done

SQL_CMD="insert into marx_job_status values ('$JOB_NAME', $PID, $date, getdate(), getdate(),$exit_status, $status,$PROCESS_NUM,'$COMMENT')"
echo $SQL_CMD

jobTmpFile=${GCDR_TMPDIR}/.$$.sql.result.tmp
$ISQL $SQL <<EOD > $jobTmpFile
$SQL_CMD
print "Job_ID %1! starts", @@identity
go
EOD

if [ $? != 0 ] ; then
echo failed to insert job;
return -1;
fi

if [ $logname = "" ] ; then return 0 ; fi
jobid=`grep Job_ID $jobTmpFile | awk '{print $2}'`

rm $jobTmpFile
if [ -n "$jobid" ] ; then
SQL_CMD_LOG="insert into marx_job_log values ($jobid, '$logname')"
echo $SQL_CMD_LOG
$ISQL $SQL <<EOD
$SQL_CMD_LOG
go
EOD
fi
}