blob: 161f0f1edf68c67864010cab906bf046e49766eb (
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
/*-------------------------------------------------------------------------
*
* aio_target.c
* AIO - Functionality related to executing IO for different targets
*
* Portions Copyright (c) 1996-2025, PostgreSQL Global Development Group
* Portions Copyright (c) 1994, Regents of the University of California
*
* IDENTIFICATION
* src/backend/storage/aio/aio_target.c
*
*-------------------------------------------------------------------------
*/
#include "postgres.h"
#include "storage/aio.h"
#include "storage/aio_internal.h"
#include "storage/smgr.h"
/*
* Registry for entities that can be the target of AIO.
*/
static const PgAioTargetInfo *pgaio_target_info[] = {
[PGAIO_TID_INVALID] = &(PgAioTargetInfo) {
.name = "invalid",
},
[PGAIO_TID_SMGR] = &aio_smgr_target_info,
};
/* --------------------------------------------------------------------------------
* Public target related functions operating on IO Handles
* --------------------------------------------------------------------------------
*/
bool
pgaio_io_has_target(PgAioHandle *ioh)
{
return ioh->target != PGAIO_TID_INVALID;
}
/*
* Return the name for the target associated with the IO. Mostly useful for
* debugging/logging.
*/
const char *
pgaio_io_get_target_name(PgAioHandle *ioh)
{
/* explicitly allow INVALID here, function used by debug messages */
Assert(ioh->target >= PGAIO_TID_INVALID && ioh->target < PGAIO_TID_COUNT);
return pgaio_target_info[ioh->target]->name;
}
/*
* Assign a target to the IO.
*
* This has to be called exactly once before pgaio_io_start_*() is called.
*/
void
pgaio_io_set_target(PgAioHandle *ioh, PgAioTargetID targetid)
{
Assert(ioh->state == PGAIO_HS_HANDED_OUT);
Assert(ioh->target == PGAIO_TID_INVALID);
ioh->target = targetid;
}
PgAioTargetData *
pgaio_io_get_target_data(PgAioHandle *ioh)
{
return &ioh->target_data;
}
/*
* Return a stringified description of the IO's target.
*
* The string is localized and allocated in the current memory context.
*/
char *
pgaio_io_get_target_description(PgAioHandle *ioh)
{
/* disallow INVALID, there wouldn't be a description */
Assert(ioh->target > PGAIO_TID_INVALID && ioh->target < PGAIO_TID_COUNT);
return pgaio_target_info[ioh->target]->describe_identity(&ioh->target_data);
}
/* --------------------------------------------------------------------------------
* Internal target related functions operating on IO Handles
* --------------------------------------------------------------------------------
*/
/*
* Internal: Check if pgaio_io_reopen() is available for the IO.
*/
bool
pgaio_io_can_reopen(PgAioHandle *ioh)
{
Assert(ioh->target > PGAIO_TID_INVALID && ioh->target < PGAIO_TID_COUNT);
return pgaio_target_info[ioh->target]->reopen != NULL;
}
/*
* Internal: Before executing an IO outside of the context of the process the
* IO has been staged in, the file descriptor has to be reopened - any FD
* referenced in the IO itself, won't be valid in the separate process.
*/
void
pgaio_io_reopen(PgAioHandle *ioh)
{
Assert(ioh->target > PGAIO_TID_INVALID && ioh->target < PGAIO_TID_COUNT);
Assert(ioh->op > PGAIO_OP_INVALID && ioh->op < PGAIO_OP_COUNT);
pgaio_target_info[ioh->target]->reopen(ioh);
}
|
