Add CHECK_FOR_INTERRUPTS() to the repeat() function

jconway · jconway · commit 887cdff4dcbd · 2020-05-28T13:19:00.000-04:00
The repeat() function loops for potentially a long time without ever checking for interrupts. This prevents, for example, a query cancel from interrupting until the work is all done. Fix by inserting a CHECK_FOR_INTERRUPTS() into the loop. Backpatch to all supported versions. Discussion: https://www.postgresql.org/message-id/flat/8692553c-7fe8-17d9-cbc1-7cddb758f4c6%40joeconway.com
diff --git a/src/backend/utils/adt/oracle_compat.c b/src/backend/utils/adt/oracle_compat.c
@@ -17,6 +17,7 @@
 
 #include "common/int.h"
 #include "mb/pg_wchar.h"
+#include "miscadmin.h"
 #include "utils/builtins.h"
 #include "utils/formatting.h"
 
@@ -1061,6 +1062,7 @@ repeat(PG_FUNCTION_ARGS)
 	{
 		memcpy(cp, sp, slen);
 		cp += slen;
+		CHECK_FOR_INTERRUPTS();
 	}
 
 	PG_RETURN_TEXT_P(result);

Original file line number	Diff line number	Diff line change
`@@ -17,6 +17,7 @@`
`17`	`17`
`18`	`18`	`#include "common/int.h"`
`19`	`19`	`#include "mb/pg_wchar.h"`
	`20`	`+#include "miscadmin.h"`
`20`	`21`	`#include "utils/builtins.h"`
`21`	`22`	`#include "utils/formatting.h"`
`22`	`23`
`@@ -1061,6 +1062,7 @@ repeat(PG_FUNCTION_ARGS)`
`1061`	`1062`	`{`
`1062`	`1063`	`memcpy(cp, sp, slen);`
`1063`	`1064`	`cp += slen;`
	`1065`	`+ CHECK_FOR_INTERRUPTS();`
`1064`	`1066`	`}`
`1065`	`1067`
`1066`	`1068`	`PG_RETURN_TEXT_P(result);`