带 if 语句的 Postgresql 函数

我怎样才能使这个伪代码在 Postgresql 中工作：

create or replace function getf(arg character varying(255)) returns int
as $$
if arg = 'a' then return 1;
else return 2;
$$ language sql;

根据参数，我需要返回一些值，并且没有其他表需要查询。只需要在函数内部构建逻辑即可。 尝试用when子句替换if，但不知道该怎么做。

Thanks!

create or replace function getf(arg character varying(255)) returns int as $$
begin
  if arg = 'a' then
    return 1;
  else 
    return 2;
  end if;
end; $$ language plpgsql;

请注意，这是一个 PL/pgSQL 函数。

在线手册中有一个精彩的章节介绍了PL/pgSQL http://www.postgresql.org/docs/current/static/plpgsql.html。这应该提供开始编写过程函数所需的一切，并为逻辑分支提供充足的支持。