Example: Create function in Postgresql

创建用户自定义函数或者存储过程，这种事在sql Server下面我们已经司空见惯了。但是在Postgresql下，我们还是要熟悉一下PL/pgsql的语法。

相关表结构：

写一个Function来插入suiteId和caseId，保证这两个Id都能正确关联到TestSuite表和TestCase表：

Create or Replace function func_TestSuite_AddTestCase(sid INTEGER,cid INTEGER) returns INTEGER as $funcbody$ Declare checker BOOLEAN; Declare cnt INTEGER; Begin -- Check if input p<a href="https://www.jb51.cc/tag/ara/" target="_blank" class="keywords">ara</a>meter sid is exist in table testsuite select count(id) into cnt from testsuite where id = $1; if cnt <> 1 then return -101; end if; -- Check if input p<a href="https://www.jb51.cc/tag/ara/" target="_blank" class="keywords">ara</a>meter cid is exist in table testcase select count(id) into cnt from testcase where id = $2; if cnt <> 1 then return -102; end if; -- Check if sid and cid pair is exist in table testsuitecase select count(*) into cnt from testsuitecase where suiteid = $1 and caseid = $2; if cnt <> 1 then insert into testsuitecase(suiteid,caseid) values($1,$2); -- Check if the insertion command is executed successfully. select count(*) into cnt from testsuitecase where suiteid = $1 and caseid = $2; if cnt <> 1 then return -2; end if; return 1; else -- the value you want to input is duplicated in table testsuitecase return -1; end if; End; $funcbody$ LANGUAGE plpg<a href="https://www.jb51.cc/tag/sql/" target="_blank" class="keywords">sql</a>;

$1、$2的用法看上去还是挺爽的。

原文地址：https://www.jb51.cc/postgresql/196983.html