LINUX下SHELL调用DB2公共函数之public_s.sh

#!/bin/sh

# 张明伟
# 20140505
# SHELL公共函数头
# 对SHELL公共部分进行处理
#

. $HOME/.bash_profile
export LANG=zh_CN
export LC_ALL=zh_CN

pub_nul=/dev/null;             #空设备
pub_splt=" ";                  #分割符
#获取系统数据日期
pub_td=$(date +%Y%m%d)         #today
pub_tm=$(date +%H%M%S)         #time
pub_dt="$pub_td $pub_tm";      #日期时间
pub_monstr=${pub_dt:0:6}01     #月开始

#kpi中使用的日期:上日中CMN_SYSDATE
pub_dt=$(date +"%Y%m%d" -d " -2 days ")  
pub_ldt=$(date +"%Y%m%d" -d " -3 days ")
pub_ndt=$(date +"%Y%m%d" -d " -1 days ")
#kpi中使用的日期

#当天数据日期
pub_dtd=$pub_ndt;

#上年年末
pub_lastyearend=$(date +"%Y%m%d" -d "${pub_dt:0:4}0101  -1 days ");

#下月
pub_nxtmss=$(date +"%Y%m%d" -d "$pub_monstr 1 months ") 

#月末
pub_monend=$(date +"%Y%m%d" -d "${pub_nxtmss:0:6}01 -1 days ")
#上月月末
pub_lastmend=$(date +"%Y%m%d" -d "${pub_ndt:0:6}01 -1 days")
#全局结果表,共享变量
pub_result=""; 
#logs目录
pub_logdir="$HOME/zmw_data/logs";
pub_logf="";
pub_debug="N"; #这个是不写日志,有些标示,日志文件为/dev/null,只打印屏幕;

#增量,一般用于日期
[ "$pub_incr" = "" ] && pub_incr=1;

#日期类型,一般是月
pub_datatype="M";
pub_datatype=$(echo $pub_datatype | tr 'a-z' 'A-Z');

#获取进程号
getpid()
{
    pub_pname="张明伟伟进程名,例如:mainETL.xml";
    [ "$1" != "" ] && pub_pname="$1";
    pub_result="";
    pub_result=$( ps -ef | grep "$pub_pname" | grep -v "grep" | tr -s " " ":" | cut -f 2 -d : );
}


#传入参数上一天
lastday()
{
    pub_result="";
    pub_ld=-1;
    pub_thisday=$pub_td;
    [ "$1" != "" ] && pub_lthisday=$1;
    [ "$2" != "" ] && pub_ld=$2;
    #date +"%Y%m%d" -d " $pub_lthisday $pub_ld days"
    pub_result=$(date +"%Y%m%d" -d "$pub_lthisday $((-1*pub_ld)) days")
    #echo "[$pub_result]"
}

#传入参数后一天
nextday()
{
    pub_result="";
    pub_d=1;
    pub_thisday=$pub_td;
    [ "$1" != "" ] && pub_nthisday=$1;
    [ "$2" != "" ] && pub_d=$2;
    lastday $pub_nthisday -$pub_d;
}

#上一年年末
getlastyearend()
{
   pub_v_datadate="$1"
   if [ "$1" = "" ] 
   then
    pub_result=$pub_lastyearend;
   else
    pub_result=$(date +"%Y%m%d" -d "${pub_v_datadate:0:4}0101  -1 days ")
   fi;
   
}

#两日期相减
fc_datediff()
{
        [ "$pub_xstrdt" = "" ] && pub_xstrdt=$pub_dtd ;
        [ "$pub_xenddt" = "" ] && pub_xenddt=$pub_dtd;
    pub_xstrdt=$1;
    pub_xenddt=$2;
    v_starts=$(date -d "$pub_xstrdt" +"%s");
    v_ends=$(date -d "$pub_xenddt" +"%s");
    v_sub=$((v_ends-v_starts));
    
    pub_result=$((v_sub/3600/24));
}

#传入参数间的月末或每天,取决于datatype
fc_getdayebtw()
{
    pub_strdt=$1;
    pub_enddt=$2;
    pub_datatype=$(echo $pub_datatype | tr 'a-z' 'A-Z');
    if [ "$pub_datatype" = "D" -o "$pub_datatype" = "M" ]
    then
      if [ "$pub_strdt" = "" -o "$pub_enddt" = "" ]
      then
        echo "请输入开始时间和结束时间参数!"
        exit;
      fi;
      
      pub_ds_1=$pub_strdt
      pub_de_1=$pub_enddt
      #echo pub_ds_1 is [$pub_ds_1] and pub_de_1 is [$pub_de_1];

      #pub_result="$pub_ds_1";
      pub_result="";
      while [ $pub_ds_1 -le $pub_de_1 ]
      do
        if [ "$pub_datatype" = "M" ]
        then
          pub_ds_1=${pub_ds_1:0:6}01
          #echo ds_1 is $ds_1
          pub_ds_1=$(date -d "$pub_ds_1 + $pub_incr months " +"%Y%m01" )
          pub_ds_1=$(date -d "$pub_ds_1 - $pub_incr days " +"%Y%m%d" )
          #echo "datatype is $datatype and ds_1 is $ds_1"
        fi;
    
        [ $pub_ds_1 -le $pub_de_1 ] && pub_result="$pub_result $pub_ds_1"
        pub_ds_1=$(date -d "$pub_ds_1 +$pub_incr day " +"%Y%m%d");
      done;
    fi;
}

#删除空格
trim()
{
    pub_str="";
    [ "$1" != "" ] && pub_str="$1";
    pub_str=$( echo "$1" | tr -d " " );
    pub_result="$pub_str";    
}

#替换
replace()
{
    pub_str="";
    pub_s="";
    pub_d="";
    [ "$1" != "" ] && pub_str="$1";
    [ "$2" != "" ] && pub_s="$2";
    [ "$3" != "" ] && pub_d="$3";
    #echo [$pub_str..$pub_s...$pub_d]
    pub_result=${pub_str//"$pub_s"/"$pub_d"}
}

#指定月上月末
getmonend()
{
    pub_g_dt=$1
    pub_i=$2;
    [ "$2" = "" ] && i=1;
    pub_g_dt=$(date +"%Y%m%d" -d "${pub_g_dt:0:6}01 $i months ago")
    pub_result=$(date +"%Y%m%d" -d "${pub_g_dt:0:6}01 1 days ago ");
}

#echo logdir is $logdir;
getlogname()
{
    pub_appnm="$1";
    
    [ "$pub_appnm" != "" ] && pub_appnm=$(basename "$pub_appnm");
    pub_appnm=${pub_appnm%.*};
    pub_result="$pub_logdir/$pub_appnm.log";
    
    [ "$pub_debug" = "Y" ] && pub_result="$pub_nul";
}

pub_sqlstate="SQLSTATE";
#db2运行,需返回数据
db2cmd()
{
  #执行函数时建议加上此句;
  pub_old_debug=$pub_debug;
  
  pub_c_sql="$1";
  pub_result="";
  pub_sqlcode=0;
  [ "$pub_c_sql" = "" ] && pub_result="SQL语句为空" && return;
  [ "$2" != "" ] && pub_debug="$2";
  #为了和前面的程序兼容,增加传入pub_debug和设置pub_debug参数都起作用!
  [ "$pub_old_debug" = "" ] && pub_debug=$pub_old_debug;
  [ "$pub_debug" = "" ] && pub_debug="N";

  if [ "$pub_debug" != "Y" ]
  then
    db2 connect to zxjxdbs>$pub_nul;
    
    pub_result=$(db2 +w -x "$pub_c_sql");

    pub_result=$(echo "$pub_result" | tr -s "  " "$pub_splt");

    pub_result=${pub_result# *};

    pub_result=${pub_result%* };
    [ "$pub_result" = "-" ] && pub_result="";
    pub_hd=$(echo "$pub_result" | head -n 1 );

    pub_sqlinfo=${pub_hd%%"${pub_sqlstate}"*};
    if [ "$pub_sqlinfo" != "" -a "$pub_sqlinfo" != "$pub_hd" ]
    then
      pub_sqlcode=$(echo "$pub_hd" | cut -d " " -f 1);
      
      #若有错误,则把错误屏蔽
      #pub_cnt=$(echo $pub_result | wc -l )
      
      #pub_result=$(tail -$((pub_cnt-1)) $pub_result)
    fi;

    #echo pub_result[$pub_result]    
    db2 connect reset>$pub_nul;
    db2 terminate>$pub_nul;
  fi;
}