数据库函数

1
2
3
4
5
6
create or replace function hello_world() -- 1returns text -- 2language sql -- 3as $$  -- 4  select 'hello world';  -- 5$$; --6

1
2
3
4
5
6
create or replace function get_planets()returns setof planetslanguage sqlas $$  select * from planets;$$;

1
2
3
4
5
6
7
8
9
10
11
12
13
14
create or replace function add_planet(name text)returns bigintlanguage plpgsqlas $$declare  new_row bigint;begin  insert into planets(name)  values (add_planet.name)  returning id into new_row;  return new_row;end;$$;

1
2
3
4
5
6
7
8
9
create function hello_world()returns textlanguage plpgsqlsecurity definer set search_path = ''as $$begin  select 'hello world';end;$$;

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
create function logging_example(  log_message text,  warning_message text,  error_message text)returns voidlanguage plpgsqlas $$begin  raise log '日志消息: %', log_message;  raise warning '警告消息: %', warning_message;  -- 立即终止函数并回滚事务  raise exception '错误消息: %', error_message;end;$$;select logging_example('日志信息', '警告信息', '错误信息');

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
create or replace function error_if_null(some_val text)returns textlanguage plpgsqlas $$begin  -- 如果 some_val 为 null 则报错  if some_val is null then    raise exception 'some_val 不应为 NULL';  end if;  -- 如果 some_val 不为 null 则返回它  return some_val;end;$$;select error_if_null(null);

1
2
-- 当条件为 false 时抛出错误assert <某个条件>, '错误信息';

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
create function assert_example(name text)returns uuidlanguage plpgsqlas $$declare  student_id uuid;begin  -- 将用户ID保存到 user_id 变量中  select    id into student_id  from attendance_table  where student = name;  -- 如果 student_id 为 null 则抛出错误  assert student_id is not null, 'assert_example() 错误：未找到学生';  -- 否则返回用户ID  return student_id;end;$$;select assert_example('Harry Potter');

1
2
3
4
5
6
7
8
9
10
11
12
13
create function error_example()returns voidlanguage plpgsqlas $$begin  -- 失败：无法从不存在的数据表读取  select * from table_that_does_not_exist;  exception      when others then          raise exception '函数 <function name> 中发生错误: %', sqlerrm;end;$$;

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
create or replace function advanced_example(num int default 10)returns textlanguage plpgsqlas $$declare    var1 int := 20;    var2 text;begin    -- 记录函数调用开始    raise log '记录函数调用开始时间: (%)', (select now());    -- 记录来自SELECT查询的变量    select      col_1 into var1    from some_table    limit 1;    raise log '记录变量值 (%)', var1;    -- 也可以不通过变量，直接将查询结果输出到日志    raise log '记录单值查询结果(%)', (select col_1 from some_table limit 1);    -- 必要时可以记录整行数据为JSON格式    raise log '记录整行数据为JSON格式 (%)', (select to_jsonb(some_table.*) from some_table limit 1);    -- 使用INSERT或UPDATE时，新值可以通过RETURNING返回    -- 使用DELETE时，被删除的值可以返回    -- 这三种操作都使用"RETURNING value(s) INTO variable(s)"语法    insert into some_table (col_2)    values ('new val')    returning col_2 into var2;    raise log '记录INSERT操作的值 (%)', var2;    return var1 || ',' || var2;exception    -- 在此处处理异常（如需要）    when others then        raise exception '函数<advanced_example>执行出错: %', sqlerrm;end;$$;select advanced_example();