PostgreSQL Source Code  git master
copy.h
Go to the documentation of this file.
1 /*-------------------------------------------------------------------------
2  *
3  * copy.h
4  * Definitions for using the POSTGRES copy command.
5  *
6  *
7  * Portions Copyright (c) 1996-2023, PostgreSQL Global Development Group
8  * Portions Copyright (c) 1994, Regents of the University of California
9  *
10  * src/include/commands/copy.h
11  *
12  *-------------------------------------------------------------------------
13  */
14 #ifndef COPY_H
15 #define COPY_H
16 
17 #include "nodes/execnodes.h"
18 #include "nodes/parsenodes.h"
19 #include "parser/parse_node.h"
20 #include "tcop/dest.h"
21 
22 /*
23  * Represents whether a header line should be present, and whether it must
24  * match the actual names (which implies "true").
25  */
26 typedef enum CopyHeaderChoice
27 {
32 
33 /*
34  * A struct to hold COPY options, in a parsed form. All of these are related
35  * to formatting, except for 'freeze', which doesn't really belong here, but
36  * it's expedient to parse it along with all the other options.
37  */
38 typedef struct CopyFormatOptions
39 {
40  /* parameters from the COPY command */
41  int file_encoding; /* file or remote side's character encoding,
42  * -1 if not specified */
43  bool binary; /* binary format? */
44  bool freeze; /* freeze rows on loading? */
45  bool csv_mode; /* Comma Separated Value format? */
46  CopyHeaderChoice header_line; /* header line? */
47  char *null_print; /* NULL marker string (server encoding!) */
48  int null_print_len; /* length of same */
49  char *null_print_client; /* same converted to file encoding */
50  char *default_print; /* DEFAULT marker string */
51  int default_print_len; /* length of same */
52  char *delim; /* column delimiter (must be 1 byte) */
53  char *quote; /* CSV quote char (must be 1 byte) */
54  char *escape; /* CSV escape char (must be 1 byte) */
55  List *force_quote; /* list of column names */
56  bool force_quote_all; /* FORCE_QUOTE *? */
57  bool *force_quote_flags; /* per-column CSV FQ flags */
58  List *force_notnull; /* list of column names */
59  bool *force_notnull_flags; /* per-column CSV FNN flags */
60  List *force_null; /* list of column names */
61  bool *force_null_flags; /* per-column CSV FN flags */
62  bool convert_selectively; /* do selective binary conversion? */
63  List *convert_select; /* list of column names (can be NIL) */
65 
66 /* These are private in commands/copy[from|to].c */
68 typedef struct CopyToStateData *CopyToState;
69 
70 typedef int (*copy_data_source_cb) (void *outbuf, int minread, int maxread);
71 typedef void (*copy_data_dest_cb) (void *data, int len);
72 
73 extern void DoCopy(ParseState *pstate, const CopyStmt *stmt,
74  int stmt_location, int stmt_len,
75  uint64 *processed);
76 
77 extern void ProcessCopyOptions(ParseState *pstate, CopyFormatOptions *opts_out, bool is_from, List *options);
79  const char *filename,
80  bool is_program, copy_data_source_cb data_source_cb, List *attnamelist, List *options);
81 extern void EndCopyFrom(CopyFromState cstate);
82 extern bool NextCopyFrom(CopyFromState cstate, ExprContext *econtext,
83  Datum *values, bool *nulls);
84 extern bool NextCopyFromRawFields(CopyFromState cstate,
85  char ***fields, int *nfields);
86 extern void CopyFromErrorCallback(void *arg);
87 
88 extern uint64 CopyFrom(CopyFromState cstate);
89 
91 
92 /*
93  * internal prototypes
94  */
95 extern CopyToState BeginCopyTo(ParseState *pstate, Relation rel, RawStmt *raw_query,
96  Oid queryRelId, const char *filename, bool is_program,
98 extern void EndCopyTo(CopyToState cstate);
99 extern uint64 DoCopyTo(CopyToState cstate);
100 extern List *CopyGetAttnums(TupleDesc tupDesc, Relation rel,
101  List *attnamelist);
102 
103 #endif /* COPY_H */
static Datum values[MAXATTR]
Definition: bootstrap.c:156
struct CopyFormatOptions CopyFormatOptions
List * CopyGetAttnums(TupleDesc tupDesc, Relation rel, List *attnamelist)
Definition: copy.c:782
uint64 DoCopyTo(CopyToState cstate)
Definition: copyto.c:749
void DoCopy(ParseState *pstate, const CopyStmt *stmt, int stmt_location, int stmt_len, uint64 *processed)
Definition: copy.c:64
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:1334
struct CopyFromStateData * CopyFromState
Definition: copy.h:67
CopyToState BeginCopyTo(ParseState *pstate, Relation rel, RawStmt *raw_query, Oid queryRelId, const char *filename, bool is_program, copy_data_dest_cb data_dest_cb, List *attnamelist, List *options)
Definition: copyto.c:357
void(* copy_data_dest_cb)(void *data, int len)
Definition: copy.h:71
uint64 CopyFrom(CopyFromState cstate)
Definition: copyfrom.c:632
int(* copy_data_source_cb)(void *outbuf, int minread, int maxread)
Definition: copy.h:70
void ProcessCopyOptions(ParseState *pstate, CopyFormatOptions *opts_out, bool is_from, List *options)
Definition: copy.c:414
void EndCopyFrom(CopyFromState cstate)
Definition: copyfrom.c:1717
struct CopyToStateData * CopyToState
Definition: copy.h:68
CopyHeaderChoice
Definition: copy.h:27
@ COPY_HEADER_TRUE
Definition: copy.h:29
@ COPY_HEADER_FALSE
Definition: copy.h:28
@ COPY_HEADER_MATCH
Definition: copy.h:30
void EndCopyTo(CopyToState cstate)
Definition: copyto.c:728
DestReceiver * CreateCopyDestReceiver(void)
Definition: copyto.c:1275
void CopyFromErrorCallback(void *arg)
Definition: copyfrom.c:116
bool NextCopyFrom(CopyFromState cstate, ExprContext *econtext, Datum *values, bool *nulls)
bool NextCopyFromRawFields(CopyFromState cstate, char ***fields, int *nfields)
#define stmt
Definition: indent_codes.h:59
void * arg
const void size_t len
const void * data
static char * filename
Definition: pg_dumpall.c:119
uintptr_t Datum
Definition: postgres.h:64
unsigned int Oid
Definition: postgres_ext.h:31
int default_print_len
Definition: copy.h:51
bool force_quote_all
Definition: copy.h:56
bool freeze
Definition: copy.h:44
bool binary
Definition: copy.h:43
int null_print_len
Definition: copy.h:48
bool convert_selectively
Definition: copy.h:62
char * quote
Definition: copy.h:53
CopyHeaderChoice header_line
Definition: copy.h:46
List * force_quote
Definition: copy.h:55
char * escape
Definition: copy.h:54
char * null_print
Definition: copy.h:47
List * force_null
Definition: copy.h:60
char * delim
Definition: copy.h:52
List * convert_select
Definition: copy.h:63
bool * force_quote_flags
Definition: copy.h:57
bool * force_notnull_flags
Definition: copy.h:59
char * null_print_client
Definition: copy.h:49
bool csv_mode
Definition: copy.h:45
int file_encoding
Definition: copy.h:41
bool * force_null_flags
Definition: copy.h:61
char * default_print
Definition: copy.h:50
List * force_notnull
Definition: copy.h:58
Node * whereClause
Definition: copyto.c:92
Relation rel
Definition: copyto.c:84
copy_data_dest_cb data_dest_cb
Definition: copyto.c:89
bool is_program
Definition: copyto.c:88
Definition: pg_list.h:54
Definition: nodes.h:129