PostgreSQL Source Code  git master
file_fdw.c File Reference
#include "postgres.h"
#include <sys/stat.h>
#include <unistd.h>
#include "access/htup_details.h"
#include "access/reloptions.h"
#include "access/sysattr.h"
#include "access/table.h"
#include "catalog/pg_authid.h"
#include "catalog/pg_foreign_table.h"
#include "commands/copy.h"
#include "commands/defrem.h"
#include "commands/explain.h"
#include "commands/vacuum.h"
#include "foreign/fdwapi.h"
#include "foreign/foreign.h"
#include "miscadmin.h"
#include "nodes/makefuncs.h"
#include "optimizer/optimizer.h"
#include "optimizer/pathnode.h"
#include "optimizer/planmain.h"
#include "optimizer/restrictinfo.h"
#include "utils/acl.h"
#include "utils/memutils.h"
#include "utils/rel.h"
#include "utils/sampling.h"
Include dependency graph for file_fdw.c:

Go to the source code of this file.

Data Structures

struct  FileFdwOption
 
struct  FileFdwPlanState
 
struct  FileFdwExecutionState
 

Typedefs

typedef struct FileFdwPlanState FileFdwPlanState
 
typedef struct FileFdwExecutionState FileFdwExecutionState
 

Functions

 PG_FUNCTION_INFO_V1 (file_fdw_handler)
 
 PG_FUNCTION_INFO_V1 (file_fdw_validator)
 
static void fileGetForeignRelSize (PlannerInfo *root, RelOptInfo *baserel, Oid foreigntableid)
 
static void fileGetForeignPaths (PlannerInfo *root, RelOptInfo *baserel, Oid foreigntableid)
 
static ForeignScanfileGetForeignPlan (PlannerInfo *root, RelOptInfo *baserel, Oid foreigntableid, ForeignPath *best_path, List *tlist, List *scan_clauses, Plan *outer_plan)
 
static void fileExplainForeignScan (ForeignScanState *node, ExplainState *es)
 
static void fileBeginForeignScan (ForeignScanState *node, int eflags)
 
static TupleTableSlotfileIterateForeignScan (ForeignScanState *node)
 
static void fileReScanForeignScan (ForeignScanState *node)
 
static void fileEndForeignScan (ForeignScanState *node)
 
static bool fileAnalyzeForeignTable (Relation relation, AcquireSampleRowsFunc *func, BlockNumber *totalpages)
 
static bool fileIsForeignScanParallelSafe (PlannerInfo *root, RelOptInfo *rel, RangeTblEntry *rte)
 
static bool is_valid_option (const char *option, Oid context)
 
static void fileGetOptions (Oid foreigntableid, char **filename, bool *is_program, List **other_options)
 
static Listget_file_fdw_attribute_options (Oid relid)
 
static bool check_selective_binary_conversion (RelOptInfo *baserel, Oid foreigntableid, List **columns)
 
static void estimate_size (PlannerInfo *root, RelOptInfo *baserel, FileFdwPlanState *fdw_private)
 
static void estimate_costs (PlannerInfo *root, RelOptInfo *baserel, FileFdwPlanState *fdw_private, Cost *startup_cost, Cost *total_cost)
 
static int file_acquire_sample_rows (Relation onerel, int elevel, HeapTuple *rows, int targrows, double *totalrows, double *totaldeadrows)
 
Datum file_fdw_handler (PG_FUNCTION_ARGS)
 
Datum file_fdw_validator (PG_FUNCTION_ARGS)
 

Variables

 PG_MODULE_MAGIC
 
static const struct FileFdwOption valid_options []
 

Typedef Documentation

◆ FileFdwExecutionState

◆ FileFdwPlanState

Function Documentation

◆ check_selective_binary_conversion()

static bool check_selective_binary_conversion ( RelOptInfo baserel,
Oid  foreigntableid,
List **  columns 
)
static

Definition at line 847 of file file_fdw.c.

References AccessShareLock, attname, attnum, RelOptInfo::baserestrictinfo, bms_first_member(), RestrictInfo::clause, defGetString(), DefElem::defname, PathTarget::exprs, FirstLowInvalidHeapAttributeNumber, format, GetForeignTable(), i, lappend(), lfirst, list_length(), makeString(), NameStr, TupleDescData::natts, NIL, ForeignTable::options, pstrdup(), pull_varattnos(), RelationGetDescr, RelOptInfo::relid, RelOptInfo::reltarget, table_close(), table_open(), and TupleDescAttr.

Referenced by fileGetForeignPaths().

850 {
851  ForeignTable *table;
852  ListCell *lc;
853  Relation rel;
854  TupleDesc tupleDesc;
856  Bitmapset *attrs_used = NULL;
857  bool has_wholerow = false;
858  int numattrs;
859  int i;
860 
861  *columns = NIL; /* default result */
862 
863  /*
864  * Check format of the file. If binary format, this is irrelevant.
865  */
866  table = GetForeignTable(foreigntableid);
867  foreach(lc, table->options)
868  {
869  DefElem *def = (DefElem *) lfirst(lc);
870 
871  if (strcmp(def->defname, "format") == 0)
872  {
873  char *format = defGetString(def);
874 
875  if (strcmp(format, "binary") == 0)
876  return false;
877  break;
878  }
879  }
880 
881  /* Collect all the attributes needed for joins or final output. */
882  pull_varattnos((Node *) baserel->reltarget->exprs, baserel->relid,
883  &attrs_used);
884 
885  /* Add all the attributes used by restriction clauses. */
886  foreach(lc, baserel->baserestrictinfo)
887  {
888  RestrictInfo *rinfo = (RestrictInfo *) lfirst(lc);
889 
890  pull_varattnos((Node *) rinfo->clause, baserel->relid,
891  &attrs_used);
892  }
893 
894  /* Convert attribute numbers to column names. */
895  rel = table_open(foreigntableid, AccessShareLock);
896  tupleDesc = RelationGetDescr(rel);
897 
898  while ((attnum = bms_first_member(attrs_used)) >= 0)
899  {
900  /* Adjust for system attributes. */
902 
903  if (attnum == 0)
904  {
905  has_wholerow = true;
906  break;
907  }
908 
909  /* Ignore system attributes. */
910  if (attnum < 0)
911  continue;
912 
913  /* Get user attributes. */
914  if (attnum > 0)
915  {
916  Form_pg_attribute attr = TupleDescAttr(tupleDesc, attnum - 1);
917  char *attname = NameStr(attr->attname);
918 
919  /* Skip dropped attributes (probably shouldn't see any here). */
920  if (attr->attisdropped)
921  continue;
922 
923  /*
924  * Skip generated columns (COPY won't accept them in the column
925  * list)
926  */
927  if (attr->attgenerated)
928  continue;
929  *columns = lappend(*columns, makeString(pstrdup(attname)));
930  }
931  }
932 
933  /* Count non-dropped user attributes while we have the tupdesc. */
934  numattrs = 0;
935  for (i = 0; i < tupleDesc->natts; i++)
936  {
937  Form_pg_attribute attr = TupleDescAttr(tupleDesc, i);
938 
939  if (attr->attisdropped)
940  continue;
941  numattrs++;
942  }
943 
945 
946  /* If there's a whole-row reference, fail: we need all the columns. */
947  if (has_wholerow)
948  {
949  *columns = NIL;
950  return false;
951  }
952 
953  /* If all the user attributes are needed, fail. */
954  if (numattrs == list_length(*columns))
955  {
956  *columns = NIL;
957  return false;
958  }
959 
960  return true;
961 }
#define NIL
Definition: pg_list.h:65
int bms_first_member(Bitmapset *a)
Definition: bitmapset.c:996
void table_close(Relation relation, LOCKMODE lockmode)
Definition: table.c:167
#define RelationGetDescr(relation)
Definition: rel.h:503
#define TupleDescAttr(tupdesc, i)
Definition: tupdesc.h:92
char * pstrdup(const char *in)
Definition: mcxt.c:1299
List * baserestrictinfo
Definition: pathnodes.h:745
#define AccessShareLock
Definition: lockdefs.h:36
Definition: nodes.h:537
ForeignTable * GetForeignTable(Oid relid)
Definition: foreign.c:248
String * makeString(char *str)
Definition: value.c:51
#define FirstLowInvalidHeapAttributeNumber
Definition: sysattr.h:27
void pull_varattnos(Node *node, Index varno, Bitmapset **varattnos)
Definition: var.c:281
NameData attname
Definition: pg_attribute.h:41
char * defGetString(DefElem *def)
Definition: define.c:49
FormData_pg_attribute * Form_pg_attribute
Definition: pg_attribute.h:207
Index relid
Definition: pathnodes.h:709
List * lappend(List *list, void *datum)
Definition: list.c:336
Expr * clause
Definition: pathnodes.h:2056
List * exprs
Definition: pathnodes.h:1110
int16 attnum
Definition: pg_attribute.h:83
#define lfirst(lc)
Definition: pg_list.h:169
static int list_length(const List *l)
Definition: pg_list.h:149
List * options
Definition: foreign.h:57
int i
#define NameStr(name)
Definition: c.h:681
char * defname
Definition: parsenodes.h:758
static char format
Relation table_open(Oid relationId, LOCKMODE lockmode)
Definition: table.c:39
struct PathTarget * reltarget
Definition: pathnodes.h:692
int16 AttrNumber
Definition: attnum.h:21

◆ estimate_costs()

static void estimate_costs ( PlannerInfo root,
RelOptInfo baserel,
FileFdwPlanState fdw_private,
Cost startup_cost,
Cost total_cost 
)
static

Definition at line 1052 of file file_fdw.c.

References RelOptInfo::baserestrictcost, cpu_tuple_cost, FileFdwPlanState::ntuples, FileFdwPlanState::pages, QualCost::per_tuple, seq_page_cost, and QualCost::startup.

Referenced by fileGetForeignPaths().

1055 {
1056  BlockNumber pages = fdw_private->pages;
1057  double ntuples = fdw_private->ntuples;
1058  Cost run_cost = 0;
1059  Cost cpu_per_tuple;
1060 
1061  /*
1062  * We estimate costs almost the same way as cost_seqscan(), thus assuming
1063  * that I/O costs are equivalent to a regular table file of the same size.
1064  * However, we take per-tuple CPU costs as 10x of a seqscan, to account
1065  * for the cost of parsing records.
1066  *
1067  * In the case of a program source, this calculation is even more divorced
1068  * from reality, but we have no good alternative; and it's not clear that
1069  * the numbers we produce here matter much anyway, since there's only one
1070  * access path for the rel.
1071  */
1072  run_cost += seq_page_cost * pages;
1073 
1074  *startup_cost = baserel->baserestrictcost.startup;
1075  cpu_per_tuple = cpu_tuple_cost * 10 + baserel->baserestrictcost.per_tuple;
1076  run_cost += cpu_per_tuple * ntuples;
1077  *total_cost = *startup_cost + run_cost;
1078 }
BlockNumber pages
Definition: file_fdw.c:95
uint32 BlockNumber
Definition: block.h:31
Cost startup
Definition: pathnodes.h:45
double ntuples
Definition: file_fdw.c:96
Cost per_tuple
Definition: pathnodes.h:46
double cpu_tuple_cost
Definition: costsize.c:121
double seq_page_cost
Definition: costsize.c:119
QualCost baserestrictcost
Definition: pathnodes.h:746
double Cost
Definition: nodes.h:671

◆ estimate_size()

static void estimate_size ( PlannerInfo root,
RelOptInfo baserel,
FileFdwPlanState fdw_private 
)
static

Definition at line 971 of file file_fdw.c.

References RelOptInfo::baserestrictinfo, clamp_row_est(), clauselist_selectivity(), FileFdwPlanState::filename, FileFdwPlanState::is_program, JOIN_INNER, MAXALIGN, FileFdwPlanState::ntuples, FileFdwPlanState::pages, RelOptInfo::pages, RelOptInfo::reltarget, RelOptInfo::rows, SizeofHeapTupleHeader, stat::st_size, RelOptInfo::tuples, and PathTarget::width.

Referenced by fileGetForeignRelSize().

973 {
974  struct stat stat_buf;
975  BlockNumber pages;
976  double ntuples;
977  double nrows;
978 
979  /*
980  * Get size of the file. It might not be there at plan time, though, in
981  * which case we have to use a default estimate. We also have to fall
982  * back to the default if using a program as the input.
983  */
984  if (fdw_private->is_program || stat(fdw_private->filename, &stat_buf) < 0)
985  stat_buf.st_size = 10 * BLCKSZ;
986 
987  /*
988  * Convert size to pages for use in I/O cost estimate later.
989  */
990  pages = (stat_buf.st_size + (BLCKSZ - 1)) / BLCKSZ;
991  if (pages < 1)
992  pages = 1;
993  fdw_private->pages = pages;
994 
995  /*
996  * Estimate the number of tuples in the file.
997  */
998  if (baserel->tuples >= 0 && baserel->pages > 0)
999  {
1000  /*
1001  * We have # of pages and # of tuples from pg_class (that is, from a
1002  * previous ANALYZE), so compute a tuples-per-page estimate and scale
1003  * that by the current file size.
1004  */
1005  double density;
1006 
1007  density = baserel->tuples / (double) baserel->pages;
1008  ntuples = clamp_row_est(density * (double) pages);
1009  }
1010  else
1011  {
1012  /*
1013  * Otherwise we have to fake it. We back into this estimate using the
1014  * planner's idea of the relation width; which is bogus if not all
1015  * columns are being read, not to mention that the text representation
1016  * of a row probably isn't the same size as its internal
1017  * representation. Possibly we could do something better, but the
1018  * real answer to anyone who complains is "ANALYZE" ...
1019  */
1020  int tuple_width;
1021 
1022  tuple_width = MAXALIGN(baserel->reltarget->width) +
1024  ntuples = clamp_row_est((double) stat_buf.st_size /
1025  (double) tuple_width);
1026  }
1027  fdw_private->ntuples = ntuples;
1028 
1029  /*
1030  * Now estimate the number of rows returned by the scan after applying the
1031  * baserestrictinfo quals.
1032  */
1033  nrows = ntuples *
1035  baserel->baserestrictinfo,
1036  0,
1037  JOIN_INNER,
1038  NULL);
1039 
1040  nrows = clamp_row_est(nrows);
1041 
1042  /* Save the output-rows estimate for the planner */
1043  baserel->rows = nrows;
1044 }
#define SizeofHeapTupleHeader
Definition: htup_details.h:184
BlockNumber pages
Definition: file_fdw.c:95
List * baserestrictinfo
Definition: pathnodes.h:745
uint32 BlockNumber
Definition: block.h:31
double ntuples
Definition: file_fdw.c:96
__int64 st_size
Definition: win32_port.h:273
char * filename
Definition: file_fdw.c:91
BlockNumber pages
Definition: pathnodes.h:720
Cardinality rows
Definition: pathnodes.h:684
#define MAXALIGN(LEN)
Definition: c.h:757
Selectivity clauselist_selectivity(PlannerInfo *root, List *clauses, int varRelid, JoinType jointype, SpecialJoinInfo *sjinfo)
Definition: clausesel.c:102
double clamp_row_est(double nrows)
Definition: costsize.c:199
struct PathTarget * reltarget
Definition: pathnodes.h:692
Cardinality tuples
Definition: pathnodes.h:721

◆ file_acquire_sample_rows()

static int file_acquire_sample_rows ( Relation  onerel,
int  elevel,
HeapTuple rows,
int  targrows,
double *  totalrows,
double *  totaldeadrows 
)
static

Definition at line 1095 of file file_fdw.c.

References ALLOCSET_DEFAULT_SIZES, AllocSetContextCreate, ErrorContextCallback::arg, Assert, BeginCopyFrom(), ErrorContextCallback::callback, CopyFromErrorCallback(), CurrentMemoryContext, EndCopyFrom(), ereport, errmsg(), error_context_stack, fileGetOptions(), filename, heap_form_tuple(), heap_freetuple(), MemoryContextDelete(), MemoryContextReset(), MemoryContextSwitchTo(), TupleDescData::natts, NextCopyFrom(), NIL, options, palloc(), pfree(), ErrorContextCallback::previous, ReservoirStateData::randstate, RelationGetDescr, RelationGetRelationName, RelationGetRelid, reservoir_get_next_S(), reservoir_init_selection_state(), sampler_random_fract(), vacuum_delay_point(), and values.

Referenced by fileAnalyzeForeignTable().

1098 {
1099  int numrows = 0;
1100  double rowstoskip = -1; /* -1 means not set yet */
1101  ReservoirStateData rstate;
1102  TupleDesc tupDesc;
1103  Datum *values;
1104  bool *nulls;
1105  bool found;
1106  char *filename;
1107  bool is_program;
1108  List *options;
1109  CopyFromState cstate;
1110  ErrorContextCallback errcallback;
1111  MemoryContext oldcontext = CurrentMemoryContext;
1112  MemoryContext tupcontext;
1113 
1114  Assert(onerel);
1115  Assert(targrows > 0);
1116 
1117  tupDesc = RelationGetDescr(onerel);
1118  values = (Datum *) palloc(tupDesc->natts * sizeof(Datum));
1119  nulls = (bool *) palloc(tupDesc->natts * sizeof(bool));
1120 
1121  /* Fetch options of foreign table */
1122  fileGetOptions(RelationGetRelid(onerel), &filename, &is_program, &options);
1123 
1124  /*
1125  * Create CopyState from FDW options.
1126  */
1127  cstate = BeginCopyFrom(NULL, onerel, NULL, filename, is_program, NULL, NIL,
1128  options);
1129 
1130  /*
1131  * Use per-tuple memory context to prevent leak of memory used to read
1132  * rows from the file with Copy routines.
1133  */
1135  "file_fdw temporary context",
1137 
1138  /* Prepare for sampling rows */
1139  reservoir_init_selection_state(&rstate, targrows);
1140 
1141  /* Set up callback to identify error line number. */
1142  errcallback.callback = CopyFromErrorCallback;
1143  errcallback.arg = (void *) cstate;
1144  errcallback.previous = error_context_stack;
1145  error_context_stack = &errcallback;
1146 
1147  *totalrows = 0;
1148  *totaldeadrows = 0;
1149  for (;;)
1150  {
1151  /* Check for user-requested abort or sleep */
1153 
1154  /* Fetch next row */
1155  MemoryContextReset(tupcontext);
1156  MemoryContextSwitchTo(tupcontext);
1157 
1158  found = NextCopyFrom(cstate, NULL, values, nulls);
1159 
1160  MemoryContextSwitchTo(oldcontext);
1161 
1162  if (!found)
1163  break;
1164 
1165  /*
1166  * The first targrows sample rows are simply copied into the
1167  * reservoir. Then we start replacing tuples in the sample until we
1168  * reach the end of the relation. This algorithm is from Jeff Vitter's
1169  * paper (see more info in commands/analyze.c).
1170  */
1171  if (numrows < targrows)
1172  {
1173  rows[numrows++] = heap_form_tuple(tupDesc, values, nulls);
1174  }
1175  else
1176  {
1177  /*
1178  * t in Vitter's paper is the number of records already processed.
1179  * If we need to compute a new S value, we must use the
1180  * not-yet-incremented value of totalrows as t.
1181  */
1182  if (rowstoskip < 0)
1183  rowstoskip = reservoir_get_next_S(&rstate, *totalrows, targrows);
1184 
1185  if (rowstoskip <= 0)
1186  {
1187  /*
1188  * Found a suitable tuple, so save it, replacing one old tuple
1189  * at random
1190  */
1191  int k = (int) (targrows * sampler_random_fract(rstate.randstate));
1192 
1193  Assert(k >= 0 && k < targrows);
1194  heap_freetuple(rows[k]);
1195  rows[k] = heap_form_tuple(tupDesc, values, nulls);
1196  }
1197 
1198  rowstoskip -= 1;
1199  }
1200 
1201  *totalrows += 1;
1202  }
1203 
1204  /* Remove error callback. */
1205  error_context_stack = errcallback.previous;
1206 
1207  /* Clean up. */
1208  MemoryContextDelete(tupcontext);
1209 
1210  EndCopyFrom(cstate);
1211 
1212  pfree(values);
1213  pfree(nulls);
1214 
1215  /*
1216  * Emit some interesting relation info
1217  */
1218  ereport(elevel,
1219  (errmsg("\"%s\": file contains %.0f rows; "
1220  "%d rows in sample",
1221  RelationGetRelationName(onerel),
1222  *totalrows, numrows)));
1223 
1224  return numrows;
1225 }
#define NIL
Definition: pg_list.h:65
void MemoryContextDelete(MemoryContext context)
Definition: mcxt.c:218
#define AllocSetContextCreate
Definition: memutils.h:173
bool NextCopyFrom(CopyFromState cstate, ExprContext *econtext, Datum *values, bool *nulls)
#define RelationGetDescr(relation)
Definition: rel.h:503
static MemoryContext MemoryContextSwitchTo(MemoryContext context)
Definition: palloc.h:109
double sampler_random_fract(SamplerRandomState randstate)
Definition: sampling.c:242
void MemoryContextReset(MemoryContext context)
Definition: mcxt.c:143
void reservoir_init_selection_state(ReservoirState rs, int n)
Definition: sampling.c:133
HeapTuple heap_form_tuple(TupleDesc tupleDescriptor, Datum *values, bool *isnull)
Definition: heaptuple.c:1020
void heap_freetuple(HeapTuple htup)
Definition: heaptuple.c:1338
void(* callback)(void *arg)
Definition: elog.h:247
struct ErrorContextCallback * previous
Definition: elog.h:246
ErrorContextCallback * error_context_stack
Definition: elog.c:93
void CopyFromErrorCallback(void *arg)
Definition: copyfrom.c:115
void pfree(void *pointer)
Definition: mcxt.c:1169
#define ALLOCSET_DEFAULT_SIZES
Definition: memutils.h:195
#define RelationGetRelationName(relation)
Definition: rel.h:511
MemoryContext CurrentMemoryContext
Definition: mcxt.c:42
static void fileGetOptions(Oid foreigntableid, char **filename, bool *is_program, List **other_options)
Definition: file_fdw.c:357
static char ** options
static int elevel
Definition: vacuumlazy.c:401
uintptr_t Datum
Definition: postgres.h:411
#define ereport(elevel,...)
Definition: elog.h:157
CopyFromState BeginCopyFrom(ParseState *pstate, Relation rel, Node *whereClause, const char *filename, bool is_program, copy_data_source_cb data_source_cb, List *attnamelist, List *options)
Definition: copyfrom.c:1186
#define Assert(condition)
Definition: c.h:804
void EndCopyFrom(CopyFromState cstate)
Definition: copyfrom.c:1564
static Datum values[MAXATTR]
Definition: bootstrap.c:156
static char * filename
Definition: pg_dumpall.c:92
void * palloc(Size size)
Definition: mcxt.c:1062
int errmsg(const char *fmt,...)
Definition: elog.c:909
void vacuum_delay_point(void)
Definition: vacuum.c:2149
Definition: pg_list.h:50
#define RelationGetRelid(relation)
Definition: rel.h:477
SamplerRandomState randstate
Definition: sampling.h:50
double reservoir_get_next_S(ReservoirState rs, double t, int n)
Definition: sampling.c:146

◆ file_fdw_handler()

Datum file_fdw_handler ( PG_FUNCTION_ARGS  )

Definition at line 171 of file file_fdw.c.

References FdwRoutine::AnalyzeForeignTable, FdwRoutine::BeginForeignScan, FdwRoutine::EndForeignScan, FdwRoutine::ExplainForeignScan, fileAnalyzeForeignTable(), fileBeginForeignScan(), fileEndForeignScan(), fileExplainForeignScan(), fileGetForeignPaths(), fileGetForeignPlan(), fileGetForeignRelSize(), fileIsForeignScanParallelSafe(), fileIterateForeignScan(), fileReScanForeignScan(), FdwRoutine::GetForeignPaths, FdwRoutine::GetForeignPlan, FdwRoutine::GetForeignRelSize, FdwRoutine::IsForeignScanParallelSafe, FdwRoutine::IterateForeignScan, makeNode, PG_RETURN_POINTER, and FdwRoutine::ReScanForeignScan.

172 {
173  FdwRoutine *fdwroutine = makeNode(FdwRoutine);
174 
176  fdwroutine->GetForeignPaths = fileGetForeignPaths;
177  fdwroutine->GetForeignPlan = fileGetForeignPlan;
182  fdwroutine->EndForeignScan = fileEndForeignScan;
185 
186  PG_RETURN_POINTER(fdwroutine);
187 }
GetForeignPlan_function GetForeignPlan
Definition: fdwapi.h:211
BeginForeignScan_function BeginForeignScan
Definition: fdwapi.h:212
#define PG_RETURN_POINTER(x)
Definition: fmgr.h:361
static bool fileIsForeignScanParallelSafe(PlannerInfo *root, RelOptInfo *rel, RangeTblEntry *rte)
Definition: file_fdw.c:831
static void fileEndForeignScan(ForeignScanState *node)
Definition: file_fdw.c:766
ExplainForeignScan_function ExplainForeignScan
Definition: fdwapi.h:252
AnalyzeForeignTable_function AnalyzeForeignTable
Definition: fdwapi.h:257
IsForeignScanParallelSafe_function IsForeignScanParallelSafe
Definition: fdwapi.h:266
static TupleTableSlot * fileIterateForeignScan(ForeignScanState *node)
Definition: file_fdw.c:706
GetForeignRelSize_function GetForeignRelSize
Definition: fdwapi.h:209
EndForeignScan_function EndForeignScan
Definition: fdwapi.h:215
static bool fileAnalyzeForeignTable(Relation relation, AcquireSampleRowsFunc *func, BlockNumber *totalpages)
Definition: file_fdw.c:780
static void fileBeginForeignScan(ForeignScanState *node, int eflags)
Definition: file_fdw.c:652
GetForeignPaths_function GetForeignPaths
Definition: fdwapi.h:210
static void fileGetForeignPaths(PlannerInfo *root, RelOptInfo *baserel, Oid foreigntableid)
Definition: file_fdw.c:532
#define makeNode(_type_)
Definition: nodes.h:585
static void fileGetForeignRelSize(PlannerInfo *root, RelOptInfo *baserel, Oid foreigntableid)
Definition: file_fdw.c:501
ReScanForeignScan_function ReScanForeignScan
Definition: fdwapi.h:214
IterateForeignScan_function IterateForeignScan
Definition: fdwapi.h:213
static void fileExplainForeignScan(ForeignScanState *node, ExplainState *es)
Definition: file_fdw.c:620
static void fileReScanForeignScan(ForeignScanState *node)
Definition: file_fdw.c:745
static ForeignScan * fileGetForeignPlan(PlannerInfo *root, RelOptInfo *baserel, Oid foreigntableid, ForeignPath *best_path, List *tlist, List *scan_clauses, Plan *outer_plan)
Definition: file_fdw.c:585

◆ file_fdw_validator()

Datum file_fdw_validator ( PG_FUNCTION_ARGS  )

Definition at line 196 of file file_fdw.c.

References appendStringInfo(), buf, StringInfoData::data, defGetBoolean(), defGetString(), DefElem::defname, ereport, errcode(), errhint(), errmsg(), ERROR, filename, GetUserId(), initStringInfo(), is_member_of_role(), is_valid_option(), lappend(), StringInfoData::len, lfirst, NIL, FileFdwOption::optcontext, FileFdwOption::optname, PG_GETARG_DATUM, PG_GETARG_OID, PG_RETURN_VOID, ProcessCopyOptions(), and untransformRelOptions().

197 {
198  List *options_list = untransformRelOptions(PG_GETARG_DATUM(0));
199  Oid catalog = PG_GETARG_OID(1);
200  char *filename = NULL;
201  DefElem *force_not_null = NULL;
202  DefElem *force_null = NULL;
203  List *other_options = NIL;
204  ListCell *cell;
205 
206  /*
207  * Check that only options supported by file_fdw, and allowed for the
208  * current object type, are given.
209  */
210  foreach(cell, options_list)
211  {
212  DefElem *def = (DefElem *) lfirst(cell);
213 
214  if (!is_valid_option(def->defname, catalog))
215  {
216  const struct FileFdwOption *opt;
218 
219  /*
220  * Unknown option specified, complain about it. Provide a hint
221  * with list of valid options for the object.
222  */
223  initStringInfo(&buf);
224  for (opt = valid_options; opt->optname; opt++)
225  {
226  if (catalog == opt->optcontext)
227  appendStringInfo(&buf, "%s%s", (buf.len > 0) ? ", " : "",
228  opt->optname);
229  }
230 
231  ereport(ERROR,
232  (errcode(ERRCODE_FDW_INVALID_OPTION_NAME),
233  errmsg("invalid option \"%s\"", def->defname),
234  buf.len > 0
235  ? errhint("Valid options in this context are: %s",
236  buf.data)
237  : errhint("There are no valid options in this context.")));
238  }
239 
240  /*
241  * Separate out filename, program, and column-specific options, since
242  * ProcessCopyOptions won't accept them.
243  */
244  if (strcmp(def->defname, "filename") == 0 ||
245  strcmp(def->defname, "program") == 0)
246  {
247  if (filename)
248  ereport(ERROR,
249  (errcode(ERRCODE_SYNTAX_ERROR),
250  errmsg("conflicting or redundant options")));
251 
252  /*
253  * Check permissions for changing which file or program is used by
254  * the file_fdw.
255  *
256  * Only members of the role 'pg_read_server_files' are allowed to
257  * set the 'filename' option of a file_fdw foreign table, while
258  * only members of the role 'pg_execute_server_program' are
259  * allowed to set the 'program' option. This is because we don't
260  * want regular users to be able to control which file gets read
261  * or which program gets executed.
262  *
263  * Putting this sort of permissions check in a validator is a bit
264  * of a crock, but there doesn't seem to be any other place that
265  * can enforce the check more cleanly.
266  *
267  * Note that the valid_options[] array disallows setting filename
268  * and program at any options level other than foreign table ---
269  * otherwise there'd still be a security hole.
270  */
271  if (strcmp(def->defname, "filename") == 0 &&
272  !is_member_of_role(GetUserId(), ROLE_PG_READ_SERVER_FILES))
273  ereport(ERROR,
274  (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
275  errmsg("only superuser or a member of the pg_read_server_files role may specify the filename option of a file_fdw foreign table")));
276 
277  if (strcmp(def->defname, "program") == 0 &&
278  !is_member_of_role(GetUserId(), ROLE_PG_EXECUTE_SERVER_PROGRAM))
279  ereport(ERROR,
280  (errcode(ERRCODE_INSUFFICIENT_PRIVILEGE),
281  errmsg("only superuser or a member of the pg_execute_server_program role may specify the program option of a file_fdw foreign table")));
282 
283  filename = defGetString(def);
284  }
285 
286  /*
287  * force_not_null is a boolean option; after validation we can discard
288  * it - it will be retrieved later in get_file_fdw_attribute_options()
289  */
290  else if (strcmp(def->defname, "force_not_null") == 0)
291  {
292  if (force_not_null)
293  ereport(ERROR,
294  (errcode(ERRCODE_SYNTAX_ERROR),
295  errmsg("conflicting or redundant options"),
296  errhint("Option \"force_not_null\" supplied more than once for a column.")));
297  force_not_null = def;
298  /* Don't care what the value is, as long as it's a legal boolean */
299  (void) defGetBoolean(def);
300  }
301  /* See comments for force_not_null above */
302  else if (strcmp(def->defname, "force_null") == 0)
303  {
304  if (force_null)
305  ereport(ERROR,
306  (errcode(ERRCODE_SYNTAX_ERROR),
307  errmsg("conflicting or redundant options"),
308  errhint("Option \"force_null\" supplied more than once for a column.")));
309  force_null = def;
310  (void) defGetBoolean(def);
311  }
312  else
313  other_options = lappend(other_options, def);
314  }
315 
316  /*
317  * Now apply the core COPY code's validation logic for more checks.
318  */
319  ProcessCopyOptions(NULL, NULL, true, other_options);
320 
321  /*
322  * Either filename or program option is required for file_fdw foreign
323  * tables.
324  */
325  if (catalog == ForeignTableRelationId && filename == NULL)
326  ereport(ERROR,
327  (errcode(ERRCODE_FDW_DYNAMIC_PARAMETER_VALUE_NEEDED),
328  errmsg("either filename or program is required for file_fdw foreign tables")));
329 
330  PG_RETURN_VOID();
331 }
#define NIL
Definition: pg_list.h:65
Oid optcontext
Definition: file_fdw.c:49
int errhint(const char *fmt,...)
Definition: elog.c:1156
static const struct FileFdwOption valid_options[]
Definition: file_fdw.c:61
Oid GetUserId(void)
Definition: miscinit.c:495
#define PG_GETARG_DATUM(n)
Definition: fmgr.h:268
int errcode(int sqlerrcode)
Definition: elog.c:698
unsigned int Oid
Definition: postgres_ext.h:31
bool defGetBoolean(DefElem *def)
Definition: define.c:106
void appendStringInfo(StringInfo str, const char *fmt,...)
Definition: stringinfo.c:91
#define ERROR
Definition: elog.h:46
char * defGetString(DefElem *def)
Definition: define.c:49
static char * buf
Definition: pg_test_fsync.c:68
#define PG_GETARG_OID(n)
Definition: fmgr.h:275
List * lappend(List *list, void *datum)
Definition: list.c:336
void initStringInfo(StringInfo str)
Definition: stringinfo.c:59
const char * optname
Definition: file_fdw.c:48
List * untransformRelOptions(Datum options)
Definition: reloptions.c:1325
#define ereport(elevel,...)
Definition: elog.h:157
#define PG_RETURN_VOID()
Definition: fmgr.h:349
bool is_member_of_role(Oid member, Oid role)
Definition: acl.c:4869
#define lfirst(lc)
Definition: pg_list.h:169
static bool is_valid_option(const char *option, Oid context)
Definition: file_fdw.c:338
static char * filename
Definition: pg_dumpall.c:92
int errmsg(const char *fmt,...)
Definition: elog.c:909
char * defname
Definition: parsenodes.h:758
Definition: pg_list.h:50
void ProcessCopyOptions(ParseState *pstate, CopyFormatOptions *opts_out, bool is_from, List *options)
Definition: copy.c:333

◆ fileAnalyzeForeignTable()

static bool fileAnalyzeForeignTable ( Relation  relation,
AcquireSampleRowsFunc func,
BlockNumber totalpages 
)
static

Definition at line 780 of file file_fdw.c.

References ereport, errcode_for_file_access(), errmsg(), ERROR, file_acquire_sample_rows(), fileGetOptions(), filename, options, RelationGetRelid, stat::st_size, and stat.

Referenced by file_fdw_handler().

783 {
784  char *filename;
785  bool is_program;
786  List *options;
787  struct stat stat_buf;
788 
789  /* Fetch options of foreign table */
790  fileGetOptions(RelationGetRelid(relation), &filename, &is_program, &options);
791 
792  /*
793  * If this is a program instead of a file, just return false to skip
794  * analyzing the table. We could run the program and collect stats on
795  * whatever it currently returns, but it seems likely that in such cases
796  * the output would be too volatile for the stats to be useful. Maybe
797  * there should be an option to enable doing this?
798  */
799  if (is_program)
800  return false;
801 
802  /*
803  * Get size of the file. (XXX if we fail here, would it be better to just
804  * return false to skip analyzing the table?)
805  */
806  if (stat(filename, &stat_buf) < 0)
807  ereport(ERROR,
809  errmsg("could not stat file \"%s\": %m",
810  filename)));
811 
812  /*
813  * Convert size to pages. Must return at least 1 so that we can tell
814  * later on that pg_class.relpages is not default.
815  */
816  *totalpages = (stat_buf.st_size + (BLCKSZ - 1)) / BLCKSZ;
817  if (*totalpages < 1)
818  *totalpages = 1;
819 
820  *func = file_acquire_sample_rows;
821 
822  return true;
823 }
static int file_acquire_sample_rows(Relation onerel, int elevel, HeapTuple *rows, int targrows, double *totalrows, double *totaldeadrows)
Definition: file_fdw.c:1095
#define ERROR
Definition: elog.h:46
int errcode_for_file_access(void)
Definition: elog.c:721
static void fileGetOptions(Oid foreigntableid, char **filename, bool *is_program, List **other_options)
Definition: file_fdw.c:357
static char ** options
#define ereport(elevel,...)
Definition: elog.h:157
static char * filename
Definition: pg_dumpall.c:92
int errmsg(const char *fmt,...)
Definition: elog.c:909
Definition: pg_list.h:50
#define RelationGetRelid(relation)
Definition: rel.h:477
#define stat
Definition: win32_port.h:283

◆ fileBeginForeignScan()

static void fileBeginForeignScan ( ForeignScanState node,
int  eflags 
)
static

Definition at line 652 of file file_fdw.c.

References BeginCopyFrom(), EXEC_FLAG_EXPLAIN_ONLY, ForeignScan::fdw_private, ForeignScanState::fdw_state, fileGetOptions(), filename, list_concat(), NIL, options, palloc(), PlanState::plan, ScanState::ps, RelationGetRelid, ForeignScanState::ss, and ScanState::ss_currentRelation.

Referenced by file_fdw_handler().

653 {
654  ForeignScan *plan = (ForeignScan *) node->ss.ps.plan;
655  char *filename;
656  bool is_program;
657  List *options;
658  CopyFromState cstate;
659  FileFdwExecutionState *festate;
660 
661  /*
662  * Do nothing in EXPLAIN (no ANALYZE) case. node->fdw_state stays NULL.
663  */
664  if (eflags & EXEC_FLAG_EXPLAIN_ONLY)
665  return;
666 
667  /* Fetch options of foreign table */
669  &filename, &is_program, &options);
670 
671  /* Add any options from the plan (currently only convert_selectively) */
673 
674  /*
675  * Create CopyState from FDW options. We always acquire all columns, so
676  * as to match the expected ScanTupleSlot signature.
677  */
678  cstate = BeginCopyFrom(NULL,
679  node->ss.ss_currentRelation,
680  NULL,
681  filename,
682  is_program,
683  NULL,
684  NIL,
685  options);
686 
687  /*
688  * Save state in node->fdw_state. We must save enough information to call
689  * BeginCopyFrom() again.
690  */
691  festate = (FileFdwExecutionState *) palloc(sizeof(FileFdwExecutionState));
692  festate->filename = filename;
693  festate->is_program = is_program;
694  festate->options = options;
695  festate->cstate = cstate;
696 
697  node->fdw_state = (void *) festate;
698 }
#define NIL
Definition: pg_list.h:65
ScanState ss
Definition: execnodes.h:1858
List * fdw_private
Definition: plannodes.h:641
List * list_concat(List *list1, const List *list2)
Definition: list.c:530
Relation ss_currentRelation
Definition: execnodes.h:1379
PlanState ps
Definition: execnodes.h:1378
static void fileGetOptions(Oid foreigntableid, char **filename, bool *is_program, List **other_options)
Definition: file_fdw.c:357
static char ** options
Plan * plan
Definition: execnodes.h:967
CopyFromState BeginCopyFrom(ParseState *pstate, Relation rel, Node *whereClause, const char *filename, bool is_program, copy_data_source_cb data_source_cb, List *attnamelist, List *options)
Definition: copyfrom.c:1186
static char * filename
Definition: pg_dumpall.c:92
void * palloc(Size size)
Definition: mcxt.c:1062
Definition: pg_list.h:50
#define EXEC_FLAG_EXPLAIN_ONLY
Definition: executor.h:56
#define RelationGetRelid(relation)
Definition: rel.h:477

◆ fileEndForeignScan()

static void fileEndForeignScan ( ForeignScanState node)
static

Definition at line 766 of file file_fdw.c.

References FileFdwExecutionState::cstate, EndCopyFrom(), and ForeignScanState::fdw_state.

Referenced by file_fdw_handler().

767 {
769 
770  /* if festate is NULL, we are in EXPLAIN; nothing to do */
771  if (festate)
772  EndCopyFrom(festate->cstate);
773 }
void EndCopyFrom(CopyFromState cstate)
Definition: copyfrom.c:1564
CopyFromState cstate
Definition: file_fdw.c:108

◆ fileExplainForeignScan()

static void fileExplainForeignScan ( ForeignScanState node,
ExplainState es 
)
static

Definition at line 620 of file file_fdw.c.

References ExplainState::costs, ExplainPropertyInteger(), ExplainPropertyText(), fileGetOptions(), filename, options, RelationGetRelid, ForeignScanState::ss, ScanState::ss_currentRelation, stat::st_size, and stat.

Referenced by file_fdw_handler().

621 {
622  char *filename;
623  bool is_program;
624  List *options;
625 
626  /* Fetch options --- we only need filename and is_program at this point */
628  &filename, &is_program, &options);
629 
630  if (is_program)
631  ExplainPropertyText("Foreign Program", filename, es);
632  else
633  ExplainPropertyText("Foreign File", filename, es);
634 
635  /* Suppress file size if we're not showing cost details */
636  if (es->costs)
637  {
638  struct stat stat_buf;
639 
640  if (!is_program &&
641  stat(filename, &stat_buf) == 0)
642  ExplainPropertyInteger("Foreign File Size", "b",
643  (int64) stat_buf.st_size, es);
644  }
645 }
ScanState ss
Definition: execnodes.h:1858
void ExplainPropertyInteger(const char *qlabel, const char *unit, int64 value, ExplainState *es)
Definition: explain.c:4433
Relation ss_currentRelation
Definition: execnodes.h:1379
bool costs
Definition: explain.h:43
void ExplainPropertyText(const char *qlabel, const char *value, ExplainState *es)
Definition: explain.c:4424
static void fileGetOptions(Oid foreigntableid, char **filename, bool *is_program, List **other_options)
Definition: file_fdw.c:357
static char ** options
static char * filename
Definition: pg_dumpall.c:92
Definition: pg_list.h:50
#define RelationGetRelid(relation)
Definition: rel.h:477
#define stat
Definition: win32_port.h:283

◆ fileGetForeignPaths()

static void fileGetForeignPaths ( PlannerInfo root,
RelOptInfo baserel,
Oid  foreigntableid 
)
static

Definition at line 532 of file file_fdw.c.

References add_path(), check_selective_binary_conversion(), create_foreignscan_path(), estimate_costs(), RelOptInfo::fdw_private, RelOptInfo::lateral_relids, list_make1, makeDefElem(), NIL, and RelOptInfo::rows.

Referenced by file_fdw_handler().

535 {
536  FileFdwPlanState *fdw_private = (FileFdwPlanState *) baserel->fdw_private;
537  Cost startup_cost;
538  Cost total_cost;
539  List *columns;
540  List *coptions = NIL;
541 
542  /* Decide whether to selectively perform binary conversion */
544  foreigntableid,
545  &columns))
546  coptions = list_make1(makeDefElem("convert_selectively",
547  (Node *) columns, -1));
548 
549  /* Estimate costs */
550  estimate_costs(root, baserel, fdw_private,
551  &startup_cost, &total_cost);
552 
553  /*
554  * Create a ForeignPath node and add it as only possible path. We use the
555  * fdw_private list of the path to carry the convert_selectively option;
556  * it will be propagated into the fdw_private list of the Plan node.
557  *
558  * We don't support pushing join clauses into the quals of this path, but
559  * it could still have required parameterization due to LATERAL refs in
560  * its tlist.
561  */
562  add_path(baserel, (Path *)
563  create_foreignscan_path(root, baserel,
564  NULL, /* default pathtarget */
565  baserel->rows,
566  startup_cost,
567  total_cost,
568  NIL, /* no pathkeys */
569  baserel->lateral_relids,
570  NULL, /* no extra plan */
571  coptions));
572 
573  /*
574  * If data file was sorted, and we knew it somehow, we could insert
575  * appropriate pathkeys into the ForeignPath node to tell the planner
576  * that.
577  */
578 }
#define NIL
Definition: pg_list.h:65
void add_path(RelOptInfo *parent_rel, Path *new_path)
Definition: pathnode.c:422
Definition: nodes.h:537
DefElem * makeDefElem(char *name, Node *arg, int location)
Definition: makefuncs.c:546
#define list_make1(x1)
Definition: pg_list.h:206
Relids lateral_relids
Definition: pathnodes.h:706
ForeignPath * create_foreignscan_path(PlannerInfo *root, RelOptInfo *rel, PathTarget *target, double rows, Cost startup_cost, Cost total_cost, List *pathkeys, Relids required_outer, Path *fdw_outerpath, List *fdw_private)
Definition: pathnode.c:2206
void * fdw_private
Definition: pathnodes.h:737
static bool check_selective_binary_conversion(RelOptInfo *baserel, Oid foreigntableid, List **columns)
Definition: file_fdw.c:847
Cardinality rows
Definition: pathnodes.h:684
static void estimate_costs(PlannerInfo *root, RelOptInfo *baserel, FileFdwPlanState *fdw_private, Cost *startup_cost, Cost *total_cost)
Definition: file_fdw.c:1052
Definition: pg_list.h:50
double Cost
Definition: nodes.h:671

◆ fileGetForeignPlan()

static ForeignScan * fileGetForeignPlan ( PlannerInfo root,
RelOptInfo baserel,
Oid  foreigntableid,
ForeignPath best_path,
List tlist,
List scan_clauses,
Plan outer_plan 
)
static

Definition at line 585 of file file_fdw.c.

References extract_actual_clauses(), ForeignPath::fdw_private, make_foreignscan(), NIL, and RelOptInfo::relid.

Referenced by file_fdw_handler().

592 {
593  Index scan_relid = baserel->relid;
594 
595  /*
596  * We have no native ability to evaluate restriction clauses, so we just
597  * put all the scan_clauses into the plan node's qual list for the
598  * executor to check. So all we have to do here is strip RestrictInfo
599  * nodes from the clauses and ignore pseudoconstants (which will be
600  * handled elsewhere).
601  */
602  scan_clauses = extract_actual_clauses(scan_clauses, false);
603 
604  /* Create the ForeignScan node */
605  return make_foreignscan(tlist,
606  scan_clauses,
607  scan_relid,
608  NIL, /* no expressions to evaluate */
609  best_path->fdw_private,
610  NIL, /* no custom tlist */
611  NIL, /* no remote quals */
612  outer_plan);
613 }
#define NIL
Definition: pg_list.h:65
Index relid
Definition: pathnodes.h:709
unsigned int Index
Definition: c.h:549
List * extract_actual_clauses(List *restrictinfo_list, bool pseudoconstant)
Definition: restrictinfo.c:443
ForeignScan * make_foreignscan(List *qptlist, List *qpqual, Index scanrelid, List *fdw_exprs, List *fdw_private, List *fdw_scan_tlist, List *fdw_recheck_quals, Plan *outer_plan)
Definition: createplan.c:5679
List * fdw_private
Definition: pathnodes.h:1400

◆ fileGetForeignRelSize()

static void fileGetForeignRelSize ( PlannerInfo root,
RelOptInfo baserel,
Oid  foreigntableid 
)
static

Definition at line 501 of file file_fdw.c.

References estimate_size(), RelOptInfo::fdw_private, fileGetOptions(), FileFdwPlanState::filename, FileFdwPlanState::is_program, FileFdwPlanState::options, and palloc().

Referenced by file_fdw_handler().

504 {
505  FileFdwPlanState *fdw_private;
506 
507  /*
508  * Fetch options. We only need filename (or program) at this point, but
509  * we might as well get everything and not need to re-fetch it later in
510  * planning.
511  */
512  fdw_private = (FileFdwPlanState *) palloc(sizeof(FileFdwPlanState));
513  fileGetOptions(foreigntableid,
514  &fdw_private->filename,
515  &fdw_private->is_program,
516  &fdw_private->options);
517  baserel->fdw_private = (void *) fdw_private;
518 
519  /* Estimate relation size */
520  estimate_size(root, baserel, fdw_private);
521 }
List * options
Definition: file_fdw.c:93
static void estimate_size(PlannerInfo *root, RelOptInfo *baserel, FileFdwPlanState *fdw_private)
Definition: file_fdw.c:971
char * filename
Definition: file_fdw.c:91
static void fileGetOptions(Oid foreigntableid, char **filename, bool *is_program, List **other_options)
Definition: file_fdw.c:357
void * fdw_private
Definition: pathnodes.h:737
void * palloc(Size size)
Definition: mcxt.c:1062

◆ fileGetOptions()

static void fileGetOptions ( Oid  foreigntableid,
char **  filename,
bool is_program,
List **  other_options 
)
static

Definition at line 357 of file file_fdw.c.

References defGetString(), DefElem::defname, elog, ERROR, ForeignServer::fdwid, foreach_delete_current, get_file_fdw_attribute_options(), GetForeignDataWrapper(), GetForeignServer(), GetForeignTable(), lfirst, list_concat(), NIL, ForeignDataWrapper::options, ForeignServer::options, options, ForeignTable::options, and ForeignTable::serverid.

Referenced by file_acquire_sample_rows(), fileAnalyzeForeignTable(), fileBeginForeignScan(), fileExplainForeignScan(), and fileGetForeignRelSize().

359 {
360  ForeignTable *table;
361  ForeignServer *server;
362  ForeignDataWrapper *wrapper;
363  List *options;
364  ListCell *lc;
365 
366  /*
367  * Extract options from FDW objects. We ignore user mappings because
368  * file_fdw doesn't have any options that can be specified there.
369  *
370  * (XXX Actually, given the current contents of valid_options[], there's
371  * no point in examining anything except the foreign table's own options.
372  * Simplify?)
373  */
374  table = GetForeignTable(foreigntableid);
375  server = GetForeignServer(table->serverid);
376  wrapper = GetForeignDataWrapper(server->fdwid);
377 
378  options = NIL;
379  options = list_concat(options, wrapper->options);
380  options = list_concat(options, server->options);
381  options = list_concat(options, table->options);
382  options = list_concat(options, get_file_fdw_attribute_options(foreigntableid));
383 
384  /*
385  * Separate out the filename or program option (we assume there is only
386  * one).
387  */
388  *filename = NULL;
389  *is_program = false;
390  foreach(lc, options)
391  {
392  DefElem *def = (DefElem *) lfirst(lc);
393 
394  if (strcmp(def->defname, "filename") == 0)
395  {
396  *filename = defGetString(def);
397  options = foreach_delete_current(options, lc);
398  break;
399  }
400  else if (strcmp(def->defname, "program") == 0)
401  {
402  *filename = defGetString(def);
403  *is_program = true;
404  options = foreach_delete_current(options, lc);
405  break;
406  }
407  }
408 
409  /*
410  * The validator should have checked that filename or program was included
411  * in the options, but check again, just in case.
412  */
413  if (*filename == NULL)
414  elog(ERROR, "either filename or program is required for file_fdw foreign tables");
415 
416  *other_options = options;
417 }
#define NIL
Definition: pg_list.h:65
static List * get_file_fdw_attribute_options(Oid relid)
Definition: file_fdw.c:428
List * list_concat(List *list1, const List *list2)
Definition: list.c:530
ForeignTable * GetForeignTable(Oid relid)
Definition: foreign.c:248
#define foreach_delete_current(lst, cell)
Definition: pg_list.h:369
ForeignDataWrapper * GetForeignDataWrapper(Oid fdwid)
Definition: foreign.c:35
#define ERROR
Definition: elog.h:46
char * defGetString(DefElem *def)
Definition: define.c:49
static char ** options
List * options
Definition: foreign.h:31
ForeignServer * GetForeignServer(Oid serverid)
Definition: foreign.c:109
#define lfirst(lc)
Definition: pg_list.h:169
Oid serverid
Definition: foreign.h:56
List * options
Definition: foreign.h:57
static char * filename
Definition: pg_dumpall.c:92
#define elog(elevel,...)
Definition: elog.h:232
char * defname
Definition: parsenodes.h:758
Definition: pg_list.h:50
List * options
Definition: foreign.h:42

◆ fileIsForeignScanParallelSafe()

static bool fileIsForeignScanParallelSafe ( PlannerInfo root,
RelOptInfo rel,
RangeTblEntry rte 
)
static

Definition at line 831 of file file_fdw.c.

Referenced by file_fdw_handler().

833 {
834  return true;
835 }

◆ fileIterateForeignScan()

static TupleTableSlot * fileIterateForeignScan ( ForeignScanState node)
static

Definition at line 706 of file file_fdw.c.

References ErrorContextCallback::callback, CopyFromErrorCallback(), FileFdwExecutionState::cstate, error_context_stack, ExecClearTuple(), ExecStoreVirtualTuple(), ForeignScanState::fdw_state, NextCopyFrom(), ErrorContextCallback::previous, ForeignScanState::ss, and ScanState::ss_ScanTupleSlot.

Referenced by file_fdw_handler().

707 {
709  TupleTableSlot *slot = node->ss.ss_ScanTupleSlot;
710  bool found;
711  ErrorContextCallback errcallback;
712 
713  /* Set up callback to identify error line number. */
714  errcallback.callback = CopyFromErrorCallback;
715  errcallback.arg = (void *) festate->cstate;
716  errcallback.previous = error_context_stack;
717  error_context_stack = &errcallback;
718 
719  /*
720  * The protocol for loading a virtual tuple into a slot is first
721  * ExecClearTuple, then fill the values/isnull arrays, then
722  * ExecStoreVirtualTuple. If we don't find another row in the file, we
723  * just skip the last step, leaving the slot empty as required.
724  *
725  * We can pass ExprContext = NULL because we read all columns from the
726  * file, so no need to evaluate default expressions.
727  */
728  ExecClearTuple(slot);
729  found = NextCopyFrom(festate->cstate, NULL,
730  slot->tts_values, slot->tts_isnull);
731  if (found)
732  ExecStoreVirtualTuple(slot);
733 
734  /* Remove error callback. */
735  error_context_stack = errcallback.previous;
736 
737  return slot;
738 }
ScanState ss
Definition: execnodes.h:1858
bool NextCopyFrom(CopyFromState cstate, ExprContext *econtext, Datum *values, bool *nulls)
static TupleTableSlot * ExecClearTuple(TupleTableSlot *slot)
Definition: tuptable.h:425
TupleTableSlot * ss_ScanTupleSlot
Definition: execnodes.h:1381
void(* callback)(void *arg)
Definition: elog.h:247
struct ErrorContextCallback * previous
Definition: elog.h:246
ErrorContextCallback * error_context_stack
Definition: elog.c:93
void CopyFromErrorCallback(void *arg)
Definition: copyfrom.c:115
CopyFromState cstate
Definition: file_fdw.c:108
TupleTableSlot * ExecStoreVirtualTuple(TupleTableSlot *slot)
Definition: execTuples.c:1552

◆ fileReScanForeignScan()

static void fileReScanForeignScan ( ForeignScanState node)
static

Definition at line 745 of file file_fdw.c.

References BeginCopyFrom(), FileFdwExecutionState::cstate, EndCopyFrom(), ForeignScanState::fdw_state, FileFdwExecutionState::filename, FileFdwExecutionState::is_program, NIL, FileFdwExecutionState::options, ForeignScanState::ss, and ScanState::ss_currentRelation.

Referenced by file_fdw_handler().

746 {
748 
749  EndCopyFrom(festate->cstate);
750 
751  festate->cstate = BeginCopyFrom(NULL,
752  node->ss.ss_currentRelation,
753  NULL,
754  festate->filename,
755  festate->is_program,
756  NULL,
757  NIL,
758  festate->options);
759 }
#define NIL
Definition: pg_list.h:65
ScanState ss
Definition: execnodes.h:1858
Relation ss_currentRelation
Definition: execnodes.h:1379
CopyFromState BeginCopyFrom(ParseState *pstate, Relation rel, Node *whereClause, const char *filename, bool is_program, copy_data_source_cb data_source_cb, List *attnamelist, List *options)
Definition: copyfrom.c:1186
void EndCopyFrom(CopyFromState cstate)
Definition: copyfrom.c:1564
CopyFromState cstate
Definition: file_fdw.c:108

◆ get_file_fdw_attribute_options()

static List * get_file_fdw_attribute_options ( Oid  relid)
static

Definition at line 428 of file file_fdw.c.

References AccessShareLock, attname, attnum, defGetBoolean(), DefElem::defname, GetForeignColumnOptions(), lappend(), lfirst, makeDefElem(), makeString(), NameStr, TupleDescData::natts, NIL, options, pstrdup(), RelationGetDescr, table_close(), table_open(), and TupleDescAttr.

Referenced by fileGetOptions().

429 {
430  Relation rel;
431  TupleDesc tupleDesc;
432  AttrNumber natts;
434  List *fnncolumns = NIL;
435  List *fncolumns = NIL;
436 
437  List *options = NIL;
438 
439  rel = table_open(relid, AccessShareLock);
440  tupleDesc = RelationGetDescr(rel);
441  natts = tupleDesc->natts;
442 
443  /* Retrieve FDW options for all user-defined attributes. */
444  for (attnum = 1; attnum <= natts; attnum++)
445  {
446  Form_pg_attribute attr = TupleDescAttr(tupleDesc, attnum - 1);
447  List *options;
448  ListCell *lc;
449 
450  /* Skip dropped attributes. */
451  if (attr->attisdropped)
452  continue;
453 
454  options = GetForeignColumnOptions(relid, attnum);
455  foreach(lc, options)
456  {
457  DefElem *def = (DefElem *) lfirst(lc);
458 
459  if (strcmp(def->defname, "force_not_null") == 0)
460  {
461  if (defGetBoolean(def))
462  {
463  char *attname = pstrdup(NameStr(attr->attname));
464 
465  fnncolumns = lappend(fnncolumns, makeString(attname));
466  }
467  }
468  else if (strcmp(def->defname, "force_null") == 0)
469  {
470  if (defGetBoolean(def))
471  {
472  char *attname = pstrdup(NameStr(attr->attname));
473 
474  fncolumns = lappend(fncolumns, makeString(attname));
475  }
476  }
477  /* maybe in future handle other options here */
478  }
479  }
480 
482 
483  /*
484  * Return DefElem only when some column(s) have force_not_null /
485  * force_null options set
486  */
487  if (fnncolumns != NIL)
488  options = lappend(options, makeDefElem("force_not_null", (Node *) fnncolumns, -1));
489 
490  if (fncolumns != NIL)
491  options = lappend(options, makeDefElem("force_null", (Node *) fncolumns, -1));
492 
493  return options;
494 }
#define NIL
Definition: pg_list.h:65
void table_close(Relation relation, LOCKMODE lockmode)
Definition: table.c:167
#define RelationGetDescr(relation)
Definition: rel.h:503
#define TupleDescAttr(tupdesc, i)
Definition: tupdesc.h:92
char * pstrdup(const char *in)
Definition: mcxt.c:1299
#define AccessShareLock
Definition: lockdefs.h:36
Definition: nodes.h:537
String * makeString(char *str)
Definition: value.c:51
DefElem * makeDefElem(char *name, Node *arg, int location)
Definition: makefuncs.c:546
bool defGetBoolean(DefElem *def)
Definition: define.c:106
List * GetForeignColumnOptions(Oid relid, AttrNumber attnum)
Definition: foreign.c:286
NameData attname
Definition: pg_attribute.h:41
FormData_pg_attribute * Form_pg_attribute
Definition: pg_attribute.h:207
List * lappend(List *list, void *datum)
Definition: list.c:336
static char ** options
int16 attnum
Definition: pg_attribute.h:83
#define lfirst(lc)
Definition: pg_list.h:169
#define NameStr(name)
Definition: c.h:681
char * defname
Definition: parsenodes.h:758
Relation table_open(Oid relationId, LOCKMODE lockmode)
Definition: table.c:39
Definition: pg_list.h:50
int16 AttrNumber
Definition: attnum.h:21

◆ is_valid_option()

static bool is_valid_option ( const char *  option,
Oid  context 
)
static

Definition at line 338 of file file_fdw.c.

References FileFdwOption::optcontext, and FileFdwOption::optname.

Referenced by file_fdw_validator().

339 {
340  const struct FileFdwOption *opt;
341 
342  for (opt = valid_options; opt->optname; opt++)
343  {
344  if (context == opt->optcontext && strcmp(opt->optname, option) == 0)
345  return true;
346  }
347  return false;
348 }
Oid optcontext
Definition: file_fdw.c:49
static const struct FileFdwOption valid_options[]
Definition: file_fdw.c:61
const char * optname
Definition: file_fdw.c:48

◆ PG_FUNCTION_INFO_V1() [1/2]

PG_FUNCTION_INFO_V1 ( file_fdw_handler  )

◆ PG_FUNCTION_INFO_V1() [2/2]

PG_FUNCTION_INFO_V1 ( file_fdw_validator  )

Variable Documentation

◆ PG_MODULE_MAGIC

PG_MODULE_MAGIC

Definition at line 41 of file file_fdw.c.

◆ valid_options

const struct FileFdwOption valid_options[]
static
Initial value:
= {
{"filename", ForeignTableRelationId},
{"program", ForeignTableRelationId},
{"format", ForeignTableRelationId},
{"header", ForeignTableRelationId},
{"delimiter", ForeignTableRelationId},
{"quote", ForeignTableRelationId},
{"escape", ForeignTableRelationId},
{"null", ForeignTableRelationId},
{"encoding", ForeignTableRelationId},
{"force_not_null", AttributeRelationId},
{"force_null", AttributeRelationId},
{NULL, InvalidOid}
}
#define InvalidOid
Definition: postgres_ext.h:36

Definition at line 61 of file file_fdw.c.