ITPub博客

首页 > Linux操作系统 > Linux操作系统 > GreenPlum存储过程的源码导出

GreenPlum存储过程的源码导出

原创 Linux操作系统 作者:myamor 时间:2012-03-24 10:10:42 0 删除 编辑
版权所有,请勿转载!!!!! 如要转载请 联系博主 !!!!谢谢合作
 
#!/bin/bash
source /usr/local/greenplum-db/greenplum_path.sh
USAGE ()
{
  echo "Usage: `basename $0` [-d database] [-p port] [-h hostip] [ -U username ] {-f procname} [-o output]";
  echo "ExpFunToFile Parameter Infomation:";
  echo "  Parameter -f must be specified                             [必须指定函数名 例如:public.test]";
  echo "  Parameter -d: Database Name,Default Value:PGDATABASE       [数据库名]";
  echo "  Parameter -p: Database Port,Default Value:5432             [数据库端口]";
  echo "  Parameter -h: Database Host,Defualt Value:PGHOST           [主机IP地址]";
  echo "  Parameter -U: User Name,Default Value:PGUSER               [连接数据库名]";
  echo "  Parameter -o: Output Path,Default Value: pwd               [目标目录]";
  echo ""
  exit 1;
}
if [ $# -lt 2 ]
  then
   USAGE
  fi
 

dbname=""
procname=""
dbport=""
dbhost=""
outpath=""
dbuser=""
#***************************Judge Parameter Begin
  while getopts :d:f:p:h:U:o: OPTION ;
  do
   case "$OPTION" in
     d) dbname="$OPTARG" ;;
     f) procname="$OPTARG" ;;
     p) dbport="$OPTARG" ;;
     h) dbhost="$OPTARG" ;;
     o) utpath="$OPTARG" ;;
     U) dbuser="$OPTARG" ;;
     \?) #usage statement
       USAGE;
       ;;
   esac
  done
  if [ "$procname" = "" ]
   then
     echo "Parameter -f must be specified "
     USAGE;
   else
   if [ `expr index $procname .` -le 0  ]
     then
       echo "The Function Name is the same as schema.funname."
       echo "Eg: public.test "
       exit 1
   fi
  fi
  strdb=""
  if [ "$dbname" != "" ]
   then
     strdb=" -d $dbname "
  fi
  if [ "$dbport" != "" ]
   then
     strdb=$strdb" -p $dbport "
  fi
  if [ "$dbhost" != "" ]
   then
     strdb=$strdb" -h $dbhost "
  fi
  if [ "$dbuser" != "" ]
   then
    strdb=$strdb" -U $dbuser "
  fi
  if [ "$outpath" != "" ]
   then
    utfile=$outpath/$procname.sql
   else
    utfile=$procname.sql
  fi
#获取函数名以及相关的参数
sqlstr="select nspname||'.'|| proname procname,usename,lanname,prorettype::regtype,proargtypes,proallargtypes,proargmodes,proargnames \
from  pg_catalog.pg_proc  a \
  left join pg_catalog.pg_namespace b on a.pronamespace=b.oid \
  left join pg_catalog.pg_user c on a.proowner=c.usesysid \
  left join pg_catalog.pg_language d on a.prolang=d.oid \
where nspname||'.'||proname=lower('$procname');"
#取出返回值
line=`psql -X -t -c "$sqlstr" -A -F#@# $strdb `
#echo $line,$strdb,$sqlstr
#分解返回值
vusername=`echo $line|awk -F#@# '{print $2}'`
vlanname=`echo $line|awk -F#@# '{print $3}'`
vretvalue=`echo $line|awk -F#@# '{print $4}'`
vargtypes=`echo $line|awk -F#@# '{print $5}'|sed 's/[{}]//g'`
vallargtypes=`echo $line|awk -F#@# '{print $6}'|sed 's/[{}]//g'`
vargmodes=`echo $line|awk -F#@# '{print $7}'|sed 's/[{}]//g'`
vargnames=`echo $line|awk -F#@# '{print $8}'|sed 's/[{}]//g'`
vfuntxt=`echo $line|awk -F#@# '{print $9}'`
#判断参数个数
fnum=`echo $vargnames|awk -F, '{print NF}'`
#echo $strarg
#重组函数的参数串
for((i=1;i<=$fnum;i++))
do
  typeid=`echo $vargtypes|cut -d' ' -f$i`
  atypeid=`echo $vallargtypes|cut -d, -f$i`
  amodes=`echo $vargmodes|cut -d, -f$i`
  argnames=`echo $vargnames|cut -d, -f$i`
  echo $typeid
  if [ "$atypeid" = "" ]
   then
      atypeid=$typeid
  fi
  atype=`psql -X -t -c "select $atypeid::regtype" $strdb `
 
  if [ "$amodes" != "o" ]
  then
   strarg="${strarg},in $argnames $atype"
  else
   strarg="${strarg},out $argnames $atype"
  fi 
done
  vstrarg=`echo $strarg|cut -c 2-`
#获取函数体
sqlstr=" select prosrc from  pg_catalog.pg_proc  a \
  left join pg_catalog.pg_namespace b on a.pronamespace=b.oid \
where nspname||'.'||proname=lower('$procname'); "
#生成文件
echo "CREATE OR REPLACE FUNCTION ${procname}($vstrarg)" >$outfile.sql
echo "RETURNS $vretvalue AS " >>$outfile.sql
echo "\$BODY\$" >>$outfile.sql
psql  -X -t -c "$sqlstr" $strdb >>$outfile.sql
echo "\$BODY\$" >>$outfile.sql
echo "LANGUAGE $vlanname VOLATILE;" >>$outfile.sql
#程序结束

exit 0
 
 

来自 “ ITPUB博客 ” ,链接:http://blog.itpub.net/177006/viewspace-719411/,如需转载,请注明出处,否则将追究法律责任。

请登录后发表评论 登录
全部评论

注册时间:2012-03-20

  • 博文量
    20
  • 访问量
    132201