PostgreSQL Source Code git master
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Pages
pgbench.h File Reference
Include dependency graph for pgbench.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Data Structures

struct  PgBenchValue
 
struct  PgBenchExpr
 
struct  PgBenchExprLink
 
struct  PgBenchExprList
 

Macros

#define yyscan_t   void *
 

Typedefs

typedef enum PgBenchExprType PgBenchExprType
 
typedef enum PgBenchFunction PgBenchFunction
 
typedef struct PgBenchExpr PgBenchExpr
 
typedef struct PgBenchExprLink PgBenchExprLink
 
typedef struct PgBenchExprList PgBenchExprList
 

Enumerations

enum  PgBenchValueType {
  PGBT_NO_VALUE = 0 , PGBT_NULL , PGBT_INT , PGBT_DOUBLE ,
  PGBT_BOOLEAN
}
 
enum  PgBenchExprType { ENODE_CONSTANT , ENODE_VARIABLE , ENODE_FUNCTION }
 
enum  PgBenchFunction {
  PGBENCH_ADD , PGBENCH_SUB , PGBENCH_MUL , PGBENCH_DIV ,
  PGBENCH_MOD , PGBENCH_DEBUG , PGBENCH_ABS , PGBENCH_LEAST ,
  PGBENCH_GREATEST , PGBENCH_INT , PGBENCH_DOUBLE , PGBENCH_PI ,
  PGBENCH_SQRT , PGBENCH_LN , PGBENCH_EXP , PGBENCH_RANDOM ,
  PGBENCH_RANDOM_GAUSSIAN , PGBENCH_RANDOM_EXPONENTIAL , PGBENCH_RANDOM_ZIPFIAN , PGBENCH_POW ,
  PGBENCH_AND , PGBENCH_OR , PGBENCH_NOT , PGBENCH_BITAND ,
  PGBENCH_BITOR , PGBENCH_BITXOR , PGBENCH_LSHIFT , PGBENCH_RSHIFT ,
  PGBENCH_EQ , PGBENCH_NE , PGBENCH_LE , PGBENCH_LT ,
  PGBENCH_IS , PGBENCH_CASE , PGBENCH_HASH_FNV1A , PGBENCH_HASH_MURMUR2 ,
  PGBENCH_PERMUTE
}
 

Functions

int expr_yyparse (yyscan_t yyscanner)
 
int expr_yylex (union YYSTYPE *yylval_param, yyscan_t yyscanner)
 
void expr_yyerror (yyscan_t yyscanner, const char *message) pg_attribute_noreturn()
 
void expr_yyerror_more (yyscan_t yyscanner, const char *message, const char *more) pg_attribute_noreturn()
 
bool expr_lex_one_word (PsqlScanState state, PQExpBuffer word_buf, int *offset)
 
yyscan_t expr_scanner_init (PsqlScanState state, const char *source, int lineno, int start_offset, const char *command)
 
void expr_scanner_finish (yyscan_t yyscanner)
 
int expr_scanner_offset (PsqlScanState state)
 
char * expr_scanner_get_substring (PsqlScanState state, int start_offset, int end_offset, bool chomp)
 
int expr_scanner_get_lineno (PsqlScanState state, int offset)
 
void syntax_error (const char *source, int lineno, const char *line, const char *command, const char *msg, const char *more, int column) pg_attribute_noreturn()
 
bool strtoint64 (const char *str, bool errorOK, int64 *result)
 
bool strtodouble (const char *str, bool errorOK, double *dv)
 

Variables

PgBenchExprexpr_parse_result
 

Macro Definition Documentation

◆ yyscan_t

#define yyscan_t   void *

Definition at line 23 of file pgbench.h.

Typedef Documentation

◆ PgBenchExpr

typedef struct PgBenchExpr PgBenchExpr

Definition at line 106 of file pgbench.h.

◆ PgBenchExprLink

Definition at line 107 of file pgbench.h.

◆ PgBenchExprList

Definition at line 108 of file pgbench.h.

◆ PgBenchExprType

◆ PgBenchFunction

Enumeration Type Documentation

◆ PgBenchExprType

Enumerator
ENODE_CONSTANT 
ENODE_VARIABLE 
ENODE_FUNCTION 

Definition at line 57 of file pgbench.h.

58{
PgBenchExprType
Definition: pgbench.h:58
@ ENODE_VARIABLE
Definition: pgbench.h:60
@ ENODE_CONSTANT
Definition: pgbench.h:59
@ ENODE_FUNCTION
Definition: pgbench.h:61

◆ PgBenchFunction

Enumerator
PGBENCH_ADD 
PGBENCH_SUB 
PGBENCH_MUL 
PGBENCH_DIV 
PGBENCH_MOD 
PGBENCH_DEBUG 
PGBENCH_ABS 
PGBENCH_LEAST 
PGBENCH_GREATEST 
PGBENCH_INT 
PGBENCH_DOUBLE 
PGBENCH_PI 
PGBENCH_SQRT 
PGBENCH_LN 
PGBENCH_EXP 
PGBENCH_RANDOM 
PGBENCH_RANDOM_GAUSSIAN 
PGBENCH_RANDOM_EXPONENTIAL 
PGBENCH_RANDOM_ZIPFIAN 
PGBENCH_POW 
PGBENCH_AND 
PGBENCH_OR 
PGBENCH_NOT 
PGBENCH_BITAND 
PGBENCH_BITOR 
PGBENCH_BITXOR 
PGBENCH_LSHIFT 
PGBENCH_RSHIFT 
PGBENCH_EQ 
PGBENCH_NE 
PGBENCH_LE 
PGBENCH_LT 
PGBENCH_IS 
PGBENCH_CASE 
PGBENCH_HASH_FNV1A 
PGBENCH_HASH_MURMUR2 
PGBENCH_PERMUTE 

Definition at line 65 of file pgbench.h.

66{
PgBenchFunction
Definition: pgbench.h:66
@ PGBENCH_DIV
Definition: pgbench.h:70
@ PGBENCH_AND
Definition: pgbench.h:87
@ PGBENCH_DOUBLE
Definition: pgbench.h:77
@ PGBENCH_LT
Definition: pgbench.h:98
@ PGBENCH_LN
Definition: pgbench.h:80
@ PGBENCH_RANDOM_EXPONENTIAL
Definition: pgbench.h:84
@ PGBENCH_RSHIFT
Definition: pgbench.h:94
@ PGBENCH_DEBUG
Definition: pgbench.h:72
@ PGBENCH_MOD
Definition: pgbench.h:71
@ PGBENCH_GREATEST
Definition: pgbench.h:75
@ PGBENCH_BITXOR
Definition: pgbench.h:92
@ PGBENCH_RANDOM_ZIPFIAN
Definition: pgbench.h:85
@ PGBENCH_INT
Definition: pgbench.h:76
@ PGBENCH_NE
Definition: pgbench.h:96
@ PGBENCH_OR
Definition: pgbench.h:88
@ PGBENCH_LE
Definition: pgbench.h:97
@ PGBENCH_EXP
Definition: pgbench.h:81
@ PGBENCH_PI
Definition: pgbench.h:78
@ PGBENCH_ADD
Definition: pgbench.h:67
@ PGBENCH_EQ
Definition: pgbench.h:95
@ PGBENCH_LSHIFT
Definition: pgbench.h:93
@ PGBENCH_RANDOM
Definition: pgbench.h:82
@ PGBENCH_POW
Definition: pgbench.h:86
@ PGBENCH_IS
Definition: pgbench.h:99
@ PGBENCH_SUB
Definition: pgbench.h:68
@ PGBENCH_HASH_MURMUR2
Definition: pgbench.h:102
@ PGBENCH_ABS
Definition: pgbench.h:73
@ PGBENCH_BITOR
Definition: pgbench.h:91
@ PGBENCH_SQRT
Definition: pgbench.h:79
@ PGBENCH_LEAST
Definition: pgbench.h:74
@ PGBENCH_PERMUTE
Definition: pgbench.h:103
@ PGBENCH_HASH_FNV1A
Definition: pgbench.h:101
@ PGBENCH_NOT
Definition: pgbench.h:89
@ PGBENCH_BITAND
Definition: pgbench.h:90
@ PGBENCH_RANDOM_GAUSSIAN
Definition: pgbench.h:83
@ PGBENCH_MUL
Definition: pgbench.h:69
@ PGBENCH_CASE
Definition: pgbench.h:100

◆ PgBenchValueType

Enumerator
PGBT_NO_VALUE 
PGBT_NULL 
PGBT_INT 
PGBT_DOUBLE 
PGBT_BOOLEAN 

Definition at line 34 of file pgbench.h.

35{
36 PGBT_NO_VALUE = 0,
41 /* add other types here */
PgBenchValueType
Definition: pgbench.h:35
@ PGBT_NO_VALUE
Definition: pgbench.h:36
@ PGBT_INT
Definition: pgbench.h:38
@ PGBT_NULL
Definition: pgbench.h:37
@ PGBT_DOUBLE
Definition: pgbench.h:39
@ PGBT_BOOLEAN
Definition: pgbench.h:40

Function Documentation

◆ expr_lex_one_word()

bool expr_lex_one_word ( PsqlScanState  state,
PQExpBuffer  word_buf,
int *  offset 
)

Definition at line 320 of file exprscan.l.

321{
322 int lexresult;
323 YYSTYPE lval;
324
325 /* Must be scanning already */
326 Assert(state->scanbufhandle != NULL);
327
328 /* Set current output target */
329 state->output_buf = word_buf;
330 resetPQExpBuffer(word_buf);
331
332 /* Set input source */
333 if (state->buffer_stack != NULL)
334 yy_switch_to_buffer(state->buffer_stack->buf, state->scanner);
335 else
336 yy_switch_to_buffer(state->scanbufhandle, state->scanner);
337
338 /* Set start state */
339 state->start_state = INITIAL;
340
341 /* And lex. */
342 lexresult = yylex(&lval, state->scanner);
343
344 /*
345 * Save start offset of word, if any. We could do this more efficiently,
346 * but for now this seems fine.
347 */
348 if (lexresult)
349 *offset = expr_scanner_offset(state) - word_buf->len;
350 else
351 *offset = -1;
352
353 /*
354 * In case the caller returns to using the regular SQL lexer, reselect the
355 * appropriate initial state.
356 */
358
359 return (bool) lexresult;
360}
#define Assert(condition)
Definition: c.h:812
int expr_scanner_offset(PsqlScanState state)
Definition: exprscan.l:420
int yylex(YYSTYPE *yylval_param, yyscan_t yyscanner)
Definition: exprscan.l:103
void resetPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:146
void psql_scan_reselect_sql_lexer(PsqlScanState state)
Definition: psqlscan.l:1304
int YYSTYPE
Definition: psqlscanslash.l:39
Definition: regguts.h:323

References Assert, expr_scanner_offset(), PQExpBufferData::len, psql_scan_reselect_sql_lexer(), resetPQExpBuffer(), and yylex().

Referenced by process_backslash_command().

◆ expr_scanner_finish()

void expr_scanner_finish ( yyscan_t  yyscanner)

Definition at line 401 of file exprscan.l.

402{
403 PsqlScanState state = yyget_extra(yyscanner);
404
405 /*
406 * Reselect appropriate initial state for SQL lexer.
407 */
409}
static core_yyscan_t yyscanner
Definition: pl_scanner.c:106

References psql_scan_reselect_sql_lexer(), and yyscanner.

Referenced by process_backslash_command().

◆ expr_scanner_get_lineno()

int expr_scanner_get_lineno ( PsqlScanState  state,
int  offset 
)

Definition at line 461 of file exprscan.l.

462{
463 int lineno = 1;
464 const char *p = state->scanbuf;
465
466 while (*p && offset > 0)
467 {
468 if (*p == '\n')
469 lineno++;
470 p++, offset--;
471 }
472 return lineno;
473}

Referenced by ParseScript(), and process_backslash_command().

◆ expr_scanner_get_substring()

char * expr_scanner_get_substring ( PsqlScanState  state,
int  start_offset,
int  end_offset,
bool  chomp 
)

Definition at line 431 of file exprscan.l.

434{
435 char *result;
436 const char *scanptr = state->scanbuf + start_offset;
437 int slen = end_offset - start_offset;
438
439 Assert(slen >= 0);
440 Assert(end_offset <= strlen(state->scanbuf));
441
442 if (chomp)
443 {
444 while (slen > 0 &&
445 (scanptr[slen - 1] == '\n' || scanptr[slen - 1] == '\r'))
446 slen--;
447 }
448
449 result = (char *) pg_malloc(slen + 1);
450 memcpy(result, scanptr, slen);
451 result[slen] = '\0';
452
453 return result;
454}
void * pg_malloc(size_t size)
Definition: fe_memutils.c:47

References Assert, and pg_malloc().

Referenced by expr_yyerror_more(), and process_backslash_command().

◆ expr_scanner_init()

yyscan_t expr_scanner_init ( PsqlScanState  state,
const char *  source,
int  lineno,
int  start_offset,
const char *  command 
)

Definition at line 369 of file exprscan.l.

372{
373 /* Save error context info */
375 expr_lineno = lineno;
376 expr_start_offset = start_offset;
377 expr_command = command;
378
379 /* Must be scanning already */
380 Assert(state->scanbufhandle != NULL);
381
382 /* Set current output target */
383 state->output_buf = NULL;
384
385 /* Set input source */
386 if (state->buffer_stack != NULL)
387 yy_switch_to_buffer(state->buffer_stack->buf, state->scanner);
388 else
389 yy_switch_to_buffer(state->scanbufhandle, state->scanner);
390
391 /* Set start state */
392 state->start_state = EXPR;
393
394 return state->scanner;
395}
static int expr_lineno
Definition: exprscan.l:40
static const char * expr_source
Definition: exprscan.l:39
static int expr_start_offset
Definition: exprscan.l:41
static const char * expr_command
Definition: exprscan.l:42
static rewind_source * source
Definition: pg_rewind.c:89

References Assert, expr_command, expr_lineno, expr_source, expr_start_offset, and source.

Referenced by process_backslash_command().

◆ expr_scanner_offset()

int expr_scanner_offset ( PsqlScanState  state)

Definition at line 420 of file exprscan.l.

421{
422 return strlen(state->scanbuf);
423}

Referenced by expr_lex_one_word(), expr_yyerror_more(), ParseScript(), and process_backslash_command().

◆ expr_yyerror()

void expr_yyerror ( yyscan_t  yyscanner,
const char *  message 
)

Definition at line 309 of file exprscan.l.

310{
311 expr_yyerror_more(yyscanner, message, NULL);
312}
void expr_yyerror_more(yyscan_t yyscanner, const char *message, const char *more)
Definition: exprscan.l:280

References expr_yyerror_more(), and yyscanner.

◆ expr_yyerror_more()

void expr_yyerror_more ( yyscan_t  yyscanner,
const char *  message,
const char *  more 
)

Definition at line 280 of file exprscan.l.

281{
282 PsqlScanState state = yyget_extra(yyscanner);
283 int error_detection_offset = expr_scanner_offset(state) - 1;
284 YYSTYPE lval;
285 char *full_line;
286
287 /*
288 * While parsing an expression, we may not have collected the whole line
289 * yet from the input source. Lex till EOL so we can report whole line.
290 * (If we're at EOF, it's okay to call yylex() an extra time.)
291 */
292 if (!last_was_newline)
293 {
294 while (yylex(&lval, yyscanner))
295 /* skip */ ;
296 }
297
298 /* Extract the line, trimming trailing newline if any */
302 true);
303
305 message, more, error_detection_offset - expr_start_offset);
306}
char * expr_scanner_get_substring(PsqlScanState state, int start_offset, int end_offset, bool chomp)
Definition: exprscan.l:431
static bool last_was_newline
Definition: exprscan.l:45
void syntax_error(const char *source, int lineno, const char *line, const char *command, const char *msg, const char *more, int column)
Definition: pgbench.c:5491

References expr_command, expr_lineno, expr_scanner_get_substring(), expr_scanner_offset(), expr_source, expr_start_offset, last_was_newline, syntax_error(), yylex(), and yyscanner.

Referenced by expr_yyerror().

◆ expr_yylex()

int expr_yylex ( union YYSTYPE yylval_param,
yyscan_t  yyscanner 
)

◆ expr_yyparse()

int expr_yyparse ( yyscan_t  yyscanner)

◆ strtodouble()

bool strtodouble ( const char *  str,
bool  errorOK,
double *  dv 
)

Definition at line 1059 of file pgbench.c.

1061{
1062 char *end;
1063
1064 errno = 0;
1065 *dv = strtod(str, &end);
1066
1067 if (unlikely(errno != 0))
1068 {
1069 if (!errorOK)
1070 pg_log_error("value \"%s\" is out of range for type double", str);
1071 return false;
1072 }
1073
1074 if (unlikely(end == str || *end != '\0'))
1075 {
1076 if (!errorOK)
1077 pg_log_error("invalid input syntax for type double: \"%s\"", str);
1078 return false;
1079 }
1080 return true;
#define unlikely(x)
Definition: c.h:330
const char * str
#define pg_log_error(...)
Definition: logging.h:106

References pg_log_error, str, and unlikely.

Referenced by makeVariableValue().

◆ strtoint64()

bool strtoint64 ( const char *  str,
bool  errorOK,
int64 result 
)

Definition at line 988 of file pgbench.c.

990{
991 const char *ptr = str;
992 int64 tmp = 0;
993 bool neg = false;
994
995 /*
996 * Do our own scan, rather than relying on sscanf which might be broken
997 * for long long.
998 *
999 * As INT64_MIN can't be stored as a positive 64 bit integer, accumulate
1000 * value as a negative number.
1001 */
1002
1003 /* skip leading spaces */
1004 while (*ptr && isspace((unsigned char) *ptr))
1005 ptr++;
1006
1007 /* handle sign */
1008 if (*ptr == '-')
1009 {
1010 ptr++;
1011 neg = true;
1012 }
1013 else if (*ptr == '+')
1014 ptr++;
1015
1016 /* require at least one digit */
1017 if (unlikely(!isdigit((unsigned char) *ptr)))
1018 goto invalid_syntax;
1019
1020 /* process digits */
1021 while (*ptr && isdigit((unsigned char) *ptr))
1022 {
1023 int8 digit = (*ptr++ - '0');
1024
1025 if (unlikely(pg_mul_s64_overflow(tmp, 10, &tmp)) ||
1026 unlikely(pg_sub_s64_overflow(tmp, digit, &tmp)))
1027 goto out_of_range;
1028 }
1029
1030 /* allow trailing whitespace, but not other trailing chars */
1031 while (*ptr != '\0' && isspace((unsigned char) *ptr))
1032 ptr++;
1033
1034 if (unlikely(*ptr != '\0'))
1035 goto invalid_syntax;
1036
1037 if (!neg)
1038 {
1039 if (unlikely(tmp == PG_INT64_MIN))
1040 goto out_of_range;
1041 tmp = -tmp;
1042 }
1043
1044 *result = tmp;
1045 return true;
1046
1047out_of_range:
1048 if (!errorOK)
1049 pg_log_error("value \"%s\" is out of range for type bigint", str);
1050 return false;
1051
1052invalid_syntax:
1053 if (!errorOK)
1054 pg_log_error("invalid input syntax for type bigint: \"%s\"", str);
1055 return false;
int64_t int64
Definition: c.h:482
int8_t int8
Definition: c.h:479
#define PG_INT64_MIN
Definition: c.h:545
static bool pg_mul_s64_overflow(int64 a, int64 b, int64 *result)
Definition: int.h:293
static bool pg_sub_s64_overflow(int64 a, int64 b, int64 *result)
Definition: int.h:262

References PG_INT64_MIN, pg_log_error, pg_mul_s64_overflow(), pg_sub_s64_overflow(), str, and unlikely.

Referenced by makeVariableValue().

◆ syntax_error()

void syntax_error ( const char *  source,
int  lineno,
const char *  line,
const char *  command,
const char *  msg,
const char *  more,
int  column 
)

Definition at line 5491 of file pgbench.c.

5495{
5497
5499
5500 printfPQExpBuffer(&buf, "%s:%d: %s", source, lineno, msg);
5501 if (more != NULL)
5502 appendPQExpBuffer(&buf, " (%s)", more);
5503 if (column >= 0 && line == NULL)
5504 appendPQExpBuffer(&buf, " at column %d", column + 1);
5505 if (command != NULL)
5506 appendPQExpBuffer(&buf, " in command \"%s\"", command);
5507
5508 pg_log_error("%s", buf.data);
5509
5511
5512 if (line != NULL)
5513 {
5514 fprintf(stderr, "%s\n", line);
5515 if (column >= 0)
5516 fprintf(stderr, "%*c error found here\n", column + 1, '^');
5517 }
5518
5519 exit(1);
#define fprintf(file, fmt, msg)
Definition: cubescan.l:21
exit(1)
static char * buf
Definition: pg_test_fsync.c:72
void printfPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:235
void initPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:90
void appendPQExpBuffer(PQExpBuffer str, const char *fmt,...)
Definition: pqexpbuffer.c:265
void termPQExpBuffer(PQExpBuffer str)
Definition: pqexpbuffer.c:129

References appendPQExpBuffer(), buf, exit(), fprintf, initPQExpBuffer(), pg_log_error, printfPQExpBuffer(), source, and termPQExpBuffer().

Referenced by expr_yyerror_more(), ParseScript(), process_backslash_command(), and string_to_uuid().

Variable Documentation

◆ expr_parse_result

PgBenchExpr* expr_parse_result
extern