sharpetronics/PostgreSQL
1
0
Fork 0
mirror of https://github.com/postgres/postgres.git synced 2025-05-11 00:03:01 -04:00
Code Issues Projects Releases Wiki Activity

aio: Use runtime arguments with injections points in tests

Browse Source 
This cleans up the code related to the testing infrastructure of AIO
that used injection points, switching the test code to use the new
facility for injection points added by 371f2db8b05e rather than tweaks
to pass and reset arguments to the callbacks run.

This removes all the dependencies to USE_INJECTION_POINTS in the AIO
code.  pgaio_io_call_inj(), pgaio_inj_io_get() and pgaio_inj_cur_handle
are now gone.

Reviewed-by: Greg Burd <greg@burd.me>
Discussion: https://postgr.es/m/Z_y9TtnXubvYAApS@paquier.xyz
This commit is contained in:
Michael Paquier 2025-05-10 12:36:57 +09:00
parent 36e5fda632
commit c259ba881c
4 changed files with 5 additions and 80 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
src/backend/storage/aio/aio.c
View File

@ -46,13 +46,10 @@
#include "storage/aio_subsys.h"
#include "utils/guc.h"
#include "utils/guc_hooks.h"
#include "utils/injection_point.h"
#include "utils/resowner.h"
#include "utils/wait_event_types.h"
#ifdef USE_INJECTION_POINTS
#include "utils/injection_point.h"
#endif
static inline void pgaio_io_update_state(PgAioHandle *ioh, PgAioHandleState new_state);
static void pgaio_io_reclaim(PgAioHandle *ioh);
@ -96,17 +93,6 @@ static const IoMethodOps *const pgaio_method_ops_table[] = {
const IoMethodOps *pgaio_method_ops;
/*
* Currently there's no infrastructure to pass arguments to injection points,
* so we instead set this up for the duration of the injection point
* invocation. See pgaio_io_call_inj().
*/
#ifdef USE_INJECTION_POINTS
static PgAioHandle *pgaio_inj_cur_handle;
#endif
/* --------------------------------------------------------------------------------
* Public Functions related to PgAioHandle
* --------------------------------------------------------------------------------
@ -507,7 +493,7 @@ pgaio_io_process_completion(PgAioHandle *ioh, int result)
pgaio_io_update_state(ioh, PGAIO_HS_COMPLETED_IO);
pgaio_io_call_inj(ioh, "aio-process-completion-before-shared");
INJECTION_POINT("aio-process-completion-before-shared", ioh);
pgaio_io_call_complete_shared(ioh);
@ -1255,43 +1241,3 @@ check_io_max_concurrency(int *newval, void **extra, GucSource source)
return true;
}
/* --------------------------------------------------------------------------------
* Injection point support
* --------------------------------------------------------------------------------
*/
#ifdef USE_INJECTION_POINTS
/*
* Call injection point with support for pgaio_inj_io_get().
*/
void
pgaio_io_call_inj(PgAioHandle *ioh, const char *injection_point)
{
pgaio_inj_cur_handle = ioh;
PG_TRY();
{
InjectionPointCached(injection_point, NULL);
}
PG_FINALLY();
{
pgaio_inj_cur_handle = NULL;
}
PG_END_TRY();
}
/*
* Return IO associated with injection point invocation. This is only needed
* as injection points currently don't support arguments.
*/
PgAioHandle *
pgaio_inj_io_get(void)
{
return pgaio_inj_cur_handle;
}
#endif

3
src/backend/storage/aio/method_worker.c
View File

@ -42,6 +42,7 @@
#include "storage/latch.h"
#include "storage/proc.h"
#include "tcop/tcopprot.h"
#include "utils/injection_point.h"
#include "utils/memdebug.h"
#include "utils/ps_status.h"
#include "utils/wait_event.h"
@ -525,7 +526,7 @@ IoWorkerMain(const void *startup_data, size_t startup_data_len)
* To be able to exercise the reopen-fails path, allow injection
* points to trigger a failure at this point.
*/
pgaio_io_call_inj(ioh, "aio-worker-after-reopen");
INJECTION_POINT("aio-worker-after-reopen", ioh);
error_errno = 0;
error_ioh = NULL;

20
src/include/storage/aio_internal.h
View File

@ -394,26 +394,6 @@ extern const char *pgaio_io_get_target_name(PgAioHandle *ioh);
pgaio_io_get_state_name(ioh), \
__VA_ARGS__)
#ifdef USE_INJECTION_POINTS
extern void pgaio_io_call_inj(PgAioHandle *ioh, const char *injection_point);
/* just for use in tests, from within injection points */
extern PgAioHandle *pgaio_inj_io_get(void);
#else
#define pgaio_io_call_inj(ioh, injection_point) (void) 0
/*
* no fallback for pgaio_inj_io_get, all code using injection points better be
* guarded by USE_INJECTION_POINTS.
*/
#endif
/* Declarations for the tables of function pointers exposed by each IO method. */
extern PGDLLIMPORT const IoMethodOps pgaio_sync_ops;
extern PGDLLIMPORT const IoMethodOps pgaio_worker_ops;

4
src/test/modules/test_aio/test_aio.c
View File

@ -684,7 +684,7 @@ extern PGDLLEXPORT void inj_io_reopen(const char *name,
void
inj_io_short_read(const char *name, const void *private_data, void *arg)
{
PgAioHandle *ioh;
PgAioHandle *ioh = (PgAioHandle *) arg;
ereport(LOG,
errmsg("short read injection point called, is enabled: %d",
@ -693,8 +693,6 @@ inj_io_short_read(const char *name, const void *private_data, void *arg)
if (inj_io_error_state->enabled_short_read)
{
ioh = pgaio_inj_io_get();
/*
* Only shorten reads that are actually longer than the target size,
* otherwise we can trigger over-reads.