script.sql

SET statement_timeout = 0;
SET lock_timeout = 0;
SET idle_in_transaction_session_timeout = 0;
SET client_encoding = 'UTF8';
SET standard_conforming_strings = on;
SET check_function_bodies = false;
SET xmloption = content;
SET client_min_messages = warning;
SET row_security = off;


DROP TABLE IF EXISTS transacao;
DROP TABLE IF EXISTS cliente;

CREATE UNLOGGED TABLE cliente(
  id SERIAL PRIMARY KEY,  
  limite INTEGER NOT NULL,
  saldo INTEGER NOT NULL DEFAULT 0
);

CREATE UNLOGGED TABLE transacao(
  id SERIAL PRIMARY KEY,
  id_cliente INTEGER REFERENCES cliente(id),
  valor INTEGER NOT NULL,
  tipo CHAR NOT NULL,
  descricao VARCHAR(10) NOT NULL,
  data_transacao TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);

CREATE INDEX idx_cliente_id ON cliente(id);
CREATE INDEX idx_transacao_cliente_id ON transacao(id_cliente);

INSERT INTO cliente (limite, saldo)
VALUES
    (1000 * 100, 0),
    (800 * 100, 0),
    (10000 * 100, 0),
    (100000 * 100, 0),
    (5000 * 100, 0);

CREATE TYPE saldo_limite AS (
    saldo_cliente integer,
    limite_cliente integer,
    success bool
);

CREATE OR REPLACE FUNCTION debito_transaction(valor INTEGER, id_cliente INTEGER, descricao varchar(10))
RETURNS saldo_limite
LANGUAGE plpgsql 
AS $$
DECLARE
    result saldo_limite;
BEGIN
   UPDATE cliente SET saldo = saldo - valor WHERE id = id_cliente AND saldo - valor >= limite * (-1)
   RETURNING saldo, limite INTO result.saldo_cliente, result.limite_cliente;
   
   IF result.saldo_cliente >= result.limite_cliente * (-1) then
   		result.success := true;
		INSERT INTO transacao (valor, tipo, descricao,  id_cliente)
		VALUES(valor, 'd', descricao, id_cliente);
	else
		result.success := false;
   END IF;
   
   RETURN result;
END;
$$;

CREATE OR REPLACE FUNCTION credito_transaction(valor INTEGER, id_cliente INTEGER, descricao varchar(10))
RETURNS saldo_limite
LANGUAGE plpgsql 
AS $$
DECLARE
    result saldo_limite;
BEGIN   
   UPDATE cliente SET saldo = saldo + valor WHERE id = id_cliente
   RETURNING saldo, limite INTO result.saldo_cliente, result.limite_cliente;

   result.success := true;
   
   INSERT INTO transacao (valor, tipo, descricao,  id_cliente)
   VALUES(valor, 'c', descricao, id_cliente);
   
   RETURN result;
END;
$$;