contrib: Raise error if pageinspect is not installed

df7cb · May 28, 2020 · fb3a90b · fb3a90b
1 parent 7eb75ac
commit fb3a90b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/contrib/bad_tuples.sql b/contrib/bad_tuples.sql
@@ -29,6 +29,10 @@ begin
       end loop;
 
     exception -- bad page
+      when undefined_function then
+        raise exception undefined_function
+          using message = SQLERRM,
+                hint = 'Use CREATE EXTENSION pageinspect; to create it';
       when others then
         bad_tuples.page := page;
         bad_tuples.sqlstate := sqlstate;

diff --git a/contrib/read_table.sql b/contrib/read_table.sql
@@ -30,6 +30,10 @@ begin
       end loop;
 
     exception -- bad page
+      when undefined_function then
+        raise exception undefined_function
+          using message = SQLERRM,
+                hint = 'Use CREATE EXTENSION pageinspect; to create it';
       when others then
         get stacked diagnostics sql_state := RETURNED_SQLSTATE;
         get stacked diagnostics error := MESSAGE_TEXT;

diff --git a/contrib/rescue_table.sql b/contrib/rescue_table.sql
@@ -46,6 +46,10 @@ begin
       end loop;
 
     exception -- bad page
+      when undefined_function then
+        raise exception undefined_function
+          using message = SQLERRM,
+                hint = 'Use CREATE EXTENSION pageinspect; to create it';
       when others then
         get stacked diagnostics sql_state := RETURNED_SQLSTATE;
         get stacked diagnostics error := MESSAGE_TEXT;
@@ -62,5 +66,5 @@ begin
 end;
 $$ language plpgsql;
 
-comment on function rescue_table(regclass, name, boolean) is
+comment on function rescue_table(regclass, name, boolean, int, int) is
   'copy all good tuples from a table to another one';