diff options
Diffstat (limited to 'challenge-264/luca-ferrari/plpgsql/ch-2.sql')
| -rw-r--r-- | challenge-264/luca-ferrari/plpgsql/ch-2.sql | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/challenge-264/luca-ferrari/plpgsql/ch-2.sql b/challenge-264/luca-ferrari/plpgsql/ch-2.sql new file mode 100644 index 0000000000..39c15556ac --- /dev/null +++ b/challenge-264/luca-ferrari/plpgsql/ch-2.sql @@ -0,0 +1,46 @@ +-- +-- Perl Weekly Challenge 264 +-- Task 2 +-- See <https://perlweeklychallenge.org/blog/perl-weekly-challenge-264> +-- + +CREATE SCHEMA IF NOT EXISTS pwc264; + +CREATE OR REPLACE FUNCTION +pwc264.task2_plpgsql( s_a int[], i_a int[] ) +RETURNS SETOF int +AS $CODE$ +DECLARE + target_index int; + current_index int; +BEGIN + CREATE TEMPORARY TABLE IF NOT EXISTS target( i int, v int ); + TRUNCATE target; + + FOR current_index IN SELECT v::int FROM unnest( i_a ) v LOOP + target_index := i_a[ current_index ]; + + PERFORM * + FROM target + WHERE i = target_index; + + IF FOUND THEN + UPDATE target + SET i = i + 1 + WHERE i >= target_index; + + END IF; + + INSERT INTO target( i, v ) + VALUES ( target_index, s_a[ current_index ] ); + + END LOOP; + + RETURN QUERY + SELECT v + FROM target + ORDER BY i; + +END +$CODE$ +LANGUAGE plpgsql; |
