#include "postgres_fe.h"
#include "getopt_long.h"
#include "libpq-fe.h"
#include "portability/instr_time.h"
#include <ctype.h>
#include <float.h>
#include <limits.h>
#include <math.h>
#include <signal.h>
#include <time.h>
#include "pgbench.h"

Include dependency graph for pgbench.c:

Data Structures
struct	Variable

struct	SimpleStats

struct	StatsData

struct	CState

struct	TState

struct	Command

struct	ParsedScript

struct	BuiltinScript

Macros
#define	M_PI 3.14159265358979323846

#define	ERRCODE_UNDEFINED_TABLE "42P01"

#define	pthread_t void *

#define	MAXCLIENTS 1024

#define	LOG_STEP_SECONDS 5 /* seconds between log messages */

#define	DEFAULT_NXACTS 10 /* default nxacts */

#define	MIN_GAUSSIAN_PARAM 2.0 /* minimum parameter for gauss */

#define	nbranches

#define	ntellers 10

#define	naccounts 100000

#define	SCALE_32BIT_THRESHOLD 20000

#define	WSEP '@' /* weight separator */

#define	MAX_SCRIPTS 128 /* max number of SQL scripts allowed */

#define	SHELL_COMMAND_SIZE 256 /* maximum size allowed for shell command */

#define	INVALID_THREAD ((pthread_t) 0)

#define	SQL_COMMAND 1

#define	META_COMMAND 2

#define	MAX_ARGS 10

#define	PARAMS_ARRAY_SIZE 7

#define	MAX_FARGS 16

#define	MAX_PREPARE_NAME 32

#define	SCALE_32BIT_THRESHOLD 20000

#define	COMMANDS_ALLOC_NUM 128

Typedefs
typedef struct SimpleStats	SimpleStats

typedef struct StatsData	StatsData

typedef enum QueryMode	QueryMode

typedef struct ParsedScript	ParsedScript

typedef struct BuiltinScript	BuiltinScript

Enumerations
enum	ConnectionStateEnum { CSTATE_CHOOSE_SCRIPT, CSTATE_START_THROTTLE, CSTATE_THROTTLE, CSTATE_START_TX, CSTATE_START_COMMAND, CSTATE_WAIT_RESULT, CSTATE_SLEEP, CSTATE_END_COMMAND, CSTATE_END_TX, CSTATE_ABORTED, CSTATE_FINISHED }

enum	QueryMode { QUERY_SIMPLE, QUERY_EXTENDED, QUERY_PREPARED, NUM_QUERYMODE }

Functions
static void	setIntValue (PgBenchValue *pv, int64 ival)

static void	setDoubleValue (PgBenchValue *pv, double dval)

static bool	evaluateExpr (TState , CState , PgBenchExpr , PgBenchValue )

static void	doLog (TState thread, CState st, StatsData *agg, bool skipped, double latency, double lag)

static void	processXactStats (TState thread, CState st, instr_time now, bool skipped, StatsData agg)

static void	pgbench_error (const char *fmt,...) pg_attribute_printf(1

static void static void	addScript (ParsedScript script)

static void *	threadRun (void *arg)

static void	setalarm (int seconds)

static void	usage (void)

static bool	is_an_int (const char *str)

int64	strtoint64 (const char *str)

static int64	getrand (TState *thread, int64 min, int64 max)

static int64	getExponentialRand (TState *thread, int64 min, int64 max, double parameter)

static int64	getGaussianRand (TState *thread, int64 min, int64 max, double parameter)

static int64	getPoissonRand (TState *thread, int64 center)

static void	initSimpleStats (SimpleStats *ss)

static void	addToSimpleStats (SimpleStats *ss, double val)

static void	mergeSimpleStats (SimpleStats acc, SimpleStats ss)

static void	initStats (StatsData *sd, time_t start_time)

static void	accumStats (StatsData *stats, bool skipped, double lat, double lag)

static void	executeStatement (PGconn con, const char sql)

static void	tryExecuteStatement (PGconn con, const char sql)

static PGconn *	doConnect (void)

static void	discard_response (CState *state)

static int	compareVariableNames (const void v1, const void v2)

static Variable *	lookupVariable (CState st, char name)

static char *	getVariable (CState st, char name)

static bool	makeVariableNumeric (Variable *var)

static bool	isLegalVariableName (const char *name)

static Variable *	lookupCreateVariable (CState st, const char context, char *name)

static bool	putVariable (CState st, const char context, char name, const char value)

static bool	putVariableNumber (CState st, const char context, char name, const PgBenchValue value)

static bool	putVariableInt (CState st, const char context, char *name, int64 value)

static char *	parseVariable (const char sql, int eaten)

static char *	replaceVariable (char *sql, char param, int len, char *value)

static char *	assignVariables (CState st, char sql)

static void	getQueryParams (CState st, const Command command, const char **params)

static bool	coerceToInt (PgBenchValue pval, int64 ival)

static bool	coerceToDouble (PgBenchValue pval, double dval)

static bool	evalFunc (TState thread, CState st, PgBenchFunction func, PgBenchExprLink args, PgBenchValue retval)

static bool	runShellCommand (CState st, char variable, char **argv, int argc)

static void	preparedStatementName (char *buffer, int file, int state)

static void	commandFailed (CState st, char message)

static int	chooseScript (TState *thread)

static bool	sendCommand (CState st, Command command)

static bool	evaluateSleep (CState st, int argc, char argv, int usecs)

static void	doCustom (TState thread, CState st, StatsData *agg)

static void	disconnect_all (CState *state, int length)

static void	init (bool is_no_vacuum)

static bool	parseQuery (Command cmd, const char raw_sql)

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

static Command *	process_sql_command (PQExpBuffer buf, const char *source)

static Command *	process_backslash_command (PsqlScanState sstate, const char *source)

static void	ParseScript (const char script, const char desc, int weight)

static char *	read_file_contents (FILE *fd)

static void	process_file (const char *filename, int weight)

static void	process_builtin (const BuiltinScript *bi, int weight)

static void	listAvailableScripts (void)

static const BuiltinScript *	findBuiltin (const char *name)

static int	parseScriptWeight (const char option, char *script)

static void	printSimpleStats (char prefix, SimpleStats ss)

static void	printResults (TState threads, StatsData total, instr_time total_time, instr_time conn_total_time, int latency_late)

int	main (int argc, char **argv)

static void	handle_sig_alarm (SIGNAL_ARGS)

Variables
int	nxacts = 0

int	duration = 0

int64	end_time = 0

int	scale = 1

int	fillfactor = 100

int	foreign_keys = 0

int	unlogged_tables = 0

double	sample_rate = 0.0

int64	throttle_delay = 0

int64	latency_limit = 0

char *	tablespace = NULL

char *	index_tablespace = NULL

bool	use_log

bool	use_quiet

int	agg_interval

bool	per_script_stats = false

int	progress = 0

bool	progress_timestamp = false

int	nclients = 1

int	nthreads = 1

bool	is_connect

bool	is_latencies

int	main_pid

char *	pghost = ""

char *	pgport = ""

char *	login = NULL

char *	dbName

char *	logfile_prefix = NULL

const char *	progname

volatile bool	timer_exceeded = false

static QueryMode	querymode = QUERY_SIMPLE

static const char *	QUERYMODE [] = {"simple", "extended", "prepared"}

static ParsedScript	sql_script [MAX_SCRIPTS]

static int	num_scripts

static int	num_commands = 0

static int64	total_weight = 0

static int	debug = 0

static const BuiltinScript	builtin_script []

static const PsqlScanCallbacks	pgbench_callbacks

Macro Definition Documentation

#define COMMANDS_ALLOC_NUM 128

Referenced by ParseScript().

#define DEFAULT_NXACTS 10 /* default nxacts */

Definition at line 94 of file pgbench.c.

Referenced by main().

#define ERRCODE_UNDEFINED_TABLE "42P01"

Definition at line 61 of file pgbench.c.

Referenced by brin_desummarize_range(), brin_summarize_range(), bt_index_check_internal(), errorMissingRTE(), main(), parserOpenTable(), pg_class_aclmask(), pg_class_ownercheck(), pg_extension_config_dump(), plpgsql_parse_wordrowtype(), RangeVarGetRelidExtended(), RemoveInheritance(), and transformLockingClause().

#define INVALID_THREAD ((pthread_t) 0)

Definition at line 356 of file pgbench.c.

Referenced by main().

#define LOG_STEP_SECONDS 5 /* seconds between log messages */

Definition at line 93 of file pgbench.c.

Referenced by init().

#define M_PI 3.14159265358979323846

Definition at line 56 of file pgbench.c.

Referenced by evalFunc(), and getGaussianRand().

#define MAX_ARGS 10

Definition at line 363 of file pgbench.c.

Referenced by parseQuery(), process_backslash_command(), and sendCommand().

#define MAX_FARGS 16

Definition at line 1286 of file pgbench.c.

Referenced by evalFunc().

#define MAX_PREPARE_NAME 32

Definition at line 1803 of file pgbench.c.

Referenced by sendCommand().

#define MAX_SCRIPTS 128 /* max number of SQL scripts allowed */

Definition at line 207 of file pgbench.c.

Referenced by addScript().

#define MAXCLIENTS 1024

Definition at line 90 of file pgbench.c.

Referenced by main().

#define META_COMMAND 2

Definition at line 362 of file pgbench.c.

Referenced by doCustom(), and process_backslash_command().

#define MIN_GAUSSIAN_PARAM 2.0 /* minimum parameter for gauss */

Definition at line 96 of file pgbench.c.

Referenced by evalFunc(), and getGaussianRand().

#define naccounts 100000

Definition at line 156 of file pgbench.c.

Referenced by init().

#define nbranches

Value:

1 /* Makes little sense to change this. Change

* -s instead */

Definition at line 153 of file pgbench.c.

Referenced by init().

#define ntellers 10

Definition at line 155 of file pgbench.c.

Referenced by init().

#define PARAMS_ARRAY_SIZE 7

Referenced by doConnect().

#define pthread_t void *

Definition at line 79 of file pgbench.c.

Referenced by main().

#define SCALE_32BIT_THRESHOLD 20000

Definition at line 165 of file pgbench.c.

Referenced by init().

#define SCALE_32BIT_THRESHOLD 20000

Definition at line 165 of file pgbench.c.

#define SHELL_COMMAND_SIZE 256 /* maximum size allowed for shell command */

Definition at line 208 of file pgbench.c.

Referenced by runShellCommand().

#define SQL_COMMAND 1

Definition at line 361 of file pgbench.c.

Referenced by doCustom(), process_sql_command(), and sendCommand().

#define WSEP '@' /* weight separator */

Definition at line 187 of file pgbench.c.

Referenced by parseScriptWeight().

Typedef Documentation

typedef struct BuiltinScript BuiltinScript

typedef struct ParsedScript ParsedScript

typedef enum QueryMode QueryMode

typedef struct SimpleStats SimpleStats

typedef struct StatsData StatsData

Enumeration Type Documentation

enum ConnectionStateEnum

Enumerator
CSTATE_CHOOSE_SCRIPT
CSTATE_START_THROTTLE
CSTATE_THROTTLE
CSTATE_START_TX
CSTATE_START_COMMAND
CSTATE_WAIT_RESULT
CSTATE_SLEEP
CSTATE_END_COMMAND
CSTATE_END_TX
CSTATE_ABORTED
CSTATE_FINISHED

Definition at line 242 of file pgbench.c.

 {
     /*
      * The client must first choose a script to execute.  Once chosen, it can
      * either be throttled (state CSTATE_START_THROTTLE under --rate) or start
      * right away (state CSTATE_START_TX).
      */
     CSTATE_CHOOSE_SCRIPT,
 
     /*
      * In CSTATE_START_THROTTLE state, we calculate when to begin the next
      * transaction, and advance to CSTATE_THROTTLE.  CSTATE_THROTTLE state
      * sleeps until that moment.  (If throttling is not enabled, doCustom()
      * falls directly through from CSTATE_START_THROTTLE to CSTATE_START_TX.)
      */
     CSTATE_START_THROTTLE,
     CSTATE_THROTTLE,
 
     /*
      * CSTATE_START_TX performs start-of-transaction processing.  Establishes
      * a new connection for the transaction, in --connect mode, and records
      * the transaction start time.
      */
     CSTATE_START_TX,
 
     /*
      * We loop through these states, to process each command in the script:
      *
      * CSTATE_START_COMMAND starts the execution of a command.  On a SQL
      * command, the command is sent to the server, and we move to
      * CSTATE_WAIT_RESULT state.  On a \sleep meta-command, the timer is set,
      * and we enter the CSTATE_SLEEP state to wait for it to expire. Other
      * meta-commands are executed immediately.
      *
      * CSTATE_WAIT_RESULT waits until we get a result set back from the server
      * for the current command.
      *
      * CSTATE_SLEEP waits until the end of \sleep.
      *
      * CSTATE_END_COMMAND records the end-of-command timestamp, increments the
      * command counter, and loops back to CSTATE_START_COMMAND state.
      */
     CSTATE_START_COMMAND,
     CSTATE_WAIT_RESULT,
     CSTATE_SLEEP,
     CSTATE_END_COMMAND,
 
     /*
      * CSTATE_END_TX performs end-of-transaction processing.  Calculates
      * latency, and logs the transaction.  In --connect mode, closes the
      * current connection.  Chooses the next script to execute and starts over
      * in CSTATE_START_THROTTLE state, or enters CSTATE_FINISHED if we have no
      * more work to do.
      */
     CSTATE_END_TX,
 
     /*
      * Final states.  CSTATE_ABORTED means that the script execution was
      * aborted because a command failed, CSTATE_FINISHED means success.
      */
     CSTATE_ABORTED,
     CSTATE_FINISHED
 } ConnectionStateEnum;

enum QueryMode

Enumerator
QUERY_SIMPLE
QUERY_EXTENDED
QUERY_PREPARED
NUM_QUERYMODE

Definition at line 365 of file pgbench.c.

 {
     QUERY_SIMPLE,               /* simple query */
     QUERY_EXTENDED,             /* extended query */
     QUERY_PREPARED,             /* extended query with prepared statements */
     NUM_QUERYMODE
 } QueryMode;

Function Documentation

static void accumStats	(	StatsData *	stats,
		bool	skipped,
		double	lat,
		double	lag
	)

static

Definition at line 797 of file pgbench.c.

References addToSimpleStats(), StatsData::cnt, StatsData::lag, StatsData::latency, and StatsData::skipped.

Referenced by doLog(), and processXactStats().

 {
     stats->cnt++;
 
     if (skipped)
     {
         /* no latency to record on skipped transactions */
         stats->skipped++;
     }
     else
     {
         addToSimpleStats(&stats->latency, lat);
 
         /* and possibly the same for schedule lag */
         if (throttle_delay)
             addToSimpleStats(&stats->lag, lag);
     }
 }

static void addScript ( ParsedScript script )

static

Definition at line 3458 of file pgbench.c.

References ParsedScript::commands, ParsedScript::desc, MAX_SCRIPTS, NULL, and num_scripts.

Referenced by ParseScript().

 {
     if (script.commands == NULL || script.commands[0] == NULL)
     {
         fprintf(stderr, "empty command list for script \"%s\"\n", script.desc);
         exit(1);
     }
 
     if (num_scripts >= MAX_SCRIPTS)
     {
         fprintf(stderr, "at most %d SQL scripts are allowed\n", MAX_SCRIPTS);
         exit(1);
     }
 
     sql_script[num_scripts] = script;
     num_scripts++;
 }

static void addToSimpleStats	(	SimpleStats *	ss,
		double	val
	)

static

Definition at line 753 of file pgbench.c.

References SimpleStats::count, SimpleStats::max, SimpleStats::min, SimpleStats::sum, SimpleStats::sum2, and val.

Referenced by accumStats(), and doCustom().

 {
     if (ss->count == 0 || val < ss->min)
         ss->min = val;
     if (ss->count == 0 || val > ss->max)
         ss->max = val;
     ss->count++;
     ss->sum += val;
     ss->sum2 += val * val;
 }

static char* assignVariables	(	CState *	st,
		char *	sql
	)

static

Definition at line 1184 of file pgbench.c.

References free, getVariable(), name, NULL, parseVariable(), replaceVariable(), and val.

Referenced by sendCommand().

 {
     char       *p,
                *name,
                *val;
 
     p = sql;
     while ((p = strchr(p, ':')) != NULL)
     {
         int         eaten;
 
         name = parseVariable(p, &eaten);
         if (name == NULL)
         {
             while (*p == ':')
             {
                 p++;
             }
             continue;
         }
 
         val = getVariable(st, name);
         free(name);
         if (val == NULL)
         {
             p++;
             continue;
         }
 
         p = replaceVariable(&sql, p, eaten, val);
     }
 
     return sql;
 }

static int chooseScript ( TState * thread )

static

Definition at line 1820 of file pgbench.c.

References getrand(), i, and ParsedScript::weight.

Referenced by doCustom().

 {
     int         i = 0;
     int64       w;
 
     if (num_scripts == 1)
         return 0;
 
     w = getrand(thread, 0, total_weight - 1);
     do
     {
         w -= sql_script[i++].weight;
     } while (w >= 0);
 
     return i - 1;
 }

static bool coerceToDouble	(	PgBenchValue *	pval,
		double *	dval
	)

static

Definition at line 1254 of file pgbench.c.

References Assert, PgBenchValue::dval, PgBenchValue::ival, PGBT_DOUBLE, PGBT_INT, PgBenchValue::type, and PgBenchValue::u.

Referenced by evalFunc().

 {
     if (pval->type == PGBT_DOUBLE)
     {
         *dval = pval->u.dval;
         return true;
     }
     else
     {
         Assert(pval->type == PGBT_INT);
         *dval = (double) pval->u.ival;
         return true;
     }
 }

static bool coerceToInt	(	PgBenchValue *	pval,
		int64 *	ival
	)

static

Definition at line 1230 of file pgbench.c.

References Assert, PgBenchValue::dval, PgBenchValue::ival, PG_INT64_MAX, PG_INT64_MIN, PGBT_DOUBLE, PGBT_INT, PgBenchValue::type, and PgBenchValue::u.

Referenced by evalFunc().

 {
     if (pval->type == PGBT_INT)
     {
         *ival = pval->u.ival;
         return true;
     }
     else
     {
         double      dval = pval->u.dval;
 
         Assert(pval->type == PGBT_DOUBLE);
         if (dval < PG_INT64_MIN || PG_INT64_MAX < dval)
         {
             fprintf(stderr, "double to int overflow for %f\n", dval);
             return false;
         }
         *ival = (int64) dval;
         return true;
     }
 }

static void commandFailed	(	CState *	st,
		char *	message
	)

static

Definition at line 1811 of file pgbench.c.

References CState::command, CState::id, and CState::use_file.

Referenced by doCustom().

 {
     fprintf(stderr,
             "client %d aborted in command %d of script %d; %s\n",
             st->id, st->command, st->use_file, message);
 }

static int compareVariableNames	(	const void *	v1,
		const void *	v2
	)

static

Definition at line 931 of file pgbench.c.

References name.

Referenced by lookupVariable().

 {
     return strcmp(((const Variable *) v1)->name,
                   ((const Variable *) v2)->name);
 }

static void discard_response ( CState * state )

static

Definition at line 917 of file pgbench.c.

References CState::con, PQclear(), and PQgetResult().

Referenced by doCustom().

 {
     PGresult   *res;
 
     do
     {
         res = PQgetResult(state->con);
         if (res)
             PQclear(res);
     } while (res);
 }

static void disconnect_all	(	CState *	state,
		int	length
	)

static

Definition at line 2555 of file pgbench.c.

References CState::con, i, length(), NULL, and PQfinish().

Referenced by main(), and threadRun().

 {
     int         i;
 
     for (i = 0; i < length; i++)
     {
         if (state[i].con)
         {
             PQfinish(state[i].con);
             state[i].con = NULL;
         }
     }
 }

static PGconn* doConnect ( void )

static

Definition at line 848 of file pgbench.c.

References conn, CONNECTION_BAD, dbName, have_password, login, NULL, PARAMS_ARRAY_SIZE, password, pghost, pgport, PQconnectdbParams(), PQconnectionNeedsPassword(), PQerrorMessage(), PQfinish(), PQstatus(), progname, simple_prompt(), and values.

Referenced by doCustom(), init(), main(), and threadRun().

 {
     PGconn     *conn;
     bool        new_pass;
     static bool have_password = false;
     static char password[100];
 
     /*
      * Start the connection.  Loop until we have a password if requested by
      * backend.
      */
     do
     {
 #define PARAMS_ARRAY_SIZE   7
 
         const char *keywords[PARAMS_ARRAY_SIZE];
         const char *values[PARAMS_ARRAY_SIZE];
 
         keywords[0] = "host";
         values[0] = pghost;
         keywords[1] = "port";
         values[1] = pgport;
         keywords[2] = "user";
         values[2] = login;
         keywords[3] = "password";
         values[3] = have_password ? password : NULL;
         keywords[4] = "dbname";
         values[4] = dbName;
         keywords[5] = "fallback_application_name";
         values[5] = progname;
         keywords[6] = NULL;
         values[6] = NULL;
 
         new_pass = false;
 
         conn = PQconnectdbParams(keywords, values, true);
 
         if (!conn)
         {
             fprintf(stderr, "connection to database \"%s\" failed\n",
                     dbName);
             return NULL;
         }
 
         if (PQstatus(conn) == CONNECTION_BAD &&
             PQconnectionNeedsPassword(conn) &&
             !have_password)
         {
             PQfinish(conn);
             simple_prompt("Password: ", password, sizeof(password), false);
             have_password = true;
             new_pass = true;
         }
     } while (new_pass);
 
     /* check to see that the backend connection was successfully made */
     if (PQstatus(conn) == CONNECTION_BAD)
     {
         fprintf(stderr, "connection to database \"%s\" failed:\n%s",
                 dbName, PQerrorMessage(conn));
         PQfinish(conn);
         return NULL;
     }
 
     return conn;
 }

static void doCustom	(	TState *	thread,
		CState *	st,
		StatsData *	agg
	)

static

Definition at line 1958 of file pgbench.c.

References addToSimpleStats(), Command::argc, Command::argv, Assert, chooseScript(), StatsData::cnt, CState::cnt, CState::command, commandFailed(), ParsedScript::commands, CState::con, TState::conn_time, CSTATE_ABORTED, CSTATE_CHOOSE_SCRIPT, CSTATE_END_COMMAND, CSTATE_END_TX, CSTATE_FINISHED, CSTATE_SLEEP, CSTATE_START_COMMAND, CSTATE_START_THROTTLE, CSTATE_START_TX, CSTATE_THROTTLE, CSTATE_WAIT_RESULT, ParsedScript::desc, discard_response(), doConnect(), evaluateExpr(), evaluateSleep(), Command::expr, getPoissonRand(), i, CState::id, INSTR_TIME_ACCUM_DIFF, INSTR_TIME_GET_DOUBLE, INSTR_TIME_GET_MICROSEC, INSTR_TIME_IS_ZERO, INSTR_TIME_SET_CURRENT, INSTR_TIME_SET_ZERO, INT64_FORMAT, META_COMMAND, now(), NULL, pg_strcasecmp(), PGRES_COMMAND_OK, PGRES_EMPTY_QUERY, PGRES_TUPLES_OK, PQclear(), PQconsumeInput(), PQerrorMessage(), PQfinish(), PQgetResult(), PQisBusy(), PQresultStatus(), CState::prepared, processXactStats(), putVariableNumber(), result, runShellCommand(), sendCommand(), CState::sleep_until, SQL_COMMAND, CState::state, TState::stats, Command::stats, CState::stmt_begin, TState::throttle_trigger, CState::txn_begin, CState::txn_scheduled, Command::type, and CState::use_file.

Referenced by threadRun().

 {
     PGresult   *res;
     Command    *command;
     instr_time  now;
     bool        end_tx_processed = false;
     int64       wait;
 
     /*
      * gettimeofday() isn't free, so we get the current timestamp lazily the
      * first time it's needed, and reuse the same value throughout this
      * function after that.  This also ensures that e.g. the calculated
      * latency reported in the log file and in the totals are the same. Zero
      * means "not set yet".  Reset "now" when we execute shell commands or
      * expressions, which might take a non-negligible amount of time, though.
      */
     INSTR_TIME_SET_ZERO(now);
 
     /*
      * Loop in the state machine, until we have to wait for a result from the
      * server (or have to sleep, for throttling or for \sleep).
      *
      * Note: In the switch-statement below, 'break' will loop back here,
      * meaning "continue in the state machine".  Return is used to return to
      * the caller.
      */
     for (;;)
     {
         switch (st->state)
         {
                 /*
                  * Select transaction to run.
                  */
             case CSTATE_CHOOSE_SCRIPT:
 
                 st->use_file = chooseScript(thread);
 
                 if (debug)
                     fprintf(stderr, "client %d executing script \"%s\"\n", st->id,
                             sql_script[st->use_file].desc);
 
                 if (throttle_delay > 0)
                     st->state = CSTATE_START_THROTTLE;
                 else
                     st->state = CSTATE_START_TX;
                 break;
 
                 /*
                  * Handle throttling once per transaction by sleeping.
                  */
             case CSTATE_START_THROTTLE:
 
                 /*
                  * Generate a delay such that the series of delays will
                  * approximate a Poisson distribution centered on the
                  * throttle_delay time.
                  *
                  * If transactions are too slow or a given wait is shorter
                  * than a transaction, the next transaction will start right
                  * away.
                  */
                 Assert(throttle_delay > 0);
                 wait = getPoissonRand(thread, throttle_delay);
 
                 thread->throttle_trigger += wait;
                 st->txn_scheduled = thread->throttle_trigger;
 
                 /*
                  * stop client if next transaction is beyond pgbench end of
                  * execution
                  */
                 if (duration > 0 && st->txn_scheduled > end_time)
                 {
                     st->state = CSTATE_FINISHED;
                     break;
                 }
 
                 /*
                  * If this --latency-limit is used, and this slot is already
                  * late so that the transaction will miss the latency limit
                  * even if it completed immediately, we skip this time slot
                  * and iterate till the next slot that isn't late yet.
                  */
                 if (latency_limit)
                 {
                     int64       now_us;
 
                     if (INSTR_TIME_IS_ZERO(now))
                         INSTR_TIME_SET_CURRENT(now);
                     now_us = INSTR_TIME_GET_MICROSEC(now);
                     while (thread->throttle_trigger < now_us - latency_limit)
                     {
                         processXactStats(thread, st, &now, true, agg);
                         /* next rendez-vous */
                         wait = getPoissonRand(thread, throttle_delay);
                         thread->throttle_trigger += wait;
                         st->txn_scheduled = thread->throttle_trigger;
                     }
                 }
 
                 st->state = CSTATE_THROTTLE;
                 if (debug)
                     fprintf(stderr, "client %d throttling " INT64_FORMAT " us\n",
                             st->id, wait);
                 break;
 
                 /*
                  * Wait until it's time to start next transaction.
                  */
             case CSTATE_THROTTLE:
                 if (INSTR_TIME_IS_ZERO(now))
                     INSTR_TIME_SET_CURRENT(now);
                 if (INSTR_TIME_GET_MICROSEC(now) < st->txn_scheduled)
                     return;     /* Still sleeping, nothing to do here */
 
                 /* Else done sleeping, start the transaction */
                 st->state = CSTATE_START_TX;
                 break;
 
                 /* Start new transaction */
             case CSTATE_START_TX:
 
                 /*
                  * Establish connection on first call, or if is_connect is
                  * true.
                  */
                 if (st->con == NULL)
                 {
                     instr_time  start;
 
                     if (INSTR_TIME_IS_ZERO(now))
                         INSTR_TIME_SET_CURRENT(now);
                     start = now;
                     if ((st->con = doConnect()) == NULL)
                     {
                         fprintf(stderr, "client %d aborted while establishing connection\n",
                                 st->id);
                         st->state = CSTATE_ABORTED;
                         break;
                     }
                     INSTR_TIME_SET_CURRENT(now);
                     INSTR_TIME_ACCUM_DIFF(thread->conn_time, now, start);
 
                     /* Reset session-local state */
                     memset(st->prepared, 0, sizeof(st->prepared));
                 }
 
                 /*
                  * Record transaction start time under logging, progress or
                  * throttling.
                  */
                 if (use_log || progress || throttle_delay || latency_limit ||
                     per_script_stats)
                 {
                     if (INSTR_TIME_IS_ZERO(now))
                         INSTR_TIME_SET_CURRENT(now);
                     st->txn_begin = now;
 
                     /*
                      * When not throttling, this is also the transaction's
                      * scheduled start time.
                      */
                     if (!throttle_delay)
                         st->txn_scheduled = INSTR_TIME_GET_MICROSEC(now);
                 }
 
                 /* Begin with the first command */
                 st->command = 0;
                 st->state = CSTATE_START_COMMAND;
                 break;
 
                 /*
                  * Send a command to server (or execute a meta-command)
                  */
             case CSTATE_START_COMMAND:
                 command = sql_script[st->use_file].commands[st->command];
 
                 /*
                  * If we reached the end of the script, move to end-of-xact
                  * processing.
                  */
                 if (command == NULL)
                 {
                     st->state = CSTATE_END_TX;
                     break;
                 }
 
                 /*
                  * Record statement start time if per-command latencies are
                  * requested
                  */
                 if (is_latencies)
                 {
                     if (INSTR_TIME_IS_ZERO(now))
                         INSTR_TIME_SET_CURRENT(now);
                     st->stmt_begin = now;
                 }
 
                 if (command->type == SQL_COMMAND)
                 {
                     if (!sendCommand(st, command))
                     {
                         /*
                          * Failed. Stay in CSTATE_START_COMMAND state, to
                          * retry. ??? What the point or retrying? Should
                          * rather abort?
                          */
                         return;
                     }
                     else
                         st->state = CSTATE_WAIT_RESULT;
                 }
                 else if (command->type == META_COMMAND)
                 {
                     int         argc = command->argc,
                                 i;
                     char      **argv = command->argv;
 
                     if (debug)
                     {
                         fprintf(stderr, "client %d executing \\%s", st->id, argv[0]);
                         for (i = 1; i < argc; i++)
                             fprintf(stderr, " %s", argv[i]);
                         fprintf(stderr, "\n");
                     }
 
                     if (pg_strcasecmp(argv[0], "sleep") == 0)
                     {
                         /*
                          * A \sleep doesn't execute anything, we just get the
                          * delay from the argument, and enter the CSTATE_SLEEP
                          * state.  (The per-command latency will be recorded
                          * in CSTATE_SLEEP state, not here, after the delay
                          * has elapsed.)
                          */
                         int         usec;
 
                         if (!evaluateSleep(st, argc, argv, &usec))
                         {
                             commandFailed(st, "execution of meta-command 'sleep' failed");
                             st->state = CSTATE_ABORTED;
                             break;
                         }
 
                         if (INSTR_TIME_IS_ZERO(now))
                             INSTR_TIME_SET_CURRENT(now);
                         st->sleep_until = INSTR_TIME_GET_MICROSEC(now) + usec;
                         st->state = CSTATE_SLEEP;
                         break;
                     }
                     else
                     {
                         if (pg_strcasecmp(argv[0], "set") == 0)
                         {
                             PgBenchExpr *expr = command->expr;
                             PgBenchValue result;
 
                             if (!evaluateExpr(thread, st, expr, &result))
                             {
                                 commandFailed(st, "evaluation of meta-command 'set' failed");
                                 st->state = CSTATE_ABORTED;
                                 break;
                             }
 
                             if (!putVariableNumber(st, argv[0], argv[1], &result))
                             {
                                 commandFailed(st, "assignment of meta-command 'set' failed");
                                 st->state = CSTATE_ABORTED;
                                 break;
                             }
                         }
                         else if (pg_strcasecmp(argv[0], "setshell") == 0)
                         {
                             bool        ret = runShellCommand(st, argv[1], argv + 2, argc - 2);
 
                             if (timer_exceeded) /* timeout */
                             {
                                 st->state = CSTATE_FINISHED;
                                 break;
                             }
                             else if (!ret)      /* on error */
                             {
                                 commandFailed(st, "execution of meta-command 'setshell' failed");
                                 st->state = CSTATE_ABORTED;
                                 break;
                             }
                             else
                             {
                                 /* succeeded */
                             }
                         }
                         else if (pg_strcasecmp(argv[0], "shell") == 0)
                         {
                             bool        ret = runShellCommand(st, NULL, argv + 1, argc - 1);
 
                             if (timer_exceeded) /* timeout */
                             {
                                 st->state = CSTATE_FINISHED;
                                 break;
                             }
                             else if (!ret)      /* on error */
                             {
                                 commandFailed(st, "execution of meta-command 'shell' failed");
                                 st->state = CSTATE_ABORTED;
                                 break;
                             }
                             else
                             {
                                 /* succeeded */
                             }
                         }
 
                         /*
                          * executing the expression or shell command might
                          * take a non-negligible amount of time, so reset
                          * 'now'
                          */
                         INSTR_TIME_SET_ZERO(now);
 
                         st->state = CSTATE_END_COMMAND;
                     }
                 }
                 break;
 
                 /*
                  * Wait for the current SQL command to complete
                  */
             case CSTATE_WAIT_RESULT:
                 command = sql_script[st->use_file].commands[st->command];
                 if (debug)
                     fprintf(stderr, "client %d receiving\n", st->id);
                 if (!PQconsumeInput(st->con))
                 {               /* there's something wrong */
                     commandFailed(st, "perhaps the backend died while processing");
                     st->state = CSTATE_ABORTED;
                     break;
                 }
                 if (PQisBusy(st->con))
                     return;     /* don't have the whole result yet */
 
                 /*
                  * Read and discard the query result;
                  */
                 res = PQgetResult(st->con);
                 switch (PQresultStatus(res))
                 {
                     case PGRES_COMMAND_OK:
                     case PGRES_TUPLES_OK:
                     case PGRES_EMPTY_QUERY:
                         /* OK */
                         PQclear(res);
                         discard_response(st);
                         st->state = CSTATE_END_COMMAND;
                         break;
                     default:
                         commandFailed(st, PQerrorMessage(st->con));
                         PQclear(res);
                         st->state = CSTATE_ABORTED;
                         break;
                 }
                 break;
 
                 /*
                  * Wait until sleep is done. This state is entered after a
                  * \sleep metacommand. The behavior is similar to
                  * CSTATE_THROTTLE, but proceeds to CSTATE_START_COMMAND
                  * instead of CSTATE_START_TX.
                  */
             case CSTATE_SLEEP:
                 if (INSTR_TIME_IS_ZERO(now))
                     INSTR_TIME_SET_CURRENT(now);
                 if (INSTR_TIME_GET_MICROSEC(now) < st->sleep_until)
                     return;     /* Still sleeping, nothing to do here */
                 /* Else done sleeping. */
                 st->state = CSTATE_END_COMMAND;
                 break;
 
                 /*
                  * End of command: record stats and proceed to next command.
                  */
             case CSTATE_END_COMMAND:
 
                 /*
                  * command completed: accumulate per-command execution times
                  * in thread-local data structure, if per-command latencies
                  * are requested.
                  */
                 if (is_latencies)
                 {
                     if (INSTR_TIME_IS_ZERO(now))
                         INSTR_TIME_SET_CURRENT(now);
 
                     /* XXX could use a mutex here, but we choose not to */
                     command = sql_script[st->use_file].commands[st->command];
                     addToSimpleStats(&command->stats,
                                      INSTR_TIME_GET_DOUBLE(now) -
                                      INSTR_TIME_GET_DOUBLE(st->stmt_begin));
                 }
 
                 /* Go ahead with next command */
                 st->command++;
                 st->state = CSTATE_START_COMMAND;
                 break;
 
                 /*
                  * End of transaction.
                  */
             case CSTATE_END_TX:
 
                 /*
                  * transaction finished: calculate latency and log the
                  * transaction
                  */
                 if (progress || throttle_delay || latency_limit ||
                     per_script_stats || use_log)
                     processXactStats(thread, st, &now, false, agg);
                 else
                     thread->stats.cnt++;
 
                 if (is_connect)
                 {
                     PQfinish(st->con);
                     st->con = NULL;
                     INSTR_TIME_SET_ZERO(now);
                 }
 
                 ++st->cnt;
                 if ((st->cnt >= nxacts && duration <= 0) || timer_exceeded)
                 {
                     /* exit success */
                     st->state = CSTATE_FINISHED;
                     break;
                 }
 
                 /*
                  * No transaction is underway anymore.
                  */
                 st->state = CSTATE_CHOOSE_SCRIPT;
 
                 /*
                  * If we paced through all commands in the script in this
                  * loop, without returning to the caller even once, do it now.
                  * This gives the thread a chance to process other
                  * connections, and to do progress reporting.  This can
                  * currently only happen if the script consists entirely of
                  * meta-commands.
                  */
                 if (end_tx_processed)
                     return;
                 else
                 {
                     end_tx_processed = true;
                     break;
                 }
 
                 /*
                  * Final states.  Close the connection if it's still open.
                  */
             case CSTATE_ABORTED:
             case CSTATE_FINISHED:
                 if (st->con != NULL)
                 {
                     PQfinish(st->con);
                     st->con = NULL;
                 }
                 return;
         }
     }
 }

static void doLog	(	TState *	thread,
		CState *	st,
		StatsData *	agg,
		bool	skipped,
		double	latency,
		double	lag
	)

static

Definition at line 2437 of file pgbench.c.

References accumStats(), Assert, StatsData::cnt, CState::cnt, gettimeofday(), CState::id, initStats(), INT64_FORMAT, StatsData::lag, StatsData::latency, logfile, TState::logfile, SimpleStats::max, SimpleStats::min, now(), NULL, pg_erand48(), TState::random_state, StatsData::skipped, StatsData::start_time, SimpleStats::sum, SimpleStats::sum2, and CState::use_file.

Referenced by processXactStats(), and threadRun().

 {
     FILE       *logfile = thread->logfile;
 
     Assert(use_log);
 
     /*
      * Skip the log entry if sampling is enabled and this row doesn't belong
      * to the random sample.
      */
     if (sample_rate != 0.0 &&
         pg_erand48(thread->random_state) > sample_rate)
         return;
 
     /* should we aggregate the results or not? */
     if (agg_interval > 0)
     {
         /*
          * Loop until we reach the interval of the current moment, and print
          * any empty intervals in between (this may happen with very low tps,
          * e.g. --rate=0.1).
          */
         time_t      now = time(NULL);
 
         while (agg->start_time + agg_interval <= now)
         {
             /* print aggregated report to logfile */
             fprintf(logfile, "%ld " INT64_FORMAT " %.0f %.0f %.0f %.0f",
                     (long) agg->start_time,
                     agg->cnt,
                     agg->latency.sum,
                     agg->latency.sum2,
                     agg->latency.min,
                     agg->latency.max);
             if (throttle_delay)
             {
                 fprintf(logfile, " %.0f %.0f %.0f %.0f",
                         agg->lag.sum,
                         agg->lag.sum2,
                         agg->lag.min,
                         agg->lag.max);
                 if (latency_limit)
                     fprintf(logfile, " " INT64_FORMAT, agg->skipped);
             }
             fputc('\n', logfile);
 
             /* reset data and move to next interval */
             initStats(agg, agg->start_time + agg_interval);
         }
 
         /* accumulate the current transaction */
         accumStats(agg, skipped, latency, lag);
     }
     else
     {
         /* no, print raw transactions */
         struct timeval tv;
 
         gettimeofday(&tv, NULL);
         if (skipped)
             fprintf(logfile, "%d " INT64_FORMAT " skipped %d %ld %ld",
                     st->id, st->cnt, st->use_file,
                     (long) tv.tv_sec, (long) tv.tv_usec);
         else
             fprintf(logfile, "%d " INT64_FORMAT " %.0f %d %ld %ld",
                     st->id, st->cnt, latency, st->use_file,
                     (long) tv.tv_sec, (long) tv.tv_usec);
         if (throttle_delay)
             fprintf(logfile, " %.0f", lag);
         fputc('\n', logfile);
     }
 }

static bool evalFunc	(	TState *	thread,
		CState *	st,
		PgBenchFunction	func,
		PgBenchExprLink *	args,
		PgBenchValue *	retval
	)

static

Definition at line 1292 of file pgbench.c.

References generate_unaccent_rules::args, Assert, coerceToDouble(), coerceToInt(), CState::command, PgBenchValue::dval, evaluateExpr(), getExponentialRand(), getGaussianRand(), getrand(), i, INT64_FORMAT, PgBenchValue::ival, M_PI, Max, MAX_FARGS, Min, MIN_GAUSSIAN_PARAM, PgBenchExprLink::next, NULL, PG_INT64_MIN, PGBENCH_ABS, PGBENCH_ADD, PGBENCH_DEBUG, PGBENCH_DIV, PGBENCH_DOUBLE, PGBENCH_GREATEST, PGBENCH_INT, PGBENCH_LEAST, PGBENCH_MOD, PGBENCH_MUL, PGBENCH_PI, PGBENCH_RANDOM, PGBENCH_RANDOM_EXPONENTIAL, PGBENCH_RANDOM_GAUSSIAN, PGBENCH_SQRT, PGBENCH_SUB, PGBT_DOUBLE, PGBT_INT, setDoubleValue(), setIntValue(), PgBenchValue::type, PgBenchValue::u, and CState::use_file.

Referenced by evaluateExpr().

 {
     /* evaluate all function arguments */
     int         nargs = 0;
     PgBenchValue vargs[MAX_FARGS];
     PgBenchExprLink *l = args;
 
     for (nargs = 0; nargs < MAX_FARGS && l != NULL; nargs++, l = l->next)
         if (!evaluateExpr(thread, st, l->expr, &vargs[nargs]))
             return false;
 
     if (l != NULL)
     {
         fprintf(stderr,
                 "too many function arguments, maximum is %d\n", MAX_FARGS);
         return false;
     }
 
     /* then evaluate function */
     switch (func)
     {
             /* overloaded operators */
         case PGBENCH_ADD:
         case PGBENCH_SUB:
         case PGBENCH_MUL:
         case PGBENCH_DIV:
         case PGBENCH_MOD:
             {
                 PgBenchValue *lval = &vargs[0],
                            *rval = &vargs[1];
 
                 Assert(nargs == 2);
 
                 /* overloaded type management, double if some double */
                 if ((lval->type == PGBT_DOUBLE ||
                      rval->type == PGBT_DOUBLE) && func != PGBENCH_MOD)
                 {
                     double      ld,
                                 rd;
 
                     if (!coerceToDouble(lval, &ld) ||
                         !coerceToDouble(rval, &rd))
                         return false;
 
                     switch (func)
                     {
                         case PGBENCH_ADD:
                             setDoubleValue(retval, ld + rd);
                             return true;
 
                         case PGBENCH_SUB:
                             setDoubleValue(retval, ld - rd);
                             return true;
 
                         case PGBENCH_MUL:
                             setDoubleValue(retval, ld * rd);
                             return true;
 
                         case PGBENCH_DIV:
                             setDoubleValue(retval, ld / rd);
                             return true;
 
                         default:
                             /* cannot get here */
                             Assert(0);
                     }
                 }
                 else    /* we have integer operands, or % */
                 {
                     int64       li,
                                 ri;
 
                     if (!coerceToInt(lval, &li) ||
                         !coerceToInt(rval, &ri))
                         return false;
 
                     switch (func)
                     {
                         case PGBENCH_ADD:
                             setIntValue(retval, li + ri);
                             return true;
 
                         case PGBENCH_SUB:
                             setIntValue(retval, li - ri);
                             return true;
 
                         case PGBENCH_MUL:
                             setIntValue(retval, li * ri);
                             return true;
 
                         case PGBENCH_DIV:
                         case PGBENCH_MOD:
                             if (ri == 0)
                             {
                                 fprintf(stderr, "division by zero\n");
                                 return false;
                             }
                             /* special handling of -1 divisor */
                             if (ri == -1)
                             {
                                 if (func == PGBENCH_DIV)
                                 {
                                     /* overflow check (needed for INT64_MIN) */
                                     if (li == PG_INT64_MIN)
                                     {
                                         fprintf(stderr, "bigint out of range\n");
                                         return false;
                                     }
                                     else
                                         setIntValue(retval, -li);
                                 }
                                 else
                                     setIntValue(retval, 0);
                                 return true;
                             }
                             /* else divisor is not -1 */
                             if (func == PGBENCH_DIV)
                                 setIntValue(retval, li / ri);
                             else    /* func == PGBENCH_MOD */
                                 setIntValue(retval, li % ri);
 
                             return true;
 
                         default:
                             /* cannot get here */
                             Assert(0);
                     }
                 }
             }
 
             /* no arguments */
         case PGBENCH_PI:
             setDoubleValue(retval, M_PI);
             return true;
 
             /* 1 overloaded argument */
         case PGBENCH_ABS:
             {
                 PgBenchValue *varg = &vargs[0];
 
                 Assert(nargs == 1);
 
                 if (varg->type == PGBT_INT)
                 {
                     int64       i = varg->u.ival;
 
                     setIntValue(retval, i < 0 ? -i : i);
                 }
                 else
                 {
                     double      d = varg->u.dval;
 
                     Assert(varg->type == PGBT_DOUBLE);
                     setDoubleValue(retval, d < 0.0 ? -d : d);
                 }
 
                 return true;
             }
 
         case PGBENCH_DEBUG:
             {
                 PgBenchValue *varg = &vargs[0];
 
                 Assert(nargs == 1);
 
                 fprintf(stderr, "debug(script=%d,command=%d): ",
                         st->use_file, st->command + 1);
 
                 if (varg->type == PGBT_INT)
                     fprintf(stderr, "int " INT64_FORMAT "\n", varg->u.ival);
                 else
                 {
                     Assert(varg->type == PGBT_DOUBLE);
                     fprintf(stderr, "double %.*g\n", DBL_DIG, varg->u.dval);
                 }
 
                 *retval = *varg;
 
                 return true;
             }
 
             /* 1 double argument */
         case PGBENCH_DOUBLE:
         case PGBENCH_SQRT:
             {
                 double      dval;
 
                 Assert(nargs == 1);
 
                 if (!coerceToDouble(&vargs[0], &dval))
                     return false;
 
                 if (func == PGBENCH_SQRT)
                     dval = sqrt(dval);
 
                 setDoubleValue(retval, dval);
                 return true;
             }
 
             /* 1 int argument */
         case PGBENCH_INT:
             {
                 int64       ival;
 
                 Assert(nargs == 1);
 
                 if (!coerceToInt(&vargs[0], &ival))
                     return false;
 
                 setIntValue(retval, ival);
                 return true;
             }
 
             /* variable number of arguments */
         case PGBENCH_LEAST:
         case PGBENCH_GREATEST:
             {
                 bool        havedouble;
                 int         i;
 
                 Assert(nargs >= 1);
 
                 /* need double result if any input is double */
                 havedouble = false;
                 for (i = 0; i < nargs; i++)
                 {
                     if (vargs[i].type == PGBT_DOUBLE)
                     {
                         havedouble = true;
                         break;
                     }
                 }
                 if (havedouble)
                 {
                     double      extremum;
 
                     if (!coerceToDouble(&vargs[0], &extremum))
                         return false;
                     for (i = 1; i < nargs; i++)
                     {
                         double      dval;
 
                         if (!coerceToDouble(&vargs[i], &dval))
                             return false;
                         if (func == PGBENCH_LEAST)
                             extremum = Min(extremum, dval);
                         else
                             extremum = Max(extremum, dval);
                     }
                     setDoubleValue(retval, extremum);
                 }
                 else
                 {
                     int64       extremum;
 
                     if (!coerceToInt(&vargs[0], &extremum))
                         return false;
                     for (i = 1; i < nargs; i++)
                     {
                         int64       ival;
 
                         if (!coerceToInt(&vargs[i], &ival))
                             return false;
                         if (func == PGBENCH_LEAST)
                             extremum = Min(extremum, ival);
                         else
                             extremum = Max(extremum, ival);
                     }
                     setIntValue(retval, extremum);
                 }
                 return true;
             }
 
             /* random functions */
         case PGBENCH_RANDOM:
         case PGBENCH_RANDOM_EXPONENTIAL:
         case PGBENCH_RANDOM_GAUSSIAN:
             {
                 int64       imin,
                             imax;
 
                 Assert(nargs >= 2);
 
                 if (!coerceToInt(&vargs[0], &imin) ||
                     !coerceToInt(&vargs[1], &imax))
                     return false;
 
                 /* check random range */
                 if (imin > imax)
                 {
                     fprintf(stderr, "empty range given to random\n");
                     return false;
                 }
                 else if (imax - imin < 0 || (imax - imin) + 1 < 0)
                 {
                     /* prevent int overflows in random functions */
                     fprintf(stderr, "random range is too large\n");
                     return false;
                 }
 
                 if (func == PGBENCH_RANDOM)
                 {
                     Assert(nargs == 2);
                     setIntValue(retval, getrand(thread, imin, imax));
                 }
                 else    /* gaussian & exponential */
                 {
                     double      param;
 
                     Assert(nargs == 3);
 
                     if (!coerceToDouble(&vargs[2], &param))
                         return false;
 
                     if (func == PGBENCH_RANDOM_GAUSSIAN)
                     {
                         if (param < MIN_GAUSSIAN_PARAM)
                         {
                             fprintf(stderr,
                                     "gaussian parameter must be at least %f "
                                     "(not %f)\n", MIN_GAUSSIAN_PARAM, param);
                             return false;
                         }
 
                         setIntValue(retval,
                                  getGaussianRand(thread, imin, imax, param));
                     }
                     else    /* exponential */
                     {
                         if (param <= 0.0)
                         {
                             fprintf(stderr,
                             "exponential parameter must be greater than zero"
                                     " (got %f)\n", param);
                             return false;
                         }
 
                         setIntValue(retval,
                               getExponentialRand(thread, imin, imax, param));
                     }
                 }
 
                 return true;
             }
 
         default:
             /* cannot get here */
             Assert(0);
             /* dead code to avoid a compiler warning */
             return false;
     }
 }

static bool evaluateExpr	(	TState *	thread,
		CState *	st,
		PgBenchExpr *	expr,
		PgBenchValue *	retval
	)

static

Definition at line 1653 of file pgbench.c.

References PgBenchExpr::constant, ENODE_CONSTANT, ENODE_FUNCTION, ENODE_VARIABLE, PgBenchExpr::etype, evalFunc(), PgBenchExpr::function, lookupVariable(), makeVariableNumeric(), NULL, Variable::num_value, PgBenchExpr::u, and PgBenchExpr::variable.

Referenced by doCustom(), and evalFunc().

 {
     switch (expr->etype)
     {
         case ENODE_CONSTANT:
             {
                 *retval = expr->u.constant;
                 return true;
             }
 
         case ENODE_VARIABLE:
             {
                 Variable   *var;
 
                 if ((var = lookupVariable(st, expr->u.variable.varname)) == NULL)
                 {
                     fprintf(stderr, "undefined variable \"%s\"\n",
                             expr->u.variable.varname);
                     return false;
                 }
 
                 if (!makeVariableNumeric(var))
                     return false;
 
                 *retval = var->num_value;
                 return true;
             }
 
         case ENODE_FUNCTION:
             return evalFunc(thread, st,
                             expr->u.function.function,
                             expr->u.function.args,
                             retval);
 
         default:
             /* internal error which should never occur */
             fprintf(stderr, "unexpected enode type in evaluation: %d\n",
                     expr->etype);
             exit(1);
     }
 }

static bool evaluateSleep	(	CState *	st,
		int	argc,
		char **	argv,
		int *	usecs
	)

static

Definition at line 1922 of file pgbench.c.

References getVariable(), NULL, and pg_strcasecmp().

Referenced by doCustom().

 {
     char       *var;
     int         usec;
 
     if (*argv[1] == ':')
     {
         if ((var = getVariable(st, argv[1] + 1)) == NULL)
         {
             fprintf(stderr, "%s: undefined variable \"%s\"\n",
                     argv[0], argv[1]);
             return false;
         }
         usec = atoi(var);
     }
     else
         usec = atoi(argv[1]);
 
     if (argc > 2)
     {
         if (pg_strcasecmp(argv[2], "ms") == 0)
             usec *= 1000;
         else if (pg_strcasecmp(argv[2], "s") == 0)
             usec *= 1000000;
     }
     else
         usec *= 1000000;
 
     *usecs = usec;
     return true;
 }

static void executeStatement	(	PGconn *	con,
		const char *	sql
	)

static

Definition at line 818 of file pgbench.c.

References PGRES_COMMAND_OK, PQclear(), PQerrorMessage(), PQexec(), and PQresultStatus().

Referenced by init().

 {
     PGresult   *res;
 
     res = PQexec(con, sql);
     if (PQresultStatus(res) != PGRES_COMMAND_OK)
     {
         fprintf(stderr, "%s", PQerrorMessage(con));
         exit(1);
     }
     PQclear(res);
 }

static const BuiltinScript* findBuiltin ( const char * name )

static

Definition at line 3377 of file pgbench.c.

References i, lengthof, listAvailableScripts(), NULL, and result.

Referenced by main().

 {
     int         i,
                 found = 0,
                 len = strlen(name);
     const BuiltinScript *result = NULL;
 
     for (i = 0; i < lengthof(builtin_script); i++)
     {
         if (strncmp(builtin_script[i].name, name, len) == 0)
         {
             result = &builtin_script[i];
             found++;
         }
     }
 
     /* ok, unambiguous result */
     if (found == 1)
         return result;
 
     /* error cases */
     if (found == 0)
         fprintf(stderr, "no builtin script found for name \"%s\"\n", name);
     else    /* found > 1 */
         fprintf(stderr,
                 "ambiguous builtin name: %d builtin scripts found for prefix \"%s\"\n", found, name);
 
     listAvailableScripts();
     exit(1);
 }

static int64 getExponentialRand	(	TState *	thread,
		int64	min,
		int64	max,
		double	parameter
	)

static

Definition at line 647 of file pgbench.c.

References Assert, pg_erand48(), and TState::random_state.

Referenced by evalFunc().

 {
     double      cut,
                 uniform,
                 rand;
 
     /* abort if wrong parameter, but must really be checked beforehand */
     Assert(parameter > 0.0);
     cut = exp(-parameter);
     /* erand in [0, 1), uniform in (0, 1] */
     uniform = 1.0 - pg_erand48(thread->random_state);
 
     /*
      * inner expression in (cut, 1] (if parameter > 0), rand in [0, 1)
      */
     Assert((1.0 - cut) != 0.0);
     rand = -log(cut + (1.0 - cut) * uniform) / parameter;
     /* return int64 random number within between min and max */
     return min + (int64) ((max - min + 1) * rand);
 }

static int64 getGaussianRand	(	TState *	thread,
		int64	min,
		int64	max,
		double	parameter
	)

static

Definition at line 670 of file pgbench.c.

References Assert, M_PI, MIN_GAUSSIAN_PARAM, pg_erand48(), and TState::random_state.

Referenced by evalFunc().

 {
     double      stdev;
     double      rand;
 
     /* abort if parameter is too low, but must really be checked beforehand */
     Assert(parameter >= MIN_GAUSSIAN_PARAM);
 
     /*
      * Get user specified random number from this loop, with -parameter <
      * stdev <= parameter
      *
      * This loop is executed until the number is in the expected range.
      *
      * As the minimum parameter is 2.0, the probability of looping is low:
      * sqrt(-2 ln(r)) <= 2 => r >= e^{-2} ~ 0.135, then when taking the
      * average sinus multiplier as 2/pi, we have a 8.6% looping probability in
      * the worst case. For a parameter value of 5.0, the looping probability
      * is about e^{-5} * 2 / pi ~ 0.43%.
      */
     do
     {
         /*
          * pg_erand48 generates [0,1), but for the basic version of the
          * Box-Muller transform the two uniformly distributed random numbers
          * are expected in (0, 1] (see
          * http://en.wikipedia.org/wiki/Box_muller)
          */
         double      rand1 = 1.0 - pg_erand48(thread->random_state);
         double      rand2 = 1.0 - pg_erand48(thread->random_state);
 
         /* Box-Muller basic form transform */
         double      var_sqrt = sqrt(-2.0 * log(rand1));
 
         stdev = var_sqrt * sin(2.0 * M_PI * rand2);
 
         /*
          * we may try with cos, but there may be a bias induced if the
          * previous value fails the test. To be on the safe side, let us try
          * over.
          */
     }
     while (stdev < -parameter || stdev >= parameter);
 
     /* stdev is in [-parameter, parameter), normalization to [0,1) */
     rand = (stdev + parameter) / (parameter * 2.0);
 
     /* return int64 random number within between min and max */
     return min + (int64) ((max - min + 1) * rand);
 }

static int64 getPoissonRand	(	TState *	thread,
		int64	center
	)

static

Definition at line 726 of file pgbench.c.

References pg_erand48(), and TState::random_state.

Referenced by doCustom().

 {
     /*
      * Use inverse transform sampling to generate a value > 0, such that the
      * expected (i.e. average) value is the given argument.
      */
     double      uniform;
 
     /* erand in [0, 1), uniform in (0, 1] */
     uniform = 1.0 - pg_erand48(thread->random_state);
 
     return (int64) (-log(uniform) * ((double) center) + 0.5);
 }

static void getQueryParams	(	CState *	st,
		const Command *	command,
		const char **	params
	)

static

Definition at line 1220 of file pgbench.c.

References Command::argc, Command::argv, getVariable(), and i.

Referenced by sendCommand().

 {
     int         i;
 
     for (i = 0; i < command->argc - 1; i++)
         params[i] = getVariable(st, command->argv[i + 1]);
 }

static int64 getrand	(	TState *	thread,
		int64	min,
		int64	max
	)

static

Definition at line 627 of file pgbench.c.

References pg_erand48(), and TState::random_state.

Referenced by chooseScript(), and evalFunc().

 {
     /*
      * Odd coding is so that min and max have approximately the same chance of
      * being selected as do numbers between them.
      *
      * pg_erand48() is thread-safe and concurrent, which is why we use it
      * rather than random(), which in glibc is non-reentrant, and therefore
      * protected by a mutex, and therefore a bottleneck on machines with many
      * CPUs.
      */
     return min + (int64) ((max - min + 1) * pg_erand48(thread->random_state));
 }

static char* getVariable	(	CState *	st,
		char *	name
	)

static

Definition at line 966 of file pgbench.c.

References Assert, PgBenchValue::dval, INT64_FORMAT, Variable::is_numeric, PgBenchValue::ival, lookupVariable(), NULL, Variable::num_value, pg_strdup(), PGBT_DOUBLE, PGBT_INT, snprintf(), PgBenchValue::type, PgBenchValue::u, and Variable::value.

Referenced by assignVariables(), evaluateSleep(), getQueryParams(), and runShellCommand().

 {
     Variable   *var;
     char        stringform[64];
 
     var = lookupVariable(st, name);
     if (var == NULL)
         return NULL;            /* not found */
 
     if (var->value)
         return var->value;      /* we have it in string form */
 
     /* We need to produce a string equivalent of the numeric value */
     Assert(var->is_numeric);
     if (var->num_value.type == PGBT_INT)
         snprintf(stringform, sizeof(stringform),
                  INT64_FORMAT, var->num_value.u.ival);
     else
     {
         Assert(var->num_value.type == PGBT_DOUBLE);
         snprintf(stringform, sizeof(stringform),
                  "%.*g", DBL_DIG, var->num_value.u.dval);
     }
     var->value = pg_strdup(stringform);
     return var->value;
 }

static void handle_sig_alarm ( SIGNAL_ARGS )

static

Definition at line 4724 of file pgbench.c.

Referenced by setalarm().

 {
     timer_exceeded = true;
 }

static void init ( bool is_no_vacuum )

static

Definition at line 2571 of file pgbench.c.

References buffer, doConnect(), executeStatement(), i, INSTR_TIME_GET_DOUBLE, INSTR_TIME_SET_CURRENT, INSTR_TIME_SUBTRACT, INT64_FORMAT, lengthof, LOG_STEP_SECONDS, naccounts, nbranches, ntellers, NULL, PGRES_COPY_IN, PQclear(), PQendcopy(), PQerrorMessage(), PQescapeIdentifier(), PQexec(), PQfinish(), PQfreemem(), PQputline(), PQresultStatus(), scale, SCALE_32BIT_THRESHOLD, snprintf(), and strlcpy().

Referenced by from_char_parse_int_len(), heap_multi_insert(), log_heap_update(), and main().

 {
 /*
  * The scale factor at/beyond which 32-bit integers are insufficient for
  * storing TPC-B account IDs.
  *
  * Although the actual threshold is 21474, we use 20000 because it is easier to
  * document and remember, and isn't that far away from the real threshold.
  */
 #define SCALE_32BIT_THRESHOLD 20000
 
     /*
      * Note: TPC-B requires at least 100 bytes per row, and the "filler"
      * fields in these table declarations were intended to comply with that.
      * The pgbench_accounts table complies with that because the "filler"
      * column is set to blank-padded empty string. But for all other tables
      * the columns default to NULL and so don't actually take any space.  We
      * could fix that by giving them non-null default values.  However, that
      * would completely break comparability of pgbench results with prior
      * versions. Since pgbench has never pretended to be fully TPC-B compliant
      * anyway, we stick with the historical behavior.
      */
     struct ddlinfo
     {
         const char *table;      /* table name */
         const char *smcols;     /* column decls if accountIDs are 32 bits */
         const char *bigcols;    /* column decls if accountIDs are 64 bits */
         int         declare_fillfactor;
     };
     static const struct ddlinfo DDLs[] = {
         {
             "pgbench_history",
             "tid int,bid int,aid    int,delta int,mtime timestamp,filler char(22)",
             "tid int,bid int,aid bigint,delta int,mtime timestamp,filler char(22)",
             0
         },
         {
             "pgbench_tellers",
             "tid int not null,bid int,tbalance int,filler char(84)",
             "tid int not null,bid int,tbalance int,filler char(84)",
             1
         },
         {
             "pgbench_accounts",
             "aid    int not null,bid int,abalance int,filler char(84)",
             "aid bigint not null,bid int,abalance int,filler char(84)",
             1
         },
         {
             "pgbench_branches",
             "bid int not null,bbalance int,filler char(88)",
             "bid int not null,bbalance int,filler char(88)",
             1
         }
     };
     static const char *const DDLINDEXes[] = {
         "alter table pgbench_branches add primary key (bid)",
         "alter table pgbench_tellers add primary key (tid)",
         "alter table pgbench_accounts add primary key (aid)"
     };
     static const char *const DDLKEYs[] = {
         "alter table pgbench_tellers add foreign key (bid) references pgbench_branches",
         "alter table pgbench_accounts add foreign key (bid) references pgbench_branches",
         "alter table pgbench_history add foreign key (bid) references pgbench_branches",
         "alter table pgbench_history add foreign key (tid) references pgbench_tellers",
         "alter table pgbench_history add foreign key (aid) references pgbench_accounts"
     };
 
     PGconn     *con;
     PGresult   *res;
     char        sql[256];
     int         i;
     int64       k;
 
     /* used to track elapsed time and estimate of the remaining time */
     instr_time  start,
                 diff;
     double      elapsed_sec,
                 remaining_sec;
     int         log_interval = 1;
 
     if ((con = doConnect()) == NULL)
         exit(1);
 
     for (i = 0; i < lengthof(DDLs); i++)
     {
         char        opts[256];
         char        buffer[256];
         const struct ddlinfo *ddl = &DDLs[i];
         const char *cols;
 
         /* Remove old table, if it exists. */
         snprintf(buffer, sizeof(buffer), "drop table if exists %s", ddl->table);
         executeStatement(con, buffer);
 
         /* Construct new create table statement. */
         opts[0] = '\0';
         if (ddl->declare_fillfactor)
             snprintf(opts + strlen(opts), sizeof(opts) - strlen(opts),
                      " with (fillfactor=%d)", fillfactor);
         if (tablespace != NULL)
         {
             char       *escape_tablespace;
 
             escape_tablespace = PQescapeIdentifier(con, tablespace,
                                                    strlen(tablespace));
             snprintf(opts + strlen(opts), sizeof(opts) - strlen(opts),
                      " tablespace %s", escape_tablespace);
             PQfreemem(escape_tablespace);
         }
 
         cols = (scale >= SCALE_32BIT_THRESHOLD) ? ddl->bigcols : ddl->smcols;
 
         snprintf(buffer, sizeof(buffer), "create%s table %s(%s)%s",
                  unlogged_tables ? " unlogged" : "",
                  ddl->table, cols, opts);
 
         executeStatement(con, buffer);
     }
 
     executeStatement(con, "begin");
 
     for (i = 0; i < nbranches * scale; i++)
     {
         /* "filler" column defaults to NULL */
         snprintf(sql, sizeof(sql),
                  "insert into pgbench_branches(bid,bbalance) values(%d,0)",
                  i + 1);
         executeStatement(con, sql);
     }
 
     for (i = 0; i < ntellers * scale; i++)
     {
         /* "filler" column defaults to NULL */
         snprintf(sql, sizeof(sql),
             "insert into pgbench_tellers(tid,bid,tbalance) values (%d,%d,0)",
                  i + 1, i / ntellers + 1);
         executeStatement(con, sql);
     }
 
     executeStatement(con, "commit");
 
     /*
      * fill the pgbench_accounts table with some data
      */
     fprintf(stderr, "creating tables...\n");
 
     executeStatement(con, "begin");
     executeStatement(con, "truncate pgbench_accounts");
 
     res = PQexec(con, "copy pgbench_accounts from stdin");
     if (PQresultStatus(res) != PGRES_COPY_IN)
     {
         fprintf(stderr, "%s", PQerrorMessage(con));
         exit(1);
     }
     PQclear(res);
 
     INSTR_TIME_SET_CURRENT(start);
 
     for (k = 0; k < (int64) naccounts * scale; k++)
     {
         int64       j = k + 1;
 
         /* "filler" column defaults to blank padded empty string */
         snprintf(sql, sizeof(sql),
                  INT64_FORMAT "\t" INT64_FORMAT "\t%d\t\n",
                  j, k / naccounts + 1, 0);
         if (PQputline(con, sql))
         {
             fprintf(stderr, "PQputline failed\n");
             exit(1);
         }
 
         /*
          * If we want to stick with the original logging, print a message each
          * 100k inserted rows.
          */
         if ((!use_quiet) && (j % 100000 == 0))
         {
             INSTR_TIME_SET_CURRENT(diff);
             INSTR_TIME_SUBTRACT(diff, start);
 
             elapsed_sec = INSTR_TIME_GET_DOUBLE(diff);
             remaining_sec = ((double) scale * naccounts - j) * elapsed_sec / j;
 
             fprintf(stderr, INT64_FORMAT " of " INT64_FORMAT " tuples (%d%%) done (elapsed %.2f s, remaining %.2f s)\n",
                     j, (int64) naccounts * scale,
                     (int) (((int64) j * 100) / (naccounts * (int64) scale)),
                     elapsed_sec, remaining_sec);
         }
         /* let's not call the timing for each row, but only each 100 rows */
         else if (use_quiet && (j % 100 == 0))
         {
             INSTR_TIME_SET_CURRENT(diff);
             INSTR_TIME_SUBTRACT(diff, start);
 
             elapsed_sec = INSTR_TIME_GET_DOUBLE(diff);
             remaining_sec = ((double) scale * naccounts - j) * elapsed_sec / j;
 
             /* have we reached the next interval (or end)? */
             if ((j == scale * naccounts) || (elapsed_sec >= log_interval * LOG_STEP_SECONDS))
             {
                 fprintf(stderr, INT64_FORMAT " of " INT64_FORMAT " tuples (%d%%) done (elapsed %.2f s, remaining %.2f s)\n",
                         j, (int64) naccounts * scale,
                         (int) (((int64) j * 100) / (naccounts * (int64) scale)), elapsed_sec, remaining_sec);
 
                 /* skip to the next interval */
                 log_interval = (int) ceil(elapsed_sec / LOG_STEP_SECONDS);
             }
         }
 
     }
     if (PQputline(con, "\\.\n"))
     {
         fprintf(stderr, "very last PQputline failed\n");
         exit(1);
     }
     if (PQendcopy(con))
     {
         fprintf(stderr, "PQendcopy failed\n");
         exit(1);
     }
     executeStatement(con, "commit");
 
     /* vacuum */
     if (!is_no_vacuum)
     {
         fprintf(stderr, "vacuum...\n");
         executeStatement(con, "vacuum analyze pgbench_branches");
         executeStatement(con, "vacuum analyze pgbench_tellers");
         executeStatement(con, "vacuum analyze pgbench_accounts");
         executeStatement(con, "vacuum analyze pgbench_history");
     }
 
     /*
      * create indexes
      */
     fprintf(stderr, "set primary keys...\n");
     for (i = 0; i < lengthof(DDLINDEXes); i++)
     {
         char        buffer[256];
 
         strlcpy(buffer, DDLINDEXes[i], sizeof(buffer));
 
         if (index_tablespace != NULL)
         {
             char       *escape_tablespace;
 
             escape_tablespace = PQescapeIdentifier(con, index_tablespace,
                                                    strlen(index_tablespace));
             snprintf(buffer + strlen(buffer), sizeof(buffer) - strlen(buffer),
                      " using index tablespace %s", escape_tablespace);
             PQfreemem(escape_tablespace);
         }
 
         executeStatement(con, buffer);
     }
 
     /*
      * create foreign keys
      */
     if (foreign_keys)
     {
         fprintf(stderr, "set foreign keys...\n");
         for (i = 0; i < lengthof(DDLKEYs); i++)
         {
             executeStatement(con, DDLKEYs[i]);
         }
     }
 
     fprintf(stderr, "done.\n");
     PQfinish(con);
 }

static void initSimpleStats ( SimpleStats * ss )

static

Definition at line 744 of file pgbench.c.

Referenced by initStats(), process_backslash_command(), and process_sql_command().

 {
     memset(ss, 0, sizeof(SimpleStats));
 }

static void initStats	(	StatsData *	sd,
		time_t	start_time
	)

static

Definition at line 784 of file pgbench.c.

References StatsData::cnt, initSimpleStats(), StatsData::lag, StatsData::latency, StatsData::skipped, start_time, and StatsData::start_time.

Referenced by doLog(), main(), ParseScript(), and threadRun().

 {
     sd->start_time = start_time;
     sd->cnt = 0;
     sd->skipped = 0;
     initSimpleStats(&sd->latency);
     initSimpleStats(&sd->lag);
 }

static bool is_an_int ( const char * str )

static

Definition at line 532 of file pgbench.c.

Referenced by makeVariableNumeric().

 {
     const char *ptr = str;
 
     /* skip leading spaces; cast is consistent with strtoint64 */
     while (*ptr && isspace((unsigned char) *ptr))
         ptr++;
 
     /* skip sign */
     if (*ptr == '+' || *ptr == '-')
         ptr++;
 
     /* at least one digit */
     if (*ptr && !isdigit((unsigned char) *ptr))
         return false;
 
     /* eat all digits */
     while (*ptr && isdigit((unsigned char) *ptr))
         ptr++;
 
     /* must have reached end of string */
     return *ptr == '\0';
 }

static bool isLegalVariableName ( const char * name )

static

Definition at line 1025 of file pgbench.c.

References i.

Referenced by lookupCreateVariable().

 {
     int         i;
 
     for (i = 0; name[i] != '\0'; i++)
     {
         if (!isalnum((unsigned char) name[i]) && name[i] != '_')
             return false;
     }
 
     return (i > 0);             /* must be non-empty */
 }

static void listAvailableScripts ( void )

static

Definition at line 3365 of file pgbench.c.

References i, lengthof, and name.

Referenced by findBuiltin(), and main().

 {
     int         i;
 
     fprintf(stderr, "Available builtin scripts:\n");
     for (i = 0; i < lengthof(builtin_script); i++)
         fprintf(stderr, "\t%s\n", builtin_script[i].name);
     fprintf(stderr, "\n");
 }

static Variable* lookupCreateVariable	(	CState *	st,
		const char *	context,
		char *	name
	)

static

Definition at line 1044 of file pgbench.c.

References isLegalVariableName(), lookupVariable(), Variable::name, NULL, CState::nvariables, pg_malloc(), pg_realloc(), pg_strdup(), Variable::value, CState::variables, and CState::vars_sorted.

Referenced by putVariable(), and putVariableNumber().

 {
     Variable   *var;
 
     var = lookupVariable(st, name);
     if (var == NULL)
     {
         Variable   *newvars;
 
         /*
          * Check for the name only when declaring a new variable to avoid
          * overhead.
          */
         if (!isLegalVariableName(name))
         {
             fprintf(stderr, "%s: invalid variable name: \"%s\"\n",
                     context, name);
             return NULL;
         }
 
         /* Create variable at the end of the array */
         if (st->variables)
             newvars = (Variable *) pg_realloc(st->variables,
                                     (st->nvariables + 1) * sizeof(Variable));
         else
             newvars = (Variable *) pg_malloc(sizeof(Variable));
 
         st->variables = newvars;
 
         var = &newvars[st->nvariables];
 
         var->name = pg_strdup(name);
         var->value = NULL;
         /* caller is expected to initialize remaining fields */
 
         st->nvariables++;
         /* we don't re-sort the array till we have to */
         st->vars_sorted = false;
     }
 
     return var;
 }

static Variable* lookupVariable	(	CState *	st,
		char *	name
	)

static

Definition at line 939 of file pgbench.c.

References compareVariableNames(), Variable::name, name, NULL, CState::nvariables, qsort, CState::variables, and CState::vars_sorted.

Referenced by evaluateExpr(), getVariable(), lookupCreateVariable(), and main().

 {
     Variable    key;
 
     /* On some versions of Solaris, bsearch of zero items dumps core */
     if (st->nvariables <= 0)
         return NULL;
 
     /* Sort if we have to */
     if (!st->vars_sorted)
     {
         qsort((void *) st->variables, st->nvariables, sizeof(Variable),
               compareVariableNames);
         st->vars_sorted = true;
     }
 
     /* Now we can search */
     key.name = name;
     return (Variable *) bsearch((void *) &key,
                                 (void *) st->variables,
                                 st->nvariables,
                                 sizeof(Variable),
                                 compareVariableNames);
 }

int main	(	int	argc,
		char **	argv
	)

Definition at line 3609 of file pgbench.c.

References _, Assert, StatsData::cnt, TState::conn_time, CONNECTION_BAD, DEFAULT_NXACTS, disconnect_all(), doConnect(), ERRCODE_UNDEFINED_TABLE, findBuiltin(), foreign_keys, get_progname(), getopt_long(), i, CState::id, init(), initStats(), INSTR_TIME_ADD, INSTR_TIME_GET_MICROSEC, INSTR_TIME_SET_CURRENT, INSTR_TIME_SET_ZERO, INSTR_TIME_SUBTRACT, INVALID_THREAD, Variable::is_numeric, StatsData::lag, StatsData::latency, TState::latency_late, listAvailableScripts(), TState::logfile, login, lookupVariable(), MAXCLIENTS, mergeSimpleStats(), Variable::name, nclients, no_argument, TState::nstate, nthreads, NULL, NUM_QUERYMODE, num_scripts, Variable::num_value, CState::nvariables, optarg, optind, parseScriptWeight(), PG_DIAG_SQLSTATE, pg_malloc(), pg_realloc(), pg_strdup(), PGRES_TUPLES_OK, PQclear(), PQdb(), PQerrorMessage(), PQexec(), PQfinish(), PQgetvalue(), PQresultErrorField(), PQresultStatus(), PQstatus(), printResults(), process_builtin(), process_file(), putVariable(), putVariableInt(), putVariableNumber(), random(), TState::random_state, required_argument, setalarm(), StatsData::skipped, srandom(), start_time, TState::start_time, TState::state, TState::stats, strerror(), TState::thread, threadRun(), TState::tid, tryExecuteStatement(), unlogged_tables, usage(), Variable::value, and CState::variables.

 {
     static struct option long_options[] = {
         /* systematic long/short named options */
         {"builtin", required_argument, NULL, 'b'},
         {"client", required_argument, NULL, 'c'},
         {"connect", no_argument, NULL, 'C'},
         {"debug", no_argument, NULL, 'd'},
         {"define", required_argument, NULL, 'D'},
         {"file", required_argument, NULL, 'f'},
         {"fillfactor", required_argument, NULL, 'F'},
         {"host", required_argument, NULL, 'h'},
         {"initialize", no_argument, NULL, 'i'},
         {"jobs", required_argument, NULL, 'j'},
         {"log", no_argument, NULL, 'l'},
         {"latency-limit", required_argument, NULL, 'L'},
         {"no-vacuum", no_argument, NULL, 'n'},
         {"port", required_argument, NULL, 'p'},
         {"progress", required_argument, NULL, 'P'},
         {"protocol", required_argument, NULL, 'M'},
         {"quiet", no_argument, NULL, 'q'},
         {"report-latencies", no_argument, NULL, 'r'},
         {"rate", required_argument, NULL, 'R'},
         {"scale", required_argument, NULL, 's'},
         {"select-only", no_argument, NULL, 'S'},
         {"skip-some-updates", no_argument, NULL, 'N'},
         {"time", required_argument, NULL, 'T'},
         {"transactions", required_argument, NULL, 't'},
         {"username", required_argument, NULL, 'U'},
         {"vacuum-all", no_argument, NULL, 'v'},
         /* long-named only options */
         {"foreign-keys", no_argument, &foreign_keys, 1},
         {"index-tablespace", required_argument, NULL, 3},
         {"tablespace", required_argument, NULL, 2},
         {"unlogged-tables", no_argument, &unlogged_tables, 1},
         {"sampling-rate", required_argument, NULL, 4},
         {"aggregate-interval", required_argument, NULL, 5},
         {"progress-timestamp", no_argument, NULL, 6},
         {"log-prefix", required_argument, NULL, 7},
         {NULL, 0, NULL, 0}
     };
 
     int         c;
     int         is_init_mode = 0;       /* initialize mode? */
     int         is_no_vacuum = 0;       /* no vacuum at all before testing? */
     int         do_vacuum_accounts = 0; /* do vacuum accounts before testing? */
     int         optindex;
     bool        scale_given = false;
 
     bool        benchmarking_option_set = false;
     bool        initialization_option_set = false;
     bool        internal_script_used = false;
 
     CState     *state;          /* status of clients */
     TState     *threads;        /* array of thread */
 
     instr_time  start_time;     /* start up time */
     instr_time  total_time;
     instr_time  conn_total_time;
     int64       latency_late = 0;
     StatsData   stats;
     int         weight;
 
     int         i;
     int         nclients_dealt;
 
 #ifdef HAVE_GETRLIMIT
     struct rlimit rlim;
 #endif
 
     PGconn     *con;
     PGresult   *res;
     char       *env;
 
     progname = get_progname(argv[0]);
 
     if (argc > 1)
     {
         if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0)
         {
             usage();
             exit(0);
         }
         if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
         {
             puts("pgbench (PostgreSQL) " PG_VERSION);
             exit(0);
         }
     }
 
 #ifdef WIN32
     /* stderr is buffered on Win32. */
     setvbuf(stderr, NULL, _IONBF, 0);
 #endif
 
     if ((env = getenv("PGHOST")) != NULL && *env != '\0')
         pghost = env;
     if ((env = getenv("PGPORT")) != NULL && *env != '\0')
         pgport = env;
     else if ((env = getenv("PGUSER")) != NULL && *env != '\0')
         login = env;
 
     state = (CState *) pg_malloc(sizeof(CState));
     memset(state, 0, sizeof(CState));
 
     while ((c = getopt_long(argc, argv, "ih:nvp:dqb:SNc:j:Crs:t:T:U:lf:D:F:M:P:R:L:", long_options, &optindex)) != -1)
     {
         char       *script;
 
         switch (c)
         {
             case 'i':
                 is_init_mode++;
                 break;
             case 'h':
                 pghost = pg_strdup(optarg);
                 break;
             case 'n':
                 is_no_vacuum++;
                 break;
             case 'v':
                 do_vacuum_accounts++;
                 break;
             case 'p':
                 pgport = pg_strdup(optarg);
                 break;
             case 'd':
                 debug++;
                 break;
             case 'c':
                 benchmarking_option_set = true;
                 nclients = atoi(optarg);
                 if (nclients <= 0 || nclients > MAXCLIENTS)
                 {
                     fprintf(stderr, "invalid number of clients: \"%s\"\n",
                             optarg);
                     exit(1);
                 }
 #ifdef HAVE_GETRLIMIT
 #ifdef RLIMIT_NOFILE            /* most platforms use RLIMIT_NOFILE */
                 if (getrlimit(RLIMIT_NOFILE, &rlim) == -1)
 #else                           /* but BSD doesn't ... */
                 if (getrlimit(RLIMIT_OFILE, &rlim) == -1)
 #endif   /* RLIMIT_NOFILE */
                 {
                     fprintf(stderr, "getrlimit failed: %s\n", strerror(errno));
                     exit(1);
                 }
                 if (rlim.rlim_cur < nclients + 3)
                 {
                     fprintf(stderr, "need at least %d open files, but system limit is %ld\n",
                             nclients + 3, (long) rlim.rlim_cur);
                     fprintf(stderr, "Reduce number of clients, or use limit/ulimit to increase the system limit.\n");
                     exit(1);
                 }
 #endif   /* HAVE_GETRLIMIT */
                 break;
             case 'j':           /* jobs */
                 benchmarking_option_set = true;
                 nthreads = atoi(optarg);
                 if (nthreads <= 0)
                 {
                     fprintf(stderr, "invalid number of threads: \"%s\"\n",
                             optarg);
                     exit(1);
                 }
 #ifndef ENABLE_THREAD_SAFETY
                 if (nthreads != 1)
                 {
                     fprintf(stderr, "threads are not supported on this platform; use -j1\n");
                     exit(1);
                 }
 #endif   /* !ENABLE_THREAD_SAFETY */
                 break;
             case 'C':
                 benchmarking_option_set = true;
                 is_connect = true;
                 break;
             case 'r':
                 benchmarking_option_set = true;
                 per_script_stats = true;
                 is_latencies = true;
                 break;
             case 's':
                 scale_given = true;
                 scale = atoi(optarg);
                 if (scale <= 0)
                 {
                     fprintf(stderr, "invalid scaling factor: \"%s\"\n", optarg);
                     exit(1);
                 }
                 break;
             case 't':
                 benchmarking_option_set = true;
                 if (duration > 0)
                 {
                     fprintf(stderr, "specify either a number of transactions (-t) or a duration (-T), not both\n");
                     exit(1);
                 }
                 nxacts = atoi(optarg);
                 if (nxacts <= 0)
                 {
                     fprintf(stderr, "invalid number of transactions: \"%s\"\n",
                             optarg);
                     exit(1);
                 }
                 break;
             case 'T':
                 benchmarking_option_set = true;
                 if (nxacts > 0)
                 {
                     fprintf(stderr, "specify either a number of transactions (-t) or a duration (-T), not both\n");
                     exit(1);
                 }
                 duration = atoi(optarg);
                 if (duration <= 0)
                 {
                     fprintf(stderr, "invalid duration: \"%s\"\n", optarg);
                     exit(1);
                 }
                 break;
             case 'U':
                 login = pg_strdup(optarg);
                 break;
             case 'l':
                 benchmarking_option_set = true;
                 use_log = true;
                 break;
             case 'q':
                 initialization_option_set = true;
                 use_quiet = true;
                 break;
 
             case 'b':
                 if (strcmp(optarg, "list") == 0)
                 {
                     listAvailableScripts();
                     exit(0);
                 }
 
                 weight = parseScriptWeight(optarg, &script);
                 process_builtin(findBuiltin(script), weight);
                 benchmarking_option_set = true;
                 internal_script_used = true;
                 break;
 
             case 'S':
                 process_builtin(findBuiltin("select-only"), 1);
                 benchmarking_option_set = true;
                 internal_script_used = true;
                 break;
             case 'N':
                 process_builtin(findBuiltin("simple-update"), 1);
                 benchmarking_option_set = true;
                 internal_script_used = true;
                 break;
             case 'f':
                 weight = parseScriptWeight(optarg, &script);
                 process_file(script, weight);
                 benchmarking_option_set = true;
                 break;
             case 'D':
                 {
                     char       *p;
 
                     benchmarking_option_set = true;
 
                     if ((p = strchr(optarg, '=')) == NULL || p == optarg || *(p + 1) == '\0')
                     {
                         fprintf(stderr, "invalid variable definition: \"%s\"\n",
                                 optarg);
                         exit(1);
                     }
 
                     *p++ = '\0';
                     if (!putVariable(&state[0], "option", optarg, p))
                         exit(1);
                 }
                 break;
             case 'F':
                 initialization_option_set = true;
                 fillfactor = atoi(optarg);
                 if (fillfactor < 10 || fillfactor > 100)
                 {
                     fprintf(stderr, "invalid fillfactor: \"%s\"\n", optarg);
                     exit(1);
                 }
                 break;
             case 'M':
                 benchmarking_option_set = true;
                 if (num_scripts > 0)
                 {
                     fprintf(stderr, "query mode (-M) should be specified before any transaction scripts (-f or -b)\n");
                     exit(1);
                 }
                 for (querymode = 0; querymode < NUM_QUERYMODE; querymode++)
                     if (strcmp(optarg, QUERYMODE[querymode]) == 0)
                         break;
                 if (querymode >= NUM_QUERYMODE)
                 {
                     fprintf(stderr, "invalid query mode (-M): \"%s\"\n",
                             optarg);
                     exit(1);
                 }
                 break;
             case 'P':
                 benchmarking_option_set = true;
                 progress = atoi(optarg);
                 if (progress <= 0)
                 {
                     fprintf(stderr, "invalid thread progress delay: \"%s\"\n",
                             optarg);
                     exit(1);
                 }
                 break;
             case 'R':
                 {
                     /* get a double from the beginning of option value */
                     double      throttle_value = atof(optarg);
 
                     benchmarking_option_set = true;
 
                     if (throttle_value <= 0.0)
                     {
                         fprintf(stderr, "invalid rate limit: \"%s\"\n", optarg);
                         exit(1);
                     }
                     /* Invert rate limit into a time offset */
                     throttle_delay = (int64) (1000000.0 / throttle_value);
                 }
                 break;
             case 'L':
                 {
                     double      limit_ms = atof(optarg);
 
                     if (limit_ms <= 0.0)
                     {
                         fprintf(stderr, "invalid latency limit: \"%s\"\n",
                                 optarg);
                         exit(1);
                     }
                     benchmarking_option_set = true;
                     latency_limit = (int64) (limit_ms * 1000);
                 }
                 break;
             case 0:
                 /* This covers long options which take no argument. */
                 if (foreign_keys || unlogged_tables)
                     initialization_option_set = true;
                 break;
             case 2:             /* tablespace */
                 initialization_option_set = true;
                 tablespace = pg_strdup(optarg);
                 break;
             case 3:             /* index-tablespace */
                 initialization_option_set = true;
                 index_tablespace = pg_strdup(optarg);
                 break;
             case 4:
                 benchmarking_option_set = true;
                 sample_rate = atof(optarg);
                 if (sample_rate <= 0.0 || sample_rate > 1.0)
                 {
                     fprintf(stderr, "invalid sampling rate: \"%s\"\n", optarg);
                     exit(1);
                 }
                 break;
             case 5:
                 benchmarking_option_set = true;
                 agg_interval = atoi(optarg);
                 if (agg_interval <= 0)
                 {
                     fprintf(stderr, "invalid number of seconds for aggregation: \"%s\"\n",
                             optarg);
                     exit(1);
                 }
                 break;
             case 6:
                 progress_timestamp = true;
                 benchmarking_option_set = true;
                 break;
             case 7:
                 benchmarking_option_set = true;
                 logfile_prefix = pg_strdup(optarg);
                 break;
             default:
                 fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname);
                 exit(1);
                 break;
         }
     }
 
     /* set default script if none */
     if (num_scripts == 0 && !is_init_mode)
     {
         process_builtin(findBuiltin("tpcb-like"), 1);
         benchmarking_option_set = true;
         internal_script_used = true;
     }
 
     /* compute total_weight */
     for (i = 0; i < num_scripts; i++)
         /* cannot overflow: weight is 32b, total_weight 64b */
         total_weight += sql_script[i].weight;
 
     if (total_weight == 0 && !is_init_mode)
     {
         fprintf(stderr, "total script weight must not be zero\n");
         exit(1);
     }
 
     /* show per script stats if several scripts are used */
     if (num_scripts > 1)
         per_script_stats = true;
 
     /*
      * Don't need more threads than there are clients.  (This is not merely an
      * optimization; throttle_delay is calculated incorrectly below if some
      * threads have no clients assigned to them.)
      */
     if (nthreads > nclients)
         nthreads = nclients;
 
     /* compute a per thread delay */
     throttle_delay *= nthreads;
 
     if (argc > optind)
         dbName = argv[optind];
     else
     {
         if ((env = getenv("PGDATABASE")) != NULL && *env != '\0')
             dbName = env;
         else if (login != NULL && *login != '\0')
             dbName = login;
         else
             dbName = "";
     }
 
     if (is_init_mode)
     {
         if (benchmarking_option_set)
         {
             fprintf(stderr, "some of the specified options cannot be used in initialization (-i) mode\n");
             exit(1);
         }
 
         init(is_no_vacuum);
         exit(0);
     }
     else
     {
         if (initialization_option_set)
         {
             fprintf(stderr, "some of the specified options cannot be used in benchmarking mode\n");
             exit(1);
         }
     }
 
     /* Use DEFAULT_NXACTS if neither nxacts nor duration is specified. */
     if (nxacts <= 0 && duration <= 0)
         nxacts = DEFAULT_NXACTS;
 
     /* --sampling-rate may be used only with -l */
     if (sample_rate > 0.0 && !use_log)
     {
         fprintf(stderr, "log sampling (--sampling-rate) is allowed only when logging transactions (-l)\n");
         exit(1);
     }
 
     /* --sampling-rate may not be used with --aggregate-interval */
     if (sample_rate > 0.0 && agg_interval > 0)
     {
         fprintf(stderr, "log sampling (--sampling-rate) and aggregation (--aggregate-interval) cannot be used at the same time\n");
         exit(1);
     }
 
     if (agg_interval > 0 && !use_log)
     {
         fprintf(stderr, "log aggregation is allowed only when actually logging transactions\n");
         exit(1);
     }
 
     if (!use_log && logfile_prefix)
     {
         fprintf(stderr, "log file prefix (--log-prefix) is allowed only when logging transactions (-l)\n");
         exit(1);
     }
 
     if (duration > 0 && agg_interval > duration)
     {
         fprintf(stderr, "number of seconds for aggregation (%d) must not be higher than test duration (%d)\n", agg_interval, duration);
         exit(1);
     }
 
     if (duration > 0 && agg_interval > 0 && duration % agg_interval != 0)
     {
         fprintf(stderr, "duration (%d) must be a multiple of aggregation interval (%d)\n", duration, agg_interval);
         exit(1);
     }
 
     /*
      * save main process id in the global variable because process id will be
      * changed after fork.
      */
     main_pid = (int) getpid();
 
     if (nclients > 1)
     {
         state = (CState *) pg_realloc(state, sizeof(CState) * nclients);
         memset(state + 1, 0, sizeof(CState) * (nclients - 1));
 
         /* copy any -D switch values to all clients */
         for (i = 1; i < nclients; i++)
         {
             int         j;
 
             state[i].id = i;
             for (j = 0; j < state[0].nvariables; j++)
             {
                 Variable   *var = &state[0].variables[j];
 
                 if (var->is_numeric)
                 {
                     if (!putVariableNumber(&state[i], "startup",
                                            var->name, &var->num_value))
                         exit(1);
                 }
                 else
                 {
                     if (!putVariable(&state[i], "startup",
                                      var->name, var->value))
                         exit(1);
                 }
             }
         }
     }
 
     if (debug)
     {
         if (duration <= 0)
             printf("pghost: %s pgport: %s nclients: %d nxacts: %d dbName: %s\n",
                    pghost, pgport, nclients, nxacts, dbName);
         else
             printf("pghost: %s pgport: %s nclients: %d duration: %d dbName: %s\n",
                    pghost, pgport, nclients, duration, dbName);
     }
 
     /* opening connection... */
     con = doConnect();
     if (con == NULL)
         exit(1);
 
     if (PQstatus(con) == CONNECTION_BAD)
     {
         fprintf(stderr, "connection to database \"%s\" failed\n", dbName);
         fprintf(stderr, "%s", PQerrorMessage(con));
         exit(1);
     }
 
     if (internal_script_used)
     {
         /*
          * get the scaling factor that should be same as count(*) from
          * pgbench_branches if this is not a custom query
          */
         res = PQexec(con, "select count(*) from pgbench_branches");
         if (PQresultStatus(res) != PGRES_TUPLES_OK)
         {
             char       *sqlState = PQresultErrorField(res, PG_DIAG_SQLSTATE);
 
             fprintf(stderr, "%s", PQerrorMessage(con));
             if (sqlState && strcmp(sqlState, ERRCODE_UNDEFINED_TABLE) == 0)
             {
                 fprintf(stderr, "Perhaps you need to do initialization (\"pgbench -i\") in database \"%s\"\n", PQdb(con));
             }
 
             exit(1);
         }
         scale = atoi(PQgetvalue(res, 0, 0));
         if (scale < 0)
         {
             fprintf(stderr, "invalid count(*) from pgbench_branches: \"%s\"\n",
                     PQgetvalue(res, 0, 0));
             exit(1);
         }
         PQclear(res);
 
         /* warn if we override user-given -s switch */
         if (scale_given)
             fprintf(stderr,
                     "scale option ignored, using count from pgbench_branches table (%d)\n",
                     scale);
     }
 
     /*
      * :scale variables normally get -s or database scale, but don't override
      * an explicit -D switch
      */
     if (lookupVariable(&state[0], "scale") == NULL)
     {
         for (i = 0; i < nclients; i++)
         {
             if (!putVariableInt(&state[i], "startup", "scale", scale))
                 exit(1);
         }
     }
 
     /*
      * Define a :client_id variable that is unique per connection. But don't
      * override an explicit -D switch.
      */
     if (lookupVariable(&state[0], "client_id") == NULL)
     {
         for (i = 0; i < nclients; i++)
         {
             if (!putVariableInt(&state[i], "startup", "client_id", i))
                 exit(1);
         }
     }
 
     if (!is_no_vacuum)
     {
         fprintf(stderr, "starting vacuum...");
         tryExecuteStatement(con, "vacuum pgbench_branches");
         tryExecuteStatement(con, "vacuum pgbench_tellers");
         tryExecuteStatement(con, "truncate pgbench_history");
         fprintf(stderr, "end.\n");
 
         if (do_vacuum_accounts)
         {
             fprintf(stderr, "starting vacuum pgbench_accounts...");
             tryExecuteStatement(con, "vacuum analyze pgbench_accounts");
             fprintf(stderr, "end.\n");
         }
     }
     PQfinish(con);
 
     /* set random seed */
     INSTR_TIME_SET_CURRENT(start_time);
     srandom((unsigned int) INSTR_TIME_GET_MICROSEC(start_time));
 
     /* set up thread data structures */
     threads = (TState *) pg_malloc(sizeof(TState) * nthreads);
     nclients_dealt = 0;
 
     for (i = 0; i < nthreads; i++)
     {
         TState     *thread = &threads[i];
 
         thread->tid = i;
         thread->state = &state[nclients_dealt];
         thread->nstate =
             (nclients - nclients_dealt + nthreads - i - 1) / (nthreads - i);
         thread->random_state[0] = random();
         thread->random_state[1] = random();
         thread->random_state[2] = random();
         thread->logfile = NULL; /* filled in later */
         thread->latency_late = 0;
         initStats(&thread->stats, 0);
 
         nclients_dealt += thread->nstate;
     }
 
     /* all clients must be assigned to a thread */
     Assert(nclients_dealt == nclients);
 
     /* get start up time */
     INSTR_TIME_SET_CURRENT(start_time);
 
     /* set alarm if duration is specified. */
     if (duration > 0)
         setalarm(duration);
 
     /* start threads */
 #ifdef ENABLE_THREAD_SAFETY
     for (i = 0; i < nthreads; i++)
     {
         TState     *thread = &threads[i];
 
         INSTR_TIME_SET_CURRENT(thread->start_time);
 
         /* compute when to stop */
         if (duration > 0)
             end_time = INSTR_TIME_GET_MICROSEC(thread->start_time) +
                 (int64) 1000000 *duration;
 
         /* the first thread (i = 0) is executed by main thread */
         if (i > 0)
         {
             int         err = pthread_create(&thread->thread, NULL, threadRun, thread);
 
             if (err != 0 || thread->thread == INVALID_THREAD)
             {
                 fprintf(stderr, "could not create thread: %s\n", strerror(err));
                 exit(1);
             }
         }
         else
         {
             thread->thread = INVALID_THREAD;
         }
     }
 #else
     INSTR_TIME_SET_CURRENT(threads[0].start_time);
     /* compute when to stop */
     if (duration > 0)
         end_time = INSTR_TIME_GET_MICROSEC(threads[0].start_time) +
             (int64) 1000000 *duration;
     threads[0].thread = INVALID_THREAD;
 #endif   /* ENABLE_THREAD_SAFETY */
 
     /* wait for threads and accumulate results */
     initStats(&stats, 0);
     INSTR_TIME_SET_ZERO(conn_total_time);
     for (i = 0; i < nthreads; i++)
     {
         TState     *thread = &threads[i];
 
 #ifdef ENABLE_THREAD_SAFETY
         if (threads[i].thread == INVALID_THREAD)
             /* actually run this thread directly in the main thread */
             (void) threadRun(thread);
         else
             /* wait of other threads. should check that 0 is returned? */
             pthread_join(thread->thread, NULL);
 #else
         (void) threadRun(thread);
 #endif   /* ENABLE_THREAD_SAFETY */
 
         /* aggregate thread level stats */
         mergeSimpleStats(&stats.latency, &thread->stats.latency);
         mergeSimpleStats(&stats.lag, &thread->stats.lag);
         stats.cnt += thread->stats.cnt;
         stats.skipped += thread->stats.skipped;
         latency_late += thread->latency_late;
         INSTR_TIME_ADD(conn_total_time, thread->conn_time);
     }
     disconnect_all(state, nclients);
 
     /*
      * XXX We compute results as though every client of every thread started
      * and finished at the same time.  That model can diverge noticeably from
      * reality for a short benchmark run involving relatively many threads.
      * The first thread may process notably many transactions before the last
      * thread begins.  Improving the model alone would bring limited benefit,
      * because performance during those periods of partial thread count can
      * easily exceed steady state performance.  This is one of the many ways
      * short runs convey deceptive performance figures.
      */
     INSTR_TIME_SET_CURRENT(total_time);
     INSTR_TIME_SUBTRACT(total_time, start_time);
     printResults(threads, &stats, total_time, conn_total_time, latency_late);
 
     return 0;
 }

static bool makeVariableNumeric ( Variable * var )

static

Definition at line 995 of file pgbench.c.

References is_an_int(), Variable::is_numeric, Variable::name, Variable::num_value, setDoubleValue(), setIntValue(), strtoint64(), and Variable::value.

Referenced by evaluateExpr().

 {
     if (var->is_numeric)
         return true;            /* no work */
 
     if (is_an_int(var->value))
     {
         setIntValue(&var->num_value, strtoint64(var->value));
         var->is_numeric = true;
     }
     else    /* type should be double */
     {
         double      dv;
         char        xs;
 
         if (sscanf(var->value, "%lf%c", &dv, &xs) != 1)
         {
             fprintf(stderr,
                     "malformed variable \"%s\" value: \"%s\"\n",
                     var->name, var->value);
             return false;
         }
         setDoubleValue(&var->num_value, dv);
         var->is_numeric = true;
     }
     return true;
 }

static void mergeSimpleStats	(	SimpleStats *	acc,
		SimpleStats *	ss
	)

static

Definition at line 768 of file pgbench.c.

References SimpleStats::count, SimpleStats::max, SimpleStats::min, SimpleStats::sum, and SimpleStats::sum2.

Referenced by main(), and threadRun().

 {
     if (acc->count == 0 || ss->min < acc->min)
         acc->min = ss->min;
     if (acc->count == 0 || ss->max > acc->max)
         acc->max = ss->max;
     acc->count += ss->count;
     acc->sum += ss->sum;
     acc->sum2 += ss->sum2;
 }

static bool parseQuery	(	Command *	cmd,
		const char *	raw_sql
	)

static

Definition at line 2850 of file pgbench.c.

References Command::argc, Command::argv, MAX_ARGS, name, NULL, parseVariable(), pg_free(), pg_strdup(), and replaceVariable().

Referenced by process_sql_command().

 {
     char       *sql,
                *p;
 
     sql = pg_strdup(raw_sql);
     cmd->argc = 1;
 
     p = sql;
     while ((p = strchr(p, ':')) != NULL)
     {
         char        var[12];
         char       *name;
         int         eaten;
 
         name = parseVariable(p, &eaten);
         if (name == NULL)
         {
             while (*p == ':')
             {
                 p++;
             }
             continue;
         }
 
         if (cmd->argc >= MAX_ARGS)
         {
             fprintf(stderr, "statement has too many arguments (maximum is %d): %s\n", MAX_ARGS - 1, raw_sql);
             pg_free(name);
             return false;
         }
 
         sprintf(var, "$%d", cmd->argc);
         p = replaceVariable(&sql, p, eaten, var);
 
         cmd->argv[cmd->argc] = name;
         cmd->argc++;
     }
 
     cmd->argv[0] = sql;
     return true;
 }

static void ParseScript	(	const char *	script,
		const char *	desc,
		int	weight
	)

static

Definition at line 3192 of file pgbench.c.

References addScript(), ParsedScript::commands, COMMANDS_ALLOC_NUM, ParsedScript::desc, initPQExpBuffer(), initStats(), NULL, pg_malloc(), pg_realloc(), pgbench_callbacks, process_backslash_command(), process_sql_command(), PSCAN_BACKSLASH, PSCAN_EOL, PSCAN_INCOMPLETE, psql_scan(), psql_scan_create(), psql_scan_destroy(), psql_scan_finish(), psql_scan_setup(), resetPQExpBuffer(), ParsedScript::stats, termPQExpBuffer(), and ParsedScript::weight.

Referenced by process_builtin(), and process_file().

 {
     ParsedScript ps;
     PsqlScanState sstate;
     PQExpBufferData line_buf;
     int         alloc_num;
     int         index;
 
 #define COMMANDS_ALLOC_NUM 128
     alloc_num = COMMANDS_ALLOC_NUM;
 
     /* Initialize all fields of ps */
     ps.desc = desc;
     ps.weight = weight;
     ps.commands = (Command **) pg_malloc(sizeof(Command *) * alloc_num);
     initStats(&ps.stats, 0);
 
     /* Prepare to parse script */
     sstate = psql_scan_create(&pgbench_callbacks);
 
     /*
      * Ideally, we'd scan scripts using the encoding and stdstrings settings
      * we get from a DB connection.  However, without major rearrangement of
      * pgbench's argument parsing, we can't have a DB connection at the time
      * we parse scripts.  Using SQL_ASCII (encoding 0) should work well enough
      * with any backend-safe encoding, though conceivably we could be fooled
      * if a script file uses a client-only encoding.  We also assume that
      * stdstrings should be true, which is a bit riskier.
      */
     psql_scan_setup(sstate, script, strlen(script), 0, true);
 
     initPQExpBuffer(&line_buf);
 
     index = 0;
 
     for (;;)
     {
         PsqlScanResult sr;
         promptStatus_t prompt;
         Command    *command;
 
         resetPQExpBuffer(&line_buf);
 
         sr = psql_scan(sstate, &line_buf, &prompt);
 
         /* If we collected a SQL command, process that */
         command = process_sql_command(&line_buf, desc);
         if (command)
         {
             ps.commands[index] = command;
             index++;
 
             if (index >= alloc_num)
             {
                 alloc_num += COMMANDS_ALLOC_NUM;
                 ps.commands = (Command **)
                     pg_realloc(ps.commands, sizeof(Command *) * alloc_num);
             }
         }
 
         /* If we reached a backslash, process that */
         if (sr == PSCAN_BACKSLASH)
         {
             command = process_backslash_command(sstate, desc);
             if (command)
             {
                 ps.commands[index] = command;
                 index++;
 
                 if (index >= alloc_num)
                 {
                     alloc_num += COMMANDS_ALLOC_NUM;
                     ps.commands = (Command **)
                         pg_realloc(ps.commands, sizeof(Command *) * alloc_num);
                 }
             }
         }
 
         /* Done if we reached EOF */
         if (sr == PSCAN_INCOMPLETE || sr == PSCAN_EOL)
             break;
     }
 
     ps.commands[index] = NULL;
 
     addScript(ps);
 
     termPQExpBuffer(&line_buf);
     psql_scan_finish(sstate);
     psql_scan_destroy(sstate);
 }

static int parseScriptWeight	(	const char *	option,
		char **	script
	)

static

Definition at line 3414 of file pgbench.c.

References INT64_FORMAT, pg_malloc(), pg_strdup(), and WSEP.

Referenced by main().

 {
     char       *sep;
     int         weight;
 
     if ((sep = strrchr(option, WSEP)))
     {
         int         namelen = sep - option;
         long        wtmp;
         char       *badp;
 
         /* generate the script name */
         *script = pg_malloc(namelen + 1);
         strncpy(*script, option, namelen);
         (*script)[namelen] = '\0';
 
         /* process digits of the weight spec */
         errno = 0;
         wtmp = strtol(sep + 1, &badp, 10);
         if (errno != 0 || badp == sep + 1 || *badp != '\0')
         {
             fprintf(stderr, "invalid weight specification: %s\n", sep);
             exit(1);
         }
         if (wtmp > INT_MAX || wtmp < 0)
         {
             fprintf(stderr,
             "weight specification out of range (0 .. %u): " INT64_FORMAT "\n",
                     INT_MAX, (int64) wtmp);
             exit(1);
         }
         weight = wtmp;
     }
     else
     {
         *script = pg_strdup(option);
         weight = 1;
     }
 
     return weight;
 }

static char* parseVariable	(	const char *	sql,
		int *	eaten
	)

static

Definition at line 1143 of file pgbench.c.

References i, name, NULL, and pg_malloc().

Referenced by assignVariables(), and parseQuery().

 {
     int         i = 0;
     char       *name;
 
     do
     {
         i++;
     } while (isalnum((unsigned char) sql[i]) || sql[i] == '_');
     if (i == 1)
         return NULL;
 
     name = pg_malloc(i);
     memcpy(name, &sql[1], i - 1);
     name[i - 1] = '\0';
 
     *eaten = i;
     return name;
 }

static void pgbench_error	(	const char *	fmt,
			...
	)

static

Definition at line 2897 of file pgbench.c.

References _.

 {
     va_list     ap;
 
     fflush(stdout);
     va_start(ap, fmt);
     vfprintf(stderr, _(fmt), ap);
     va_end(ap);
 }

static void preparedStatementName	(	char *	buffer,
		int	file,
		int	state
	)

static

Definition at line 1805 of file pgbench.c.

Referenced by sendCommand().

 {
     sprintf(buffer, "P%d_%d", file, state);
 }

static void printResults	(	TState *	threads,
		StatsData *	total,
		instr_time	total_time,
		instr_time	conn_total_time,
		int	latency_late
	)

static

Definition at line 3489 of file pgbench.c.

References StatsData::cnt, i, INSTR_TIME_GET_DOUBLE, INT64_FORMAT, StatsData::lag, StatsData::latency, SimpleStats::max, nclients, NULL, num_scripts, printSimpleStats(), StatsData::skipped, ParsedScript::stats, and SimpleStats::sum.

Referenced by main().

 {
     double      time_include,
                 tps_include,
                 tps_exclude;
 
     time_include = INSTR_TIME_GET_DOUBLE(total_time);
     tps_include = total->cnt / time_include;
     tps_exclude = total->cnt / (time_include -
                         (INSTR_TIME_GET_DOUBLE(conn_total_time) / nclients));
 
     /* Report test parameters. */
     printf("transaction type: %s\n",
            num_scripts == 1 ? sql_script[0].desc : "multiple scripts");
     printf("scaling factor: %d\n", scale);
     printf("query mode: %s\n", QUERYMODE[querymode]);
     printf("number of clients: %d\n", nclients);
     printf("number of threads: %d\n", nthreads);
     if (duration <= 0)
     {
         printf("number of transactions per client: %d\n", nxacts);
         printf("number of transactions actually processed: " INT64_FORMAT "/%d\n",
                total->cnt, nxacts * nclients);
     }
     else
     {
         printf("duration: %d s\n", duration);
         printf("number of transactions actually processed: " INT64_FORMAT "\n",
                total->cnt);
     }
 
     /* Remaining stats are nonsensical if we failed to execute any xacts */
     if (total->cnt <= 0)
         return;
 
     if (throttle_delay && latency_limit)
         printf("number of transactions skipped: " INT64_FORMAT " (%.3f %%)\n",
                total->skipped,
                100.0 * total->skipped / (total->skipped + total->cnt));
 
     if (latency_limit)
         printf("number of transactions above the %.1f ms latency limit: %d (%.3f %%)\n",
                latency_limit / 1000.0, latency_late,
                100.0 * latency_late / (total->skipped + total->cnt));
 
     if (throttle_delay || progress || latency_limit)
         printSimpleStats("latency", &total->latency);
     else
     {
         /* no measurement, show average latency computed from run time */
         printf("latency average = %.3f ms\n",
                1000.0 * time_include * nclients / total->cnt);
     }
 
     if (throttle_delay)
     {
         /*
          * Report average transaction lag under rate limit throttling.  This
          * is the delay between scheduled and actual start times for the
          * transaction.  The measured lag may be caused by thread/client load,
          * the database load, or the Poisson throttling process.
          */
         printf("rate limit schedule lag: avg %.3f (max %.3f) ms\n",
                0.001 * total->lag.sum / total->cnt, 0.001 * total->lag.max);
     }
 
     printf("tps = %f (including connections establishing)\n", tps_include);
     printf("tps = %f (excluding connections establishing)\n", tps_exclude);
 
     /* Report per-script/command statistics */
     if (per_script_stats || latency_limit || is_latencies)
     {
         int         i;
 
         for (i = 0; i < num_scripts; i++)
         {
             if (num_scripts > 1)
                 printf("SQL script %d: %s\n"
                        " - weight: %d (targets %.1f%% of total)\n"
                        " - " INT64_FORMAT " transactions (%.1f%% of total, tps = %f)\n",
                        i + 1, sql_script[i].desc,
                        sql_script[i].weight,
                        100.0 * sql_script[i].weight / total_weight,
                        sql_script[i].stats.cnt,
                        100.0 * sql_script[i].stats.cnt / total->cnt,
                        sql_script[i].stats.cnt / time_include);
             else
                 printf("script statistics:\n");
 
             if (latency_limit)
                 printf(" - number of transactions skipped: " INT64_FORMAT " (%.3f%%)\n",
                        sql_script[i].stats.skipped,
                        100.0 * sql_script[i].stats.skipped /
                     (sql_script[i].stats.skipped + sql_script[i].stats.cnt));
 
             if (num_scripts > 1)
                 printSimpleStats(" - latency", &sql_script[i].stats.latency);
 
             /* Report per-command latencies */
             if (is_latencies)
             {
                 Command   **commands;
 
                 printf(" - statement latencies in milliseconds:\n");
 
                 for (commands = sql_script[i].commands;
                      *commands != NULL;
                      commands++)
                     printf("   %11.3f  %s\n",
                            1000.0 * (*commands)->stats.sum /
                            (*commands)->stats.count,
                            (*commands)->line);
             }
         }
     }
 }

static void printSimpleStats	(	char *	prefix,
		SimpleStats *	ss
	)

static

Definition at line 3477 of file pgbench.c.

References SimpleStats::count, SimpleStats::sum, and SimpleStats::sum2.

Referenced by printResults().

 {
     /* print NaN if no transactions where executed */
     double      latency = ss->sum / ss->count;
     double      stddev = sqrt(ss->sum2 / ss->count - latency * latency);
 
     printf("%s average = %.3f ms\n", prefix, 0.001 * latency);
     printf("%s stddev = %.3f ms\n", prefix, 0.001 * stddev);
 }

static Command* process_backslash_command	(	PsqlScanState	sstate,
		const char *	source
	)

static

Definition at line 3027 of file pgbench.c.

References Command::argc, Command::argv, Command::command_num, PQExpBufferData::data, Command::expr, expr_lex_one_word(), expr_parse_result, expr_scanner_finish(), expr_scanner_get_lineno(), expr_scanner_get_substring(), expr_scanner_init(), expr_scanner_offset(), expr_yyparse(), initPQExpBuffer(), initSimpleStats(), Command::line, MAX_ARGS, META_COMMAND, NULL, pg_malloc0(), pg_strcasecmp(), pg_strdup(), Command::stats, syntax_error(), termPQExpBuffer(), Command::type, and yyscanner.

Referenced by ParseScript().

 {
     Command    *my_command;
     PQExpBufferData word_buf;
     int         word_offset;
     int         offsets[MAX_ARGS];      /* offsets of argument words */
     int         start_offset,
                 end_offset;
     int         lineno;
     int         j;
 
     initPQExpBuffer(&word_buf);
 
     /* Remember location of the backslash */
     start_offset = expr_scanner_offset(sstate) - 1;
     lineno = expr_scanner_get_lineno(sstate, start_offset);
 
     /* Collect first word of command */
     if (!expr_lex_one_word(sstate, &word_buf, &word_offset))
     {
         termPQExpBuffer(&word_buf);
         return NULL;
     }
 
     /* Allocate and initialize Command structure */
     my_command = (Command *) pg_malloc0(sizeof(Command));
     my_command->command_num = num_commands++;
     my_command->type = META_COMMAND;
     my_command->argc = 0;
     initSimpleStats(&my_command->stats);
 
     /* Save first word (command name) */
     j = 0;
     offsets[j] = word_offset;
     my_command->argv[j++] = pg_strdup(word_buf.data);
     my_command->argc++;
 
     if (pg_strcasecmp(my_command->argv[0], "set") == 0)
     {
         /* For \set, collect var name, then lex the expression. */
         yyscan_t    yyscanner;
 
         if (!expr_lex_one_word(sstate, &word_buf, &word_offset))
             syntax_error(source, lineno, my_command->line, my_command->argv[0],
                          "missing argument", NULL, -1);
 
         offsets[j] = word_offset;
         my_command->argv[j++] = pg_strdup(word_buf.data);
         my_command->argc++;
 
         yyscanner = expr_scanner_init(sstate, source, lineno, start_offset,
                                       my_command->argv[0]);
 
         if (expr_yyparse(yyscanner) != 0)
         {
             /* dead code: exit done from syntax_error called by yyerror */
             exit(1);
         }
 
         my_command->expr = expr_parse_result;
 
         /* Get location of the ending newline */
         end_offset = expr_scanner_offset(sstate) - 1;
 
         /* Save line */
         my_command->line = expr_scanner_get_substring(sstate,
                                                       start_offset,
                                                       end_offset);
 
         expr_scanner_finish(yyscanner);
 
         termPQExpBuffer(&word_buf);
 
         return my_command;
     }
 
     /* For all other commands, collect remaining words. */
     while (expr_lex_one_word(sstate, &word_buf, &word_offset))
     {
         if (j >= MAX_ARGS)
             syntax_error(source, lineno, my_command->line, my_command->argv[0],
                          "too many arguments", NULL, -1);
 
         offsets[j] = word_offset;
         my_command->argv[j++] = pg_strdup(word_buf.data);
         my_command->argc++;
     }
 
     /* Get location of the ending newline */
     end_offset = expr_scanner_offset(sstate) - 1;
 
     /* Save line */
     my_command->line = expr_scanner_get_substring(sstate,
                                                   start_offset,
                                                   end_offset);
 
     if (pg_strcasecmp(my_command->argv[0], "sleep") == 0)
     {
         if (my_command->argc < 2)
             syntax_error(source, lineno, my_command->line, my_command->argv[0],
                          "missing argument", NULL, -1);
 
         if (my_command->argc > 3)
             syntax_error(source, lineno, my_command->line, my_command->argv[0],
                          "too many arguments", NULL,
                          offsets[3] - start_offset);
 
         /*
          * Split argument into number and unit to allow "sleep 1ms" etc. We
          * don't have to terminate the number argument with null because it
          * will be parsed with atoi, which ignores trailing non-digit
          * characters.
          */
         if (my_command->argc == 2 && my_command->argv[1][0] != ':')
         {
             char       *c = my_command->argv[1];
 
             while (isdigit((unsigned char) *c))
                 c++;
             if (*c)
             {
                 my_command->argv[2] = c;
                 offsets[2] = offsets[1] + (c - my_command->argv[1]);
                 my_command->argc = 3;
             }
         }
 
         if (my_command->argc == 3)
         {
             if (pg_strcasecmp(my_command->argv[2], "us") != 0 &&
                 pg_strcasecmp(my_command->argv[2], "ms") != 0 &&
                 pg_strcasecmp(my_command->argv[2], "s") != 0)
                 syntax_error(source, lineno, my_command->line, my_command->argv[0],
                              "unrecognized time unit, must be us, ms or s",
                              my_command->argv[2], offsets[2] - start_offset);
         }
     }
     else if (pg_strcasecmp(my_command->argv[0], "setshell") == 0)
     {
         if (my_command->argc < 3)
             syntax_error(source, lineno, my_command->line, my_command->argv[0],
                          "missing argument", NULL, -1);
     }
     else if (pg_strcasecmp(my_command->argv[0], "shell") == 0)
     {
         if (my_command->argc < 2)
             syntax_error(source, lineno, my_command->line, my_command->argv[0],
                          "missing command", NULL, -1);
     }
     else
     {
         syntax_error(source, lineno, my_command->line, my_command->argv[0],
                      "invalid command", NULL, -1);
     }
 
     termPQExpBuffer(&word_buf);
 
     return my_command;
 }

static void process_builtin	(	const BuiltinScript *	bi,
		int	weight
	)

static

Definition at line 3358 of file pgbench.c.

References BuiltinScript::desc, ParseScript(), and BuiltinScript::script.

Referenced by main().

 {
     ParseScript(bi->script, bi->desc, weight);
 }

static void process_file	(	const char *	filename,
		int	weight
	)

static

Definition at line 3324 of file pgbench.c.

References buf, fd(), free, NULL, ParseScript(), read_file_contents(), and strerror().

Referenced by main(), and process_psqlrc_file().

 {
     FILE       *fd;
     char       *buf;
 
     /* Slurp the file contents into "buf" */
     if (strcmp(filename, "-") == 0)
         fd = stdin;
     else if ((fd = fopen(filename, "r")) == NULL)
     {
         fprintf(stderr, "could not open file \"%s\": %s\n",
                 filename, strerror(errno));
         exit(1);
     }
 
     buf = read_file_contents(fd);
 
     if (ferror(fd))
     {
         fprintf(stderr, "could not read file \"%s\": %s\n",
                 filename, strerror(errno));
         exit(1);
     }
 
     if (fd != stdin)
         fclose(fd);
 
     ParseScript(buf, filename, weight);
 
     free(buf);
 }

static Command* process_sql_command	(	PQExpBuffer	buf,
		const char *	source
	)

static

Definition at line 2955 of file pgbench.c.

References Command::argc, Command::argv, Command::command_num, PQExpBufferData::data, initSimpleStats(), Command::line, NULL, parseQuery(), pg_malloc(), pg_malloc0(), pg_strdup(), QUERY_EXTENDED, QUERY_PREPARED, QUERY_SIMPLE, SQL_COMMAND, Command::stats, and Command::type.

Referenced by ParseScript().

 {
     Command    *my_command;
     char       *p;
     char       *nlpos;
 
     /* Skip any leading whitespace, as well as "--" style comments */
     p = buf->data;
     for (;;)
     {
         if (isspace((unsigned char) *p))
             p++;
         else if (strncmp(p, "--", 2) == 0)
         {
             p = strchr(p, '\n');
             if (p == NULL)
                 return NULL;
             p++;
         }
         else
             break;
     }
 
     /* If there's nothing but whitespace and comments, we're done */
     if (*p == '\0')
         return NULL;
 
     /* Allocate and initialize Command structure */
     my_command = (Command *) pg_malloc0(sizeof(Command));
     my_command->command_num = num_commands++;
     my_command->type = SQL_COMMAND;
     my_command->argc = 0;
     initSimpleStats(&my_command->stats);
 
     /*
      * If SQL command is multi-line, we only want to save the first line as
      * the "line" label.
      */
     nlpos = strchr(p, '\n');
     if (nlpos)
     {
         my_command->line = pg_malloc(nlpos - p + 1);
         memcpy(my_command->line, p, nlpos - p);
         my_command->line[nlpos - p] = '\0';
     }
     else
         my_command->line = pg_strdup(p);
 
     switch (querymode)
     {
         case QUERY_SIMPLE:
             my_command->argv[0] = pg_strdup(p);
             my_command->argc++;
             break;
         case QUERY_EXTENDED:
         case QUERY_PREPARED:
             if (!parseQuery(my_command, p))
                 exit(1);
             break;
         default:
             exit(1);
     }
 
     return my_command;
 }

static void processXactStats	(	TState *	thread,
		CState *	st,
		instr_time *	now,
		bool	skipped,
		StatsData *	agg
	)

static

Definition at line 2517 of file pgbench.c.

References accumStats(), StatsData::cnt, doLog(), INSTR_TIME_GET_MICROSEC, INSTR_TIME_IS_ZERO, INSTR_TIME_SET_CURRENT, TState::latency_late, TState::stats, ParsedScript::stats, CState::txn_begin, CState::txn_scheduled, and CState::use_file.

Referenced by doCustom().

 {
     double      latency = 0.0,
                 lag = 0.0;
 
     if ((!skipped) && INSTR_TIME_IS_ZERO(*now))
         INSTR_TIME_SET_CURRENT(*now);
 
     if (!skipped)
     {
         /* compute latency & lag */
         latency = INSTR_TIME_GET_MICROSEC(*now) - st->txn_scheduled;
         lag = INSTR_TIME_GET_MICROSEC(st->txn_begin) - st->txn_scheduled;
     }
 
     if (progress || throttle_delay || latency_limit)
     {
         accumStats(&thread->stats, skipped, latency, lag);
 
         /* count transactions over the latency limit, if needed */
         if (latency_limit && latency > latency_limit)
             thread->latency_late++;
     }
     else
         thread->stats.cnt++;
 
     if (use_log)
         doLog(thread, st, agg, skipped, latency, lag);
 
     /* XXX could use a mutex here, but we choose not to */
     if (per_script_stats)
         accumStats(&sql_script[st->use_file].stats, skipped, latency, lag);
 }

static bool putVariable	(	CState *	st,
		const char *	context,
		char *	name,
		const char *	value
	)

static

Definition at line 1090 of file pgbench.c.

References free, Variable::is_numeric, lookupCreateVariable(), pg_strdup(), val, and Variable::value.

Referenced by main().

 {
     Variable   *var;
     char       *val;
 
     var = lookupCreateVariable(st, context, name);
     if (!var)
         return false;
 
     /* dup then free, in case value is pointing at this variable */
     val = pg_strdup(value);
 
     if (var->value)
         free(var->value);
     var->value = val;
     var->is_numeric = false;
 
     return true;
 }

static bool putVariableInt	(	CState *	st,
		const char *	context,
		char *	name,
		int64	value
	)

static

Definition at line 1134 of file pgbench.c.

References putVariableNumber(), setIntValue(), and val.

Referenced by main(), and runShellCommand().

 {
     PgBenchValue val;
 
     setIntValue(&val, value);
     return putVariableNumber(st, context, name, &val);
 }

static bool putVariableNumber	(	CState *	st,
		const char *	context,
		char *	name,
		const PgBenchValue *	value
	)

static

Definition at line 1113 of file pgbench.c.

References free, Variable::is_numeric, lookupCreateVariable(), NULL, Variable::num_value, Variable::value, and value.

Referenced by doCustom(), main(), and putVariableInt().

 {
     Variable   *var;
 
     var = lookupCreateVariable(st, context, name);
     if (!var)
         return false;
 
     if (var->value)
         free(var->value);
     var->value = NULL;
     var->is_numeric = true;
     var->num_value = *value;
 
     return true;
 }

static char* read_file_contents ( FILE * fd )

static

Definition at line 3291 of file pgbench.c.

References buf, pg_malloc(), and pg_realloc().

Referenced by process_file().

 {
     char       *buf;
     size_t      buflen = BUFSIZ;
     size_t      used = 0;
 
     buf = (char *) pg_malloc(buflen);
 
     for (;;)
     {
         size_t      nread;
 
         nread = fread(buf + used, 1, BUFSIZ, fd);
         used += nread;
         /* If fread() read less than requested, must be EOF or error */
         if (nread < BUFSIZ)
             break;
         /* Enlarge buf so we can read some more */
         buflen += BUFSIZ;
         buf = (char *) pg_realloc(buf, buflen);
     }
     /* There is surely room for a terminator */
     buf[used] = '\0';
 
     return buf;
 }

static char* replaceVariable	(	char **	sql,
		char *	param,
		int	len,
		char *	value
	)

static

Definition at line 1164 of file pgbench.c.

References memmove, and pg_realloc().

Referenced by assignVariables(), and parseQuery().

 {
     int         valueln = strlen(value);
 
     if (valueln > len)
     {
         size_t      offset = param - *sql;
 
         *sql = pg_realloc(*sql, strlen(*sql) - len + valueln + 1);
         param = *sql + offset;
     }
 
     if (valueln != len)
         memmove(param + valueln, param + len, strlen(param + len) + 1);
     memcpy(param, value, valueln);
 
     return param + valueln;
 }

static bool runShellCommand	(	CState *	st,
		char *	variable,
		char **	argv,
		int	argc
	)

static

Definition at line 1700 of file pgbench.c.

References arg, getVariable(), i, NULL, putVariableInt(), and SHELL_COMMAND_SIZE.

Referenced by doCustom().

 {
     char        command[SHELL_COMMAND_SIZE];
     int         i,
                 len = 0;
     FILE       *fp;
     char        res[64];
     char       *endptr;
     int         retval;
 
     /*----------
      * Join arguments with whitespace separators. Arguments starting with
      * exactly one colon are treated as variables:
      *  name - append a string "name"
      *  :var - append a variable named 'var'
      *  ::name - append a string ":name"
      *----------
      */
     for (i = 0; i < argc; i++)
     {
         char       *arg;
         int         arglen;
 
         if (argv[i][0] != ':')
         {
             arg = argv[i];      /* a string literal */
         }
         else if (argv[i][1] == ':')
         {
             arg = argv[i] + 1;  /* a string literal starting with colons */
         }
         else if ((arg = getVariable(st, argv[i] + 1)) == NULL)
         {
             fprintf(stderr, "%s: undefined variable \"%s\"\n",
                     argv[0], argv[i]);
             return false;
         }
 
         arglen = strlen(arg);
         if (len + arglen + (i > 0 ? 1 : 0) >= SHELL_COMMAND_SIZE - 1)
         {
             fprintf(stderr, "%s: shell command is too long\n", argv[0]);
             return false;
         }
 
         if (i > 0)
             command[len++] = ' ';
         memcpy(command + len, arg, arglen);
         len += arglen;
     }
 
     command[len] = '\0';
 
     /* Fast path for non-assignment case */
     if (variable == NULL)
     {
         if (system(command))
         {
             if (!timer_exceeded)
                 fprintf(stderr, "%s: could not launch shell command\n", argv[0]);
             return false;
         }
         return true;
     }
 
     /* Execute the command with pipe and read the standard output. */
     if ((fp = popen(command, "r")) == NULL)
     {
         fprintf(stderr, "%s: could not launch shell command\n", argv[0]);
         return false;
     }
     if (fgets(res, sizeof(res), fp) == NULL)
     {
         if (!timer_exceeded)
             fprintf(stderr, "%s: could not read result of shell command\n", argv[0]);
         (void) pclose(fp);
         return false;
     }
     if (pclose(fp) < 0)
     {
         fprintf(stderr, "%s: could not close shell command\n", argv[0]);
         return false;
     }
 
     /* Check whether the result is an integer and assign it to the variable */
     retval = (int) strtol(res, &endptr, 10);
     while (*endptr != '\0' && isspace((unsigned char) *endptr))
         endptr++;
     if (*res == '\0' || *endptr != '\0')
     {
         fprintf(stderr, "%s: shell command must return an integer (not \"%s\")\n",
                 argv[0], res);
         return false;
     }
     if (!putVariableInt(st, "setshell", variable, retval))
         return false;
 
 #ifdef DEBUG
     printf("shell parameter name: \"%s\", value: \"%s\"\n", argv[1], res);
 #endif
     return true;
 }

Data Structures

Macros

Typedefs

Enumerations

Functions

Variables

Macro Definition Documentation

Typedef Documentation

Enumeration Type Documentation

Function Documentation