Add handling for plpgsql LOOP and EXIT

src/pgspot/visitors.py

@@ -138,6 +138,12 @@ def visit(self, node):
                                 self.state,
                                 "SELECT " + value["expr"]["PLpgSQL_expr"]["query"],
                             )
+                    case "PLpgSQL_stmt_exit":
+                        if "cond" in value:
+                            visit_sql(
+                                self.state,
+                                "SELECT " + value["cond"]["PLpgSQL_expr"]["query"],
+                            )
                     case "PLpgSQL_stmt_if":
                         if "cond" in value:
                             visit_sql(
@@ -169,6 +175,9 @@ def visit(self, node):
                             )
                         if "body" in value:
                             self.visit(value["body"])
+                    case "PLpgSQL_stmt_loop":
+                        if "body" in value:
+                            self.visit(value["body"])
                     case "PLpgSQL_stmt_raise":
                         if "params" in value:
                             for item in value["params"]:

testdata/expected/loop.out

@@ -0,0 +1,6 @@
+PS002: Unsafe function creation: l() at line 1
+PS005: Function without explicit search_path: l() at line 1
+PS016: Unqualified function call: f1 at line 1
+
+ Errors: 0 Warnings: 3 Unknown: 0 
+

testdata/loop.sql

@@ -0,0 +1,11 @@
+create or replace function l() returns void as $func$
+declare
+begin
+ 	loop
+ 	end loop;
+ 	loop
+ 		PERFORM f1('select 1');
+    exit when true;
+ 	end loop;
+end
+$func$ language plpgsql;