39.2. Funções escritas em PL/Ruby

Exemplo 39-1. Função PL/Ruby para retornar o maior entre dois valores inteiros

A função abaixo, escrita em PL/Ruby, retorna o maior valor entre os seus dois argumentos inteiros:

CREATE OR REPLACE FUNCTION ruby_max(int4, int4) RETURNS int4 AS $$
    if args[0].to_i > args[1].to_i
        return args[0]
    else
        return args[1]
    end
$$ LANGUAGE 'plruby' IMMUTABLE STRICT;

Exemplos de utilização da função:

plruby=# SELECT ruby_max (4,5);

 ruby_max
----------
        5
(1 linha)

plruby=# SELECT ruby_max (-4,-5);

 ruby_max
----------
       -4
(1 linha)

Exemplo 39-2. Função PL/Ruby para retornar o fatorial de um número

A função abaixo, escrita em PL/Ruby, retorna o fatorial do argumento inteiro, utilizando a definição do método fatorial_incorporado escrito em Ruby presente no comando de criação da função:

CREATE OR REPLACE FUNCTION ruby_fatorial_incorporado(int4) RETURNS int4 AS $$
    return fatorial_incorporado(args[0].to_i)
end

def PLtemp.fatorial_incorporado(n)
    if n < 0
        nil
    elsif n == 0
        1
    else
        n * fatorial_incorporado(n-1)
    end
$$ LANGUAGE 'plruby' IMMUTABLE STRICT;

Exemplos de utilização da função:

plruby=# SELECT ruby_fatorial_incorporado(6);

 ruby_fatorial_incorporado
---------------------------
                       720
(1 linha)

plruby=# SELECT ruby_fatorial_incorporado(1);

 ruby_fatorial_incorporado
---------------------------
                         1
(1 linha)

plruby=# SELECT ruby_fatorial_incorporado(0);

 ruby_fatorial_incorporado
---------------------------
                         1
(1 linha)

plruby=# SELECT ruby_fatorial_incorporado(-1);

 ruby_fatorial_incorporado
---------------------------

(1 linha)

Exemplo 39-3. Função PL/Ruby para retornar o fatorial de um número

Neste exemplo, antes de definir a função ruby_fatorial em PL/Ruby é inserido o método singleton para cálculo do fatorial na tabela plruby_singleton_methods, conforme mostrado abaixo:

INSERT INTO plruby_singleton_methods VALUES
('fatorial', 'n', '
    if n < 0
        nil
    elsif n == 0
        1
    else
        n * fatorial(n-1)
    end
','Cálculo do fatorial');

Para definir o método é realizada uma nova conexão com o banco de dados:

plruby=# \c plruby
Conectado ao banco de dados "plruby".

Após definir o método é definida a função, escrita em PL/Ruby, que faz uso do método:

CREATE OR REPLACE FUNCTION ruby_fatorial(int4) RETURNS int4 AS $$
    return fatorial(args[0].to_i)
$$ LANGUAGE 'plruby' IMMUTABLE STRICT;

Por fim, são feitas chamadas à função para testar seu funcionamento:

plruby=# SELECT ruby_fatorial(6);

 ruby_fatorial
---------------
           720
(1 linha)

plruby=# SELECT ruby_fatorial(1);

 ruby_fatorial
---------------
             1
(1 linha)

plruby=# SELECT ruby_fatorial(0);

 ruby_fatorial
---------------
             1
(1 linha)

plruby=# SELECT ruby_fatorial(-1);

 ruby_fatorial
---------------

(1 linha)

Para ver a definição da função pode ser consultada a tabela plruby_singleton_methods conforme mostrado abaixo:

plruby=# \x
Ativada a exibição expandida.
plruby=# SELECT * FROM plruby_singleton_methods;
-[ LINHA 1 ]------------------------------------
name    | fatorial
args    | n
body    |
    if n < 0
        nil
    elsif n == 0
        1
    else
        n * fatorial(n-1)
    end

comment | Cálculo do fatorial

Exemplo 39-4. Função PL/Ruby para calcular o desconto do IRPF mensal

Abaixo está um exemplo do cálculo do valor do imposto de renda mensal baseado no rendimento tributável, desconto para a previdência oficial, dependentes, pensão alimentícia judicial e outras deduções. Até R$ 1.164,00 o contribuinte está isento, deste valor até R$ 2.326,00 incide uma alíquota de 15% e um desconto de R$ 174,60, e acima deste valor incide uma alíquota de 27,5% e um desconto de R$ 465,35, conforme tabela progressiva mensal para fatos geradores a partir de 01/01/2005. Foi criada a tabela ir_mensal para guardar os dados do contribuinte utilizando o comando:

CREATE TABLE ir_mensal (
    nome        TEXT,           -- Nome
    rend_trib   DECIMAL(14,2),  -- Rendimento Tributável
    prev_ofic   DECIMAL(14,2),  -- Previdência Oficial
    dependentes INTEGER,        -- Número de Dependentes
    pens_alim   DECIMAL(14,2),  -- Pensão Alimentícia Judicial
    outr_deduc  DECIMAL(14,2)   -- Outras Deduções
);

A tabela foi carregada com os valores:

INSERT INTO ir_mensal VALUES ('José',1200,120,2,0,0);
INSERT INTO ir_mensal VALUES ('Joaquim',2200,220,3,0,0);
INSERT INTO ir_mensal VALUES ('Manuel',3200,320,0,0,100);
INSERT INTO ir_mensal VALUES ('Maria',12200.77,1220,0,4000,200);

A função round2dec para arredondar os números de ponto flutuante com duas casas decimais foi adicionada à tabela plruby_singleton_methods através do comando:

INSERT INTO plruby_singleton_methods VALUES
('round2dec', 'n', '("%.2f" % n).to_f','Arredondar com duas casas decimais');

A função escrita em PL/Ruby foi definida através do comando:

CREATE OR REPLACE FUNCTION ruby_ir(ir_mensal) RETURNS DECIMAL(14,2) AS $$
    base_calc =  args[0]["rend_trib"].to_f - (
                 args[0]["prev_ofic"].to_f +
                 (args[0]["dependentes"].to_f * 117.00) +
                 args[0]["pens_alim"].to_f +
                 args[0]["outr_deduc"].to_f)
    if base_calc <= 1164
        return 0.00
    elsif base_calc <= 2326
        return round2dec(base_calc * 0.15) - 174.60
    else
        return round2dec(base_calc * 0.275) - 465.35
    end
$$ LANGUAGE 'plruby' IMMUTABLE STRICT;

O teste da função foi feito através da execução do comando:

plruby=# \c plruby
Conectado ao banco de dados "plruby".
plruby=# SELECT nome,
plruby-#      rend_trib - (prev_ofic+117.00*dependentes+pens_alim+outr_deduc) AS base_calculo,
plruby-#      ruby_ir(ir_mensal) AS ir
plruby-# FROM   ir_mensal;

  nome   | base_calculo |   ir
---------+--------------+---------
 José    |       846.00 |       0
 Joaquim |      1629.00 |   69.75
 Manuel  |      2780.00 |  299.15
 Maria   |      6780.77 | 1399.36
(4 linhas)