ApsaraDB RDS:Horizontal splitting (PL/Proxy)

CREATE EXTENSION plproxy;

CREATE TABLE users (userid int, name text);

Sharding horizontal

A cláusula RUN ON ANY roteia cada INSERT aleatoriamente para um dos nós de dados e distribui as gravações pelo cluster.

1. Crie a função insert_user em cada nó de dados:

   CREATE OR REPLACE FUNCTION insert_user(i_id int, i_name text)
   RETURNS integer AS $$
       INSERT INTO users (userid, name) VALUES ($1, $2);
       SELECT 1;
   $$ LANGUAGE SQL;

2. Crie a função de proxy no nó de proxy. O PL/Proxy associa chamadas pelo nome da função; portanto, a função de proxy e a função do nó de dados devem ter o mesmo nome.

   CREATE OR REPLACE FUNCTION insert_user(i_id int, i_name text)
   RETURNS integer AS $$
       CLUSTER 'cluster_srv1';
       RUN ON ANY;
   $$ LANGUAGE plproxy;

3. Crie uma função de leitura no nó de proxy. A instrução RUN ON ALL consulta todos os nós de dados e mescla os resultados.

   CREATE OR REPLACE FUNCTION get_user_name()
   RETURNS TABLE(userid int, name text) AS $$
       CLUSTER 'cluster_srv1';
       RUN ON ALL;
       SELECT userid, name FROM users;
   $$ LANGUAGE plproxy;

4. Insira 10 registros a partir do nó de proxy:

   SELECT insert_user(1001, 'Sven');
   SELECT insert_user(1002, 'Marko');
   SELECT insert_user(1003, 'Steve');
   SELECT insert_user(1004, 'lottu');
   SELECT insert_user(1005, 'rax');
   SELECT insert_user(1006, 'ak');
   SELECT insert_user(1007, 'jack');
   SELECT insert_user(1008, 'molica');
   SELECT insert_user(1009, 'pg');
   SELECT insert_user(1010, 'oracle');

5. Consulte cada nó de dados para verificar a distribuição dos dados:

   Como RUN ON ANY roteia gravações aleatoriamente, a distribuição é desigual.

   -- On pl_db0
   SELECT * FROM users;
    userid |  name
   --------+--------
      1001 | Sven
      1003 | Steve
      1004 | lottu
      1005 | rax
      1006 | ak
      1007 | jack
      1008 | molica
      1009 | pg
   (8 rows)
   
   -- On pl_db1
   SELECT * FROM users;
    userid |  name
   --------+--------
      1002 | Marko
      1010 | oracle
   (2 rows)

6. Leia todas as linhas pelo nó de proxy:

   SELECT userid, name FROM get_user_name();
    userid |  name
   --------+--------
      1001 | Sven
      1003 | Steve
      1004 | lottu
      1005 | rax
      1006 | ak
      1007 | jack
      1008 | molica
      1009 | pg
      1002 | Marko
      1010 | oracle
   (10 rows)

Replicação SQL

A diretiva RUN ON ALL transmite cada gravação para todos os nós de dados e os mantém sincronizados.

1. Crie uma função trunc_user em cada nó de dados e também no nó de proxy:

   -- On pl_db0 and pl_db1
   CREATE OR REPLACE FUNCTION trunc_user()
   RETURNS integer AS $$
       TRUNCATE TABLE users;
       SELECT 1;
   $$ LANGUAGE SQL;
   
   -- On the proxy node
   CREATE OR REPLACE FUNCTION trunc_user()
   RETURNS SETOF integer AS $$
       CLUSTER 'cluster_srv1';
       RUN ON ALL;
   $$ LANGUAGE plproxy;

2. Trunque a tabela em todos os nós:

   SELECT trunc_user();
    trunc_user
   ------------
             1
             1
   (2 rows)

3. Crie a função de inserção com replicação no nó de proxy. O parâmetro TARGET insert_user instrui o PL/Proxy a chamar a função insert_user (já definida nos nós de dados) em cada nó:

   CREATE OR REPLACE FUNCTION insert_user_2(i_id int, i_name text)
   RETURNS SETOF integer AS $$
       CLUSTER 'cluster_srv1';
       RUN ON ALL;
       TARGET insert_user;
   $$ LANGUAGE plproxy;

4. Insira 4 registros:

   SELECT insert_user_2(1004, 'lottu');
   SELECT insert_user_2(1005, 'rax');
   SELECT insert_user_2(1006, 'ak');
   SELECT insert_user_2(1007, 'jack');

5. Verifique se ambos os nós de dados contêm dados idênticos:

   -- On pl_db0
   SELECT * FROM users;
    userid | name
   --------+-------
      1004 | lottu
      1005 | rax
      1006 | ak
      1007 | jack
   (4 rows)
   
   -- On pl_db1
   SELECT * FROM users;
    userid | name
   --------+-------
      1004 | lottu
      1005 | rax
      1006 | ak
      1007 | jack
   (4 rows)

6. Leia os dados pelo nó de proxy usando RUN ON ANY — como todos os nós possuem dados idênticos, qualquer um deles retorna o resultado completo:

   CREATE OR REPLACE FUNCTION get_user_name_2()
   RETURNS TABLE(userid int, name text) AS $$
       CLUSTER 'cluster_srv1';
       RUN ON ANY;
       SELECT userid, name FROM users;
   $$ LANGUAGE plproxy;
   
   SELECT userid, name FROM get_user_name_2();
    userid | name
   --------+-------
      1004 | lottu
      1005 | rax
      1006 | ak
      1007 | jack
   (4 rows)

CREATE OR REPLACE FUNCTION get_user_name_3()
RETURNS TABLE(userid int, name text) AS $$
    CONNECT 'dbname=pl_db0 host=100.81.137.72 port=56789';
    SELECT userid, name FROM users;
$$ LANGUAGE plproxy;

SELECT userid, name FROM get_user_name_3();
 userid | name
--------+-------
   1004 | lottu
   1005 | rax
   1006 | ak
   1007 | jack
(4 rows)