aboutsummaryrefslogtreecommitdiff
path: root/challenge-333/luca-ferrari/plpgsql/ch-2.sql
blob: 363466f8240258245491704b8c239f994ab95ea6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
--
-- Perl Weekly Challenge 333
-- Task 2
-- See <https://perlweeklychallenge.org/blog/perl-weekly-challenge-333>
--

CREATE SCHEMA IF NOT EXISTS pwc333;

CREATE OR REPLACE FUNCTION
pwc333.task2_plpgsql( n int[] )
RETURNS SETOF int
AS $CODE$
DECLARE
	v int;
	l int;
BEGIN

	l := array_length( n, 1 );

	FOREACH v IN ARRAY n LOOP
		IF l = 0 THEN
		   RETURN;
		END IF;


		RETURN NEXT v;
		l := l - 1;

		IF l = 0 THEN
		   RETURN;
		END IF;

		IF v = 0 THEN
		   RETURN NEXT v;
		   l := l - 1;
		END IF;
	END LOOP;

	RETURN;
END
$CODE$
LANGUAGE plpgsql;