开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数据库系统 > 查看源码
pl_funcs.c
资源名称：postgresql-6.5.2.tar.gz　[点击查看]
上传用户：blenddy
上传日期：2007-01-07
资源大小：6495k
文件大小：15k
源码类别：

数据库系统

开发平台：

Unix_Linux

pl_funcs.c：源码内容
  1. /**********************************************************************
  2.  * pl_funcs.c - Misc functins for the PL/pgSQL
  3.  *   procedural language
  4.  *
  5.  * IDENTIFICATION
  6.  *   $Header: /usr/local/cvsroot/pgsql/src/pl/plpgsql/src/pl_funcs.c,v 1.5 1999/05/25 16:15:18 momjian Exp $
  7.  *
  8.  *   This software is copyrighted by Jan Wieck - Hamburg.
  9.  *
  10.  *   The author hereby grants permission  to  use,  copy, modify,
  11.  *   distribute,  and license this software and its documentation
  12.  *   for any purpose, provided that existing copyright notices are
  13.  *   retained in all  copies  and  that this notice is included
  14.  *   verbatim in any distributions. No written agreement, license,
  15.  *   or  royalty  fee is required for any of the authorized uses.
  16.  *   Modifications to this software may be  copyrighted  by  their
  17.  *   author  and  need  not  follow  the licensing terms described
  18.  *   here, provided that the new terms are  clearly  indicated  on
  19.  *   the first page of each file where they apply.
  20.  *
  21.  *   IN NO EVENT SHALL THE AUTHOR OR DISTRIBUTORS BE LIABLE TO ANY
  22.  *   PARTY  FOR  DIRECT, INDIRECT, SPECIAL,   INCIDENTAL,  OR
  23.  *   CONSEQUENTIAL   DAMAGES  ARISING OUT  OF  THE  USE  OF  THIS
  24.  *   SOFTWARE, ITS DOCUMENTATION, OR ANY DERIVATIVES THEREOF, EVEN
  25.  *   IF  THE  AUTHOR  HAVE BEEN ADVISED OF THE POSSIBILITY OF SUCH
  26.  *   DAMAGE.
  27.  *
  28.  *   THE  AUTHOR  AND DISTRIBUTORS  SPECIFICALLY  DISCLAIM ANY
  29.  *   WARRANTIES,  INCLUDING,  BUT NOT  LIMITED  TO,  THE IMPLIED
  30.  *   WARRANTIES  OF  MERCHANTABILITY, FITNESS  FOR  A  PARTICULAR
  31.  *   PURPOSE, AND NON-INFRINGEMENT.  THIS SOFTWARE IS PROVIDED ON
  32.  *   AN "AS IS" BASIS, AND THE AUTHOR AND  DISTRIBUTORS  HAVE  NO
  33.  *   OBLIGATION   TO PROVIDE   MAINTENANCE,  SUPPORT,  UPDATES,
  34.  *   ENHANCEMENTS, OR MODIFICATIONS.
  35.  *
  36.  **********************************************************************/
  38. #include <stdio.h>
  39. #include <stdlib.h>
  40. #include <stdarg.h>
  41. #include <unistd.h>
  42. #include <fcntl.h>
  43. #include <string.h>
  44. #include <ctype.h>
  46. #include "plpgsql.h"
  47. #include "pl.tab.h"
  50. /* ----------
  51.  * Local variables for the namestack handling
  52.  * ----------
  53.  */
  54. static PLpgSQL_ns *ns_current = NULL;
  55. static bool ns_localmode = false;
  58. /* ----------
  59.  * plpgsql_dstring_init Dynamic string initialization
  60.  * ----------
  61.  */
  62. void
  63. plpgsql_dstring_init(PLpgSQL_dstring * ds)
  64. {
  65. ds->value = palloc(ds->alloc = 512);
  66. ds->used = 0;
  67. }
  70. /* ----------
  71.  * plpgsql_dstring_free Dynamic string destruction
  72.  * ----------
  73.  */
  74. void
  75. plpgsql_dstring_free(PLpgSQL_dstring * ds)
  76. {
  77. pfree(ds->value);
  78. }
  81. /* ----------
  82.  * plpgsql_dstring_append Dynamic string extending
  83.  * ----------
  84.  */
  85. void
  86. plpgsql_dstring_append(PLpgSQL_dstring * ds, char *str)
  87. {
  88. int len = strlen(str);
  90. if (ds->used + len + 1 > ds->alloc)
  91. {
  92. ds->alloc *= 2;
  93. ds->value = repalloc(ds->value, ds->alloc);
  94. }
  96. strcpy(&(ds->value[ds->used]), str);
  97. ds->used += len;
  98. }
  101. /* ----------
  102.  * plpgsql_dstring_get Dynamic string get value
  103.  * ----------
  104.  */
  105. char *
  106. plpgsql_dstring_get(PLpgSQL_dstring * ds)
  107. {
  108. return ds->value;
  109. }
  112. /* ----------
  113.  * plpgsql_ns_init Initialize the namestack
  114.  * ----------
  115.  */
  116. void
  117. plpgsql_ns_init(void)
  118. {
  119. ns_current = NULL;
  120. ns_localmode = false;
  121. }
  124. /* ----------
  125.  * plpgsql_ns_setlocal Tell plpgsql_ns_lookup to or to
  126.  * not look into the current level
  127.  * only.
  128.  * ----------
  129.  */
  130. bool
  131. plpgsql_ns_setlocal(bool flag)
  132. {
  133. bool oldstate;
  135. oldstate = ns_localmode;
  136. ns_localmode = flag;
  137. return oldstate;
  138. }
  141. /* ----------
  142.  * plpgsql_ns_push Enter a new namestack level
  143.  * ----------
  144.  */
  145. void
  146. plpgsql_ns_push(char *label)
  147. {
  148. PLpgSQL_ns *new;
  150. new = palloc(sizeof(PLpgSQL_ns));
  151. memset(new, 0, sizeof(PLpgSQL_ns));
  152. new->upper = ns_current;
  153. ns_current = new;
  155. plpgsql_ns_additem(PLPGSQL_NSTYPE_LABEL, 0, label);
  156. }
  159. /* ----------
  160.  * plpgsql_ns_pop Return to the previous level
  161.  * ----------
  162.  */
  163. void
  164. plpgsql_ns_pop()
  165. {
  166. int i;
  167. PLpgSQL_ns *old;
  169. old = ns_current;
  170. ns_current = old->upper;
  172. for (i = 0; i < old->items_used; i++)
  173. pfree(old->items[i]);
  174. pfree(old->items);
  175. pfree(old);
  176. }
  179. /* ----------
  180.  * plpgsql_ns_additem Add an item to the current
  181.  * namestack level
  182.  * ----------
  183.  */
  184. void
  185. plpgsql_ns_additem(int itemtype, int itemno, char *name)
  186. {
  187. PLpgSQL_ns *ns = ns_current;
  188. PLpgSQL_nsitem *nse;
  190. if (name == NULL)
  191. name = "";
  192. name = plpgsql_tolower(name);
  194. if (ns->items_used == ns->items_alloc)
  195. {
  196. if (ns->items_alloc == 0)
  197. {
  198. ns->items_alloc = 32;
  199. ns->items = palloc(sizeof(PLpgSQL_nsitem *) * ns->items_alloc);
  200. }
  201. else
  202. {
  203. ns->items_alloc *= 2;
  204. ns->items = repalloc(ns->items,
  205.  sizeof(PLpgSQL_nsitem *) * ns->items_alloc);
  206. }
  207. }
  209. nse = palloc(sizeof(PLpgSQL_nsitem) + strlen(name));
  210. nse->itemtype = itemtype;
  211. nse->itemno = itemno;
  212. strcpy(nse->name, name);
  213. ns->items[ns->items_used++] = nse;
  214. }
  217. /* ----------
  218.  * plpgsql_ns_lookup Lookup for a word in the namestack
  219.  * ----------
  220.  */
  221. PLpgSQL_nsitem *
  222. plpgsql_ns_lookup(char *name, char *label)
  223. {
  224. PLpgSQL_ns *ns;
  225. int i;
  227. /* ----------
  228.  * If a label is specified, lookup only in that
  229.  * ----------
  230.  */
  231. if (label != NULL)
  232. {
  233. for (ns = ns_current; ns != NULL; ns = ns->upper)
  234. {
  235. if (!strcmp(ns->items[0]->name, label))
  236. {
  237. for (i = 1; i < ns->items_used; i++)
  238. {
  239. if (!strcmp(ns->items[i]->name, name))
  240. return ns->items[i];
  241. }
  242. return NULL; /* name not found in specified label */
  243. }
  244. }
  245. return NULL; /* label not found */
  246. }
  248. /* ----------
  249.  * No label given, lookup for visible labels ignoring localmode
  250.  * ----------
  251.  */
  252. for (ns = ns_current; ns != NULL; ns = ns->upper)
  253. {
  254. if (!strcmp(ns->items[0]->name, name))
  255. return ns->items[0];
  256. }
  258. /* ----------
  259.  * Finally lookup name in the namestack
  260.  * ----------
  261.  */
  262. for (ns = ns_current; ns != NULL; ns = ns->upper)
  263. {
  264. for (i = 1; i < ns->items_used; i++)
  265. {
  266. if (!strcmp(ns->items[i]->name, name))
  267. return ns->items[i];
  268. }
  269. if (ns_localmode)
  270. {
  271. return NULL; /* name not found in current namespace */
  272. }
  273. }
  275. return NULL;
  276. }
  279. /* ----------
  280.  * plpgsql_ns_rename Rename a namespace entry
  281.  * ----------
  282.  */
  283. void
  284. plpgsql_ns_rename(char *oldname, char *newname)
  285. {
  286. PLpgSQL_ns *ns;
  287. PLpgSQL_nsitem *newitem;
  288. int i;
  290. /* ----------
  291.  * Lookup in the current namespace only
  292.  * ----------
  293.  */
  294. /* ----------
  295.  * Lookup name in the namestack
  296.  * ----------
  297.  */
  298. for (ns = ns_current; ns != NULL; ns = ns->upper)
  299. {
  300. for (i = 1; i < ns->items_used; i++)
  301. {
  302. if (!strcmp(ns->items[i]->name, oldname))
  303. {
  304. newitem = palloc(sizeof(PLpgSQL_nsitem) + strlen(newname));
  305. newitem->itemtype = ns->items[i]->itemtype;
  306. newitem->itemno = ns->items[i]->itemno;
  307. strcpy(newitem->name, newname);
  309. pfree(oldname);
  310. pfree(newname);
  312. pfree(ns->items[i]);
  313. ns->items[i] = newitem;
  314. return;
  315. }
  316. }
  317. }
  319. elog(ERROR, "there is no variable '%s' in the current block", oldname);
  320. }
  323. /* ----------
  324.  * plpgsql_tolower Translate a string to lower case
  325.  * but honor "" escaping.
  326.  * ----------
  327.  */
  328. char *
  329. plpgsql_tolower(char *s)
  330. {
  331. char    *ret;
  332. char    *cp;
  334. ret = palloc(strlen(s) + 1);
  335. cp = ret;
  337. while (*s)
  338. {
  339. if (*s == '"')
  340. {
  341. s++;
  342. while (*s)
  343. {
  344. if (*s == '"')
  345. break;
  346. *cp++ = *s++;
  347. }
  348. if (*s != '"')
  349. {
  350. plpgsql_comperrinfo();
  351. elog(ERROR, "unterminated "");
  352. }
  353. s++;
  354. }
  355. else
  356. {
  357. if (isupper(*s))
  358. *cp++ = tolower(*s++);
  359. else
  360. *cp++ = *s++;
  361. }
  362. }
  363. *cp = '';
  365. return ret;
  366. }
  372. /**********************************************************************
  373.  * Debug functions for analyzing the compiled code
  374.  **********************************************************************/
  375. static int dump_indent;
  377. static void dump_ind();
  378. static void dump_stmt(PLpgSQL_stmt * stmt);
  379. static void dump_block(PLpgSQL_stmt_block * block);
  380. static void dump_assign(PLpgSQL_stmt_assign * stmt);
  381. static void dump_if(PLpgSQL_stmt_if * stmt);
  382. static void dump_loop(PLpgSQL_stmt_loop * stmt);
  383. static void dump_while(PLpgSQL_stmt_while * stmt);
  384. static void dump_fori(PLpgSQL_stmt_fori * stmt);
  385. static void dump_fors(PLpgSQL_stmt_fors * stmt);
  386. static void dump_select(PLpgSQL_stmt_select * stmt);
  387. static void dump_exit(PLpgSQL_stmt_exit * stmt);
  388. static void dump_return(PLpgSQL_stmt_return * stmt);
  389. static void dump_raise(PLpgSQL_stmt_raise * stmt);
  390. static void dump_execsql(PLpgSQL_stmt_execsql * stmt);
  391. static void dump_expr(PLpgSQL_expr * expr);
  394. static void
  395. dump_ind()
  396. {
  397. int i;
  399. for (i = 0; i < dump_indent; i++)
  400. printf(" ");
  401. }
  403. static void
  404. dump_stmt(PLpgSQL_stmt * stmt)
  405. {
  406. printf("%3d:", stmt->lineno);
  407. switch (stmt->cmd_type)
  408. {
  409. case PLPGSQL_STMT_BLOCK:
  410. dump_block((PLpgSQL_stmt_block *) stmt);
  411. break;
  412. case PLPGSQL_STMT_ASSIGN:
  413. dump_assign((PLpgSQL_stmt_assign *) stmt);
  414. break;
  415. case PLPGSQL_STMT_IF:
  416. dump_if((PLpgSQL_stmt_if *) stmt);
  417. break;
  418. case PLPGSQL_STMT_LOOP:
  419. dump_loop((PLpgSQL_stmt_loop *) stmt);
  420. break;
  421. case PLPGSQL_STMT_WHILE:
  422. dump_while((PLpgSQL_stmt_while *) stmt);
  423. break;
  424. case PLPGSQL_STMT_FORI:
  425. dump_fori((PLpgSQL_stmt_fori *) stmt);
  426. break;
  427. case PLPGSQL_STMT_FORS:
  428. dump_fors((PLpgSQL_stmt_fors *) stmt);
  429. break;
  430. case PLPGSQL_STMT_SELECT:
  431. dump_select((PLpgSQL_stmt_select *) stmt);
  432. break;
  433. case PLPGSQL_STMT_EXIT:
  434. dump_exit((PLpgSQL_stmt_exit *) stmt);
  435. break;
  436. case PLPGSQL_STMT_RETURN:
  437. dump_return((PLpgSQL_stmt_return *) stmt);
  438. break;
  439. case PLPGSQL_STMT_RAISE:
  440. dump_raise((PLpgSQL_stmt_raise *) stmt);
  441. break;
  442. case PLPGSQL_STMT_EXECSQL:
  443. dump_execsql((PLpgSQL_stmt_execsql *) stmt);
  444. break;
  445. default:
  446. elog(ERROR, "plpgsql_dump: unknown cmd_type %dn", stmt->cmd_type);
  447. break;
  448. }
  449. }
  451. static void
  452. dump_block(PLpgSQL_stmt_block * block)
  453. {
  454. int i;
  455. char    *name;
  457. if (block->label == NULL)
  458. name = "*unnamed*";
  459. else
  460. name = block->label;
  462. dump_ind();
  463. printf("BLOCK <<%s>>n", name);
  465. dump_indent += 2;
  466. for (i = 0; i < block->body->stmts_used; i++)
  467. dump_stmt((PLpgSQL_stmt *) (block->body->stmts[i]));
  468. dump_indent -= 2;
  470. dump_ind();
  471. printf("    END -- %sn", name);
  472. }
  474. static void
  475. dump_assign(PLpgSQL_stmt_assign * stmt)
  476. {
  477. dump_ind();
  478. printf("ASSIGN var %d := ", stmt->varno);
  479. dump_expr(stmt->expr);
  480. printf("n");
  481. }
  483. static void
  484. dump_if(PLpgSQL_stmt_if * stmt)
  485. {
  486. int i;
  488. dump_ind();
  489. printf("IF ");
  490. dump_expr(stmt->cond);
  491. printf(" THENn");
  493. dump_indent += 2;
  494. for (i = 0; i < stmt->true_body->stmts_used; i++)
  495. dump_stmt((PLpgSQL_stmt *) (stmt->true_body->stmts[i]));
  496. dump_indent -= 2;
  498. dump_ind();
  499. printf("    ELSEn");
  501. dump_indent += 2;
  502. for (i = 0; i < stmt->false_body->stmts_used; i++)
  503. dump_stmt((PLpgSQL_stmt *) (stmt->false_body->stmts[i]));
  504. dump_indent -= 2;
  506. dump_ind();
  507. printf("    ENDIFn");
  508. }
  510. static void
  511. dump_loop(PLpgSQL_stmt_loop * stmt)
  512. {
  513. int i;
  515. dump_ind();
  516. printf("LOOPn");
  518. dump_indent += 2;
  519. for (i = 0; i < stmt->body->stmts_used; i++)
  520. dump_stmt((PLpgSQL_stmt *) (stmt->body->stmts[i]));
  521. dump_indent -= 2;
  523. dump_ind();
  524. printf("    ENDLOOPn");
  525. }
  527. static void
  528. dump_while(PLpgSQL_stmt_while * stmt)
  529. {
  530. int i;
  532. dump_ind();
  533. printf("WHILE ");
  534. dump_expr(stmt->cond);
  535. printf("n");
  537. dump_indent += 2;
  538. for (i = 0; i < stmt->body->stmts_used; i++)
  539. dump_stmt((PLpgSQL_stmt *) (stmt->body->stmts[i]));
  540. dump_indent -= 2;
  542. dump_ind();
  543. printf("    ENDWHILEn");
  544. }
  546. static void
  547. dump_fori(PLpgSQL_stmt_fori * stmt)
  548. {
  549. int i;
  551. dump_ind();
  552. printf("FORI %s %sn", stmt->var->refname, (stmt->reverse) ? "REVERSE" : "NORMAL");
  554. dump_indent += 2;
  555. dump_ind();
  556. printf("    lower = ");
  557. dump_expr(stmt->lower);
  558. printf("n");
  559. dump_ind();
  560. printf("    upper = ");
  561. dump_expr(stmt->upper);
  562. printf("n");
  564. for (i = 0; i < stmt->body->stmts_used; i++)
  565. dump_stmt((PLpgSQL_stmt *) (stmt->body->stmts[i]));
  566. dump_indent -= 2;
  568. dump_ind();
  569. printf("    ENDFORIn");
  570. }
  572. static void
  573. dump_fors(PLpgSQL_stmt_fors * stmt)
  574. {
  575. int i;
  577. dump_ind();
  578. printf("FORS %s ", (stmt->rec != NULL) ? stmt->rec->refname : stmt->row->refname);
  579. dump_expr(stmt->query);
  580. printf("n");
  582. dump_indent += 2;
  583. for (i = 0; i < stmt->body->stmts_used; i++)
  584. dump_stmt((PLpgSQL_stmt *) (stmt->body->stmts[i]));
  585. dump_indent -= 2;
  587. dump_ind();
  588. printf("    ENDFORSn");
  589. }
  591. static void
  592. dump_select(PLpgSQL_stmt_select * stmt)
  593. {
  594. dump_ind();
  595. printf("SELECT ");
  596. dump_expr(stmt->query);
  597. printf("n");
  599. dump_indent += 2;
  600. if (stmt->rec != NULL)
  601. {
  602. dump_ind();
  603. printf("    target = %d %sn", stmt->rec->recno, stmt->rec->refname);
  604. }
  605. if (stmt->row != NULL)
  606. {
  607. dump_ind();
  608. printf("    target = %d %sn", stmt->row->rowno, stmt->row->refname);
  609. }
  610. dump_indent -= 2;
  612. }
  614. static void
  615. dump_exit(PLpgSQL_stmt_exit * stmt)
  616. {
  617. dump_ind();
  618. printf("EXIT lbl='%s'", stmt->label);
  619. if (stmt->cond != NULL)
  620. {
  621. printf(" WHEN ");
  622. dump_expr(stmt->cond);
  623. }
  624. printf("n");
  625. }
  627. static void
  628. dump_return(PLpgSQL_stmt_return * stmt)
  629. {
  630. dump_ind();
  631. printf("RETURN ");
  632. if (stmt->retrecno >= 0)
  633. printf("record %d", stmt->retrecno);
  634. else
  635. {
  636. if (stmt->expr == NULL)
  637. printf("NULL");
  638. else
  639. dump_expr(stmt->expr);
  640. }
  641. printf("n");
  642. }
  644. static void
  645. dump_raise(PLpgSQL_stmt_raise * stmt)
  646. {
  647. int i;
  649. dump_ind();
  650. printf("RAISE '%s'", stmt->message);
  651. for (i = 0; i < stmt->nparams; i++)
  652. printf(" %d", stmt->params[i]);
  653. printf("n");
  654. }
  656. static void
  657. dump_execsql(PLpgSQL_stmt_execsql * stmt)
  658. {
  659. dump_ind();
  660. printf("EXECSQL ");
  661. dump_expr(stmt->sqlstmt);
  662. printf("n");
  663. }
  665. static void
  666. dump_expr(PLpgSQL_expr * expr)
  667. {
  668. int i;
  670. printf("'%s", expr->query);
  671. if (expr->nparams > 0)
  672. {
  673. printf(" {");
  674. for (i = 0; i < expr->nparams; i++)
  675. {
  676. if (i > 0)
  677. printf(", ");
  678. printf("$%d=%d", i + 1, expr->params[i]);
  679. }
  680. printf("}");
  681. }
  682. printf("'");
  683. }
  685. void
  686. plpgsql_dumptree(PLpgSQL_function * func)
  687. {
  688. int i;
  689. PLpgSQL_datum *d;
  691. printf("nExecution tree of successfully compiled PL/pgSQL function %s:n",
  692.    func->fn_name);
  694. printf("nFunctions data area:n");
  695. for (i = 0; i < func->ndatums; i++)
  696. {
  697. d = func->datums[i];
  699. printf("    entry %d: ", i);
  700. switch (d->dtype)
  701. {
  702. case PLPGSQL_DTYPE_VAR:
  703. {
  704. PLpgSQL_var *var = (PLpgSQL_var *) d;
  706. printf("VAR %-16s type %s (typoid %u) atttypmod %dn",
  707.    var->refname, var->datatype->typname,
  708.    var->datatype->typoid,
  709.    var->datatype->atttypmod);
  710. }
  711. break;
  712. case PLPGSQL_DTYPE_ROW:
  713. {
  714. PLpgSQL_row *row = (PLpgSQL_row *) d;
  715. int i;
  717. printf("ROW %-16s fields", row->refname);
  718. for (i = 0; i < row->nfields; i++)
  719. {
  720. printf(" %s=var %d", row->fieldnames[i],
  721.    row->varnos[i]);
  722. }
  723. printf("n");
  724. }
  725. break;
  726. case PLPGSQL_DTYPE_REC:
  727. printf("REC %sn", ((PLpgSQL_rec *) d)->refname);
  728. break;
  729. case PLPGSQL_DTYPE_RECFIELD:
  730. printf("RECFIELD %-16s of REC %dn", ((PLpgSQL_recfield *) d)->fieldname, ((PLpgSQL_recfield *) d)->recno);
  731. break;
  732. case PLPGSQL_DTYPE_TRIGARG:
  733. printf("TRIGARG ");
  734. dump_expr(((PLpgSQL_trigarg *) d)->argnum);
  735. printf("n");
  736. break;
  737. default:
  738. printf("??? unknown data type %dn", d->dtype);
  739. }
  740. }
  741. printf("nFunctions statements:n");
  743. dump_indent = 0;
  744. printf("%3d:", func->action->lineno);
  745. dump_block(func->action);
  746. printf("nEnd of execution tree of function %snn", func->fn_name);
  747. }