LCOV - code coverage report
Current view: top level - src/bin/pgbench - pgbench.c (source / functions) Hit Total Coverage
Test: PostgreSQL 11devel Lines: 1551 1785 86.9 %
Date: 2017-11-22 12:18:04 Functions: 70 73 95.9 %
Legend: Lines: hit not hit | Branches: + taken - not taken # not executed Branches: 901 1166 77.3 %

           Branch data     Line data    Source code
       1                 :            : /*
       2                 :            :  * pgbench.c
       3                 :            :  *
       4                 :            :  * A simple benchmark program for PostgreSQL
       5                 :            :  * Originally written by Tatsuo Ishii and enhanced by many contributors.
       6                 :            :  *
       7                 :            :  * src/bin/pgbench/pgbench.c
       8                 :            :  * Copyright (c) 2000-2017, PostgreSQL Global Development Group
       9                 :            :  * ALL RIGHTS RESERVED;
      10                 :            :  *
      11                 :            :  * Permission to use, copy, modify, and distribute this software and its
      12                 :            :  * documentation for any purpose, without fee, and without a written agreement
      13                 :            :  * is hereby granted, provided that the above copyright notice and this
      14                 :            :  * paragraph and the following two paragraphs appear in all copies.
      15                 :            :  *
      16                 :            :  * IN NO EVENT SHALL THE AUTHOR OR DISTRIBUTORS BE LIABLE TO ANY PARTY FOR
      17                 :            :  * DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING
      18                 :            :  * LOST PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS
      19                 :            :  * DOCUMENTATION, EVEN IF THE AUTHOR OR DISTRIBUTORS HAVE BEEN ADVISED OF THE
      20                 :            :  * POSSIBILITY OF SUCH DAMAGE.
      21                 :            :  *
      22                 :            :  * THE AUTHOR AND DISTRIBUTORS SPECIFICALLY DISCLAIMS ANY WARRANTIES,
      23                 :            :  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
      24                 :            :  * AND FITNESS FOR A PARTICULAR PURPOSE.  THE SOFTWARE PROVIDED HEREUNDER IS
      25                 :            :  * ON AN "AS IS" BASIS, AND THE AUTHOR AND DISTRIBUTORS HAS NO OBLIGATIONS TO
      26                 :            :  * PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
      27                 :            :  *
      28                 :            :  */
      29                 :            : 
      30                 :            : #ifdef WIN32
      31                 :            : #define FD_SETSIZE 1024         /* set before winsock2.h is included */
      32                 :            : #endif                          /* ! WIN32 */
      33                 :            : 
      34                 :            : #include "postgres_fe.h"
      35                 :            : 
      36                 :            : #include "getopt_long.h"
      37                 :            : #include "libpq-fe.h"
      38                 :            : #include "portability/instr_time.h"
      39                 :            : 
      40                 :            : #include <ctype.h>
      41                 :            : #include <float.h>
      42                 :            : #include <limits.h>
      43                 :            : #include <math.h>
      44                 :            : #include <signal.h>
      45                 :            : #include <time.h>
      46                 :            : #include <sys/time.h>
      47                 :            : #ifdef HAVE_SYS_SELECT_H
      48                 :            : #include <sys/select.h>
      49                 :            : #endif
      50                 :            : 
      51                 :            : #ifdef HAVE_SYS_RESOURCE_H
      52                 :            : #include <sys/resource.h>     /* for getrlimit */
      53                 :            : #endif
      54                 :            : 
      55                 :            : #ifndef M_PI
      56                 :            : #define M_PI 3.14159265358979323846
      57                 :            : #endif
      58                 :            : 
      59                 :            : #include "pgbench.h"
      60                 :            : 
      61                 :            : #define ERRCODE_UNDEFINED_TABLE  "42P01"
      62                 :            : 
      63                 :            : /*
      64                 :            :  * Multi-platform pthread implementations
      65                 :            :  */
      66                 :            : 
      67                 :            : #ifdef WIN32
      68                 :            : /* Use native win32 threads on Windows */
      69                 :            : typedef struct win32_pthread *pthread_t;
      70                 :            : typedef int pthread_attr_t;
      71                 :            : 
      72                 :            : static int  pthread_create(pthread_t *thread, pthread_attr_t *attr, void *(*start_routine) (void *), void *arg);
      73                 :            : static int  pthread_join(pthread_t th, void **thread_return);
      74                 :            : #elif defined(ENABLE_THREAD_SAFETY)
      75                 :            : /* Use platform-dependent pthread capability */
      76                 :            : #include <pthread.h>
      77                 :            : #else
      78                 :            : /* No threads implementation, use none (-j 1) */
      79                 :            : #define pthread_t void *
      80                 :            : #endif
      81                 :            : 
      82                 :            : 
      83                 :            : /********************************************************************
      84                 :            :  * some configurable parameters */
      85                 :            : 
      86                 :            : /* max number of clients allowed */
      87                 :            : #ifdef FD_SETSIZE
      88                 :            : #define MAXCLIENTS  (FD_SETSIZE - 10)
      89                 :            : #else
      90                 :            : #define MAXCLIENTS  1024
      91                 :            : #endif
      92                 :            : 
      93                 :            : #define DEFAULT_INIT_STEPS "dtgvp"    /* default -I setting */
      94                 :            : 
      95                 :            : #define LOG_STEP_SECONDS    5   /* seconds between log messages */
      96                 :            : #define DEFAULT_NXACTS  10      /* default nxacts */
      97                 :            : 
      98                 :            : #define MIN_GAUSSIAN_PARAM      2.0 /* minimum parameter for gauss */
      99                 :            : 
     100                 :            : int         nxacts = 0;         /* number of transactions per client */
     101                 :            : int         duration = 0;       /* duration in seconds */
     102                 :            : int64       end_time = 0;       /* when to stop in micro seconds, under -T */
     103                 :            : 
     104                 :            : /*
     105                 :            :  * scaling factor. for example, scale = 10 will make 1000000 tuples in
     106                 :            :  * pgbench_accounts table.
     107                 :            :  */
     108                 :            : int         scale = 1;
     109                 :            : 
     110                 :            : /*
     111                 :            :  * fillfactor. for example, fillfactor = 90 will use only 90 percent
     112                 :            :  * space during inserts and leave 10 percent free.
     113                 :            :  */
     114                 :            : int         fillfactor = 100;
     115                 :            : 
     116                 :            : /*
     117                 :            :  * use unlogged tables?
     118                 :            :  */
     119                 :            : bool        unlogged_tables = false;
     120                 :            : 
     121                 :            : /*
     122                 :            :  * log sampling rate (1.0 = log everything, 0.0 = option not given)
     123                 :            :  */
     124                 :            : double      sample_rate = 0.0;
     125                 :            : 
     126                 :            : /*
     127                 :            :  * When threads are throttled to a given rate limit, this is the target delay
     128                 :            :  * to reach that rate in usec.  0 is the default and means no throttling.
     129                 :            :  */
     130                 :            : int64       throttle_delay = 0;
     131                 :            : 
     132                 :            : /*
     133                 :            :  * Transactions which take longer than this limit (in usec) are counted as
     134                 :            :  * late, and reported as such, although they are completed anyway. When
     135                 :            :  * throttling is enabled, execution time slots that are more than this late
     136                 :            :  * are skipped altogether, and counted separately.
     137                 :            :  */
     138                 :            : int64       latency_limit = 0;
     139                 :            : 
     140                 :            : /*
     141                 :            :  * tablespace selection
     142                 :            :  */
     143                 :            : char       *tablespace = NULL;
     144                 :            : char       *index_tablespace = NULL;
     145                 :            : 
     146                 :            : /*
     147                 :            :  * end of configurable parameters
     148                 :            :  *********************************************************************/
     149                 :            : 
     150                 :            : #define nbranches   1           /* Makes little sense to change this.  Change
     151                 :            :                                  * -s instead */
     152                 :            : #define ntellers    10
     153                 :            : #define naccounts   100000
     154                 :            : 
     155                 :            : /*
     156                 :            :  * The scale factor at/beyond which 32bit integers are incapable of storing
     157                 :            :  * 64bit values.
     158                 :            :  *
     159                 :            :  * Although the actual threshold is 21474, we use 20000 because it is easier to
     160                 :            :  * document and remember, and isn't that far away from the real threshold.
     161                 :            :  */
     162                 :            : #define SCALE_32BIT_THRESHOLD 20000
     163                 :            : 
     164                 :            : bool        use_log;            /* log transaction latencies to a file */
     165                 :            : bool        use_quiet;          /* quiet logging onto stderr */
     166                 :            : int         agg_interval;       /* log aggregates instead of individual
     167                 :            :                                  * transactions */
     168                 :            : bool        per_script_stats = false;   /* whether to collect stats per script */
     169                 :            : int         progress = 0;       /* thread progress report every this seconds */
     170                 :            : bool        progress_timestamp = false; /* progress report with Unix time */
     171                 :            : int         nclients = 1;       /* number of clients */
     172                 :            : int         nthreads = 1;       /* number of threads */
     173                 :            : bool        is_connect;         /* establish connection for each transaction */
     174                 :            : bool        is_latencies;       /* report per-command latencies */
     175                 :            : int         main_pid;           /* main process id used in log filename */
     176                 :            : 
     177                 :            : char       *pghost = "";
     178                 :            : char       *pgport = "";
     179                 :            : char       *login = NULL;
     180                 :            : char       *dbName;
     181                 :            : char       *logfile_prefix = NULL;
     182                 :            : const char *progname;
     183                 :            : 
     184                 :            : #define WSEP '@'                /* weight separator */
     185                 :            : 
     186                 :            : volatile bool timer_exceeded = false;   /* flag from signal handler */
     187                 :            : 
     188                 :            : /*
     189                 :            :  * Variable definitions.  If a variable has a string value, "value" is that
     190                 :            :  * value, is_numeric is false, and num_value is undefined.  If the value is
     191                 :            :  * known to be numeric, is_numeric is true and num_value contains the value
     192                 :            :  * (in any permitted numeric variant).  In this case "value" contains the
     193                 :            :  * string equivalent of the number, if we've had occasion to compute that,
     194                 :            :  * or NULL if we haven't.
     195                 :            :  */
     196                 :            : typedef struct
     197                 :            : {
     198                 :            :     char       *name;           /* variable's name */
     199                 :            :     char       *value;          /* its value in string form, if known */
     200                 :            :     bool        is_numeric;     /* is numeric value known? */
     201                 :            :     PgBenchValue num_value;     /* variable's value in numeric form */
     202                 :            : } Variable;
     203                 :            : 
     204                 :            : #define MAX_SCRIPTS     128     /* max number of SQL scripts allowed */
     205                 :            : #define SHELL_COMMAND_SIZE  256 /* maximum size allowed for shell command */
     206                 :            : 
     207                 :            : /*
     208                 :            :  * Simple data structure to keep stats about something.
     209                 :            :  *
     210                 :            :  * XXX probably the first value should be kept and used as an offset for
     211                 :            :  * better numerical stability...
     212                 :            :  */
     213                 :            : typedef struct SimpleStats
     214                 :            : {
     215                 :            :     int64       count;          /* how many values were encountered */
     216                 :            :     double      min;            /* the minimum seen */
     217                 :            :     double      max;            /* the maximum seen */
     218                 :            :     double      sum;            /* sum of values */
     219                 :            :     double      sum2;           /* sum of squared values */
     220                 :            : } SimpleStats;
     221                 :            : 
     222                 :            : /*
     223                 :            :  * Data structure to hold various statistics: per-thread and per-script stats
     224                 :            :  * are maintained and merged together.
     225                 :            :  */
     226                 :            : typedef struct StatsData
     227                 :            : {
     228                 :            :     time_t      start_time;     /* interval start time, for aggregates */
     229                 :            :     int64       cnt;            /* number of transactions, including skipped */
     230                 :            :     int64       skipped;        /* number of transactions skipped under --rate
     231                 :            :                                  * and --latency-limit */
     232                 :            :     SimpleStats latency;
     233                 :            :     SimpleStats lag;
     234                 :            : } StatsData;
     235                 :            : 
     236                 :            : /*
     237                 :            :  * Connection state machine states.
     238                 :            :  */
     239                 :            : typedef enum
     240                 :            : {
     241                 :            :     /*
     242                 :            :      * The client must first choose a script to execute.  Once chosen, it can
     243                 :            :      * either be throttled (state CSTATE_START_THROTTLE under --rate) or start
     244                 :            :      * right away (state CSTATE_START_TX).
     245                 :            :      */
     246                 :            :     CSTATE_CHOOSE_SCRIPT,
     247                 :            : 
     248                 :            :     /*
     249                 :            :      * In CSTATE_START_THROTTLE state, we calculate when to begin the next
     250                 :            :      * transaction, and advance to CSTATE_THROTTLE.  CSTATE_THROTTLE state
     251                 :            :      * sleeps until that moment.  (If throttling is not enabled, doCustom()
     252                 :            :      * falls directly through from CSTATE_START_THROTTLE to CSTATE_START_TX.)
     253                 :            :      */
     254                 :            :     CSTATE_START_THROTTLE,
     255                 :            :     CSTATE_THROTTLE,
     256                 :            : 
     257                 :            :     /*
     258                 :            :      * CSTATE_START_TX performs start-of-transaction processing.  Establishes
     259                 :            :      * a new connection for the transaction, in --connect mode, and records
     260                 :            :      * the transaction start time.
     261                 :            :      */
     262                 :            :     CSTATE_START_TX,
     263                 :            : 
     264                 :            :     /*
     265                 :            :      * We loop through these states, to process each command in the script:
     266                 :            :      *
     267                 :            :      * CSTATE_START_COMMAND starts the execution of a command.  On a SQL
     268                 :            :      * command, the command is sent to the server, and we move to
     269                 :            :      * CSTATE_WAIT_RESULT state.  On a \sleep meta-command, the timer is set,
     270                 :            :      * and we enter the CSTATE_SLEEP state to wait for it to expire. Other
     271                 :            :      * meta-commands are executed immediately.
     272                 :            :      *
     273                 :            :      * CSTATE_WAIT_RESULT waits until we get a result set back from the server
     274                 :            :      * for the current command.
     275                 :            :      *
     276                 :            :      * CSTATE_SLEEP waits until the end of \sleep.
     277                 :            :      *
     278                 :            :      * CSTATE_END_COMMAND records the end-of-command timestamp, increments the
     279                 :            :      * command counter, and loops back to CSTATE_START_COMMAND state.
     280                 :            :      */
     281                 :            :     CSTATE_START_COMMAND,
     282                 :            :     CSTATE_WAIT_RESULT,
     283                 :            :     CSTATE_SLEEP,
     284                 :            :     CSTATE_END_COMMAND,
     285                 :            : 
     286                 :            :     /*
     287                 :            :      * CSTATE_END_TX performs end-of-transaction processing.  Calculates
     288                 :            :      * latency, and logs the transaction.  In --connect mode, closes the
     289                 :            :      * current connection.  Chooses the next script to execute and starts over
     290                 :            :      * in CSTATE_START_THROTTLE state, or enters CSTATE_FINISHED if we have no
     291                 :            :      * more work to do.
     292                 :            :      */
     293                 :            :     CSTATE_END_TX,
     294                 :            : 
     295                 :            :     /*
     296                 :            :      * Final states.  CSTATE_ABORTED means that the script execution was
     297                 :            :      * aborted because a command failed, CSTATE_FINISHED means success.
     298                 :            :      */
     299                 :            :     CSTATE_ABORTED,
     300                 :            :     CSTATE_FINISHED
     301                 :            : } ConnectionStateEnum;
     302                 :            : 
     303                 :            : /*
     304                 :            :  * Connection state.
     305                 :            :  */
     306                 :            : typedef struct
     307                 :            : {
     308                 :            :     PGconn     *con;            /* connection handle to DB */
     309                 :            :     int         id;             /* client No. */
     310                 :            :     ConnectionStateEnum state;  /* state machine's current state. */
     311                 :            : 
     312                 :            :     int         use_file;       /* index in sql_script for this client */
     313                 :            :     int         command;        /* command number in script */
     314                 :            : 
     315                 :            :     /* client variables */
     316                 :            :     Variable   *variables;      /* array of variable definitions */
     317                 :            :     int         nvariables;     /* number of variables */
     318                 :            :     bool        vars_sorted;    /* are variables sorted by name? */
     319                 :            : 
     320                 :            :     /* various times about current transaction */
     321                 :            :     int64       txn_scheduled;  /* scheduled start time of transaction (usec) */
     322                 :            :     int64       sleep_until;    /* scheduled start time of next cmd (usec) */
     323                 :            :     instr_time  txn_begin;      /* used for measuring schedule lag times */
     324                 :            :     instr_time  stmt_begin;     /* used for measuring statement latencies */
     325                 :            : 
     326                 :            :     bool        prepared[MAX_SCRIPTS];  /* whether client prepared the script */
     327                 :            : 
     328                 :            :     /* per client collected stats */
     329                 :            :     int64       cnt;            /* client transaction count, for -t */
     330                 :            :     int         ecnt;           /* error count */
     331                 :            : } CState;
     332                 :            : 
     333                 :            : /*
     334                 :            :  * Thread state
     335                 :            :  */
     336                 :            : typedef struct
     337                 :            : {
     338                 :            :     int         tid;            /* thread id */
     339                 :            :     pthread_t   thread;         /* thread handle */
     340                 :            :     CState     *state;          /* array of CState */
     341                 :            :     int         nstate;         /* length of state[] */
     342                 :            :     unsigned short random_state[3]; /* separate randomness for each thread */
     343                 :            :     int64       throttle_trigger;   /* previous/next throttling (us) */
     344                 :            :     FILE       *logfile;        /* where to log, or NULL */
     345                 :            : 
     346                 :            :     /* per thread collected stats */
     347                 :            :     instr_time  start_time;     /* thread start time */
     348                 :            :     instr_time  conn_time;
     349                 :            :     StatsData   stats;
     350                 :            :     int64       latency_late;   /* executed but late transactions */
     351                 :            : } TState;
     352                 :            : 
     353                 :            : #define INVALID_THREAD      ((pthread_t) 0)
     354                 :            : 
     355                 :            : /*
     356                 :            :  * queries read from files
     357                 :            :  */
     358                 :            : #define SQL_COMMAND     1
     359                 :            : #define META_COMMAND    2
     360                 :            : #define MAX_ARGS        10
     361                 :            : 
     362                 :            : typedef enum MetaCommand
     363                 :            : {
     364                 :            :     META_NONE,                  /* not a known meta-command */
     365                 :            :     META_SET,                   /* \set */
     366                 :            :     META_SETSHELL,              /* \setshell */
     367                 :            :     META_SHELL,                 /* \shell */
     368                 :            :     META_SLEEP                  /* \sleep */
     369                 :            : } MetaCommand;
     370                 :            : 
     371                 :            : typedef enum QueryMode
     372                 :            : {
     373                 :            :     QUERY_SIMPLE,               /* simple query */
     374                 :            :     QUERY_EXTENDED,             /* extended query */
     375                 :            :     QUERY_PREPARED,             /* extended query with prepared statements */
     376                 :            :     NUM_QUERYMODE
     377                 :            : } QueryMode;
     378                 :            : 
     379                 :            : static QueryMode querymode = QUERY_SIMPLE;
     380                 :            : static const char *QUERYMODE[] = {"simple", "extended", "prepared"};
     381                 :            : 
     382                 :            : typedef struct
     383                 :            : {
     384                 :            :     char       *line;           /* text of command line */
     385                 :            :     int         command_num;    /* unique index of this Command struct */
     386                 :            :     int         type;           /* command type (SQL_COMMAND or META_COMMAND) */
     387                 :            :     MetaCommand meta;           /* meta command identifier, or META_NONE */
     388                 :            :     int         argc;           /* number of command words */
     389                 :            :     char       *argv[MAX_ARGS]; /* command word list */
     390                 :            :     PgBenchExpr *expr;          /* parsed expression, if needed */
     391                 :            :     SimpleStats stats;          /* time spent in this command */
     392                 :            : } Command;
     393                 :            : 
     394                 :            : typedef struct ParsedScript
     395                 :            : {
     396                 :            :     const char *desc;           /* script descriptor (eg, file name) */
     397                 :            :     int         weight;         /* selection weight */
     398                 :            :     Command   **commands;       /* NULL-terminated array of Commands */
     399                 :            :     StatsData   stats;          /* total time spent in script */
     400                 :            : } ParsedScript;
     401                 :            : 
     402                 :            : static ParsedScript sql_script[MAX_SCRIPTS];    /* SQL script files */
     403                 :            : static int  num_scripts;        /* number of scripts in sql_script[] */
     404                 :            : static int  num_commands = 0;   /* total number of Command structs */
     405                 :            : static int64 total_weight = 0;
     406                 :            : 
     407                 :            : static int  debug = 0;          /* debug flag */
     408                 :            : 
     409                 :            : /* Builtin test scripts */
     410                 :            : typedef struct BuiltinScript
     411                 :            : {
     412                 :            :     const char *name;           /* very short name for -b ... */
     413                 :            :     const char *desc;           /* short description */
     414                 :            :     const char *script;         /* actual pgbench script */
     415                 :            : } BuiltinScript;
     416                 :            : 
     417                 :            : static const BuiltinScript builtin_script[] =
     418                 :            : {
     419                 :            :     {
     420                 :            :         "tpcb-like",
     421                 :            :         "<builtin: TPC-B (sort of)>",
     422                 :            :         "\\set aid random(1, " CppAsString2(naccounts) " * :scale)\n"
     423                 :            :         "\\set bid random(1, " CppAsString2(nbranches) " * :scale)\n"
     424                 :            :         "\\set tid random(1, " CppAsString2(ntellers) " * :scale)\n"
     425                 :            :         "\\set delta random(-5000, 5000)\n"
     426                 :            :         "BEGIN;\n"
     427                 :            :         "UPDATE pgbench_accounts SET abalance = abalance + :delta WHERE aid = :aid;\n"
     428                 :            :         "SELECT abalance FROM pgbench_accounts WHERE aid = :aid;\n"
     429                 :            :         "UPDATE pgbench_tellers SET tbalance = tbalance + :delta WHERE tid = :tid;\n"
     430                 :            :         "UPDATE pgbench_branches SET bbalance = bbalance + :delta WHERE bid = :bid;\n"
     431                 :            :         "INSERT INTO pgbench_history (tid, bid, aid, delta, mtime) VALUES (:tid, :bid, :aid, :delta, CURRENT_TIMESTAMP);\n"
     432                 :            :         "END;\n"
     433                 :            :     },
     434                 :            :     {
     435                 :            :         "simple-update",
     436                 :            :         "<builtin: simple update>",
     437                 :            :         "\\set aid random(1, " CppAsString2(naccounts) " * :scale)\n"
     438                 :            :         "\\set bid random(1, " CppAsString2(nbranches) " * :scale)\n"
     439                 :            :         "\\set tid random(1, " CppAsString2(ntellers) " * :scale)\n"
     440                 :            :         "\\set delta random(-5000, 5000)\n"
     441                 :            :         "BEGIN;\n"
     442                 :            :         "UPDATE pgbench_accounts SET abalance = abalance + :delta WHERE aid = :aid;\n"
     443                 :            :         "SELECT abalance FROM pgbench_accounts WHERE aid = :aid;\n"
     444                 :            :         "INSERT INTO pgbench_history (tid, bid, aid, delta, mtime) VALUES (:tid, :bid, :aid, :delta, CURRENT_TIMESTAMP);\n"
     445                 :            :         "END;\n"
     446                 :            :     },
     447                 :            :     {
     448                 :            :         "select-only",
     449                 :            :         "<builtin: select only>",
     450                 :            :         "\\set aid random(1, " CppAsString2(naccounts) " * :scale)\n"
     451                 :            :         "SELECT abalance FROM pgbench_accounts WHERE aid = :aid;\n"
     452                 :            :     }
     453                 :            : };
     454                 :            : 
     455                 :            : 
     456                 :            : /* Function prototypes */
     457                 :            : static void setIntValue(PgBenchValue *pv, int64 ival);
     458                 :            : static void setDoubleValue(PgBenchValue *pv, double dval);
     459                 :            : static bool evaluateExpr(TState *, CState *, PgBenchExpr *, PgBenchValue *);
     460                 :            : static void doLog(TState *thread, CState *st,
     461                 :            :       StatsData *agg, bool skipped, double latency, double lag);
     462                 :            : static void processXactStats(TState *thread, CState *st, instr_time *now,
     463                 :            :                  bool skipped, StatsData *agg);
     464                 :            : static void pgbench_error(const char *fmt,...) pg_attribute_printf(1, 2);
     465                 :            : static void addScript(ParsedScript script);
     466                 :            : static void *threadRun(void *arg);
     467                 :            : static void setalarm(int seconds);
     468                 :            : 
     469                 :            : 
     470                 :            : /* callback functions for our flex lexer */
     471                 :            : static const PsqlScanCallbacks pgbench_callbacks = {
     472                 :            :     NULL,                       /* don't need get_variable functionality */
     473                 :            :     pgbench_error
     474                 :            : };
     475                 :            : 
     476                 :            : 
     477                 :            : static void
     478                 :          2 : usage(void)
     479                 :            : {
     480                 :          2 :     printf("%s is a benchmarking tool for PostgreSQL.\n\n"
     481                 :            :            "Usage:\n"
     482                 :            :            "  %s [OPTION]... [DBNAME]\n"
     483                 :            :            "\nInitialization options:\n"
     484                 :            :            "  -i, --initialize         invokes initialization mode\n"
     485                 :            :            "  -I, --init-steps=[dtgvpf]+ (default \"dtgvp\")\n"
     486                 :            :            "                           run selected initialization steps\n"
     487                 :            :            "  -F, --fillfactor=NUM     set fill factor\n"
     488                 :            :            "  -n, --no-vacuum          do not run VACUUM during initialization\n"
     489                 :            :            "  -q, --quiet              quiet logging (one message each 5 seconds)\n"
     490                 :            :            "  -s, --scale=NUM          scaling factor\n"
     491                 :            :            "  --foreign-keys           create foreign key constraints between tables\n"
     492                 :            :            "  --index-tablespace=TABLESPACE\n"
     493                 :            :            "                           create indexes in the specified tablespace\n"
     494                 :            :            "  --tablespace=TABLESPACE  create tables in the specified tablespace\n"
     495                 :            :            "  --unlogged-tables        create tables as unlogged tables\n"
     496                 :            :            "\nOptions to select what to run:\n"
     497                 :            :            "  -b, --builtin=NAME[@W]   add builtin script NAME weighted at W (default: 1)\n"
     498                 :            :            "                           (use \"-b list\" to list available scripts)\n"
     499                 :            :            "  -f, --file=FILENAME[@W]  add script FILENAME weighted at W (default: 1)\n"
     500                 :            :            "  -N, --skip-some-updates  skip updates of pgbench_tellers and pgbench_branches\n"
     501                 :            :            "                           (same as \"-b simple-update\")\n"
     502                 :            :            "  -S, --select-only        perform SELECT-only transactions\n"
     503                 :            :            "                           (same as \"-b select-only\")\n"
     504                 :            :            "\nBenchmarking options:\n"
     505                 :            :            "  -c, --client=NUM         number of concurrent database clients (default: 1)\n"
     506                 :            :            "  -C, --connect            establish new connection for each transaction\n"
     507                 :            :            "  -D, --define=VARNAME=VALUE\n"
     508                 :            :            "                           define variable for use by custom script\n"
     509                 :            :            "  -j, --jobs=NUM           number of threads (default: 1)\n"
     510                 :            :            "  -l, --log                write transaction times to log file\n"
     511                 :            :            "  -L, --latency-limit=NUM  count transactions lasting more than NUM ms as late\n"
     512                 :            :            "  -M, --protocol=simple|extended|prepared\n"
     513                 :            :            "                           protocol for submitting queries (default: simple)\n"
     514                 :            :            "  -n, --no-vacuum          do not run VACUUM before tests\n"
     515                 :            :            "  -P, --progress=NUM       show thread progress report every NUM seconds\n"
     516                 :            :            "  -r, --report-latencies   report average latency per command\n"
     517                 :            :            "  -R, --rate=NUM           target rate in transactions per second\n"
     518                 :            :            "  -s, --scale=NUM          report this scale factor in output\n"
     519                 :            :            "  -t, --transactions=NUM   number of transactions each client runs (default: 10)\n"
     520                 :            :            "  -T, --time=NUM           duration of benchmark test in seconds\n"
     521                 :            :            "  -v, --vacuum-all         vacuum all four standard tables before tests\n"
     522                 :            :            "  --aggregate-interval=NUM aggregate data over NUM seconds\n"
     523                 :            :            "  --log-prefix=PREFIX      prefix for transaction time log file\n"
     524                 :            :            "                           (default: \"pgbench_log\")\n"
     525                 :            :            "  --progress-timestamp     use Unix epoch timestamps for progress\n"
     526                 :            :            "  --sampling-rate=NUM      fraction of transactions to log (e.g., 0.01 for 1%%)\n"
     527                 :            :            "\nCommon options:\n"
     528                 :            :            "  -d, --debug              print debugging output\n"
     529                 :            :            "  -h, --host=HOSTNAME      database server host or socket directory\n"
     530                 :            :            "  -p, --port=PORT          database server port number\n"
     531                 :            :            "  -U, --username=USERNAME  connect as specified database user\n"
     532                 :            :            "  -V, --version            output version information, then exit\n"
     533                 :            :            "  -?, --help               show this help, then exit\n"
     534                 :            :            "\n"
     535                 :            :            "Report bugs to <pgsql-bugs@postgresql.org>.\n",
     536                 :            :            progname, progname);
     537                 :          2 : }
     538                 :            : 
     539                 :            : /* return whether str matches "^\s*[-+]?[0-9]+$" */
     540                 :            : static bool
     541                 :          6 : is_an_int(const char *str)
     542                 :            : {
     543                 :          6 :     const char *ptr = str;
     544                 :            : 
     545                 :            :     /* skip leading spaces; cast is consistent with strtoint64 */
     546 [ +  - ][ -  + ]:          6 :     while (*ptr && isspace((unsigned char) *ptr))
     547                 :          0 :         ptr++;
     548                 :            : 
     549                 :            :     /* skip sign */
     550 [ +  - ][ +  + ]:          6 :     if (*ptr == '+' || *ptr == '-')
     551                 :          4 :         ptr++;
     552                 :            : 
     553                 :            :     /* at least one digit */
     554 [ +  - ][ +  + ]:          6 :     if (*ptr && !isdigit((unsigned char) *ptr))
     555                 :          2 :         return false;
     556                 :            : 
     557                 :            :     /* eat all digits */
     558 [ +  + ][ +  + ]:         46 :     while (*ptr && isdigit((unsigned char) *ptr))
     559                 :         42 :         ptr++;
     560                 :            : 
     561                 :            :     /* must have reached end of string */
     562                 :          4 :     return *ptr == '\0';
     563                 :            : }
     564                 :            : 
     565                 :            : 
     566                 :            : /*
     567                 :            :  * strtoint64 -- convert a string to 64-bit integer
     568                 :            :  *
     569                 :            :  * This function is a modified version of scanint8() from
     570                 :            :  * src/backend/utils/adt/int8.c.
     571                 :            :  */
     572                 :            : int64
     573                 :        958 : strtoint64(const char *str)
     574                 :            : {
     575                 :        958 :     const char *ptr = str;
     576                 :        958 :     int64       result = 0;
     577                 :        958 :     int         sign = 1;
     578                 :            : 
     579                 :            :     /*
     580                 :            :      * Do our own scan, rather than relying on sscanf which might be broken
     581                 :            :      * for long long.
     582                 :            :      */
     583                 :            : 
     584                 :            :     /* skip leading spaces */
     585 [ +  - ][ -  + ]:        958 :     while (*ptr && isspace((unsigned char) *ptr))
     586                 :          0 :         ptr++;
     587                 :            : 
     588                 :            :     /* handle sign */
     589         [ +  + ]:        958 :     if (*ptr == '-')
     590                 :            :     {
     591                 :          2 :         ptr++;
     592                 :            : 
     593                 :            :         /*
     594                 :            :          * Do an explicit check for INT64_MIN.  Ugly though this is, it's
     595                 :            :          * cleaner than trying to get the loop below to handle it portably.
     596                 :            :          */
     597         [ +  - ]:          2 :         if (strncmp(ptr, "9223372036854775808", 19) == 0)
     598                 :            :         {
     599                 :          2 :             result = PG_INT64_MIN;
     600                 :          2 :             ptr += 19;
     601                 :          2 :             goto gotdigits;
     602                 :            :         }
     603                 :          0 :         sign = -1;
     604                 :            :     }
     605         [ -  + ]:        956 :     else if (*ptr == '+')
     606                 :          0 :         ptr++;
     607                 :            : 
     608                 :            :     /* require at least one digit */
     609         [ -  + ]:        956 :     if (!isdigit((unsigned char) *ptr))
     610                 :          0 :         fprintf(stderr, "invalid input syntax for integer: \"%s\"\n", str);
     611                 :            : 
     612                 :            :     /* process digits */
     613 [ +  + ][ +  - ]:       3822 :     while (*ptr && isdigit((unsigned char) *ptr))
     614                 :            :     {
     615                 :       2866 :         int64       tmp = result * 10 + (*ptr++ - '0');
     616                 :            : 
     617         [ +  + ]:       2866 :         if ((tmp / 10) != result)   /* overflow? */
     618                 :          4 :             fprintf(stderr, "value \"%s\" is out of range for type bigint\n", str);
     619                 :       2866 :         result = tmp;
     620                 :            :     }
     621                 :            : 
     622                 :            : gotdigits:
     623                 :            : 
     624                 :            :     /* allow trailing whitespace, but not other trailing chars */
     625 [ -  + ][ #  # ]:        958 :     while (*ptr != '\0' && isspace((unsigned char) *ptr))
     626                 :          0 :         ptr++;
     627                 :            : 
     628         [ -  + ]:        958 :     if (*ptr != '\0')
     629                 :          0 :         fprintf(stderr, "invalid input syntax for integer: \"%s\"\n", str);
     630                 :            : 
     631         [ -  + ]:        958 :     return ((sign < 0) ? -result : result);
     632                 :            : }
     633                 :            : 
     634                 :            : /* random number generator: uniform distribution from min to max inclusive */
     635                 :            : static int64
     636                 :       3240 : getrand(TState *thread, int64 min, int64 max)
     637                 :            : {
     638                 :            :     /*
     639                 :            :      * Odd coding is so that min and max have approximately the same chance of
     640                 :            :      * being selected as do numbers between them.
     641                 :            :      *
     642                 :            :      * pg_erand48() is thread-safe and concurrent, which is why we use it
     643                 :            :      * rather than random(), which in glibc is non-reentrant, and therefore
     644                 :            :      * protected by a mutex, and therefore a bottleneck on machines with many
     645                 :            :      * CPUs.
     646                 :            :      */
     647                 :       3240 :     return min + (int64) ((max - min + 1) * pg_erand48(thread->random_state));
     648                 :            : }
     649                 :            : 
     650                 :            : /*
     651                 :            :  * random number generator: exponential distribution from min to max inclusive.
     652                 :            :  * the parameter is so that the density of probability for the last cut-off max
     653                 :            :  * value is exp(-parameter).
     654                 :            :  */
     655                 :            : static int64
     656                 :          2 : getExponentialRand(TState *thread, int64 min, int64 max, double parameter)
     657                 :            : {
     658                 :            :     double      cut,
     659                 :            :                 uniform,
     660                 :            :                 rand;
     661                 :            : 
     662                 :            :     /* abort if wrong parameter, but must really be checked beforehand */
     663                 :            :     Assert(parameter > 0.0);
     664                 :          2 :     cut = exp(-parameter);
     665                 :            :     /* erand in [0, 1), uniform in (0, 1] */
     666                 :          2 :     uniform = 1.0 - pg_erand48(thread->random_state);
     667                 :            : 
     668                 :            :     /*
     669                 :            :      * inner expression in (cut, 1] (if parameter > 0), rand in [0, 1)
     670                 :            :      */
     671                 :            :     Assert((1.0 - cut) != 0.0);
     672                 :          2 :     rand = -log(cut + (1.0 - cut) * uniform) / parameter;
     673                 :            :     /* return int64 random number within between min and max */
     674                 :          2 :     return min + (int64) ((max - min + 1) * rand);
     675                 :            : }
     676                 :            : 
     677                 :            : /* random number generator: gaussian distribution from min to max inclusive */
     678                 :            : static int64
     679                 :          2 : getGaussianRand(TState *thread, int64 min, int64 max, double parameter)
     680                 :            : {
     681                 :            :     double      stdev;
     682                 :            :     double      rand;
     683                 :            : 
     684                 :            :     /* abort if parameter is too low, but must really be checked beforehand */
     685                 :            :     Assert(parameter >= MIN_GAUSSIAN_PARAM);
     686                 :            : 
     687                 :            :     /*
     688                 :            :      * Get user specified random number from this loop, with -parameter <
     689                 :            :      * stdev <= parameter
     690                 :            :      *
     691                 :            :      * This loop is executed until the number is in the expected range.
     692                 :            :      *
     693                 :            :      * As the minimum parameter is 2.0, the probability of looping is low:
     694                 :            :      * sqrt(-2 ln(r)) <= 2 => r >= e^{-2} ~ 0.135, then when taking the
     695                 :            :      * average sinus multiplier as 2/pi, we have a 8.6% looping probability in
     696                 :            :      * the worst case. For a parameter value of 5.0, the looping probability
     697                 :            :      * is about e^{-5} * 2 / pi ~ 0.43%.
     698                 :            :      */
     699                 :            :     do
     700                 :            :     {
     701                 :            :         /*
     702                 :            :          * pg_erand48 generates [0,1), but for the basic version of the
     703                 :            :          * Box-Muller transform the two uniformly distributed random numbers
     704                 :            :          * are expected in (0, 1] (see
     705                 :            :          * http://en.wikipedia.org/wiki/Box_muller)
     706                 :            :          */
     707                 :          2 :         double      rand1 = 1.0 - pg_erand48(thread->random_state);
     708                 :          2 :         double      rand2 = 1.0 - pg_erand48(thread->random_state);
     709                 :            : 
     710                 :            :         /* Box-Muller basic form transform */
     711                 :          2 :         double      var_sqrt = sqrt(-2.0 * log(rand1));
     712                 :            : 
     713                 :          2 :         stdev = var_sqrt * sin(2.0 * M_PI * rand2);
     714                 :            : 
     715                 :            :         /*
     716                 :            :          * we may try with cos, but there may be a bias induced if the
     717                 :            :          * previous value fails the test. To be on the safe side, let us try
     718                 :            :          * over.
     719                 :            :          */
     720                 :            :     }
     721 [ -  + ][ -  + ]:          2 :     while (stdev < -parameter || stdev >= parameter);
     722                 :            : 
     723                 :            :     /* stdev is in [-parameter, parameter), normalization to [0,1) */
     724                 :          2 :     rand = (stdev + parameter) / (parameter * 2.0);
     725                 :            : 
     726                 :            :     /* return int64 random number within between min and max */
     727                 :          2 :     return min + (int64) ((max - min + 1) * rand);
     728                 :            : }
     729                 :            : 
     730                 :            : /*
     731                 :            :  * random number generator: generate a value, such that the series of values
     732                 :            :  * will approximate a Poisson distribution centered on the given value.
     733                 :            :  */
     734                 :            : static int64
     735                 :        422 : getPoissonRand(TState *thread, int64 center)
     736                 :            : {
     737                 :            :     /*
     738                 :            :      * Use inverse transform sampling to generate a value > 0, such that the
     739                 :            :      * expected (i.e. average) value is the given argument.
     740                 :            :      */
     741                 :            :     double      uniform;
     742                 :            : 
     743                 :            :     /* erand in [0, 1), uniform in (0, 1] */
     744                 :        422 :     uniform = 1.0 - pg_erand48(thread->random_state);
     745                 :            : 
     746                 :        422 :     return (int64) (-log(uniform) * ((double) center) + 0.5);
     747                 :            : }
     748                 :            : 
     749                 :            : /*
     750                 :            :  * Initialize the given SimpleStats struct to all zeroes
     751                 :            :  */
     752                 :            : static void
     753                 :       2146 : initSimpleStats(SimpleStats *ss)
     754                 :            : {
     755                 :       2146 :     memset(ss, 0, sizeof(SimpleStats));
     756                 :       2146 : }
     757                 :            : 
     758                 :            : /*
     759                 :            :  * Accumulate one value into a SimpleStats struct.
     760                 :            :  */
     761                 :            : static void
     762                 :       1800 : addToSimpleStats(SimpleStats *ss, double val)
     763                 :            : {
     764 [ +  + ][ +  + ]:       1800 :     if (ss->count == 0 || val < ss->min)
     765                 :         56 :         ss->min = val;
     766 [ +  + ][ +  + ]:       1800 :     if (ss->count == 0 || val > ss->max)
     767                 :        786 :         ss->max = val;
     768                 :       1800 :     ss->count++;
     769                 :       1800 :     ss->sum += val;
     770                 :       1800 :     ss->sum2 += val * val;
     771                 :       1800 : }
     772                 :            : 
     773                 :            : /*
     774                 :            :  * Merge two SimpleStats objects
     775                 :            :  */
     776                 :            : static void
     777                 :        120 : mergeSimpleStats(SimpleStats *acc, SimpleStats *ss)
     778                 :            : {
     779 [ -  + ][ #  # ]:        120 :     if (acc->count == 0 || ss->min < acc->min)
     780                 :        120 :         acc->min = ss->min;
     781 [ -  + ][ #  # ]:        120 :     if (acc->count == 0 || ss->max > acc->max)
     782                 :        120 :         acc->max = ss->max;
     783                 :        120 :     acc->count += ss->count;
     784                 :        120 :     acc->sum += ss->sum;
     785                 :        120 :     acc->sum2 += ss->sum2;
     786                 :        120 : }
     787                 :            : 
     788                 :            : /*
     789                 :            :  * Initialize a StatsData struct to mostly zeroes, with its start time set to
     790                 :            :  * the given value.
     791                 :            :  */
     792                 :            : static void
     793                 :        562 : initStats(StatsData *sd, time_t start_time)
     794                 :            : {
     795                 :        562 :     sd->start_time = start_time;
     796                 :        562 :     sd->cnt = 0;
     797                 :        562 :     sd->skipped = 0;
     798                 :        562 :     initSimpleStats(&sd->latency);
     799                 :        562 :     initSimpleStats(&sd->lag);
     800                 :        562 : }
     801                 :            : 
     802                 :            : /*
     803                 :            :  * Accumulate one additional item into the given stats object.
     804                 :            :  */
     805                 :            : static void
     806                 :        620 : accumStats(StatsData *stats, bool skipped, double lat, double lag)
     807                 :            : {
     808                 :        620 :     stats->cnt++;
     809                 :            : 
     810         [ +  + ]:        620 :     if (skipped)
     811                 :            :     {
     812                 :            :         /* no latency to record on skipped transactions */
     813                 :         20 :         stats->skipped++;
     814                 :            :     }
     815                 :            :     else
     816                 :            :     {
     817                 :        600 :         addToSimpleStats(&stats->latency, lat);
     818                 :            : 
     819                 :            :         /* and possibly the same for schedule lag */
     820         [ +  + ]:        600 :         if (throttle_delay)
     821                 :        400 :             addToSimpleStats(&stats->lag, lag);
     822                 :            :     }
     823                 :        620 : }
     824                 :            : 
     825                 :            : /* call PQexec() and exit() on failure */
     826                 :            : static void
     827                 :        168 : executeStatement(PGconn *con, const char *sql)
     828                 :            : {
     829                 :            :     PGresult   *res;
     830                 :            : 
     831                 :        168 :     res = PQexec(con, sql);
     832         [ -  + ]:        168 :     if (PQresultStatus(res) != PGRES_COMMAND_OK)
     833                 :            :     {
     834                 :          0 :         fprintf(stderr, "%s", PQerrorMessage(con));
     835                 :          0 :         exit(1);
     836                 :            :     }
     837                 :        168 :     PQclear(res);
     838                 :        168 : }
     839                 :            : 
     840                 :            : /* call PQexec() and complain, but without exiting, on failure */
     841                 :            : static void
     842                 :         18 : tryExecuteStatement(PGconn *con, const char *sql)
     843                 :            : {
     844                 :            :     PGresult   *res;
     845                 :            : 
     846                 :         18 :     res = PQexec(con, sql);
     847         [ -  + ]:         18 :     if (PQresultStatus(res) != PGRES_COMMAND_OK)
     848                 :            :     {
     849                 :          0 :         fprintf(stderr, "%s", PQerrorMessage(con));
     850                 :          0 :         fprintf(stderr, "(ignoring this error and continuing anyway)\n");
     851                 :            :     }
     852                 :         18 :     PQclear(res);
     853                 :         18 : }
     854                 :            : 
     855                 :            : /* set up a connection to the backend */
     856                 :            : static PGconn *
     857                 :        372 : doConnect(void)
     858                 :            : {
     859                 :            :     PGconn     *conn;
     860                 :            :     bool        new_pass;
     861                 :            :     static bool have_password = false;
     862                 :            :     static char password[100];
     863                 :            : 
     864                 :            :     /*
     865                 :            :      * Start the connection.  Loop until we have a password if requested by
     866                 :            :      * backend.
     867                 :            :      */
     868                 :            :     do
     869                 :            :     {
     870                 :            : #define PARAMS_ARRAY_SIZE   7
     871                 :            : 
     872                 :            :         const char *keywords[PARAMS_ARRAY_SIZE];
     873                 :            :         const char *values[PARAMS_ARRAY_SIZE];
     874                 :            : 
     875                 :        372 :         keywords[0] = "host";
     876                 :        372 :         values[0] = pghost;
     877                 :        372 :         keywords[1] = "port";
     878                 :        372 :         values[1] = pgport;
     879                 :        372 :         keywords[2] = "user";
     880                 :        372 :         values[2] = login;
     881                 :        372 :         keywords[3] = "password";
     882         [ -  + ]:        372 :         values[3] = have_password ? password : NULL;
     883                 :        372 :         keywords[4] = "dbname";
     884                 :        372 :         values[4] = dbName;
     885                 :        372 :         keywords[5] = "fallback_application_name";
     886                 :        372 :         values[5] = progname;
     887                 :        372 :         keywords[6] = NULL;
     888                 :        372 :         values[6] = NULL;
     889                 :            : 
     890                 :        372 :         new_pass = false;
     891                 :            : 
     892                 :        372 :         conn = PQconnectdbParams(keywords, values, true);
     893                 :            : 
     894         [ -  + ]:        372 :         if (!conn)
     895                 :            :         {
     896                 :          0 :             fprintf(stderr, "connection to database \"%s\" failed\n",
     897                 :            :                     dbName);
     898                 :          0 :             return NULL;
     899                 :            :         }
     900                 :            : 
     901   [ +  +  -  + ]:        374 :         if (PQstatus(conn) == CONNECTION_BAD &&
     902         [ #  # ]:          2 :             PQconnectionNeedsPassword(conn) &&
     903                 :          0 :             !have_password)
     904                 :            :         {
     905                 :          0 :             PQfinish(conn);
     906                 :          0 :             simple_prompt("Password: ", password, sizeof(password), false);
     907                 :          0 :             have_password = true;
     908                 :          0 :             new_pass = true;
     909                 :            :         }
     910         [ -  + ]:        372 :     } while (new_pass);
     911                 :            : 
     912                 :            :     /* check to see that the backend connection was successfully made */
     913         [ +  + ]:        372 :     if (PQstatus(conn) == CONNECTION_BAD)
     914                 :            :     {
     915                 :          2 :         fprintf(stderr, "connection to database \"%s\" failed:\n%s",
     916                 :            :                 dbName, PQerrorMessage(conn));
     917                 :          2 :         PQfinish(conn);
     918                 :          2 :         return NULL;
     919                 :            :     }
     920                 :            : 
     921                 :        370 :     return conn;
     922                 :            : }
     923                 :            : 
     924                 :            : /* throw away response from backend */
     925                 :            : static void
     926                 :       4034 : discard_response(CState *state)
     927                 :            : {
     928                 :            :     PGresult   *res;
     929                 :            : 
     930                 :            :     do
     931                 :            :     {
     932                 :       4034 :         res = PQgetResult(state->con);
     933         [ -  + ]:       4034 :         if (res)
     934                 :          0 :             PQclear(res);
     935         [ -  + ]:       4034 :     } while (res);
     936                 :       4034 : }
     937                 :            : 
     938                 :            : /* qsort comparator for Variable array */
     939                 :            : static int
     940                 :      26380 : compareVariableNames(const void *v1, const void *v2)
     941                 :            : {
     942                 :      26380 :     return strcmp(((const Variable *) v1)->name,
     943                 :      26380 :                   ((const Variable *) v2)->name);
     944                 :            : }
     945                 :            : 
     946                 :            : /* Locate a variable by name; returns NULL if unknown */
     947                 :            : static Variable *
     948                 :      10586 : lookupVariable(CState *st, char *name)
     949                 :            : {
     950                 :            :     Variable    key;
     951                 :            : 
     952                 :            :     /* On some versions of Solaris, bsearch of zero items dumps core */
     953         [ +  + ]:      10586 :     if (st->nvariables <= 0)
     954                 :        146 :         return NULL;
     955                 :            : 
     956                 :            :     /* Sort if we have to */
     957         [ +  + ]:      10440 :     if (!st->vars_sorted)
     958                 :            :     {
     959                 :        340 :         qsort((void *) st->variables, st->nvariables, sizeof(Variable),
     960                 :            :               compareVariableNames);
     961                 :        340 :         st->vars_sorted = true;
     962                 :            :     }
     963                 :            : 
     964                 :            :     /* Now we can search */
     965                 :      10440 :     key.name = name;
     966                 :      10586 :     return (Variable *) bsearch((void *) &key,
     967                 :      10440 :                                 (void *) st->variables,
     968                 :      10440 :                                 st->nvariables,
     969                 :            :                                 sizeof(Variable),
     970                 :            :                                 compareVariableNames);
     971                 :            : }
     972                 :            : 
     973                 :            : /* Get the value of a variable, in string form; returns NULL if unknown */
     974                 :            : static char *
     975                 :       4154 : getVariable(CState *st, char *name)
     976                 :            : {
     977                 :            :     Variable   *var;
     978                 :            :     char        stringform[64];
     979                 :            : 
     980                 :       4154 :     var = lookupVariable(st, name);
     981         [ +  + ]:       4154 :     if (var == NULL)
     982                 :          4 :         return NULL;            /* not found */
     983                 :            : 
     984         [ +  + ]:       4150 :     if (var->value)
     985                 :        982 :         return var->value;       /* we have it in string form */
     986                 :            : 
     987                 :            :     /* We need to produce a string equivalent of the numeric value */
     988                 :            :     Assert(var->is_numeric);
     989         [ +  + ]:       3168 :     if (var->num_value.type == PGBT_INT)
     990                 :       3166 :         snprintf(stringform, sizeof(stringform),
     991                 :            :                  INT64_FORMAT, var->num_value.u.ival);
     992                 :            :     else
     993                 :            :     {
     994                 :            :         Assert(var->num_value.type == PGBT_DOUBLE);
     995                 :          2 :         snprintf(stringform, sizeof(stringform),
     996                 :            :                  "%.*g", DBL_DIG, var->num_value.u.dval);
     997                 :            :     }
     998                 :       3168 :     var->value = pg_strdup(stringform);
     999                 :       4154 :     return var->value;
    1000                 :            : }
    1001                 :            : 
    1002                 :            : /* Try to convert variable to numeric form; return false on failure */
    1003                 :            : static bool
    1004                 :       2830 : makeVariableNumeric(Variable *var)
    1005                 :            : {
    1006         [ +  + ]:       2830 :     if (var->is_numeric)
    1007                 :       2824 :         return true;            /* no work */
    1008                 :            : 
    1009         [ +  + ]:          6 :     if (is_an_int(var->value))
    1010                 :            :     {
    1011                 :          2 :         setIntValue(&var->num_value, strtoint64(var->value));
    1012                 :          2 :         var->is_numeric = true;
    1013                 :            :     }
    1014                 :            :     else                        /* type should be double */
    1015                 :            :     {
    1016                 :            :         double      dv;
    1017                 :            :         char        xs;
    1018                 :            : 
    1019         [ +  + ]:          4 :         if (sscanf(var->value, "%lf%c", &dv, &xs) != 1)
    1020                 :            :         {
    1021                 :          2 :             fprintf(stderr,
    1022                 :            :                     "malformed variable \"%s\" value: \"%s\"\n",
    1023                 :            :                     var->name, var->value);
    1024                 :          2 :             return false;
    1025                 :            :         }
    1026                 :          2 :         setDoubleValue(&var->num_value, dv);
    1027                 :          2 :         var->is_numeric = true;
    1028                 :            :     }
    1029                 :          4 :     return true;
    1030                 :            : }
    1031                 :            : 
    1032                 :            : /*
    1033                 :            :  * Check whether a variable's name is allowed.
    1034                 :            :  *
    1035                 :            :  * We allow any non-ASCII character, as well as ASCII letters, digits, and
    1036                 :            :  * underscore.
    1037                 :            :  *
    1038                 :            :  * Keep this in sync with the definitions of variable name characters in
    1039                 :            :  * "src/fe_utils/psqlscan.l", "src/bin/psql/psqlscanslash.l" and
    1040                 :            :  * "src/bin/pgbench/exprscan.l".  Also see parseVariable(), below.
    1041                 :            :  *
    1042                 :            :  * Note: this static function is copied from "src/bin/psql/variables.c"
    1043                 :            :  */
    1044                 :            : static bool
    1045                 :        408 : valid_variable_name(const char *name)
    1046                 :            : {
    1047                 :        408 :     const unsigned char *ptr = (const unsigned char *) name;
    1048                 :            : 
    1049                 :            :     /* Mustn't be zero-length */
    1050         [ -  + ]:        408 :     if (*ptr == '\0')
    1051                 :          0 :         return false;
    1052                 :            : 
    1053         [ +  + ]:       2388 :     while (*ptr)
    1054                 :            :     {
    1055 [ +  - ][ +  + ]:       1982 :         if (IS_HIGHBIT_SET(*ptr) ||
    1056                 :       1982 :             strchr("ABCDEFGHIJKLMNOPQRSTUVWXYZ" "abcdefghijklmnopqrstuvwxyz"
    1057                 :       1982 :                    "_0123456789", *ptr) != NULL)
    1058                 :       1980 :             ptr++;
    1059                 :            :         else
    1060                 :          2 :             return false;
    1061                 :            :     }
    1062                 :            : 
    1063                 :        406 :     return true;
    1064                 :            : }
    1065                 :            : 
    1066                 :            : /*
    1067                 :            :  * Lookup a variable by name, creating it if need be.
    1068                 :            :  * Caller is expected to assign a value to the variable.
    1069                 :            :  * Returns NULL on failure (bad name).
    1070                 :            :  */
    1071                 :            : static Variable *
    1072                 :       3480 : lookupCreateVariable(CState *st, const char *context, char *name)
    1073                 :            : {
    1074                 :            :     Variable   *var;
    1075                 :            : 
    1076                 :       3480 :     var = lookupVariable(st, name);
    1077         [ +  + ]:       3480 :     if (var == NULL)
    1078                 :            :     {
    1079                 :            :         Variable   *newvars;
    1080                 :            : 
    1081                 :            :         /*
    1082                 :            :          * Check for the name only when declaring a new variable to avoid
    1083                 :            :          * overhead.
    1084                 :            :          */
    1085         [ +  + ]:        408 :         if (!valid_variable_name(name))
    1086                 :            :         {
    1087                 :          2 :             fprintf(stderr, "%s: invalid variable name: \"%s\"\n",
    1088                 :            :                     context, name);
    1089                 :          2 :             return NULL;
    1090                 :            :         }
    1091                 :            : 
    1092                 :            :         /* Create variable at the end of the array */
    1093         [ +  + ]:        406 :         if (st->variables)
    1094                 :        282 :             newvars = (Variable *) pg_realloc(st->variables,
    1095                 :        282 :                                               (st->nvariables + 1) * sizeof(Variable));
    1096                 :            :         else
    1097                 :        124 :             newvars = (Variable *) pg_malloc(sizeof(Variable));
    1098                 :            : 
    1099                 :        406 :         st->variables = newvars;
    1100                 :            : 
    1101                 :        406 :         var = &newvars[st->nvariables];
    1102                 :            : 
    1103                 :        406 :         var->name = pg_strdup(name);
    1104                 :        406 :         var->value = NULL;
    1105                 :            :         /* caller is expected to initialize remaining fields */
    1106                 :            : 
    1107                 :        406 :         st->nvariables++;
    1108                 :            :         /* we don't re-sort the array till we have to */
    1109                 :        406 :         st->vars_sorted = false;
    1110                 :            :     }
    1111                 :            : 
    1112                 :       3478 :     return var;
    1113                 :            : }
    1114                 :            : 
    1115                 :            : /* Assign a string value to a variable, creating it if need be */
    1116                 :            : /* Returns false on failure (bad name) */
    1117                 :            : static bool
    1118                 :         74 : putVariable(CState *st, const char *context, char *name, const char *value)
    1119                 :            : {
    1120                 :            :     Variable   *var;
    1121                 :            :     char       *val;
    1122                 :            : 
    1123                 :         74 :     var = lookupCreateVariable(st, context, name);
    1124         [ -  + ]:         74 :     if (!var)
    1125                 :          0 :         return false;
    1126                 :            : 
    1127                 :            :     /* dup then free, in case value is pointing at this variable */
    1128                 :         74 :     val = pg_strdup(value);
    1129                 :            : 
    1130         [ -  + ]:         74 :     if (var->value)
    1131                 :          0 :         free(var->value);
    1132                 :         74 :     var->value = val;
    1133                 :         74 :     var->is_numeric = false;
    1134                 :            : 
    1135                 :         74 :     return true;
    1136                 :            : }
    1137                 :            : 
    1138                 :            : /* Assign a numeric value to a variable, creating it if need be */
    1139                 :            : /* Returns false on failure (bad name) */
    1140                 :            : static bool
    1141                 :       3406 : putVariableNumber(CState *st, const char *context, char *name,
    1142                 :            :                   const PgBenchValue *value)
    1143                 :            : {
    1144                 :            :     Variable   *var;
    1145                 :            : 
    1146                 :       3406 :     var = lookupCreateVariable(st, context, name);
    1147         [ +  + ]:       3406 :     if (!var)
    1148                 :          2 :         return false;
    1149                 :            : 
    1150         [ +  + ]:       3404 :     if (var->value)
    1151                 :       3070 :         free(var->value);
    1152                 :       3404 :     var->value = NULL;
    1153                 :       3404 :     var->is_numeric = true;
    1154                 :       3404 :     var->num_value = *value;
    1155                 :            : 
    1156                 :       3404 :     return true;
    1157                 :            : }
    1158                 :            : 
    1159                 :            : /* Assign an integer value to a variable, creating it if need be */
    1160                 :            : /* Returns false on failure (bad name) */
    1161                 :            : static bool
    1162                 :        194 : putVariableInt(CState *st, const char *context, char *name, int64 value)
    1163                 :            : {
    1164                 :            :     PgBenchValue val;
    1165                 :            : 
    1166                 :        194 :     setIntValue(&val, value);
    1167                 :        194 :     return putVariableNumber(st, context, name, &val);
    1168                 :            : }
    1169                 :            : 
    1170                 :            : /*
    1171                 :            :  * Parse a possible variable reference (:varname).
    1172                 :            :  *
    1173                 :            :  * "sql" points at a colon.  If what follows it looks like a valid
    1174                 :            :  * variable name, return a malloc'd string containing the variable name,
    1175                 :            :  * and set *eaten to the number of characters consumed.
    1176                 :            :  * Otherwise, return NULL.
    1177                 :            :  */
    1178                 :            : static char *
    1179                 :        936 : parseVariable(const char *sql, int *eaten)
    1180                 :            : {
    1181                 :        936 :     int         i = 0;
    1182                 :            :     char       *name;
    1183                 :            : 
    1184                 :            :     do
    1185                 :            :     {
    1186                 :       3798 :         i++;
    1187         [ +  + ]:       3798 :     } while (IS_HIGHBIT_SET(sql[i]) ||
    1188                 :       3798 :              strchr("ABCDEFGHIJKLMNOPQRSTUVWXYZ" "abcdefghijklmnopqrstuvwxyz"
    1189         [ -  + ]:       3798 :                     "_0123456789", sql[i]) != NULL);
    1190         [ +  + ]:        936 :     if (i == 1)
    1191                 :         30 :         return NULL;            /* no valid variable name chars */
    1192                 :            : 
    1193                 :        906 :     name = pg_malloc(i);
    1194                 :        906 :     memcpy(name, &sql[1], i - 1);
    1195                 :        906 :     name[i - 1] = '\0';
    1196                 :            : 
    1197                 :        906 :     *eaten = i;
    1198                 :        906 :     return name;
    1199                 :            : }
    1200                 :            : 
    1201                 :            : static char *
    1202                 :        904 : replaceVariable(char **sql, char *param, int len, char *value)
    1203                 :            : {
    1204                 :        904 :     int         valueln = strlen(value);
    1205                 :            : 
    1206         [ +  + ]:        904 :     if (valueln > len)
    1207                 :            :     {
    1208                 :        622 :         size_t      offset = param - *sql;
    1209                 :            : 
    1210                 :        622 :         *sql = pg_realloc(*sql, strlen(*sql) - len + valueln + 1);
    1211                 :        622 :         param = *sql + offset;
    1212                 :            :     }
    1213                 :            : 
    1214         [ +  + ]:        904 :     if (valueln != len)
    1215                 :        816 :         memmove(param + valueln, param + len, strlen(param + len) + 1);
    1216                 :        904 :     memcpy(param, value, valueln);
    1217                 :            : 
    1218                 :        904 :     return param + valueln;
    1219                 :            : }
    1220                 :            : 
    1221                 :            : static char *
    1222                 :        820 : assignVariables(CState *st, char *sql)
    1223                 :            : {
    1224                 :            :     char       *p,
    1225                 :            :                *name,
    1226                 :            :                *val;
    1227                 :            : 
    1228                 :        820 :     p = sql;
    1229         [ +  + ]:       1700 :     while ((p = strchr(p, ':')) != NULL)
    1230                 :            :     {
    1231                 :            :         int         eaten;
    1232                 :            : 
    1233                 :        880 :         name = parseVariable(p, &eaten);
    1234         [ +  + ]:        880 :         if (name == NULL)
    1235                 :            :         {
    1236         [ +  + ]:         60 :             while (*p == ':')
    1237                 :            :             {
    1238                 :         40 :                 p++;
    1239                 :            :             }
    1240                 :         20 :             continue;
    1241                 :            :         }
    1242                 :            : 
    1243                 :        860 :         val = getVariable(st, name);
    1244                 :        860 :         free(name);
    1245         [ -  + ]:        860 :         if (val == NULL)
    1246                 :            :         {
    1247                 :          0 :             p++;
    1248                 :          0 :             continue;
    1249                 :            :         }
    1250                 :            : 
    1251                 :        860 :         p = replaceVariable(&sql, p, eaten, val);
    1252                 :            :     }
    1253                 :            : 
    1254                 :        820 :     return sql;
    1255                 :            : }
    1256                 :            : 
    1257                 :            : static void
    1258                 :       3216 : getQueryParams(CState *st, const Command *command, const char **params)
    1259                 :            : {
    1260                 :            :     int         i;
    1261                 :            : 
    1262         [ +  + ]:       6500 :     for (i = 0; i < command->argc - 1; i++)
    1263                 :       3284 :         params[i] = getVariable(st, command->argv[i + 1]);
    1264                 :       3216 : }
    1265                 :            : 
    1266                 :            : /* get a value as an int, tell if there is a problem */
    1267                 :            : static bool
    1268                 :      12320 : coerceToInt(PgBenchValue *pval, int64 *ival)
    1269                 :            : {
    1270         [ +  + ]:      12320 :     if (pval->type == PGBT_INT)
    1271                 :            :     {
    1272                 :      12316 :         *ival = pval->u.ival;
    1273                 :      12316 :         return true;
    1274                 :            :     }
    1275                 :            :     else
    1276                 :            :     {
    1277                 :          4 :         double      dval = pval->u.dval;
    1278                 :            : 
    1279                 :            :         Assert(pval->type == PGBT_DOUBLE);
    1280 [ +  - ][ +  + ]:          4 :         if (dval < PG_INT64_MIN || PG_INT64_MAX < dval)
    1281                 :            :         {
    1282                 :          2 :             fprintf(stderr, "double to int overflow for %f\n", dval);
    1283                 :          2 :             return false;
    1284                 :            :         }
    1285                 :          2 :         *ival = (int64) dval;
    1286                 :          2 :         return true;
    1287                 :            :     }
    1288                 :            : }
    1289                 :            : 
    1290                 :            : /* get a value as a double, or tell if there is a problem */
    1291                 :            : static bool
    1292                 :         88 : coerceToDouble(PgBenchValue *pval, double *dval)
    1293                 :            : {
    1294         [ +  + ]:         88 :     if (pval->type == PGBT_DOUBLE)
    1295                 :            :     {
    1296                 :         66 :         *dval = pval->u.dval;
    1297                 :         66 :         return true;
    1298                 :            :     }
    1299                 :            :     else
    1300                 :            :     {
    1301                 :            :         Assert(pval->type == PGBT_INT);
    1302                 :         22 :         *dval = (double) pval->u.ival;
    1303                 :         22 :         return true;
    1304                 :            :     }
    1305                 :            : }
    1306                 :            : 
    1307                 :            : /* assign an integer value */
    1308                 :            : static void
    1309                 :       6342 : setIntValue(PgBenchValue *pv, int64 ival)
    1310                 :            : {
    1311                 :       6342 :     pv->type = PGBT_INT;
    1312                 :       6342 :     pv->u.ival = ival;
    1313                 :       6342 : }
    1314                 :            : 
    1315                 :            : /* assign a double value */
    1316                 :            : static void
    1317                 :         46 : setDoubleValue(PgBenchValue *pv, double dval)
    1318                 :            : {
    1319                 :         46 :     pv->type = PGBT_DOUBLE;
    1320                 :         46 :     pv->u.dval = dval;
    1321                 :         46 : }
    1322                 :            : 
    1323                 :            : /* maximum number of function arguments */
    1324                 :            : #define MAX_FARGS 16
    1325                 :            : 
    1326                 :            : /*
    1327                 :            :  * Recursive evaluation of functions
    1328                 :            :  */
    1329                 :            : static bool
    1330                 :       6252 : evalFunc(TState *thread, CState *st,
    1331                 :            :          PgBenchFunction func, PgBenchExprLink *args, PgBenchValue *retval)
    1332                 :            : {
    1333                 :            :     /* evaluate all function arguments */
    1334                 :       6252 :     int         nargs = 0;
    1335                 :            :     PgBenchValue vargs[MAX_FARGS];
    1336                 :       6252 :     PgBenchExprLink *l = args;
    1337                 :            : 
    1338 [ +  + ][ +  + ]:      18740 :     for (nargs = 0; nargs < MAX_FARGS && l != NULL; nargs++, l = l->next)
    1339         [ +  + ]:      12490 :         if (!evaluateExpr(thread, st, l->expr, &vargs[nargs]))
    1340                 :          2 :             return false;
    1341                 :            : 
    1342         [ +  + ]:       6250 :     if (l != NULL)
    1343                 :            :     {
    1344                 :          2 :         fprintf(stderr,
    1345                 :            :                 "too many function arguments, maximum is %d\n", MAX_FARGS);
    1346                 :          2 :         return false;
    1347                 :            :     }
    1348                 :            : 
    1349                 :            :     /* then evaluate function */
    1350   [ +  +  +  +  :       6248 :     switch (func)
             +  +  +  +  
                      - ]
    1351                 :            :     {
    1352                 :            :             /* overloaded operators */
    1353                 :            :         case PGBENCH_ADD:
    1354                 :            :         case PGBENCH_SUB:
    1355                 :            :         case PGBENCH_MUL:
    1356                 :            :         case PGBENCH_DIV:
    1357                 :            :         case PGBENCH_MOD:
    1358                 :            :             {
    1359                 :       3130 :                 PgBenchValue *lval = &vargs[0],
    1360                 :       3130 :                            *rval = &vargs[1];
    1361                 :            : 
    1362                 :            :                 Assert(nargs == 2);
    1363                 :            : 
    1364                 :            :                 /* overloaded type management, double if some double */
    1365 [ +  + ][ +  + ]:       3130 :                 if ((lval->type == PGBT_DOUBLE ||
    1366         [ +  - ]:         32 :                      rval->type == PGBT_DOUBLE) && func != PGBENCH_MOD)
    1367                 :          0 :                 {
    1368                 :            :                     double      ld,
    1369                 :            :                                 rd;
    1370                 :            : 
    1371   [ +  -  -  + ]:         64 :                     if (!coerceToDouble(lval, &ld) ||
    1372                 :         32 :                         !coerceToDouble(rval, &rd))
    1373                 :         32 :                         return false;
    1374                 :            : 
    1375   [ +  +  +  +  :         32 :                     switch (func)
                      - ]
    1376                 :            :                     {
    1377                 :            :                         case PGBENCH_ADD:
    1378                 :          2 :                             setDoubleValue(retval, ld + rd);
    1379                 :          2 :                             return true;
    1380                 :            : 
    1381                 :            :                         case PGBENCH_SUB:
    1382                 :         10 :                             setDoubleValue(retval, ld - rd);
    1383                 :         10 :                             return true;
    1384                 :            : 
    1385                 :            :                         case PGBENCH_MUL:
    1386                 :         16 :                             setDoubleValue(retval, ld * rd);
    1387                 :         16 :                             return true;
    1388                 :            : 
    1389                 :            :                         case PGBENCH_DIV:
    1390                 :          4 :                             setDoubleValue(retval, ld / rd);
    1391                 :          4 :                             return true;
    1392                 :            : 
    1393                 :            :                         default:
    1394                 :            :                             /* cannot get here */
    1395                 :            :                             Assert(0);
    1396                 :            :                     }
    1397                 :            :                 }
    1398                 :            :                 else            /* we have integer operands, or % */
    1399                 :            :                 {
    1400                 :            :                     int64       li,
    1401                 :            :                                 ri;
    1402                 :            : 
    1403   [ +  -  -  + ]:       6196 :                     if (!coerceToInt(lval, &li) ||
    1404                 :       3098 :                         !coerceToInt(rval, &ri))
    1405                 :       3098 :                         return false;
    1406                 :            : 
    1407   [ +  +  +  +  :       3098 :                     switch (func)
                      - ]
    1408                 :            :                     {
    1409                 :            :                         case PGBENCH_ADD:
    1410                 :         14 :                             setIntValue(retval, li + ri);
    1411                 :         14 :                             return true;
    1412                 :            : 
    1413                 :            :                         case PGBENCH_SUB:
    1414                 :        248 :                             setIntValue(retval, li - ri);
    1415                 :        248 :                             return true;
    1416                 :            : 
    1417                 :            :                         case PGBENCH_MUL:
    1418                 :       2822 :                             setIntValue(retval, li * ri);
    1419                 :       2822 :                             return true;
    1420                 :            : 
    1421                 :            :                         case PGBENCH_DIV:
    1422                 :            :                         case PGBENCH_MOD:
    1423         [ +  + ]:         14 :                             if (ri == 0)
    1424                 :            :                             {
    1425                 :          2 :                                 fprintf(stderr, "division by zero\n");
    1426                 :          2 :                                 return false;
    1427                 :            :                             }
    1428                 :            :                             /* special handling of -1 divisor */
    1429         [ +  + ]:         12 :                             if (ri == -1)
    1430                 :            :                             {
    1431         [ +  + ]:          6 :                                 if (func == PGBENCH_DIV)
    1432                 :            :                                 {
    1433                 :            :                                     /* overflow check (needed for INT64_MIN) */
    1434         [ +  + ]:          4 :                                     if (li == PG_INT64_MIN)
    1435                 :            :                                     {
    1436                 :          2 :                                         fprintf(stderr, "bigint out of range\n");
    1437                 :          2 :                                         return false;
    1438                 :            :                                     }
    1439                 :            :                                     else
    1440                 :          2 :                                         setIntValue(retval, -li);
    1441                 :            :                                 }
    1442                 :            :                                 else
    1443                 :          2 :                                     setIntValue(retval, 0);
    1444                 :          4 :                                 return true;
    1445                 :            :                             }
    1446                 :            :                             /* else divisor is not -1 */
    1447         [ +  + ]:          6 :                             if (func == PGBENCH_DIV)
    1448                 :          4 :                                 setIntValue(retval, li / ri);
    1449                 :            :                             else    /* func == PGBENCH_MOD */
    1450                 :          2 :                                 setIntValue(retval, li % ri);
    1451                 :            : 
    1452                 :          6 :                             return true;
    1453                 :            : 
    1454                 :            :                         default:
    1455                 :            :                             /* cannot get here */
    1456                 :            :                             Assert(0);
    1457                 :            :                     }
    1458                 :            :                 }
    1459                 :            :             }
    1460                 :            : 
    1461                 :            :             /* no arguments */
    1462                 :            :         case PGBENCH_PI:
    1463                 :          2 :             setDoubleValue(retval, M_PI);
    1464                 :          2 :             return true;
    1465                 :            : 
    1466                 :            :             /* 1 overloaded argument */
    1467                 :            :         case PGBENCH_ABS:
    1468                 :            :             {
    1469                 :          4 :                 PgBenchValue *varg = &vargs[0];
    1470                 :            : 
    1471                 :            :                 Assert(nargs == 1);
    1472                 :            : 
    1473         [ +  + ]:          4 :                 if (varg->type == PGBT_INT)
    1474                 :            :                 {
    1475                 :          2 :                     int64       i = varg->u.ival;
    1476                 :            : 
    1477                 :          2 :                     setIntValue(retval, i < 0 ? -i : i);
    1478                 :            :                 }
    1479                 :            :                 else
    1480                 :            :                 {
    1481                 :          2 :                     double      d = varg->u.dval;
    1482                 :            : 
    1483                 :            :                     Assert(varg->type == PGBT_DOUBLE);
    1484         [ +  - ]:          2 :                     setDoubleValue(retval, d < 0.0 ? -d : d);
    1485                 :            :                 }
    1486                 :            : 
    1487                 :          4 :                 return true;
    1488                 :            :             }
    1489                 :            : 
    1490                 :            :         case PGBENCH_DEBUG:
    1491                 :            :             {
    1492                 :         44 :                 PgBenchValue *varg = &vargs[0];
    1493                 :            : 
    1494                 :            :                 Assert(nargs == 1);
    1495                 :            : 
    1496                 :         44 :                 fprintf(stderr, "debug(script=%d,command=%d): ",
    1497                 :         44 :                         st->use_file, st->command + 1);
    1498                 :            : 
    1499         [ +  + ]:         44 :                 if (varg->type == PGBT_INT)
    1500                 :         28 :                     fprintf(stderr, "int " INT64_FORMAT "\n", varg->u.ival);
    1501                 :            :                 else
    1502                 :            :                 {
    1503                 :            :                     Assert(varg->type == PGBT_DOUBLE);
    1504                 :         16 :                     fprintf(stderr, "double %.*g\n", DBL_DIG, varg->u.dval);
    1505                 :            :                 }
    1506                 :            : 
    1507                 :         44 :                 *retval = *varg;
    1508                 :            : 
    1509                 :         44 :                 return true;
    1510                 :            :             }
    1511                 :            : 
    1512                 :            :             /* 1 double argument */
    1513                 :            :         case PGBENCH_DOUBLE:
    1514                 :            :         case PGBENCH_SQRT:
    1515                 :            :             {
    1516                 :            :                 double      dval;
    1517                 :            : 
    1518                 :            :                 Assert(nargs == 1);
    1519                 :            : 
    1520         [ -  + ]:          4 :                 if (!coerceToDouble(&vargs[0], &dval))
    1521                 :          0 :                     return false;
    1522                 :            : 
    1523         [ +  + ]:          4 :                 if (func == PGBENCH_SQRT)
    1524                 :          2 :                     dval = sqrt(dval);
    1525                 :            : 
    1526                 :          4 :                 setDoubleValue(retval, dval);
    1527                 :          4 :                 return true;
    1528                 :            :             }
    1529                 :            : 
    1530                 :            :             /* 1 int argument */
    1531                 :            :         case PGBENCH_INT:
    1532                 :            :             {
    1533                 :            :                 int64       ival;
    1534                 :            : 
    1535                 :            :                 Assert(nargs == 1);
    1536                 :            : 
    1537         [ +  + ]:          4 :                 if (!coerceToInt(&vargs[0], &ival))
    1538                 :          2 :                     return false;
    1539                 :            : 
    1540                 :          2 :                 setIntValue(retval, ival);
    1541                 :          4 :                 return true;
    1542                 :            :             }
    1543                 :            : 
    1544                 :            :             /* variable number of arguments */
    1545                 :            :         case PGBENCH_LEAST:
    1546                 :            :         case PGBENCH_GREATEST:
    1547                 :            :             {
    1548                 :            :                 bool        havedouble;
    1549                 :            :                 int         i;
    1550                 :            : 
    1551                 :            :                 Assert(nargs >= 1);
    1552                 :            : 
    1553                 :            :                 /* need double result if any input is double */
    1554                 :          8 :                 havedouble = false;
    1555         [ +  + ]:         28 :                 for (i = 0; i < nargs; i++)
    1556                 :            :                 {
    1557         [ +  + ]:         24 :                     if (vargs[i].type == PGBT_DOUBLE)
    1558                 :            :                     {
    1559                 :          4 :                         havedouble = true;
    1560                 :          4 :                         break;
    1561                 :            :                     }
    1562                 :            :                 }
    1563         [ +  + ]:          8 :                 if (havedouble)
    1564                 :            :                 {
    1565                 :            :                     double      extremum;
    1566                 :            : 
    1567         [ -  + ]:          4 :                     if (!coerceToDouble(&vargs[0], &extremum))
    1568                 :          0 :                         return false;
    1569         [ +  + ]:         12 :                     for (i = 1; i < nargs; i++)
    1570                 :            :                     {
    1571                 :            :                         double      dval;
    1572                 :            : 
    1573         [ -  + ]:          8 :                         if (!coerceToDouble(&vargs[i], &dval))
    1574                 :          0 :                             return false;
    1575         [ +  + ]:          8 :                         if (func == PGBENCH_LEAST)
    1576         [ +  - ]:          4 :                             extremum = Min(extremum, dval);
    1577                 :            :                         else
    1578         [ +  - ]:          4 :                             extremum = Max(extremum, dval);
    1579                 :            :                     }
    1580                 :          4 :                     setDoubleValue(retval, extremum);
    1581                 :            :                 }
    1582                 :            :                 else
    1583                 :            :                 {
    1584                 :            :                     int64       extremum;
    1585                 :            : 
    1586         [ -  + ]:          4 :                     if (!coerceToInt(&vargs[0], &extremum))
    1587                 :          0 :                         return false;
    1588         [ +  + ]:         16 :                     for (i = 1; i < nargs; i++)
    1589                 :            :                     {
    1590                 :            :                         int64       ival;
    1591                 :            : 
    1592         [ -  + ]:         12 :                         if (!coerceToInt(&vargs[i], &ival))
    1593                 :          0 :                             return false;
    1594         [ +  + ]:         12 :                         if (func == PGBENCH_LEAST)
    1595                 :          6 :                             extremum = Min(extremum, ival);
    1596                 :            :                         else
    1597                 :          6 :                             extremum = Max(extremum, ival);
    1598                 :            :                     }
    1599                 :          4 :                     setIntValue(retval, extremum);
    1600                 :            :                 }
    1601                 :          8 :                 return true;
    1602                 :            :             }
    1603                 :            : 
    1604                 :            :             /* random functions */
    1605                 :            :         case PGBENCH_RANDOM:
    1606                 :            :         case PGBENCH_RANDOM_EXPONENTIAL:
    1607                 :            :         case PGBENCH_RANDOM_GAUSSIAN:
    1608                 :            :             {
    1609                 :            :                 int64       imin,
    1610                 :            :                             imax;
    1611                 :            : 
    1612                 :            :                 Assert(nargs >= 2);
    1613                 :            : 
    1614   [ +  -  -  + ]:       6104 :                 if (!coerceToInt(&vargs[0], &imin) ||
    1615                 :       3052 :                     !coerceToInt(&vargs[1], &imax))
    1616                 :          0 :                     return false;
    1617                 :            : 
    1618                 :            :                 /* check random range */
    1619         [ +  + ]:       3052 :                 if (imin > imax)
    1620                 :            :                 {
    1621                 :          2 :                     fprintf(stderr, "empty range given to random\n");
    1622                 :          2 :                     return false;
    1623                 :            :                 }
    1624 [ +  + ][ -  + ]:       3050 :                 else if (imax - imin < 0 || (imax - imin) + 1 < 0)
    1625                 :            :                 {
    1626                 :            :                     /* prevent int overflows in random functions */
    1627                 :          2 :                     fprintf(stderr, "random range is too large\n");
    1628                 :          2 :                     return false;
    1629                 :            :                 }
    1630                 :            : 
    1631         [ +  + ]:       3048 :                 if (func == PGBENCH_RANDOM)
    1632                 :            :                 {
    1633                 :            :                     Assert(nargs == 2);
    1634                 :       3040 :                     setIntValue(retval, getrand(thread, imin, imax));
    1635                 :            :                 }
    1636                 :            :                 else            /* gaussian & exponential */
    1637                 :            :                 {
    1638                 :            :                     double      param;
    1639                 :            : 
    1640                 :            :                     Assert(nargs == 3);
    1641                 :            : 
    1642         [ -  + ]:          8 :                     if (!coerceToDouble(&vargs[2], &param))
    1643                 :          4 :                         return false;
    1644                 :            : 
    1645         [ +  + ]:          8 :                     if (func == PGBENCH_RANDOM_GAUSSIAN)
    1646                 :            :                     {
    1647         [ +  + ]:          4 :                         if (param < MIN_GAUSSIAN_PARAM)
    1648                 :            :                         {
    1649                 :          2 :                             fprintf(stderr,
    1650                 :            :                                     "gaussian parameter must be at least %f "
    1651                 :            :                                     "(not %f)\n", MIN_GAUSSIAN_PARAM, param);
    1652                 :          2 :                             return false;
    1653                 :            :                         }
    1654                 :            : 
    1655                 :          2 :                         setIntValue(retval,
    1656                 :            :                                     getGaussianRand(thread, imin, imax, param));
    1657                 :            :                     }
    1658                 :            :                     else        /* exponential */
    1659                 :            :                     {
    1660         [ +  + ]:          4 :                         if (param <= 0.0)
    1661                 :            :                         {
    1662                 :          2 :                             fprintf(stderr,
    1663                 :            :                                     "exponential parameter must be greater than zero"
    1664                 :            :                                     " (got %f)\n", param);
    1665                 :          2 :                             return false;
    1666                 :            :                         }
    1667                 :            : 
    1668                 :          4 :                         setIntValue(retval,
    1669                 :            :                                     getExponentialRand(thread, imin, imax, param));
    1670                 :            :                     }
    1671                 :            :                 }
    1672                 :            : 
    1673                 :       3052 :                 return true;
    1674                 :            :             }
    1675                 :            : 
    1676                 :            :         default:
    1677                 :            :             /* cannot get here */
    1678                 :            :             Assert(0);
    1679                 :            :             /* dead code to avoid a compiler warning */
    1680                 :       6252 :             return false;
    1681                 :            :     }
    1682                 :            : }
    1683                 :            : 
    1684                 :            : /*
    1685                 :            :  * Recursive evaluation of an expression in a pgbench script
    1686                 :            :  * using the current state of variables.
    1687                 :            :  * Returns whether the evaluation was ok,
    1688                 :            :  * the value itself is returned through the retval pointer.
    1689                 :            :  */
    1690                 :            : static bool
    1691                 :      15722 : evaluateExpr(TState *thread, CState *st, PgBenchExpr *expr, PgBenchValue *retval)
    1692                 :            : {
    1693   [ +  +  +  - ]:      15722 :     switch (expr->etype)
    1694                 :            :     {
    1695                 :            :         case ENODE_CONSTANT:
    1696                 :            :             {
    1697                 :       6638 :                 *retval = expr->u.constant;
    1698                 :       6638 :                 return true;
    1699                 :            :             }
    1700                 :            : 
    1701                 :            :         case ENODE_VARIABLE:
    1702                 :            :             {
    1703                 :            :                 Variable   *var;
    1704                 :            : 
    1705         [ +  + ]:       2832 :                 if ((var = lookupVariable(st, expr->u.variable.varname)) == NULL)
    1706                 :            :                 {
    1707                 :          2 :                     fprintf(stderr, "undefined variable \"%s\"\n",
    1708                 :            :                             expr->u.variable.varname);
    1709                 :          2 :                     return false;
    1710                 :            :                 }
    1711                 :            : 
    1712         [ +  + ]:       2830 :                 if (!makeVariableNumeric(var))
    1713                 :          2 :                     return false;
    1714                 :            : 
    1715                 :       2828 :                 *retval = var->num_value;
    1716                 :       2828 :                 return true;
    1717                 :            :             }
    1718                 :            : 
    1719                 :            :         case ENODE_FUNCTION:
    1720                 :       6252 :             return evalFunc(thread, st,
    1721                 :            :                             expr->u.function.function,
    1722                 :            :                             expr->u.function.args,
    1723                 :            :                             retval);
    1724                 :            : 
    1725                 :            :         default:
    1726                 :            :             /* internal error which should never occur */
    1727                 :          0 :             fprintf(stderr, "unexpected enode type in evaluation: %d\n",
    1728                 :          0 :                     expr->etype);
    1729                 :          0 :             exit(1);
    1730                 :            :     }
    1731                 :            : }
    1732                 :            : 
    1733                 :            : /*
    1734                 :            :  * Convert command name to meta-command enum identifier
    1735                 :            :  */
    1736                 :            : static MetaCommand
    1737                 :        524 : getMetaCommand(const char *cmd)
    1738                 :            : {
    1739                 :            :     MetaCommand mc;
    1740                 :            : 
    1741         [ -  + ]:        524 :     if (cmd == NULL)
    1742                 :          0 :         mc = META_NONE;
    1743         [ +  + ]:        524 :     else if (pg_strcasecmp(cmd, "set") == 0)
    1744                 :        486 :         mc = META_SET;
    1745         [ +  + ]:         38 :     else if (pg_strcasecmp(cmd, "setshell") == 0)
    1746                 :          8 :         mc = META_SETSHELL;
    1747         [ +  + ]:         30 :     else if (pg_strcasecmp(cmd, "shell") == 0)
    1748                 :         10 :         mc = META_SHELL;
    1749         [ +  + ]:         20 :     else if (pg_strcasecmp(cmd, "sleep") == 0)
    1750                 :         18 :         mc = META_SLEEP;
    1751                 :            :     else
    1752                 :          2 :         mc = META_NONE;
    1753                 :        524 :     return mc;
    1754                 :            : }
    1755                 :            : 
    1756                 :            : /*
    1757                 :            :  * Run a shell command. The result is assigned to the variable if not NULL.
    1758                 :            :  * Return true if succeeded, or false on error.
    1759                 :            :  */
    1760                 :            : static bool
    1761                 :         12 : runShellCommand(CState *st, char *variable, char **argv, int argc)
    1762                 :            : {
    1763                 :            :     char        command[SHELL_COMMAND_SIZE];
    1764                 :            :     int         i,
    1765                 :         12 :                 len = 0;
    1766                 :            :     FILE       *fp;
    1767                 :            :     char        res[64];
    1768                 :            :     char       *endptr;
    1769                 :            :     int         retval;
    1770                 :            : 
    1771                 :            :     /*----------
    1772                 :            :      * Join arguments with whitespace separators. Arguments starting with
    1773                 :            :      * exactly one colon are treated as variables:
    1774                 :            :      *  name - append a string "name"
    1775                 :            :      *  :var - append a variable named 'var'
    1776                 :            :      *  ::name - append a string ":name"
    1777                 :            :      *----------
    1778                 :            :      */
    1779         [ +  + ]:         38 :     for (i = 0; i < argc; i++)
    1780                 :            :     {
    1781                 :            :         char       *arg;
    1782                 :            :         int         arglen;
    1783                 :            : 
    1784         [ +  + ]:         28 :         if (argv[i][0] != ':')
    1785                 :            :         {
    1786                 :         22 :             arg = argv[i];      /* a string literal */
    1787                 :            :         }
    1788         [ +  + ]:          6 :         else if (argv[i][1] == ':')
    1789                 :            :         {
    1790                 :          2 :             arg = argv[i] + 1;  /* a string literal starting with colons */
    1791                 :            :         }
    1792         [ +  + ]:          4 :         else if ((arg = getVariable(st, argv[i] + 1)) == NULL)
    1793                 :            :         {
    1794                 :          2 :             fprintf(stderr, "%s: undefined variable \"%s\"\n",
    1795                 :          2 :                     argv[0], argv[i]);
    1796                 :          2 :             return false;
    1797                 :            :         }
    1798                 :            : 
    1799                 :         26 :         arglen = strlen(arg);
    1800         [ -  + ]:         26 :         if (len + arglen + (i > 0 ? 1 : 0) >= SHELL_COMMAND_SIZE - 1)
    1801                 :            :         {
    1802                 :          0 :             fprintf(stderr, "%s: shell command is too long\n", argv[0]);
    1803                 :          0 :             return false;
    1804                 :            :         }
    1805                 :            : 
    1806         [ +  + ]:         26 :         if (i > 0)
    1807                 :         14 :             command[len++] = ' ';
    1808                 :         26 :         memcpy(command + len, arg, arglen);
    1809                 :         26 :         len += arglen;
    1810                 :            :     }
    1811                 :            : 
    1812                 :         10 :     command[len] = '\0';
    1813                 :            : 
    1814                 :            :     /* Fast path for non-assignment case */
    1815         [ +  + ]:         10 :     if (variable == NULL)
    1816                 :            :     {
    1817         [ +  + ]:          4 :         if (system(command))
    1818                 :            :         {
    1819         [ +  - ]:          2 :             if (!timer_exceeded)
    1820                 :          2 :                 fprintf(stderr, "%s: could not launch shell command\n", argv[0]);
    1821                 :          2 :             return false;
    1822                 :            :         }
    1823                 :          2 :         return true;
    1824                 :            :     }
    1825                 :            : 
    1826                 :            :     /* Execute the command with pipe and read the standard output. */
    1827         [ -  + ]:          6 :     if ((fp = popen(command, "r")) == NULL)
    1828                 :            :     {
    1829                 :          0 :         fprintf(stderr, "%s: could not launch shell command\n", argv[0]);
    1830                 :          0 :         return false;
    1831                 :            :     }
    1832         [ +  + ]:          6 :     if (fgets(res, sizeof(res), fp) == NULL)
    1833                 :            :     {
    1834         [ +  - ]:          2 :         if (!timer_exceeded)
    1835                 :          2 :             fprintf(stderr, "%s: could not read result of shell command\n", argv[0]);
    1836                 :          2 :         (void) pclose(fp);
    1837                 :          2 :         return false;
    1838                 :            :     }
    1839         [ -  + ]:          4 :     if (pclose(fp) < 0)
    1840                 :            :     {
    1841                 :          0 :         fprintf(stderr, "%s: could not close shell command\n", argv[0]);
    1842                 :          0 :         return false;
    1843                 :            :     }
    1844                 :            : 
    1845                 :            :     /* Check whether the result is an integer and assign it to the variable */
    1846                 :          4 :     retval = (int) strtol(res, &endptr, 10);
    1847 [ +  + ][ +  + ]:          6 :     while (*endptr != '\0' && isspace((unsigned char) *endptr))
    1848                 :          2 :         endptr++;
    1849 [ +  - ][ +  + ]:          4 :     if (*res == '\0' || *endptr != '\0')
    1850                 :            :     {
    1851                 :          2 :         fprintf(stderr, "%s: shell command must return an integer (not \"%s\")\n",
    1852                 :            :                 argv[0], res);
    1853                 :          2 :         return false;
    1854                 :            :     }
    1855         [ -  + ]:          2 :     if (!putVariableInt(st, "setshell", variable, retval))
    1856                 :          0 :         return false;
    1857                 :            : 
    1858                 :            : #ifdef DEBUG
    1859                 :            :     printf("shell parameter name: \"%s\", value: \"%s\"\n", argv[1], res);
    1860                 :            : #endif
    1861                 :         12 :     return true;
    1862                 :            : }
    1863                 :            : 
    1864                 :            : #define MAX_PREPARE_NAME        32
    1865                 :            : static void
    1866                 :       2130 : preparedStatementName(char *buffer, int file, int state)
    1867                 :            : {
    1868                 :       2130 :     sprintf(buffer, "P%d_%d", file, state);
    1869                 :       2130 : }
    1870                 :            : 
    1871                 :            : static void
    1872                 :         34 : commandFailed(CState *st, const char *message)
    1873                 :            : {
    1874                 :         34 :     fprintf(stderr,
    1875                 :            :             "client %d aborted in command %d of script %d; %s\n",
    1876                 :            :             st->id, st->command, st->use_file, message);
    1877                 :         34 : }
    1878                 :            : 
    1879                 :            : /* return a script number with a weighted choice. */
    1880                 :            : static int
    1881                 :       2790 : chooseScript(TState *thread)
    1882                 :            : {
    1883                 :       2790 :     int         i = 0;
    1884                 :            :     int64       w;
    1885                 :            : 
    1886         [ +  + ]:       2790 :     if (num_scripts == 1)
    1887                 :       2590 :         return 0;
    1888                 :            : 
    1889                 :        200 :     w = getrand(thread, 0, total_weight - 1);
    1890                 :            :     do
    1891                 :            :     {
    1892                 :        322 :         w -= sql_script[i++].weight;
    1893         [ +  + ]:        322 :     } while (w >= 0);
    1894                 :            : 
    1895                 :        200 :     return i - 1;
    1896                 :            : }
    1897                 :            : 
    1898                 :            : /* Send a SQL command, using the chosen querymode */
    1899                 :            : static bool
    1900                 :       4036 : sendCommand(CState *st, Command *command)
    1901                 :            : {
    1902                 :            :     int         r;
    1903                 :            : 
    1904         [ +  + ]:       4036 :     if (querymode == QUERY_SIMPLE)
    1905                 :            :     {
    1906                 :            :         char       *sql;
    1907                 :            : 
    1908                 :        820 :         sql = pg_strdup(command->argv[0]);
    1909                 :        820 :         sql = assignVariables(st, sql);
    1910                 :            : 
    1911         [ -  + ]:        820 :         if (debug)
    1912                 :          0 :             fprintf(stderr, "client %d sending %s\n", st->id, sql);
    1913                 :        820 :         r = PQsendQuery(st->con, sql);
    1914                 :        820 :         free(sql);
    1915                 :            :     }
    1916         [ +  + ]:       3216 :     else if (querymode == QUERY_EXTENDED)
    1917                 :            :     {
    1918                 :       1120 :         const char *sql = command->argv[0];
    1919                 :            :         const char *params[MAX_ARGS];
    1920                 :            : 
    1921                 :       1120 :         getQueryParams(st, command, params);
    1922                 :            : 
    1923         [ -  + ]:       1120 :         if (debug)
    1924                 :          0 :             fprintf(stderr, "client %d sending %s\n", st->id, sql);
    1925                 :       1120 :         r = PQsendQueryParams(st->con, sql, command->argc - 1,
    1926                 :            :                               NULL, params, NULL, NULL, 0);
    1927                 :            :     }
    1928         [ +  - ]:       2096 :     else if (querymode == QUERY_PREPARED)
    1929                 :            :     {
    1930                 :            :         char        name[MAX_PREPARE_NAME];
    1931                 :            :         const char *params[MAX_ARGS];
    1932                 :            : 
    1933         [ +  + ]:       2096 :         if (!st->prepared[st->use_file])
    1934                 :            :         {
    1935                 :            :             int         j;
    1936                 :         30 :             Command   **commands = sql_script[st->use_file].commands;
    1937                 :            : 
    1938         [ +  + ]:         82 :             for (j = 0; commands[j] != NULL; j++)
    1939                 :            :             {
    1940                 :            :                 PGresult   *res;
    1941                 :            :                 char        name[MAX_PREPARE_NAME];
    1942                 :            : 
    1943         [ +  + ]:         52 :                 if (commands[j]->type != SQL_COMMAND)
    1944                 :         18 :                     continue;
    1945                 :         34 :                 preparedStatementName(name, st->use_file, j);
    1946                 :         34 :                 res = PQprepare(st->con, name,
    1947                 :         68 :                                 commands[j]->argv[0], commands[j]->argc - 1, NULL);
    1948         [ +  + ]:         34 :                 if (PQresultStatus(res) != PGRES_COMMAND_OK)
    1949                 :          2 :                     fprintf(stderr, "%s", PQerrorMessage(st->con));
    1950                 :         34 :                 PQclear(res);
    1951                 :            :             }
    1952                 :         30 :             st->prepared[st->use_file] = true;
    1953                 :            :         }
    1954                 :            : 
    1955                 :       2096 :         getQueryParams(st, command, params);
    1956                 :       2096 :         preparedStatementName(name, st->use_file, st->command);
    1957                 :            : 
    1958         [ +  + ]:       2096 :         if (debug)
    1959                 :       1400 :             fprintf(stderr, "client %d sending %s\n", st->id, name);
    1960                 :       2096 :         r = PQsendQueryPrepared(st->con, name, command->argc - 1,
    1961                 :            :                                 params, NULL, NULL, 0);
    1962                 :            :     }
    1963                 :            :     else                        /* unknown sql mode */
    1964                 :          0 :         r = 0;
    1965                 :            : 
    1966         [ -  + ]:       4036 :     if (r == 0)
    1967                 :            :     {
    1968         [ #  # ]:          0 :         if (debug)
    1969                 :          0 :             fprintf(stderr, "client %d could not send %s\n",
    1970                 :            :                     st->id, command->argv[0]);
    1971                 :          0 :         st->ecnt++;
    1972                 :          0 :         return false;
    1973                 :            :     }
    1974                 :            :     else
    1975                 :       4036 :         return true;
    1976                 :            : }
    1977                 :            : 
    1978                 :            : /*
    1979                 :            :  * Parse the argument to a \sleep command, and return the requested amount
    1980                 :            :  * of delay, in microseconds.  Returns true on success, false on error.
    1981                 :            :  */
    1982                 :            : static bool
    1983                 :         10 : evaluateSleep(CState *st, int argc, char **argv, int *usecs)
    1984                 :            : {
    1985                 :            :     char       *var;
    1986                 :            :     int         usec;
    1987                 :            : 
    1988         [ +  + ]:         10 :     if (*argv[1] == ':')
    1989                 :            :     {
    1990         [ +  + ]:          6 :         if ((var = getVariable(st, argv[1] + 1)) == NULL)
    1991                 :            :         {
    1992                 :          2 :             fprintf(stderr, "%s: undefined variable \"%s\"\n",
    1993                 :          2 :                     argv[0], argv[1]);
    1994                 :          2 :             return false;
    1995                 :            :         }
    1996                 :          4 :         usec = atoi(var);
    1997                 :            :     }
    1998                 :            :     else
    1999                 :          4 :         usec = atoi(argv[1]);
    2000                 :            : 
    2001         [ +  + ]:          8 :     if (argc > 2)
    2002                 :            :     {
    2003         [ +  + ]:          6 :         if (pg_strcasecmp(argv[2], "ms") == 0)
    2004                 :          2 :             usec *= 1000;
    2005         [ +  + ]:          4 :         else if (pg_strcasecmp(argv[2], "s") == 0)
    2006                 :          6 :             usec *= 1000000;
    2007                 :            :     }
    2008                 :            :     else
    2009                 :          2 :         usec *= 1000000;
    2010                 :            : 
    2011                 :          8 :     *usecs = usec;
    2012                 :          8 :     return true;
    2013                 :            : }
    2014                 :            : 
    2015                 :            : /*
    2016                 :            :  * Advance the state machine of a connection, if possible.
    2017                 :            :  */
    2018                 :            : static void
    2019                 :       4136 : doCustom(TState *thread, CState *st, StatsData *agg)
    2020                 :            : {
    2021                 :            :     PGresult   *res;
    2022                 :            :     Command    *command;
    2023                 :            :     instr_time  now;
    2024                 :       4136 :     bool        end_tx_processed = false;
    2025                 :            :     int64       wait;
    2026                 :            : 
    2027                 :            :     /*
    2028                 :            :      * gettimeofday() isn't free, so we get the current timestamp lazily the
    2029                 :            :      * first time it's needed, and reuse the same value throughout this
    2030                 :            :      * function after that.  This also ensures that e.g. the calculated
    2031                 :            :      * latency reported in the log file and in the totals are the same. Zero
    2032                 :            :      * means "not set yet".  Reset "now" when we execute shell commands or
    2033                 :            :      * expressions, which might take a non-negligible amount of time, though.
    2034                 :            :      */
    2035                 :       4136 :     INSTR_TIME_SET_ZERO(now);
    2036                 :            : 
    2037                 :            :     /*
    2038                 :            :      * Loop in the state machine, until we have to wait for a result from the
    2039                 :            :      * server (or have to sleep, for throttling or for \sleep).
    2040                 :            :      *
    2041                 :            :      * Note: In the switch-statement below, 'break' will loop back here,
    2042                 :            :      * meaning "continue in the state machine".  Return is used to return to
    2043                 :            :      * the caller.
    2044                 :            :      */
    2045                 :            :     for (;;)
    2046                 :            :     {
    2047   [ +  +  +  +  :      38750 :         switch (st->state)
          +  +  +  +  +  
                   +  - ]
    2048                 :            :         {
    2049                 :            :                 /*
    2050                 :            :                  * Select transaction to run.
    2051                 :            :                  */
    2052                 :            :             case CSTATE_CHOOSE_SCRIPT:
    2053                 :            : 
    2054                 :       2790 :                 st->use_file = chooseScript(thread);
    2055                 :            : 
    2056         [ +  + ]:       2790 :                 if (debug)
    2057                 :       1400 :                     fprintf(stderr, "client %d executing script \"%s\"\n", st->id,
    2058                 :       1400 :                             sql_script[st->use_file].desc);
    2059                 :            : 
    2060         [ +  + ]:       2790 :                 if (throttle_delay > 0)
    2061                 :        402 :                     st->state = CSTATE_START_THROTTLE;
    2062                 :            :                 else
    2063                 :       2388 :                     st->state = CSTATE_START_TX;
    2064                 :       2790 :                 break;
    2065                 :            : 
    2066                 :            :                 /*
    2067                 :            :                  * Handle throttling once per transaction by sleeping.
    2068                 :            :                  */
    2069                 :            :             case CSTATE_START_THROTTLE:
    2070                 :            : 
    2071                 :            :                 /*
    2072                 :            :                  * Generate a delay such that the series of delays will
    2073                 :            :                  * approximate a Poisson distribution centered on the
    2074                 :            :                  * throttle_delay time.
    2075                 :            :                  *
    2076                 :            :                  * If transactions are too slow or a given wait is shorter
    2077                 :            :                  * than a transaction, the next transaction will start right
    2078                 :            :                  * away.
    2079                 :            :                  */
    2080                 :            :                 Assert(throttle_delay > 0);
    2081                 :        402 :                 wait = getPoissonRand(thread, throttle_delay);
    2082                 :            : 
    2083                 :        402 :                 thread->throttle_trigger += wait;
    2084                 :        402 :                 st->txn_scheduled = thread->throttle_trigger;
    2085                 :            : 
    2086                 :            :                 /*
    2087                 :            :                  * stop client if next transaction is beyond pgbench end of
    2088                 :            :                  * execution
    2089                 :            :                  */
    2090 [ -  + ][ #  # ]:        402 :                 if (duration > 0 && st->txn_scheduled > end_time)
    2091                 :            :                 {
    2092                 :          0 :                     st->state = CSTATE_FINISHED;
    2093                 :          0 :                     break;
    2094                 :            :                 }
    2095                 :            : 
    2096                 :            :                 /*
    2097                 :            :                  * If --latency-limit is used, and this slot is already late
    2098                 :            :                  * so that the transaction will miss the latency limit even if
    2099                 :            :                  * it completed immediately, we skip this time slot and
    2100                 :            :                  * iterate till the next slot that isn't late yet.  But don't
    2101                 :            :                  * iterate beyond the -t limit, if one is given.
    2102                 :            :                  */
    2103         [ +  - ]:        402 :                 if (latency_limit)
    2104                 :            :                 {
    2105                 :            :                     int64       now_us;
    2106                 :            : 
    2107 [ +  + ][ +  - ]:        402 :                     if (INSTR_TIME_IS_ZERO(now))
    2108                 :          6 :                         INSTR_TIME_SET_CURRENT(now);
    2109                 :        402 :                     now_us = INSTR_TIME_GET_MICROSEC(now);
    2110 [ +  + ][ -  + ]:        422 :                     while (thread->throttle_trigger < now_us - latency_limit &&
    2111         [ +  + ]:         22 :                            (nxacts <= 0 || st->cnt < nxacts))
    2112                 :            :                     {
    2113                 :         20 :                         processXactStats(thread, st, &now, true, agg);
    2114                 :            :                         /* next rendez-vous */
    2115                 :         20 :                         wait = getPoissonRand(thread, throttle_delay);
    2116                 :         20 :                         thread->throttle_trigger += wait;
    2117                 :         20 :                         st->txn_scheduled = thread->throttle_trigger;
    2118                 :            :                     }
    2119                 :            :                     /* stop client if -t exceeded */
    2120 [ +  - ][ +  + ]:        402 :                     if (nxacts > 0 && st->cnt >= nxacts)
    2121                 :            :                     {
    2122                 :          2 :                         st->state = CSTATE_FINISHED;
    2123                 :          2 :                         break;
    2124                 :            :                     }
    2125                 :            :                 }
    2126                 :            : 
    2127                 :        400 :                 st->state = CSTATE_THROTTLE;
    2128         [ -  + ]:        400 :                 if (debug)
    2129                 :          0 :                     fprintf(stderr, "client %d throttling " INT64_FORMAT " us\n",
    2130                 :            :                             st->id, wait);
    2131                 :        400 :                 break;
    2132                 :            : 
    2133                 :            :                 /*
    2134                 :            :                  * Wait until it's time to start next transaction.
    2135                 :            :                  */
    2136                 :            :             case CSTATE_THROTTLE:
    2137 [ -  + ][ #  # ]:        400 :                 if (INSTR_TIME_IS_ZERO(now))
    2138                 :          0 :                     INSTR_TIME_SET_CURRENT(now);
    2139         [ -  + ]:        400 :                 if (INSTR_TIME_GET_MICROSEC(now) < st->txn_scheduled)
    2140                 :          0 :                     return;     /* Still sleeping, nothing to do here */
    2141                 :            : 
    2142                 :            :                 /* Else done sleeping, start the transaction */
    2143                 :        400 :                 st->state = CSTATE_START_TX;
    2144                 :        400 :                 break;
    2145                 :            : 
    2146                 :            :                 /* Start new transaction */
    2147                 :            :             case CSTATE_START_TX:
    2148                 :            : 
    2149                 :            :                 /*
    2150                 :            :                  * Establish connection on first call, or if is_connect is
    2151                 :            :                  * true.
    2152                 :            :                  */
    2153         [ +  + ]:       2788 :                 if (st->con == NULL)
    2154                 :            :                 {
    2155                 :            :                     instr_time  start;
    2156                 :            : 
    2157 [ +  - ][ +  - ]:        220 :                     if (INSTR_TIME_IS_ZERO(now))
    2158                 :        220 :                         INSTR_TIME_SET_CURRENT(now);
    2159                 :        220 :                     start = now;
    2160         [ -  + ]:        220 :                     if ((st->con = doConnect()) == NULL)
    2161                 :            :                     {
    2162                 :          0 :                         fprintf(stderr, "client %d aborted while establishing connection\n",
    2163                 :            :                                 st->id);
    2164                 :          0 :                         st->state = CSTATE_ABORTED;
    2165                 :          0 :                         break;
    2166                 :            :                     }
    2167                 :        220 :                     INSTR_TIME_SET_CURRENT(now);
    2168 [ +  + ][ +  + ]:        236 :                     INSTR_TIME_ACCUM_DIFF(thread->conn_time, now, start);
    2169                 :            : 
    2170                 :            :                     /* Reset session-local state */
    2171                 :        220 :                     memset(st->prepared, 0, sizeof(st->prepared));
    2172                 :            :                 }
    2173                 :            : 
    2174                 :            :                 /*
    2175                 :            :                  * Record transaction start time under logging, progress or
    2176                 :            :                  * throttling.
    2177                 :            :                  */
    2178 [ +  + ][ +  - ]:       2788 :                 if (use_log || progress || throttle_delay || latency_limit ||
         [ +  + ][ +  - ]
                 [ +  + ]
    2179                 :            :                     per_script_stats)
    2180                 :            :                 {
    2181 [ +  + ][ +  - ]:        820 :                     if (INSTR_TIME_IS_ZERO(now))
    2182                 :          8 :                         INSTR_TIME_SET_CURRENT(now);
    2183                 :        820 :                     st->txn_begin = now;
    2184                 :            : 
    2185                 :            :                     /*
    2186                 :            :                      * When not throttling, this is also the transaction's
    2187                 :            :                      * scheduled start time.
    2188                 :            :                      */
    2189         [ +  + ]:        820 :                     if (!throttle_delay)
    2190                 :        420 :                         st->txn_scheduled = INSTR_TIME_GET_MICROSEC(now);
    2191                 :            :                 }
    2192                 :            : 
    2193                 :            :                 /* Begin with the first command */
    2194                 :       2788 :                 st->command = 0;
    2195                 :       2788 :                 st->state = CSTATE_START_COMMAND;
    2196                 :       2788 :                 break;
    2197                 :            : 
    2198                 :            :                 /*
    2199                 :            :                  * Send a command to server (or execute a meta-command)
    2200                 :            :                  */
    2201                 :            :             case CSTATE_START_COMMAND:
    2202                 :      10044 :                 command = sql_script[st->use_file].commands[st->command];
    2203                 :            : 
    2204                 :            :                 /*
    2205                 :            :                  * If we reached the end of the script, move to end-of-xact
    2206                 :            :                  * processing.
    2207                 :            :                  */
    2208         [ +  + ]:      10044 :                 if (command == NULL)
    2209                 :            :                 {
    2210                 :       2754 :                     st->state = CSTATE_END_TX;
    2211                 :       2754 :                     break;
    2212                 :            :                 }
    2213                 :            : 
    2214                 :            :                 /*
    2215                 :            :                  * Record statement start time if per-command latencies are
    2216                 :            :                  * requested
    2217                 :            :                  */
    2218         [ +  + ]:       7290 :                 if (is_latencies)
    2219                 :            :                 {
    2220 [ -  + ][ #  # ]:        800 :                     if (INSTR_TIME_IS_ZERO(now))
    2221                 :          0 :                         INSTR_TIME_SET_CURRENT(now);
    2222                 :        800 :                     st->stmt_begin = now;
    2223                 :            :                 }
    2224                 :            : 
    2225         [ +  + ]:       7290 :                 if (command->type == SQL_COMMAND)
    2226                 :            :                 {
    2227         [ -  + ]:       4036 :                     if (!sendCommand(st, command))
    2228                 :            :                     {
    2229                 :          0 :                         commandFailed(st, "SQL command send failed");
    2230                 :          0 :                         st->state = CSTATE_ABORTED;
    2231                 :            :                     }
    2232                 :            :                     else
    2233                 :       4036 :                         st->state = CSTATE_WAIT_RESULT;
    2234                 :            :                 }
    2235         [ +  - ]:       3254 :                 else if (command->type == META_COMMAND)
    2236                 :            :                 {
    2237                 :       3254 :                     int         argc = command->argc,
    2238                 :            :                                 i;
    2239                 :       3254 :                     char      **argv = command->argv;
    2240                 :            : 
    2241         [ +  + ]:       3254 :                     if (debug)
    2242                 :            :                     {
    2243                 :       1400 :                         fprintf(stderr, "client %d executing \\%s", st->id, argv[0]);
    2244         [ +  + ]:       2800 :                         for (i = 1; i < argc; i++)
    2245                 :       1400 :                             fprintf(stderr, " %s", argv[i]);
    2246                 :       1400 :                         fprintf(stderr, "\n");
    2247                 :            :                     }
    2248                 :            : 
    2249         [ +  + ]:       3254 :                     if (command->meta == META_SLEEP)
    2250                 :            :                     {
    2251                 :            :                         /*
    2252                 :            :                          * A \sleep doesn't execute anything, we just get the
    2253                 :            :                          * delay from the argument, and enter the CSTATE_SLEEP
    2254                 :            :                          * state.  (The per-command latency will be recorded
    2255                 :            :                          * in CSTATE_SLEEP state, not here, after the delay
    2256                 :            :                          * has elapsed.)
    2257                 :            :                          */
    2258                 :            :                         int         usec;
    2259                 :            : 
    2260         [ +  + ]:         10 :                         if (!evaluateSleep(st, argc, argv, &usec))
    2261                 :            :                         {
    2262                 :          2 :                             commandFailed(st, "execution of meta-command 'sleep' failed");
    2263                 :          2 :                             st->state = CSTATE_ABORTED;
    2264                 :          2 :                             break;
    2265                 :            :                         }
    2266                 :            : 
    2267 [ +  + ][ +  - ]:          8 :                         if (INSTR_TIME_IS_ZERO(now))
    2268                 :          2 :                             INSTR_TIME_SET_CURRENT(now);
    2269                 :          8 :                         st->sleep_until = INSTR_TIME_GET_MICROSEC(now) + usec;
    2270                 :          8 :                         st->state = CSTATE_SLEEP;
    2271                 :         10 :                         break;
    2272                 :            :                     }
    2273                 :            :                     else
    2274                 :            :                     {
    2275         [ +  + ]:       3244 :                         if (command->meta == META_SET)
    2276                 :            :                         {
    2277                 :       3232 :                             PgBenchExpr *expr = command->expr;
    2278                 :            :                             PgBenchValue result;
    2279                 :            : 
    2280         [ +  + ]:       3232 :                             if (!evaluateExpr(thread, st, expr, &result))
    2281                 :            :                             {
    2282                 :         20 :                                 commandFailed(st, "evaluation of meta-command 'set' failed");
    2283                 :         20 :                                 st->state = CSTATE_ABORTED;
    2284                 :         22 :                                 break;
    2285                 :            :                             }
    2286                 :            : 
    2287         [ +  + ]:       3212 :                             if (!putVariableNumber(st, argv[0], argv[1], &result))
    2288                 :            :                             {
    2289                 :          2 :                                 commandFailed(st, "assignment of meta-command 'set' failed");
    2290                 :          2 :                                 st->state = CSTATE_ABORTED;
    2291                 :       3212 :                                 break;
    2292                 :            :                             }
    2293                 :            :                         }
    2294         [ +  + ]:         12 :                         else if (command->meta == META_SETSHELL)
    2295                 :            :                         {
    2296                 :          6 :                             bool        ret = runShellCommand(st, argv[1], argv + 2, argc - 2);
    2297                 :            : 
    2298         [ -  + ]:          6 :                             if (timer_exceeded) /* timeout */
    2299                 :            :                             {
    2300                 :          0 :                                 st->state = CSTATE_FINISHED;
    2301                 :          0 :                                 break;
    2302                 :            :                             }
    2303         [ +  + ]:          6 :                             else if (!ret)  /* on error */
    2304                 :            :                             {
    2305                 :          4 :                                 commandFailed(st, "execution of meta-command 'setshell' failed");
    2306                 :          4 :                                 st->state = CSTATE_ABORTED;
    2307                 :          6 :                                 break;
    2308                 :            :                             }
    2309                 :            :                             else
    2310                 :            :                             {
    2311                 :            :                                 /* succeeded */
    2312                 :            :                             }
    2313                 :            :                         }
    2314         [ +  - ]:          6 :                         else if (command->meta == META_SHELL)
    2315                 :            :                         {
    2316                 :          6 :                             bool        ret = runShellCommand(st, NULL, argv + 1, argc - 1);
    2317                 :            : 
    2318         [ -  + ]:          6 :                             if (timer_exceeded) /* timeout */
    2319                 :            :                             {
    2320                 :          0 :                                 st->state = CSTATE_FINISHED;
    2321                 :          0 :                                 break;
    2322                 :            :                             }
    2323         [ +  + ]:          6 :                             else if (!ret)  /* on error */
    2324                 :            :                             {
    2325                 :          4 :                                 commandFailed(st, "execution of meta-command 'shell' failed");
    2326                 :          4 :                                 st->state = CSTATE_ABORTED;
    2327                 :          4 :                                 break;
    2328                 :            :                             }
    2329                 :            :                             else
    2330                 :            :                             {
    2331                 :            :                                 /* succeeded */
    2332                 :            :                             }
    2333                 :            :                         }
    2334                 :            : 
    2335                 :            :                         /*
    2336                 :            :                          * executing the expression or shell command might
    2337                 :            :                          * take a non-negligible amount of time, so reset
    2338                 :            :                          * 'now'
    2339                 :            :                          */
    2340                 :       3214 :                         INSTR_TIME_SET_ZERO(now);
    2341                 :            : 
    2342                 :       3214 :                         st->state = CSTATE_END_COMMAND;
    2343                 :            :                     }
    2344                 :            :                 }
    2345                 :       7250 :                 break;
    2346                 :            : 
    2347                 :            :                 /*
    2348                 :            :                  * Wait for the current SQL command to complete
    2349                 :            :                  */
    2350                 :            :             case CSTATE_WAIT_RESULT:
    2351                 :       8072 :                 command = sql_script[st->use_file].commands[st->command];
    2352         [ +  + ]:       8072 :                 if (debug)
    2353                 :       2800 :                     fprintf(stderr, "client %d receiving\n", st->id);
    2354         [ -  + ]:       8072 :                 if (!PQconsumeInput(st->con))
    2355                 :            :                 {               /* there's something wrong */
    2356                 :          0 :                     commandFailed(st, "perhaps the backend died while processing");
    2357                 :          0 :                     st->state = CSTATE_ABORTED;
    2358                 :          0 :                     break;
    2359                 :            :                 }
    2360         [ +  + ]:       8072 :                 if (PQisBusy(st->con))
    2361                 :       4036 :                     return;     /* don't have the whole result yet */
    2362                 :            : 
    2363                 :            :                 /*
    2364                 :            :                  * Read and discard the query result;
    2365                 :            :                  */
    2366                 :       4036 :                 res = PQgetResult(st->con);
    2367         [ +  + ]:       4036 :                 switch (PQresultStatus(res))
    2368                 :            :                 {
    2369                 :            :                     case PGRES_COMMAND_OK:
    2370                 :            :                     case PGRES_TUPLES_OK:
    2371                 :            :                     case PGRES_EMPTY_QUERY:
    2372                 :            :                         /* OK */
    2373                 :       4034 :                         PQclear(res);
    2374                 :       4034 :                         discard_response(st);
    2375                 :       4034 :                         st->state = CSTATE_END_COMMAND;
    2376                 :       4034 :                         break;
    2377                 :            :                     default:
    2378                 :          2 :                         commandFailed(st, PQerrorMessage(st->con));
    2379                 :          2 :                         PQclear(res);
    2380                 :          2 :                         st->state = CSTATE_ABORTED;
    2381                 :          2 :                         break;
    2382                 :            :                 }
    2383                 :       4036 :                 break;
    2384                 :            : 
    2385                 :            :                 /*
    2386                 :            :                  * Wait until sleep is done. This state is entered after a
    2387                 :            :                  * \sleep metacommand. The behavior is similar to
    2388                 :            :                  * CSTATE_THROTTLE, but proceeds to CSTATE_START_COMMAND
    2389                 :            :                  * instead of CSTATE_START_TX.
    2390                 :            :                  */
    2391                 :            :             case CSTATE_SLEEP:
    2392 [ +  + ][ +  - ]:         12 :                 if (INSTR_TIME_IS_ZERO(now))
    2393                 :          4 :                     INSTR_TIME_SET_CURRENT(now);
    2394         [ +  + ]:         12 :                 if (INSTR_TIME_GET_MICROSEC(now) < st->sleep_until)
    2395                 :          4 :                     return;     /* Still sleeping, nothing to do here */
    2396                 :            :                 /* Else done sleeping. */
    2397                 :          8 :                 st->state = CSTATE_END_COMMAND;
    2398                 :          8 :                 break;
    2399                 :            : 
    2400                 :            :                 /*
    2401                 :            :                  * End of command: record stats and proceed to next command.
    2402                 :            :                  */
    2403                 :            :             case CSTATE_END_COMMAND:
    2404                 :            : 
    2405                 :            :                 /*
    2406                 :            :                  * command completed: accumulate per-command execution times
    2407                 :            :                  * in thread-local data structure, if per-command latencies
    2408                 :            :                  * are requested.
    2409                 :            :                  */
    2410         [ +  + ]:       7256 :                 if (is_latencies)
    2411                 :            :                 {
    2412 [ +  - ][ +  - ]:        800 :                     if (INSTR_TIME_IS_ZERO(now))
    2413                 :        800 :                         INSTR_TIME_SET_CURRENT(now);
    2414                 :            : 
    2415                 :            :                     /* XXX could use a mutex here, but we choose not to */
    2416                 :        800 :                     command = sql_script[st->use_file].commands[st->command];
    2417                 :        800 :                     addToSimpleStats(&command->stats,
    2418                 :        800 :                                      INSTR_TIME_GET_DOUBLE(now) -
    2419                 :        800 :                                      INSTR_TIME_GET_DOUBLE(st->stmt_begin));
    2420                 :            :                 }
    2421                 :            : 
    2422                 :            :                 /* Go ahead with next command */
    2423                 :       7256 :                 st->command++;
    2424                 :       7256 :                 st->state = CSTATE_START_COMMAND;
    2425                 :       7256 :                 break;
    2426                 :            : 
    2427                 :            :                 /*
    2428                 :            :                  * End of transaction.
    2429                 :            :                  */
    2430                 :            :             case CSTATE_END_TX:
    2431                 :            : 
    2432                 :            :                 /* transaction finished: calculate latency and do log */
    2433                 :       2754 :                 processXactStats(thread, st, &now, false, agg);
    2434                 :            : 
    2435         [ +  + ]:       2754 :                 if (is_connect)
    2436                 :            :                 {
    2437                 :        220 :                     PQfinish(st->con);
    2438                 :        220 :                     st->con = NULL;
    2439                 :        220 :                     INSTR_TIME_SET_ZERO(now);
    2440                 :            :                 }
    2441                 :            : 
    2442 [ +  + ][ -  + ]:       2754 :                 if ((st->cnt >= nxacts && duration <= 0) || timer_exceeded)
                 [ -  + ]
    2443                 :            :                 {
    2444                 :            :                     /* exit success */
    2445                 :         60 :                     st->state = CSTATE_FINISHED;
    2446                 :         60 :                     break;
    2447                 :            :                 }
    2448                 :            : 
    2449                 :            :                 /*
    2450                 :            :                  * No transaction is underway anymore.
    2451                 :            :                  */
    2452                 :       2694 :                 st->state = CSTATE_CHOOSE_SCRIPT;
    2453                 :            : 
    2454                 :            :                 /*
    2455                 :            :                  * If we paced through all commands in the script in this
    2456                 :            :                  * loop, without returning to the caller even once, do it now.
    2457                 :            :                  * This gives the thread a chance to process other
    2458                 :            :                  * connections, and to do progress reporting.  This can
    2459                 :            :                  * currently only happen if the script consists entirely of
    2460                 :            :                  * meta-commands.
    2461                 :            :                  */
    2462         [ -  + ]:       2694 :                 if (end_tx_processed)
    2463                 :          0 :                     return;
    2464                 :            :                 else
    2465                 :            :                 {
    2466                 :       2694 :                     end_tx_processed = true;
    2467                 :       2694 :                     break;
    2468                 :            :                 }
    2469                 :            : 
    2470                 :            :                 /*
    2471                 :            :                  * Final states.  Close the connection if it's still open.
    2472                 :            :                  */
    2473                 :            :             case CSTATE_ABORTED:
    2474                 :            :             case CSTATE_FINISHED:
    2475         [ +  + ]:         96 :                 if (st->con != NULL)
    2476                 :            :                 {
    2477                 :         82 :                     PQfinish(st->con);
    2478                 :         82 :                     st->con = NULL;
    2479                 :            :                 }
    2480                 :         96 :                 return;
    2481                 :            :         }
    2482                 :            :     }
    2483                 :            : }
    2484                 :            : 
    2485                 :            : /*
    2486                 :            :  * Print log entry after completing one transaction.
    2487                 :            :  *
    2488                 :            :  * We print Unix-epoch timestamps in the log, so that entries can be
    2489                 :            :  * correlated against other logs.  On some platforms this could be obtained
    2490                 :            :  * from the instr_time reading the caller has, but rather than get entangled
    2491                 :            :  * with that, we just eat the cost of an extra syscall in all cases.
    2492                 :            :  */
    2493                 :            : static void
    2494                 :        220 : doLog(TState *thread, CState *st,
    2495                 :            :       StatsData *agg, bool skipped, double latency, double lag)
    2496                 :            : {
    2497                 :        220 :     FILE       *logfile = thread->logfile;
    2498                 :            : 
    2499                 :            :     Assert(use_log);
    2500                 :            : 
    2501                 :            :     /*
    2502                 :            :      * Skip the log entry if sampling is enabled and this row doesn't belong
    2503                 :            :      * to the random sample.
    2504                 :            :      */
    2505   [ +  +  +  + ]:        420 :     if (sample_rate != 0.0 &&
    2506                 :        200 :         pg_erand48(thread->random_state) > sample_rate)
    2507                 :         84 :         return;
    2508                 :            : 
    2509                 :            :     /* should we aggregate the results or not? */
    2510         [ -  + ]:        136 :     if (agg_interval > 0)
    2511                 :            :     {
    2512                 :            :         /*
    2513                 :            :          * Loop until we reach the interval of the current moment, and print
    2514                 :            :          * any empty intervals in between (this may happen with very low tps,
    2515                 :            :          * e.g. --rate=0.1).
    2516                 :            :          */
    2517                 :          0 :         time_t      now = time(NULL);
    2518                 :            : 
    2519         [ #  # ]:          0 :         while (agg->start_time + agg_interval <= now)
    2520                 :            :         {
    2521                 :            :             /* print aggregated report to logfile */
    2522                 :          0 :             fprintf(logfile, "%ld " INT64_FORMAT " %.0f %.0f %.0f %.0f",
    2523                 :          0 :                     (long) agg->start_time,
    2524                 :            :                     agg->cnt,
    2525                 :            :                     agg->latency.sum,
    2526                 :            :                     agg->latency.sum2,
    2527                 :            :                     agg->latency.min,
    2528                 :            :                     agg->latency.max);
    2529         [ #  # ]:          0 :             if (throttle_delay)
    2530                 :            :             {
    2531                 :          0 :                 fprintf(logfile, " %.0f %.0f %.0f %.0f",
    2532                 :            :                         agg->lag.sum,
    2533                 :            :                         agg->lag.sum2,
    2534                 :            :                         agg->lag.min,
    2535                 :            :                         agg->lag.max);
    2536         [ #  # ]:          0 :                 if (latency_limit)
    2537                 :          0 :                     fprintf(logfile, " " INT64_FORMAT, agg->skipped);
    2538                 :            :             }
    2539                 :          0 :             fputc('\n', logfile);
    2540                 :            : 
    2541                 :            :             /* reset data and move to next interval */
    2542                 :          0 :             initStats(agg, agg->start_time + agg_interval);
    2543                 :            :         }
    2544                 :            : 
    2545                 :            :         /* accumulate the current transaction */
    2546                 :          0 :         accumStats(agg, skipped, latency, lag);
    2547                 :            :     }
    2548                 :            :     else
    2549                 :            :     {
    2550                 :            :         /* no, print raw transactions */
    2551                 :            :         struct timeval tv;
    2552                 :            : 
    2553                 :        136 :         gettimeofday(&tv, NULL);
    2554         [ -  + ]:        136 :         if (skipped)
    2555                 :          0 :             fprintf(logfile, "%d " INT64_FORMAT " skipped %d %ld %ld",
    2556                 :            :                     st->id, st->cnt, st->use_file,
    2557                 :          0 :                     (long) tv.tv_sec, (long) tv.tv_usec);
    2558                 :            :         else
    2559                 :        136 :             fprintf(logfile, "%d " INT64_FORMAT " %.0f %d %ld %ld",
    2560                 :            :                     st->id, st->cnt, latency, st->use_file,
    2561                 :        136 :                     (long) tv.tv_sec, (long) tv.tv_usec);
    2562         [ -  + ]:        136 :         if (throttle_delay)
    2563                 :          0 :             fprintf(logfile, " %.0f", lag);
    2564                 :        136 :         fputc('\n', logfile);
    2565                 :            :     }
    2566                 :            : }
    2567                 :            : 
    2568                 :            : /*
    2569                 :            :  * Accumulate and report statistics at end of a transaction.
    2570                 :            :  *
    2571                 :            :  * (This is also called when a transaction is late and thus skipped.
    2572                 :            :  * Note that even skipped transactions are counted in the "cnt" fields.)
    2573                 :            :  */
    2574                 :            : static void
    2575                 :       2774 : processXactStats(TState *thread, CState *st, instr_time *now,
    2576                 :            :                  bool skipped, StatsData *agg)
    2577                 :            : {
    2578                 :       2774 :     double      latency = 0.0,
    2579                 :       2774 :                 lag = 0.0;
    2580 [ +  - ][ +  + ]:       2774 :     bool        thread_details = progress || throttle_delay || latency_limit,
                 [ -  + ]
    2581 [ +  + ][ +  + ]:       2774 :                 detailed = thread_details || use_log || per_script_stats;
                 [ +  + ]
    2582                 :            : 
    2583 [ +  + ][ +  + ]:       2774 :     if (detailed && !skipped)
    2584                 :            :     {
    2585 [ +  + ][ +  - ]:        820 :         if (INSTR_TIME_IS_ZERO(*now))
    2586                 :        420 :             INSTR_TIME_SET_CURRENT(*now);
    2587                 :            : 
    2588                 :            :         /* compute latency & lag */
    2589                 :        820 :         latency = INSTR_TIME_GET_MICROSEC(*now) - st->txn_scheduled;
    2590                 :        820 :         lag = INSTR_TIME_GET_MICROSEC(st->txn_begin) - st->txn_scheduled;
    2591                 :            :     }
    2592                 :            : 
    2593         [ +  + ]:       2774 :     if (thread_details)
    2594                 :            :     {
    2595                 :            :         /* keep detailed thread stats */
    2596                 :        420 :         accumStats(&thread->stats, skipped, latency, lag);
    2597                 :            : 
    2598                 :            :         /* count transactions over the latency limit, if needed */
    2599 [ +  - ][ -  + ]:        420 :         if (latency_limit && latency > latency_limit)
    2600                 :        420 :             thread->latency_late++;
    2601                 :            :     }
    2602                 :            :     else
    2603                 :            :     {
    2604                 :            :         /* no detailed stats, just count */
    2605                 :       2354 :         thread->stats.cnt++;
    2606                 :            :     }
    2607                 :            : 
    2608                 :            :     /* client stat is just counting */
    2609                 :       2774 :     st->cnt++;
    2610                 :            : 
    2611         [ +  + ]:       2774 :     if (use_log)
    2612                 :        220 :         doLog(thread, st, agg, skipped, latency, lag);
    2613                 :            : 
    2614                 :            :     /* XXX could use a mutex here, but we choose not to */
    2615         [ +  + ]:       2774 :     if (per_script_stats)
    2616                 :        200 :         accumStats(&sql_script[st->use_file].stats, skipped, latency, lag);
    2617                 :       2774 : }
    2618                 :            : 
    2619                 :            : 
    2620                 :            : /* discard connections */
    2621                 :            : static void
    2622                 :        120 : disconnect_all(CState *state, int length)
    2623                 :            : {
    2624                 :            :     int         i;
    2625                 :            : 
    2626         [ +  + ]:        312 :     for (i = 0; i < length; i++)
    2627                 :            :     {
    2628         [ -  + ]:        192 :         if (state[i].con)
    2629                 :            :         {
    2630                 :          0 :             PQfinish(state[i].con);
    2631                 :          0 :             state[i].con = NULL;
    2632                 :            :         }
    2633                 :            :     }
    2634                 :        120 : }
    2635                 :            : 
    2636                 :            : /*
    2637                 :            :  * Remove old pgbench tables, if any exist
    2638                 :            :  */
    2639                 :            : static void
    2640                 :          6 : initDropTables(PGconn *con)
    2641                 :            : {
    2642                 :          6 :     fprintf(stderr, "dropping old tables...\n");
    2643                 :            : 
    2644                 :            :     /*
    2645                 :            :      * We drop all the tables in one command, so that whether there are
    2646                 :            :      * foreign key dependencies or not doesn't matter.
    2647                 :            :      */
    2648                 :          6 :     executeStatement(con, "drop table if exists "
    2649                 :            :                      "pgbench_accounts, "
    2650                 :            :                      "pgbench_branches, "
    2651                 :            :                      "pgbench_history, "
    2652                 :            :                      "pgbench_tellers");
    2653                 :          6 : }
    2654                 :            : 
    2655                 :            : /*
    2656                 :            :  * Create pgbench's standard tables
    2657                 :            :  */
    2658                 :            : static void
    2659                 :          6 : initCreateTables(PGconn *con)
    2660                 :            : {
    2661                 :            :     /*
    2662                 :            :      * The scale factor at/beyond which 32-bit integers are insufficient for
    2663                 :            :      * storing TPC-B account IDs.
    2664                 :            :      *
    2665                 :            :      * Although the actual threshold is 21474, we use 20000 because it is
    2666                 :            :      * easier to document and remember, and isn't that far away from the real
    2667                 :            :      * threshold.
    2668                 :            :      */
    2669                 :            : #define SCALE_32BIT_THRESHOLD 20000
    2670                 :            : 
    2671                 :            :     /*
    2672                 :            :      * Note: TPC-B requires at least 100 bytes per row, and the "filler"
    2673                 :            :      * fields in these table declarations were intended to comply with that.
    2674                 :            :      * The pgbench_accounts table complies with that because the "filler"
    2675                 :            :      * column is set to blank-padded empty string. But for all other tables
    2676                 :            :      * the columns default to NULL and so don't actually take any space.  We
    2677                 :            :      * could fix that by giving them non-null default values.  However, that
    2678                 :            :      * would completely break comparability of pgbench results with prior
    2679                 :            :      * versions. Since pgbench has never pretended to be fully TPC-B compliant
    2680                 :            :      * anyway, we stick with the historical behavior.
    2681                 :            :      */
    2682                 :            :     struct ddlinfo
    2683                 :            :     {
    2684                 :            :         const char *table;      /* table name */
    2685                 :            :         const char *smcols;     /* column decls if accountIDs are 32 bits */
    2686                 :            :         const char *bigcols;    /* column decls if accountIDs are 64 bits */
    2687                 :            :         int         declare_fillfactor;
    2688                 :            :     };
    2689                 :            :     static const struct ddlinfo DDLs[] = {
    2690                 :            :         {
    2691                 :            :             "pgbench_history",
    2692                 :            :             "tid int,bid int,aid    int,delta int,mtime timestamp,filler char(22)",
    2693                 :            :             "tid int,bid int,aid bigint,delta int,mtime timestamp,filler char(22)",
    2694                 :            :             0
    2695                 :            :         },
    2696                 :            :         {
    2697                 :            :             "pgbench_tellers",
    2698                 :            :             "tid int not null,bid int,tbalance int,filler char(84)",
    2699                 :            :             "tid int not null,bid int,tbalance int,filler char(84)",
    2700                 :            :             1
    2701                 :            :         },
    2702                 :            :         {
    2703                 :            :             "pgbench_accounts",
    2704                 :            :             "aid    int not null,bid int,abalance int,filler char(84)",
    2705                 :            :             "aid bigint not null,bid int,abalance int,filler char(84)",
    2706                 :            :             1
    2707                 :            :         },
    2708                 :            :         {
    2709                 :            :             "pgbench_branches",
    2710                 :            :             "bid int not null,bbalance int,filler char(88)",
    2711                 :            :             "bid int not null,bbalance int,filler char(88)",
    2712                 :            :             1
    2713                 :            :         }
    2714                 :            :     };
    2715                 :            :     int         i;
    2716                 :            : 
    2717                 :          6 :     fprintf(stderr, "creating tables...\n");
    2718                 :            : 
    2719         [ +  + ]:         30 :     for (i = 0; i < lengthof(DDLs); i++)
    2720                 :            :     {
    2721                 :            :         char        opts[256];
    2722                 :            :         char        buffer[256];
    2723                 :         24 :         const struct ddlinfo *ddl = &DDLs[i];
    2724                 :            :         const char *cols;
    2725                 :            : 
    2726                 :            :         /* Construct new create table statement. */
    2727                 :         24 :         opts[0] = '\0';
    2728         [ +  + ]:         24 :         if (ddl->declare_fillfactor)
    2729                 :         18 :             snprintf(opts + strlen(opts), sizeof(opts) - strlen(opts),
    2730                 :            :                      " with (fillfactor=%d)", fillfactor);
    2731         [ +  + ]:         24 :         if (tablespace != NULL)
    2732                 :            :         {
    2733                 :            :             char       *escape_tablespace;
    2734                 :            : 
    2735                 :          8 :             escape_tablespace = PQescapeIdentifier(con, tablespace,
    2736                 :            :                                                    strlen(tablespace));
    2737                 :          8 :             snprintf(opts + strlen(opts), sizeof(opts) - strlen(opts),
    2738                 :            :                      " tablespace %s", escape_tablespace);
    2739                 :          8 :             PQfreemem(escape_tablespace);
    2740                 :            :         }
    2741                 :            : 
    2742         [ -  + ]:         24 :         cols = (scale >= SCALE_32BIT_THRESHOLD) ? ddl->bigcols : ddl->smcols;
    2743                 :            : 
    2744         [ +  + ]:         24 :         snprintf(buffer, sizeof(buffer), "create%s table %s(%s)%s",
    2745                 :         24 :                  unlogged_tables ? " unlogged" : "",
    2746                 :            :                  ddl->table, cols, opts);
    2747                 :            : 
    2748                 :         24 :         executeStatement(con, buffer);
    2749                 :            :     }
    2750                 :          6 : }
    2751                 :            : 
    2752                 :            : /*
    2753                 :            :  * Fill the standard tables with some data
    2754                 :            :  */
    2755                 :            : static void
    2756                 :          6 : initGenerateData(PGconn *con)
    2757                 :            : {
    2758                 :            :     char        sql[256];
    2759                 :            :     PGresult   *res;
    2760                 :            :     int         i;
    2761                 :            :     int64       k;
    2762                 :            : 
    2763                 :            :     /* used to track elapsed time and estimate of the remaining time */
    2764                 :            :     instr_time  start,
    2765                 :            :                 diff;
    2766                 :            :     double      elapsed_sec,
    2767                 :            :                 remaining_sec;
    2768                 :          6 :     int         log_interval = 1;
    2769                 :            : 
    2770                 :          6 :     fprintf(stderr, "generating data...\n");
    2771                 :            : 
    2772                 :            :     /*
    2773                 :            :      * we do all of this in one transaction to enable the backend's
    2774                 :            :      * data-loading optimizations
    2775                 :            :      */
    2776                 :          6 :     executeStatement(con, "begin");
    2777                 :            : 
    2778                 :            :     /*
    2779                 :            :      * truncate away any old data, in one command in case there are foreign
    2780                 :            :      * keys
    2781                 :            :      */
    2782                 :          6 :     executeStatement(con, "truncate table "
    2783                 :            :                      "pgbench_accounts, "
    2784                 :            :                      "pgbench_branches, "
    2785                 :            :                      "pgbench_history, "
    2786                 :            :                      "pgbench_tellers");
    2787                 :            : 
    2788                 :            :     /*
    2789                 :            :      * fill branches, tellers, accounts in that order in case foreign keys
    2790                 :            :      * already exist
    2791                 :            :      */
    2792         [ +  + ]:         12 :     for (i = 0; i < nbranches * scale; i++)
    2793                 :            :     {
    2794                 :            :         /* "filler" column defaults to NULL */
    2795                 :          6 :         snprintf(sql, sizeof(sql),
    2796                 :            :                  "insert into pgbench_branches(bid,bbalance) values(%d,0)",
    2797                 :            :                  i + 1);
    2798                 :          6 :         executeStatement(con, sql);
    2799                 :            :     }
    2800                 :            : 
    2801         [ +  + ]:         66 :     for (i = 0; i < ntellers * scale; i++)
    2802                 :            :     {
    2803                 :            :         /* "filler" column defaults to NULL */
    2804                 :         60 :         snprintf(sql, sizeof(sql),
    2805                 :            :                  "insert into pgbench_tellers(tid,bid,tbalance) values (%d,%d,0)",
    2806                 :         60 :                  i + 1, i / ntellers + 1);
    2807                 :         60 :         executeStatement(con, sql);
    2808                 :            :     }
    2809                 :            : 
    2810                 :            :     /*
    2811                 :            :      * accounts is big enough to be worth using COPY and tracking runtime
    2812                 :            :      */
    2813                 :          6 :     res = PQexec(con, "copy pgbench_accounts from stdin");
    2814         [ -  + ]:          6 :     if (PQresultStatus(res) != PGRES_COPY_IN)
    2815                 :            :     {
    2816                 :          0 :         fprintf(stderr, "%s", PQerrorMessage(con));
    2817                 :          0 :         exit(1);
    2818                 :            :     }
    2819                 :          6 :     PQclear(res);
    2820                 :            : 
    2821                 :          6 :     INSTR_TIME_SET_CURRENT(start);
    2822                 :            : 
    2823         [ +  + ]:     600006 :     for (k = 0; k < (int64) naccounts * scale; k++)
    2824                 :            :     {
    2825                 :     600000 :         int64       j = k + 1;
    2826                 :            : 
    2827                 :            :         /* "filler" column defaults to blank padded empty string */
    2828                 :     600000 :         snprintf(sql, sizeof(sql),
    2829                 :            :                  INT64_FORMAT "\t" INT64_FORMAT "\t%d\t\n",
    2830                 :     600000 :                  j, k / naccounts + 1, 0);
    2831         [ -  + ]:     600000 :         if (PQputline(con, sql))
    2832                 :            :         {
    2833                 :          0 :             fprintf(stderr, "PQputline failed\n");
    2834                 :          0 :             exit(1);
    2835                 :            :         }
    2836                 :            : 
    2837                 :            :         /*
    2838                 :            :          * If we want to stick with the original logging, print a message each
    2839                 :            :          * 100k inserted rows.
    2840                 :            :          */
    2841 [ +  + ][ +  + ]:     600000 :         if ((!use_quiet) && (j % 100000 == 0))
    2842                 :            :         {
    2843                 :          4 :             INSTR_TIME_SET_CURRENT(diff);
    2844         [ -  + ]:          4 :             INSTR_TIME_SUBTRACT(diff, start);
    2845                 :            : 
    2846                 :          4 :             elapsed_sec = INSTR_TIME_GET_DOUBLE(diff);
    2847                 :          4 :             remaining_sec = ((double) scale * naccounts - j) * elapsed_sec / j;
    2848                 :            : 
    2849                 :          4 :             fprintf(stderr, INT64_FORMAT " of " INT64_FORMAT " tuples (%d%%) done (elapsed %.2f s, remaining %.2f s)\n",
    2850                 :            :                     j, (int64) naccounts * scale,
    2851                 :          4 :                     (int) (((int64) j * 100) / (naccounts * (int64) scale)),
    2852                 :            :                     elapsed_sec, remaining_sec);
    2853                 :            :         }
    2854                 :            :         /* let's not call the timing for each row, but only each 100 rows */
    2855 [ +  + ][ +  + ]:     599996 :         else if (use_quiet && (j % 100 == 0))
    2856                 :            :         {
    2857                 :       2000 :             INSTR_TIME_SET_CURRENT(diff);
    2858         [ -  + ]:       2000 :             INSTR_TIME_SUBTRACT(diff, start);
    2859                 :            : 
    2860                 :       2000 :             elapsed_sec = INSTR_TIME_GET_DOUBLE(diff);
    2861                 :       2000 :             remaining_sec = ((double) scale * naccounts - j) * elapsed_sec / j;
    2862                 :            : 
    2863                 :            :             /* have we reached the next interval (or end)? */
    2864 [ +  + ][ -  + ]:       2000 :             if ((j == scale * naccounts) || (elapsed_sec >= log_interval * LOG_STEP_SECONDS))
    2865                 :            :             {
    2866                 :          2 :                 fprintf(stderr, INT64_FORMAT " of " INT64_FORMAT " tuples (%d%%) done (elapsed %.2f s, remaining %.2f s)\n",
    2867                 :            :                         j, (int64) naccounts * scale,
    2868                 :          2 :                         (int) (((int64) j * 100) / (naccounts * (int64) scale)), elapsed_sec, remaining_sec);
    2869                 :            : 
    2870                 :            :                 /* skip to the next interval */
    2871                 :          2 :                 log_interval = (int) ceil(elapsed_sec / LOG_STEP_SECONDS);
    2872                 :            :             }
    2873                 :            :         }
    2874                 :            : 
    2875                 :            :     }
    2876         [ -  + ]:          6 :     if (PQputline(con, "\\.\n"))
    2877                 :            :     {
    2878                 :          0 :         fprintf(stderr, "very last PQputline failed\n");
    2879                 :          0 :         exit(1);
    2880                 :            :     }
    2881         [ -  + ]:          6 :     if (PQendcopy(con))
    2882                 :            :     {
    2883                 :          0 :         fprintf(stderr, "PQendcopy failed\n");
    2884                 :          0 :         exit(1);
    2885                 :            :     }
    2886                 :            : 
    2887                 :          6 :     executeStatement(con, "commit");
    2888                 :          6 : }
    2889                 :            : 
    2890                 :            : /*
    2891                 :            :  * Invoke vacuum on the standard tables
    2892                 :            :  */
    2893                 :            : static void
    2894                 :          4 : initVacuum(PGconn *con)
    2895                 :            : {
    2896                 :          4 :     fprintf(stderr, "vacuuming...\n");
    2897                 :          4 :     executeStatement(con, "vacuum analyze pgbench_branches");
    2898                 :          4 :     executeStatement(con, "vacuum analyze pgbench_tellers");
    2899                 :          4 :     executeStatement(con, "vacuum analyze pgbench_accounts");
    2900                 :          4 :     executeStatement(con, "vacuum analyze pgbench_history");
    2901                 :          4 : }
    2902                 :            : 
    2903                 :            : /*
    2904                 :            :  * Create primary keys on the standard tables
    2905                 :            :  */
    2906                 :            : static void
    2907                 :          6 : initCreatePKeys(PGconn *con)
    2908                 :            : {
    2909                 :            :     static const char *const DDLINDEXes[] = {
    2910                 :            :         "alter table pgbench_branches add primary key (bid)",
    2911                 :            :         "alter table pgbench_tellers add primary key (tid)",
    2912                 :            :         "alter table pgbench_accounts add primary key (aid)"
    2913                 :            :     };
    2914                 :            :     int         i;
    2915                 :            : 
    2916                 :          6 :     fprintf(stderr, "creating primary keys...\n");
    2917         [ +  + ]:         24 :     for (i = 0; i < lengthof(DDLINDEXes); i++)
    2918                 :            :     {
    2919                 :            :         char        buffer[256];
    2920                 :            : 
    2921                 :         18 :         strlcpy(buffer, DDLINDEXes[i], sizeof(buffer));
    2922                 :            : 
    2923         [ +  + ]:         18 :         if (index_tablespace != NULL)
    2924                 :            :         {
    2925                 :            :             char       *escape_tablespace;
    2926                 :            : 
    2927                 :          6 :             escape_tablespace = PQescapeIdentifier(con, index_tablespace,
    2928                 :            :                                                    strlen(index_tablespace));
    2929                 :          6 :             snprintf(buffer + strlen(buffer), sizeof(buffer) - strlen(buffer),
    2930                 :            :                      " using index tablespace %s", escape_tablespace);
    2931                 :          6 :             PQfreemem(escape_tablespace);
    2932                 :            :         }
    2933                 :            : 
    2934                 :         18 :         executeStatement(con, buffer);
    2935                 :            :     }
    2936                 :          6 : }
    2937                 :            : 
    2938                 :            : /*
    2939                 :            :  * Create foreign key constraints between the standard tables
    2940                 :            :  */
    2941                 :            : static void
    2942                 :          4 : initCreateFKeys(PGconn *con)
    2943                 :            : {
    2944                 :            :     static const char *const DDLKEYs[] = {
    2945                 :            :         "alter table pgbench_tellers add constraint pgbench_tellers_bid_fkey foreign key (bid) references pgbench_branches",
    2946                 :            :         "alter table pgbench_accounts add constraint pgbench_accounts_bid_fkey foreign key (bid) references pgbench_branches",
    2947                 :            :         "alter table pgbench_history add constraint pgbench_history_bid_fkey foreign key (bid) references pgbench_branches",
    2948                 :            :         "alter table pgbench_history add constraint pgbench_history_tid_fkey foreign key (tid) references pgbench_tellers",
    2949                 :            :         "alter table pgbench_history add constraint pgbench_history_aid_fkey foreign key (aid) references pgbench_accounts"
    2950                 :            :     };
    2951                 :            :     int         i;
    2952                 :            : 
    2953                 :          4 :     fprintf(stderr, "creating foreign keys...\n");
    2954         [ +  + ]:         24 :     for (i = 0; i < lengthof(DDLKEYs); i++)
    2955                 :            :     {
    2956                 :         20 :         executeStatement(con, DDLKEYs[i]);
    2957                 :            :     }
    2958                 :          4 : }
    2959                 :            : 
    2960                 :            : /*
    2961                 :            :  * Validate an initialization-steps string
    2962                 :            :  *
    2963                 :            :  * (We could just leave it to runInitSteps() to fail if there are wrong
    2964                 :            :  * characters, but since initialization can take awhile, it seems friendlier
    2965                 :            :  * to check during option parsing.)
    2966                 :            :  */
    2967                 :            : static void
    2968                 :          8 : checkInitSteps(const char *initialize_steps)
    2969                 :            : {
    2970                 :            :     const char *step;
    2971                 :            : 
    2972         [ -  + ]:          8 :     if (initialize_steps[0] == '\0')
    2973                 :            :     {
    2974                 :          0 :         fprintf(stderr, "no initialization steps specified\n");
    2975                 :          0 :         exit(1);
    2976                 :            :     }
    2977                 :            : 
    2978         [ +  + ]:         42 :     for (step = initialize_steps; *step != '\0'; step++)
    2979                 :            :     {
    2980         [ +  + ]:         36 :         if (strchr("dtgvpf ", *step) == NULL)
    2981                 :            :         {
    2982                 :          2 :             fprintf(stderr, "unrecognized initialization step \"%c\"\n",
    2983                 :          2 :                     *step);
    2984                 :          2 :             fprintf(stderr, "allowed steps are: \"d\", \"t\", \"g\", \"v\", \"p\", \"f\"\n");
    2985                 :          2 :             exit(1);
    2986                 :            :         }
    2987                 :            :     }
    2988                 :          6 : }
    2989                 :            : 
    2990                 :            : /*
    2991                 :            :  * Invoke each initialization step in the given string
    2992                 :            :  */
    2993                 :            : static void
    2994                 :          6 : runInitSteps(const char *initialize_steps)
    2995                 :            : {
    2996                 :            :     PGconn     *con;
    2997                 :            :     const char *step;
    2998                 :            : 
    2999         [ -  + ]:          6 :     if ((con = doConnect()) == NULL)
    3000                 :          0 :         exit(1);
    3001                 :            : 
    3002         [ +  + ]:         44 :     for (step = initialize_steps; *step != '\0'; step++)
    3003                 :            :     {
    3004   [ +  +  +  +  :         38 :         switch (*step)
             +  +  +  - ]
    3005                 :            :         {
    3006                 :            :             case 'd':
    3007                 :          6 :                 initDropTables(con);
    3008                 :          6 :                 break;
    3009                 :            :             case 't':
    3010                 :          6 :                 initCreateTables(con);
    3011                 :          6 :                 break;
    3012                 :            :             case 'g':
    3013                 :          6 :                 initGenerateData(con);
    3014                 :          6 :                 break;
    3015                 :            :             case 'v':
    3016                 :          4 :                 initVacuum(con);
    3017                 :          4 :                 break;
    3018                 :            :             case 'p':
    3019                 :          6 :                 initCreatePKeys(con);
    3020                 :          6 :                 break;
    3021                 :            :             case 'f':
    3022                 :          4 :                 initCreateFKeys(con);
    3023                 :          4 :                 break;
    3024                 :            :             case ' ':
    3025                 :          6 :                 break;          /* ignore */
    3026                 :            :             default:
    3027                 :          0 :                 fprintf(stderr, "unrecognized initialization step \"%c\"\n",
    3028                 :          0 :                         *step);
    3029                 :          0 :                 PQfinish(con);
    3030                 :          0 :                 exit(1);
    3031                 :            :         }
    3032                 :            :     }
    3033                 :            : 
    3034                 :          6 :     fprintf(stderr, "done.\n");
    3035                 :          6 :     PQfinish(con);
    3036                 :          6 : }
    3037                 :            : 
    3038                 :            : /*
    3039                 :            :  * Replace :param with $n throughout the command's SQL text, which
    3040                 :            :  * is a modifiable string in cmd->argv[0].
    3041                 :            :  */
    3042                 :            : static bool
    3043                 :         32 : parseQuery(Command *cmd)
    3044                 :            : {
    3045                 :            :     char       *sql,
    3046                 :            :                *p;
    3047                 :            : 
    3048                 :            :     /* We don't want to scribble on cmd->argv[0] until done */
    3049                 :         32 :     sql = pg_strdup(cmd->argv[0]);
    3050                 :            : 
    3051                 :         32 :     cmd->argc = 1;
    3052                 :            : 
    3053                 :         32 :     p = sql;
    3054         [ +  + ]:         86 :     while ((p = strchr(p, ':')) != NULL)
    3055                 :            :     {
    3056                 :            :         char        var[12];
    3057                 :            :         char       *name;
    3058                 :            :         int         eaten;
    3059                 :            : 
    3060                 :         56 :         name = parseVariable(p, &eaten);
    3061         [ +  + ]:         56 :         if (name == NULL)
    3062                 :            :         {
    3063         [ +  + ]:         30 :             while (*p == ':')
    3064                 :            :             {
    3065                 :         20 :                 p++;
    3066                 :            :             }
    3067                 :         10 :             continue;
    3068                 :            :         }
    3069                 :            : 
    3070         [ +  + ]:         46 :         if (cmd->argc >= MAX_ARGS)
    3071                 :            :         {
    3072                 :          2 :             fprintf(stderr, "statement has too many arguments (maximum is %d): %s\n",
    3073                 :            :                     MAX_ARGS - 1, cmd->argv[0]);
    3074                 :          2 :             pg_free(name);
    3075                 :          2 :             return false;
    3076                 :            :         }
    3077                 :            : 
    3078                 :         44 :         sprintf(var, "$%d", cmd->argc);
    3079                 :         44 :         p = replaceVariable(&sql, p, eaten, var);
    3080                 :            : 
    3081                 :         44 :         cmd->argv[cmd->argc] = name;
    3082                 :         44 :         cmd->argc++;
    3083                 :            :     }
    3084                 :            : 
    3085                 :         30 :     pg_free(cmd->argv[0]);
    3086                 :         30 :     cmd->argv[0] = sql;
    3087                 :         32 :     return true;
    3088                 :            : }
    3089                 :            : 
    3090                 :            : /*
    3091                 :            :  * Simple error-printing function, might be needed by lexer
    3092                 :            :  */
    3093                 :            : static void
    3094                 :          0 : pgbench_error(const char *fmt,...)
    3095                 :            : {
    3096                 :            :     va_list     ap;
    3097                 :            : 
    3098                 :          0 :     fflush(stdout);
    3099                 :          0 :     va_start(ap, fmt);
    3100                 :          0 :     vfprintf(stderr, _(fmt), ap);
    3101                 :          0 :     va_end(ap);
    3102                 :          0 : }
    3103                 :            : 
    3104                 :            : /*
    3105                 :            :  * syntax error while parsing a script (in practice, while parsing a
    3106                 :            :  * backslash command, because we don't detect syntax errors in SQL)
    3107                 :            :  *
    3108                 :            :  * source: source of script (filename or builtin-script ID)
    3109                 :            :  * lineno: line number within script (count from 1)
    3110                 :            :  * line: whole line of backslash command, if available
    3111                 :            :  * command: backslash command name, if available
    3112                 :            :  * msg: the actual error message
    3113                 :            :  * more: optional extra message
    3114                 :            :  * column: zero-based column number, or -1 if unknown
    3115                 :            :  */
    3116                 :            : void
    3117                 :         24 : syntax_error(const char *source, int lineno,
    3118                 :            :              const char *line, const char *command,
    3119                 :            :              const char *msg, const char *more, int column)
    3120                 :            : {
    3121                 :         24 :     fprintf(stderr, "%s:%d: %s", source, lineno, msg);
    3122         [ +  + ]:         24 :     if (more != NULL)
    3123                 :          6 :         fprintf(stderr, " (%s)", more);
    3124 [ +  + ][ -  + ]:         24 :     if (column >= 0 && line == NULL)
    3125                 :          0 :         fprintf(stderr, " at column %d", column + 1);
    3126         [ +  - ]:         24 :     if (command != NULL)
    3127                 :         24 :         fprintf(stderr, " in command \"%s\"", command);
    3128                 :         24 :     fprintf(stderr, "\n");
    3129         [ +  + ]:         24 :     if (line != NULL)
    3130                 :            :     {
    3131                 :         20 :         fprintf(stderr, "%s\n", line);
    3132         [ +  + ]:         20 :         if (column >= 0)
    3133                 :            :         {
    3134                 :            :             int         i;
    3135                 :            : 
    3136         [ +  + ]:        144 :             for (i = 0; i < column; i++)
    3137                 :        132 :                 fprintf(stderr, " ");
    3138                 :         12 :             fprintf(stderr, "^ error found here\n");
    3139                 :            :         }
    3140                 :            :     }
    3141                 :         24 :     exit(1);
    3142                 :            : }
    3143                 :            : 
    3144                 :            : /*
    3145                 :            :  * Parse a SQL command; return a Command struct, or NULL if it's a comment
    3146                 :            :  *
    3147                 :            :  * On entry, psqlscan.l has collected the command into "buf", so we don't
    3148                 :            :  * really need to do much here except check for comment and set up a
    3149                 :            :  * Command struct.
    3150                 :            :  */
    3151                 :            : static Command *
    3152                 :       1380 : process_sql_command(PQExpBuffer buf, const char *source)
    3153                 :            : {
    3154                 :            :     Command    *my_command;
    3155                 :            :     char       *p;
    3156                 :            :     char       *nlpos;
    3157                 :            : 
    3158                 :            :     /* Skip any leading whitespace, as well as "--" style comments */
    3159                 :       1380 :     p = buf->data;
    3160                 :            :     for (;;)
    3161                 :            :     {
    3162         [ -  + ]:       1380 :         if (isspace((unsigned char) *p))
    3163                 :          0 :             p++;
    3164         [ -  + ]:       1380 :         else if (strncmp(p, "--", 2) == 0)
    3165                 :            :         {
    3166                 :          0 :             p = strchr(p, '\n');
    3167         [ #  # ]:          0 :             if (p == NULL)
    3168                 :          0 :                 return NULL;
    3169                 :          0 :             p++;
    3170                 :            :         }
    3171                 :            :         else
    3172                 :       1380 :             break;
    3173                 :            :     }
    3174                 :            : 
    3175                 :            :     /* If there's nothing but whitespace and comments, we're done */
    3176         [ +  + ]:       1380 :     if (*p == '\0')
    3177                 :        882 :         return NULL;
    3178                 :            : 
    3179                 :            :     /* Allocate and initialize Command structure */
    3180                 :        498 :     my_command = (Command *) pg_malloc0(sizeof(Command));
    3181                 :        498 :     my_command->command_num = num_commands++;
    3182                 :        498 :     my_command->type = SQL_COMMAND;
    3183                 :        498 :     my_command->meta = META_NONE;
    3184                 :        498 :     initSimpleStats(&my_command->stats);
    3185                 :            : 
    3186                 :            :     /*
    3187                 :            :      * Install query text as the sole argv string.  If we are using a
    3188                 :            :      * non-simple query mode, we'll extract parameters from it later.
    3189                 :            :      */
    3190                 :        498 :     my_command->argv[0] = pg_strdup(p);
    3191                 :        498 :     my_command->argc = 1;
    3192                 :            : 
    3193                 :            :     /*
    3194                 :            :      * If SQL command is multi-line, we only want to save the first line as
    3195                 :            :      * the "line" label.
    3196                 :            :      */
    3197                 :        498 :     nlpos = strchr(p, '\n');
    3198         [ +  + ]:        498 :     if (nlpos)
    3199                 :            :     {
    3200                 :          6 :         my_command->line = pg_malloc(nlpos - p + 1);
    3201                 :          6 :         memcpy(my_command->line, p, nlpos - p);
    3202                 :          6 :         my_command->line[nlpos - p] = '\0';
    3203                 :            :     }
    3204                 :            :     else
    3205                 :        492 :         my_command->line = pg_strdup(p);
    3206                 :            : 
    3207                 :        498 :     return my_command;
    3208                 :            : }
    3209                 :            : 
    3210                 :            : /*
    3211                 :            :  * Parse a backslash command; return a Command struct, or NULL if comment
    3212                 :            :  *
    3213                 :            :  * At call, we have scanned only the initial backslash.
    3214                 :            :  */
    3215                 :            : static Command *
    3216                 :        524 : process_backslash_command(PsqlScanState sstate, const char *source)
    3217                 :            : {
    3218                 :            :     Command    *my_command;
    3219                 :            :     PQExpBufferData word_buf;
    3220                 :            :     int         word_offset;
    3221                 :            :     int         offsets[MAX_ARGS];  /* offsets of argument words */
    3222                 :            :     int         start_offset;
    3223                 :            :     int         lineno;
    3224                 :            :     int         j;
    3225                 :            : 
    3226                 :        524 :     initPQExpBuffer(&word_buf);
    3227                 :            : 
    3228                 :            :     /* Remember location of the backslash */
    3229                 :        524 :     start_offset = expr_scanner_offset(sstate) - 1;
    3230                 :        524 :     lineno = expr_scanner_get_lineno(sstate, start_offset);
    3231                 :            : 
    3232                 :            :     /* Collect first word of command */
    3233         [ -  + ]:        524 :     if (!expr_lex_one_word(sstate, &word_buf, &word_offset))
    3234                 :            :     {
    3235                 :          0 :         termPQExpBuffer(&word_buf);
    3236                 :          0 :         return NULL;
    3237                 :            :     }
    3238                 :            : 
    3239                 :            :     /* Allocate and initialize Command structure */
    3240                 :        524 :     my_command = (Command *) pg_malloc0(sizeof(Command));
    3241                 :        524 :     my_command->command_num = num_commands++;
    3242                 :        524 :     my_command->type = META_COMMAND;
    3243                 :        524 :     my_command->argc = 0;
    3244                 :        524 :     initSimpleStats(&my_command->stats);
    3245                 :            : 
    3246                 :            :     /* Save first word (command name) */
    3247                 :        524 :     j = 0;
    3248                 :        524 :     offsets[j] = word_offset;
    3249                 :        524 :     my_command->argv[j++] = pg_strdup(word_buf.data);
    3250                 :        524 :     my_command->argc++;
    3251                 :            : 
    3252                 :            :     /* ... and convert it to enum form */
    3253                 :        524 :     my_command->meta = getMetaCommand(my_command->argv[0]);
    3254                 :            : 
    3255         [ +  + ]:        524 :     if (my_command->meta == META_SET)
    3256                 :            :     {
    3257                 :            :         /* For \set, collect var name, then lex the expression. */
    3258                 :            :         yyscan_t    yyscanner;
    3259                 :            : 
    3260         [ +  + ]:        486 :         if (!expr_lex_one_word(sstate, &word_buf, &word_offset))
    3261                 :          2 :             syntax_error(source, lineno, my_command->line, my_command->argv[0],
    3262                 :            :                          "missing argument", NULL, -1);
    3263                 :            : 
    3264                 :        484 :         offsets[j] = word_offset;
    3265                 :        484 :         my_command->argv[j++] = pg_strdup(word_buf.data);
    3266                 :        484 :         my_command->argc++;
    3267                 :            : 
    3268                 :        484 :         yyscanner = expr_scanner_init(sstate, source, lineno, start_offset,
    3269                 :        484 :                                       my_command->argv[0]);
    3270                 :            : 
    3271         [ -  + ]:        484 :         if (expr_yyparse(yyscanner) != 0)
    3272                 :            :         {
    3273                 :            :             /* dead code: exit done from syntax_error called by yyerror */
    3274                 :          0 :             exit(1);
    3275                 :            :         }
    3276                 :            : 
    3277                 :        476 :         my_command->expr = expr_parse_result;
    3278                 :            : 
    3279                 :            :         /* Save line, trimming any trailing newline */
    3280                 :        476 :         my_command->line = expr_scanner_get_substring(sstate,
    3281                 :            :                                                       start_offset,
    3282                 :            :                                                       expr_scanner_offset(sstate),
    3283                 :            :                                                       true);
    3284                 :            : 
    3285                 :        476 :         expr_scanner_finish(yyscanner);
    3286                 :            : 
    3287                 :        476 :         termPQExpBuffer(&word_buf);
    3288                 :            : 
    3289                 :        476 :         return my_command;
    3290                 :            :     }
    3291                 :            : 
    3292                 :            :     /* For all other commands, collect remaining words. */
    3293         [ +  + ]:        120 :     while (expr_lex_one_word(sstate, &word_buf, &word_offset))
    3294                 :            :     {
    3295         [ +  + ]:         84 :         if (j >= MAX_ARGS)
    3296                 :          2 :             syntax_error(source, lineno, my_command->line, my_command->argv[0],
    3297                 :            :                          "too many arguments", NULL, -1);
    3298                 :            : 
    3299                 :         82 :         offsets[j] = word_offset;
    3300                 :         82 :         my_command->argv[j++] = pg_strdup(word_buf.data);
    3301                 :         82 :         my_command->argc++;
    3302                 :            :     }
    3303                 :            : 
    3304                 :            :     /* Save line, trimming any trailing newline */
    3305                 :         36 :     my_command->line = expr_scanner_get_substring(sstate,
    3306                 :            :                                                   start_offset,
    3307                 :            :                                                   expr_scanner_offset(sstate),
    3308                 :            :                                                   true);
    3309                 :            : 
    3310         [ +  + ]:         36 :     if (my_command->meta == META_SLEEP)
    3311                 :            :     {
    3312         [ +  + ]:         18 :         if (my_command->argc < 2)
    3313                 :          2 :             syntax_error(source, lineno, my_command->line, my_command->argv[0],
    3314                 :            :                          "missing argument", NULL, -1);
    3315                 :            : 
    3316         [ +  + ]:         16 :         if (my_command->argc > 3)
    3317                 :          2 :             syntax_error(source, lineno, my_command->line, my_command->argv[0],
    3318                 :            :                          "too many arguments", NULL,
    3319                 :          2 :                          offsets[3] - start_offset);
    3320                 :            : 
    3321                 :            :         /*
    3322                 :            :          * Split argument into number and unit to allow "sleep 1ms" etc. We
    3323                 :            :          * don't have to terminate the number argument with null because it
    3324                 :            :          * will be parsed with atoi, which ignores trailing non-digit
    3325                 :            :          * characters.
    3326                 :            :          */
    3327 [ +  + ][ +  + ]:         14 :         if (my_command->argc == 2 && my_command->argv[1][0] != ':')
    3328                 :            :         {
    3329                 :          2 :             char       *c = my_command->argv[1];
    3330                 :            : 
    3331         [ +  + ]:          4 :             while (isdigit((unsigned char) *c))
    3332                 :          2 :                 c++;
    3333         [ +  - ]:          2 :             if (*c)
    3334                 :            :             {
    3335                 :          2 :                 my_command->argv[2] = c;
    3336                 :          2 :                 offsets[2] = offsets[1] + (c - my_command->argv[1]);
    3337                 :          2 :                 my_command->argc = 3;
    3338                 :            :             }
    3339                 :            :         }
    3340                 :            : 
    3341         [ +  + ]:         14 :         if (my_command->argc == 3)
    3342                 :            :         {
    3343   [ +  +  +  + ]:         18 :             if (pg_strcasecmp(my_command->argv[2], "us") != 0 &&
    3344         [ +  + ]:         12 :                 pg_strcasecmp(my_command->argv[2], "ms") != 0 &&
    3345                 :          4 :                 pg_strcasecmp(my_command->argv[2], "s") != 0)
    3346                 :          2 :                 syntax_error(source, lineno, my_command->line, my_command->argv[0],
    3347                 :            :                              "unrecognized time unit, must be us, ms or s",
    3348                 :          4 :                              my_command->argv[2], offsets[2] - start_offset);
    3349                 :            :         }
    3350                 :            :     }
    3351         [ +  + ]:         18 :     else if (my_command->meta == META_SETSHELL)
    3352                 :            :     {
    3353         [ +  + ]:          8 :         if (my_command->argc < 3)
    3354                 :          2 :             syntax_error(source, lineno, my_command->line, my_command->argv[0],
    3355                 :            :                          "missing argument", NULL, -1);
    3356                 :            :     }
    3357         [ +  + ]:         10 :     else if (my_command->meta == META_SHELL)
    3358                 :            :     {
    3359         [ +  + ]:          8 :         if (my_command->argc < 2)
    3360                 :          2 :             syntax_error(source, lineno, my_command->line, my_command->argv[0],
    3361                 :            :                          "missing command", NULL, -1);
    3362                 :            :     }
    3363                 :            :     else
    3364                 :            :     {
    3365                 :            :         /* my_command->meta == META_NONE */
    3366                 :          2 :         syntax_error(source, lineno, my_command->line, my_command->argv[0],
    3367                 :            :                      "invalid command", NULL, -1);
    3368                 :            :     }
    3369                 :            : 
    3370                 :         24 :     termPQExpBuffer(&word_buf);
    3371                 :            : 
    3372                 :        500 :     return my_command;
    3373                 :            : }
    3374                 :            : 
    3375                 :            : /*
    3376                 :            :  * Parse a script (either the contents of a file, or a built-in script)
    3377                 :            :  * and add it to the list of scripts.
    3378                 :            :  */
    3379                 :            : static void
    3380                 :        382 : ParseScript(const char *script, const char *desc, int weight)
    3381                 :            : {
    3382                 :            :     ParsedScript ps;
    3383                 :            :     PsqlScanState sstate;
    3384                 :            :     PQExpBufferData line_buf;
    3385                 :            :     int         alloc_num;
    3386                 :            :     int         index;
    3387                 :            : 
    3388                 :            : #define COMMANDS_ALLOC_NUM 128
    3389                 :        382 :     alloc_num = COMMANDS_ALLOC_NUM;
    3390                 :            : 
    3391                 :            :     /* Initialize all fields of ps */
    3392                 :        382 :     ps.desc = desc;
    3393                 :        382 :     ps.weight = weight;
    3394                 :        382 :     ps.commands = (Command **) pg_malloc(sizeof(Command *) * alloc_num);
    3395                 :        382 :     initStats(&ps.stats, 0);
    3396                 :            : 
    3397                 :            :     /* Prepare to parse script */
    3398                 :        382 :     sstate = psql_scan_create(&pgbench_callbacks);
    3399                 :            : 
    3400                 :            :     /*
    3401                 :            :      * Ideally, we'd scan scripts using the encoding and stdstrings settings
    3402                 :            :      * we get from a DB connection.  However, without major rearrangement of
    3403                 :            :      * pgbench's argument parsing, we can't have a DB connection at the time
    3404                 :            :      * we parse scripts.  Using SQL_ASCII (encoding 0) should work well enough
    3405                 :            :      * with any backend-safe encoding, though conceivably we could be fooled
    3406                 :            :      * if a script file uses a client-only encoding.  We also assume that
    3407                 :            :      * stdstrings should be true, which is a bit riskier.
    3408                 :            :      */
    3409                 :        382 :     psql_scan_setup(sstate, script, strlen(script), 0, true);
    3410                 :            : 
    3411                 :        382 :     initPQExpBuffer(&line_buf);
    3412                 :            : 
    3413                 :        382 :     index = 0;
    3414                 :            : 
    3415                 :            :     for (;;)
    3416                 :        998 :     {
    3417                 :            :         PsqlScanResult sr;
    3418                 :            :         promptStatus_t prompt;
    3419                 :            :         Command    *command;
    3420                 :            : 
    3421                 :       1380 :         resetPQExpBuffer(&line_buf);
    3422                 :            : 
    3423                 :       1380 :         sr = psql_scan(sstate, &line_buf, &prompt);
    3424                 :            : 
    3425                 :            :         /* If we collected a SQL command, process that */
    3426                 :       1380 :         command = process_sql_command(&line_buf, desc);
    3427         [ +  + ]:       1380 :         if (command)
    3428                 :            :         {
    3429                 :        498 :             ps.commands[index] = command;
    3430                 :        498 :             index++;
    3431                 :            : 
    3432         [ -  + ]:        498 :             if (index >= alloc_num)
    3433                 :            :             {
    3434                 :          0 :                 alloc_num += COMMANDS_ALLOC_NUM;
    3435                 :          0 :                 ps.commands = (Command **)
    3436                 :          0 :                     pg_realloc(ps.commands, sizeof(Command *) * alloc_num);
    3437                 :            :             }
    3438                 :            :         }
    3439                 :            : 
    3440                 :            :         /* If we reached a backslash, process that */
    3441         [ +  + ]:       1380 :         if (sr == PSCAN_BACKSLASH)
    3442                 :            :         {
    3443                 :        524 :             command = process_backslash_command(sstate, desc);
    3444         [ +  - ]:        500 :             if (command)
    3445                 :            :             {
    3446                 :        500 :                 ps.commands[index] = command;
    3447                 :        500 :                 index++;
    3448                 :            : 
    3449         [ -  + ]:        500 :                 if (index >= alloc_num)
    3450                 :            :                 {
    3451                 :          0 :                     alloc_num += COMMANDS_ALLOC_NUM;
    3452                 :          0 :                     ps.commands = (Command **)
    3453                 :          0 :                         pg_realloc(ps.commands, sizeof(Command *) * alloc_num);
    3454                 :            :                 }
    3455                 :            :             }
    3456                 :            :         }
    3457                 :            : 
    3458                 :            :         /* Done if we reached EOF */
    3459 [ +  + ][ +  - ]:       1356 :         if (sr == PSCAN_INCOMPLETE || sr == PSCAN_EOL)
    3460                 :            :             break;
    3461                 :            :     }
    3462                 :            : 
    3463                 :        358 :     ps.commands[index] = NULL;
    3464                 :            : 
    3465                 :        358 :     addScript(ps);
    3466                 :            : 
    3467                 :        354 :     termPQExpBuffer(&line_buf);
    3468                 :        354 :     psql_scan_finish(sstate);
    3469                 :        354 :     psql_scan_destroy(sstate);
    3470                 :        354 : }
    3471                 :            : 
    3472                 :            : /*
    3473                 :            :  * Read the entire contents of file fd, and return it in a malloc'd buffer.
    3474                 :            :  *
    3475                 :            :  * The buffer will typically be larger than necessary, but we don't care
    3476                 :            :  * in this program, because we'll free it as soon as we've parsed the script.
    3477                 :            :  */
    3478                 :            : static char *
    3479                 :         78 : read_file_contents(FILE *fd)
    3480                 :            : {
    3481                 :            :     char       *buf;
    3482                 :         78 :     size_t      buflen = BUFSIZ;
    3483                 :         78 :     size_t      used = 0;
    3484                 :            : 
    3485                 :         78 :     buf = (char *) pg_malloc(buflen);
    3486                 :            : 
    3487                 :            :     for (;;)
    3488                 :          0 :     {
    3489                 :            :         size_t      nread;
    3490                 :            : 
    3491                 :         78 :         nread = fread(buf + used, 1, BUFSIZ, fd);
    3492                 :         78 :         used += nread;
    3493                 :            :         /* If fread() read less than requested, must be EOF or error */
    3494         [ +  - ]:         78 :         if (nread < BUFSIZ)
    3495                 :         78 :             break;
    3496                 :            :         /* Enlarge buf so we can read some more */
    3497                 :          0 :         buflen += BUFSIZ;
    3498                 :          0 :         buf = (char *) pg_realloc(buf, buflen);
    3499                 :            :     }
    3500                 :            :     /* There is surely room for a terminator */
    3501                 :         78 :     buf[used] = '\0';
    3502                 :            : 
    3503                 :         78 :     return buf;
    3504                 :            : }
    3505                 :            : 
    3506                 :            : /*
    3507                 :            :  * Given a file name, read it and add its script to the list.
    3508                 :            :  * "-" means to read stdin.
    3509                 :            :  * NB: filename must be storage that won't disappear.
    3510                 :            :  */
    3511                 :            : static void
    3512                 :         80 : process_file(const char *filename, int weight)
    3513                 :            : {
    3514                 :            :     FILE       *fd;
    3515                 :            :     char       *buf;
    3516                 :            : 
    3517                 :            :     /* Slurp the file contents into "buf" */
    3518         [ -  + ]:         80 :     if (strcmp(filename, "-") == 0)
    3519                 :          0 :         fd = stdin;
    3520         [ +  + ]:         80 :     else if ((fd = fopen(filename, "r")) == NULL)
    3521                 :            :     {
    3522                 :          2 :         fprintf(stderr, "could not open file \"%s\": %s\n",
    3523                 :          2 :                 filename, strerror(errno));
    3524                 :          2 :         exit(1);
    3525                 :            :     }
    3526                 :            : 
    3527                 :         78 :     buf = read_file_contents(fd);
    3528                 :            : 
    3529         [ -  + ]:         78 :     if (ferror(fd))
    3530                 :            :     {
    3531                 :          0 :         fprintf(stderr, "could not read file \"%s\": %s\n",
    3532                 :          0 :                 filename, strerror(errno));
    3533                 :          0 :         exit(1);
    3534                 :            :     }
    3535                 :            : 
    3536         [ +  - ]:         78 :     if (fd != stdin)
    3537                 :         78 :         fclose(fd);
    3538                 :            : 
    3539                 :         78 :     ParseScript(buf, filename, weight);
    3540                 :            : 
    3541                 :         52 :     free(buf);
    3542                 :         52 : }
    3543                 :            : 
    3544                 :            : /* Parse the given builtin script and add it to the list. */
    3545                 :            : static void
    3546                 :        304 : process_builtin(const BuiltinScript *bi, int weight)
    3547                 :            : {
    3548                 :        304 :     ParseScript(bi->script, bi->desc, weight);
    3549                 :        302 : }
    3550                 :            : 
    3551                 :            : /* show available builtin scripts */
    3552                 :            : static void
    3553                 :          6 : listAvailableScripts(void)
    3554                 :            : {
    3555                 :            :     int         i;
    3556                 :            : 
    3557                 :          6 :     fprintf(stderr, "Available builtin scripts:\n");
    3558         [ +  + ]:         24 :     for (i = 0; i < lengthof(builtin_script); i++)
    3559                 :         18 :         fprintf(stderr, "\t%s\n", builtin_script[i].name);
    3560                 :          6 :     fprintf(stderr, "\n");
    3561                 :          6 : }
    3562                 :            : 
    3563                 :            : /* return builtin script "name" if unambiguous, fails if not found */
    3564                 :            : static const BuiltinScript *
    3565                 :        308 : findBuiltin(const char *name)
    3566                 :            : {
    3567                 :            :     int         i,
    3568                 :        308 :                 found = 0,
    3569                 :        308 :                 len = strlen(name);
    3570                 :        308 :     const BuiltinScript *result = NULL;
    3571                 :            : 
    3572         [ +  + ]:       1232 :     for (i = 0; i < lengthof(builtin_script); i++)
    3573                 :            :     {
    3574         [ +  + ]:        924 :         if (strncmp(builtin_script[i].name, name, len) == 0)
    3575                 :            :         {
    3576                 :        308 :             result = &builtin_script[i];
    3577                 :        308 :             found++;
    3578                 :            :         }
    3579                 :            :     }
    3580                 :            : 
    3581                 :            :     /* ok, unambiguous result */
    3582         [ +  + ]:        308 :     if (found == 1)
    3583                 :        304 :         return result;
    3584                 :            : 
    3585                 :            :     /* error cases */
    3586         [ +  + ]:          4 :     if (found == 0)
    3587                 :          2 :         fprintf(stderr, "no builtin script found for name \"%s\"\n", name);
    3588                 :            :     else                        /* found > 1 */
    3589                 :          2 :         fprintf(stderr,
    3590                 :            :                 "ambiguous builtin name: %d builtin scripts found for prefix \"%s\"\n", found, name);
    3591                 :            : 
    3592                 :          4 :     listAvailableScripts();
    3593                 :          4 :     exit(1);
    3594                 :            : }
    3595                 :            : 
    3596                 :            : /*
    3597                 :            :  * Determine the weight specification from a script option (-b, -f), if any,
    3598                 :            :  * and return it as an integer (1 is returned if there's no weight).  The
    3599                 :            :  * script name is returned in *script as a malloc'd string.
    3600                 :            :  */
    3601                 :            : static int
    3602                 :        102 : parseScriptWeight(const char *option, char **script)
    3603                 :            : {
    3604                 :            :     char       *sep;
    3605                 :            :     int         weight;
    3606                 :            : 
    3607         [ +  + ]:        102 :     if ((sep = strrchr(option, WSEP)))
    3608                 :            :     {
    3609                 :         14 :         int         namelen = sep - option;
    3610                 :            :         long        wtmp;
    3611                 :            :         char       *badp;
    3612                 :            : 
    3613                 :            :         /* generate the script name */
    3614                 :         14 :         *script = pg_malloc(namelen + 1);
    3615                 :         14 :         strncpy(*script, option, namelen);
    3616                 :         14 :         (*script)[namelen] = '\0';
    3617                 :            : 
    3618                 :            :         /* process digits of the weight spec */
    3619                 :         14 :         errno = 0;
    3620                 :         14 :         wtmp = strtol(sep + 1, &badp, 10);
    3621 [ +  - ][ +  + ]:         14 :         if (errno != 0 || badp == sep + 1 || *badp != '\0')
                 [ -  + ]
    3622                 :            :         {
    3623                 :          2 :             fprintf(stderr, "invalid weight specification: %s\n", sep);
    3624                 :          2 :             exit(1);
    3625                 :            :         }
    3626 [ +  - ][ +  + ]:         12 :         if (wtmp > INT_MAX || wtmp < 0)
    3627                 :            :         {
    3628                 :          2 :             fprintf(stderr,
    3629                 :            :                     "weight specification out of range (0 .. %u): " INT64_FORMAT "\n",
    3630                 :            :                     INT_MAX, (int64) wtmp);
    3631                 :          2 :             exit(1);
    3632                 :            :         }
    3633                 :         10 :         weight = wtmp;
    3634                 :            :     }
    3635                 :            :     else
    3636                 :            :     {
    3637                 :         88 :         *script = pg_strdup(option);
    3638                 :         88 :         weight = 1;
    3639                 :            :     }
    3640                 :            : 
    3641                 :         98 :     return weight;
    3642                 :            : }
    3643                 :            : 
    3644                 :            : /* append a script to the list of scripts to process */
    3645                 :            : static void
    3646                 :        358 : addScript(ParsedScript script)
    3647                 :            : {
    3648 [ +  - ][ +  + ]:        358 :     if (script.commands == NULL || script.commands[0] == NULL)
    3649                 :            :     {
    3650                 :          2 :         fprintf(stderr, "empty command list for script \"%s\"\n", script.desc);
    3651                 :          2 :         exit(1);
    3652                 :            :     }
    3653                 :            : 
    3654         [ +  + ]:        356 :     if (num_scripts >= MAX_SCRIPTS)
    3655                 :            :     {
    3656                 :          2 :         fprintf(stderr, "at most %d SQL scripts are allowed\n", MAX_SCRIPTS);
    3657                 :          2 :         exit(1);
    3658                 :            :     }
    3659                 :            : 
    3660                 :        354 :     sql_script[num_scripts] = script;
    3661                 :        354 :     num_scripts++;
    3662                 :        354 : }
    3663                 :            : 
    3664                 :            : static void
    3665                 :          8 : printSimpleStats(const char *prefix, SimpleStats *ss)
    3666                 :            : {
    3667         [ +  + ]:          8 :     if (ss->count > 0)
    3668                 :            :     {
    3669                 :          6 :         double      latency = ss->sum / ss->count;
    3670                 :          6 :         double      stddev = sqrt(ss->sum2 / ss->count - latency * latency);
    3671                 :            : 
    3672                 :          6 :         printf("%s average = %.3f ms\n", prefix, 0.001 * latency);
    3673                 :          6 :         printf("%s stddev = %.3f ms\n", prefix, 0.001 * stddev);
    3674                 :            :     }
    3675                 :          8 : }
    3676                 :            : 
    3677                 :            : /* print out results */
    3678                 :            : static void
    3679                 :         60 : printResults(TState *threads, StatsData *total, instr_time total_time,
    3680                 :            :              instr_time conn_total_time, int64 latency_late)
    3681                 :            : {
    3682                 :            :     double      time_include,
    3683                 :            :                 tps_include,
    3684                 :            :                 tps_exclude;
    3685                 :         60 :     int64       ntx = total->cnt - total->skipped;
    3686                 :            : 
    3687                 :         60 :     time_include = INSTR_TIME_GET_DOUBLE(total_time);
    3688                 :            : 
    3689                 :            :     /* tps is about actually executed transactions */
    3690                 :         60 :     tps_include = ntx / time_include;
    3691                 :        120 :     tps_exclude = ntx /
    3692                 :         60 :         (time_include - (INSTR_TIME_GET_DOUBLE(conn_total_time) / nclients));
    3693                 :            : 
    3694                 :            :     /* Report test parameters. */
    3695         [ +  + ]:         60 :     printf("transaction type: %s\n",
    3696                 :         60 :            num_scripts == 1 ? sql_script[0].desc : "multiple scripts");
    3697                 :         60 :     printf("scaling factor: %d\n", scale);
    3698                 :         60 :     printf("query mode: %s\n", QUERYMODE[querymode]);
    3699                 :         60 :     printf("number of clients: %d\n", nclients);
    3700                 :         60 :     printf("number of threads: %d\n", nthreads);
    3701         [ +  - ]:         60 :     if (duration <= 0)
    3702                 :            :     {
    3703                 :         60 :         printf("number of transactions per client: %d\n", nxacts);
    3704                 :         60 :         printf("number of transactions actually processed: " INT64_FORMAT "/%d\n",
    3705                 :            :                ntx, nxacts * nclients);
    3706                 :            :     }
    3707                 :            :     else
    3708                 :            :     {
    3709                 :          0 :         printf("duration: %d s\n", duration);
    3710                 :          0 :         printf("number of transactions actually processed: " INT64_FORMAT "\n",
    3711                 :            :                ntx);
    3712                 :            :     }
    3713                 :            : 
    3714                 :            :     /* Remaining stats are nonsensical if we failed to execute any xacts */
    3715         [ +  + ]:         60 :     if (total->cnt <= 0)
    3716                 :         34 :         return;
    3717                 :            : 
    3718 [ +  + ][ +  - ]:         26 :     if (throttle_delay && latency_limit)
    3719                 :          4 :         printf("number of transactions skipped: " INT64_FORMAT " (%.3f %%)\n",
    3720                 :            :                total->skipped,
    3721                 :          4 :                100.0 * total->skipped / total->cnt);
    3722                 :            : 
    3723         [ +  + ]:         26 :     if (latency_limit)
    3724         [ +  + ]:          4 :         printf("number of transactions above the %.1f ms latency limit: " INT64_FORMAT "/" INT64_FORMAT " (%.3f %%)\n",
    3725                 :            :                latency_limit / 1000.0, latency_late, ntx,
    3726                 :          2 :                (ntx > 0) ? 100.0 * latency_late / ntx : 0.0);
    3727                 :            : 
    3728 [ +  + ][ +  - ]:         26 :     if (throttle_delay || progress || latency_limit)
                 [ -  + ]
    3729                 :          4 :         printSimpleStats("latency", &total->latency);
    3730                 :            :     else
    3731                 :            :     {
    3732                 :            :         /* no measurement, show average latency computed from run time */
    3733                 :         22 :         printf("latency average = %.3f ms\n",
    3734                 :         22 :                1000.0 * time_include * nclients / total->cnt);
    3735                 :            :     }
    3736                 :            : 
    3737         [ +  + ]:         26 :     if (throttle_delay)
    3738                 :            :     {
    3739                 :            :         /*
    3740                 :            :          * Report average transaction lag under rate limit throttling.  This
    3741                 :            :          * is the delay between scheduled and actual start times for the
    3742                 :            :          * transaction.  The measured lag may be caused by thread/client load,
    3743                 :            :          * the database load, or the Poisson throttling process.
    3744                 :            :          */
    3745                 :          4 :         printf("rate limit schedule lag: avg %.3f (max %.3f) ms\n",
    3746                 :          8 :                0.001 * total->lag.sum / total->cnt, 0.001 * total->lag.max);
    3747                 :            :     }
    3748                 :            : 
    3749                 :         26 :     printf("tps = %f (including connections establishing)\n", tps_include);
    3750                 :         26 :     printf("tps = %f (excluding connections establishing)\n", tps_exclude);
    3751                 :            : 
    3752                 :            :     /* Report per-script/command statistics */
    3753 [ +  + ][ +  + ]:         26 :     if (per_script_stats || is_latencies)
    3754                 :            :     {
    3755                 :            :         int         i;
    3756                 :            : 
    3757         [ +  + ]:         14 :         for (i = 0; i < num_scripts; i++)
    3758                 :            :         {
    3759         [ +  + ]:          8 :             if (per_script_stats)
    3760                 :            :             {
    3761                 :          4 :                 StatsData  *sstats = &sql_script[i].stats;
    3762                 :            : 
    3763                 :          4 :                 printf("SQL script %d: %s\n"
    3764                 :            :                        " - weight: %d (targets %.1f%% of total)\n"
    3765                 :            :                        " - " INT64_FORMAT " transactions (%.1f%% of total, tps = %f)\n",
    3766                 :            :                        i + 1, sql_script[i].desc,
    3767                 :            :                        sql_script[i].weight,
    3768                 :          4 :                        100.0 * sql_script[i].weight / total_weight,
    3769                 :            :                        sstats->cnt,
    3770                 :          4 :                        100.0 * sstats->cnt / total->cnt,
    3771                 :          4 :                        (sstats->cnt - sstats->skipped) / time_include);
    3772                 :            : 
    3773 [ -  + ][ #  # ]:          4 :                 if (throttle_delay && latency_limit && sstats->cnt > 0)
                 [ #  # ]
    3774                 :          0 :                     printf(" - number of transactions skipped: " INT64_FORMAT " (%.3f%%)\n",
    3775                 :            :                            sstats->skipped,
    3776                 :          0 :                            100.0 * sstats->skipped / sstats->cnt);
    3777                 :            : 
    3778                 :          4 :                 printSimpleStats(" - latency", &sstats->latency);
    3779                 :            :             }
    3780                 :            : 
    3781                 :            :             /* Report per-command latencies */
    3782         [ +  + ]:          8 :             if (is_latencies)
    3783                 :            :             {
    3784                 :            :                 Command   **commands;
    3785                 :            : 
    3786         [ -  + ]:          4 :                 if (per_script_stats)
    3787                 :          0 :                     printf(" - statement latencies in milliseconds:\n");
    3788                 :            :                 else
    3789                 :          4 :                     printf("statement latencies in milliseconds:\n");
    3790                 :            : 
    3791         [ +  + ]:         10 :                 for (commands = sql_script[i].commands;
    3792                 :         10 :                      *commands != NULL;
    3793                 :          6 :                      commands++)
    3794                 :            :                 {
    3795                 :          6 :                     SimpleStats *cstats = &(*commands)->stats;
    3796                 :            : 
    3797         [ +  + ]:          6 :                     printf("   %11.3f  %s\n",
    3798                 :          6 :                            (cstats->count > 0) ?
    3799                 :          4 :                            1000.0 * cstats->sum / cstats->count : 0.0,
    3800                 :          6 :                            (*commands)->line);
    3801                 :            :                 }
    3802                 :            :             }
    3803                 :            :         }
    3804                 :            :     }
    3805                 :            : }
    3806                 :            : 
    3807                 :            : 
    3808                 :            : int
    3809                 :        174 : main(int argc, char **argv)
    3810                 :            : {
    3811                 :            :     static struct option long_options[] = {
    3812                 :            :         /* systematic long/short named options */
    3813                 :            :         {"builtin", required_argument, NULL, 'b'},
    3814                 :            :         {"client", required_argument, NULL, 'c'},
    3815                 :            :         {"connect", no_argument, NULL, 'C'},
    3816                 :            :         {"debug", no_argument, NULL, 'd'},
    3817                 :            :         {"define", required_argument, NULL, 'D'},
    3818                 :            :         {"file", required_argument, NULL, 'f'},
    3819                 :            :         {"fillfactor", required_argument, NULL, 'F'},
    3820                 :            :         {"host", required_argument, NULL, 'h'},
    3821                 :            :         {"initialize", no_argument, NULL, 'i'},
    3822                 :            :         {"init-steps", required_argument, NULL, 'I'},
    3823                 :            :         {"jobs", required_argument, NULL, 'j'},
    3824                 :            :         {"log", no_argument, NULL, 'l'},
    3825                 :            :         {"latency-limit", required_argument, NULL, 'L'},
    3826                 :            :         {"no-vacuum", no_argument, NULL, 'n'},
    3827                 :            :         {"port", required_argument, NULL, 'p'},
    3828                 :            :         {"progress", required_argument, NULL, 'P'},
    3829                 :            :         {"protocol", required_argument, NULL, 'M'},
    3830                 :            :         {"quiet", no_argument, NULL, 'q'},
    3831                 :            :         {"report-latencies", no_argument, NULL, 'r'},
    3832                 :            :         {"rate", required_argument, NULL, 'R'},
    3833                 :            :         {"scale", required_argument, NULL, 's'},
    3834                 :            :         {"select-only", no_argument, NULL, 'S'},
    3835                 :            :         {"skip-some-updates", no_argument, NULL, 'N'},
    3836                 :            :         {"time", required_argument, NULL, 'T'},
    3837                 :            :         {"transactions", required_argument, NULL, 't'},
    3838                 :            :         {"username", required_argument, NULL, 'U'},
    3839                 :            :         {"vacuum-all", no_argument, NULL, 'v'},
    3840                 :            :         /* long-named only options */
    3841                 :            :         {"unlogged-tables", no_argument, NULL, 1},
    3842                 :            :         {"tablespace", required_argument, NULL, 2},
    3843                 :            :         {"index-tablespace", required_argument, NULL, 3},
    3844                 :            :         {"sampling-rate", required_argument, NULL, 4},
    3845                 :            :         {"aggregate-interval", required_argument, NULL, 5},
    3846                 :            :         {"progress-timestamp", no_argument, NULL, 6},
    3847                 :            :         {"log-prefix", required_argument, NULL, 7},
    3848                 :            :         {"foreign-keys", no_argument, NULL, 8},
    3849                 :            :         {NULL, 0, NULL, 0}
    3850                 :            :     };
    3851                 :            : 
    3852                 :            :     int         c;
    3853                 :        174 :     bool        is_init_mode = false;   /* initialize mode? */
    3854                 :        174 :     char       *initialize_steps = NULL;
    3855                 :        174 :     bool        foreign_keys = false;
    3856                 :        174 :     bool        is_no_vacuum = false;
    3857                 :        174 :     bool        do_vacuum_accounts = false; /* vacuum accounts table? */
    3858                 :            :     int         optindex;
    3859                 :        174 :     bool        scale_given = false;
    3860                 :            : 
    3861                 :        174 :     bool        benchmarking_option_set = false;
    3862                 :        174 :     bool        initialization_option_set = false;
    3863                 :        174 :     bool        internal_script_used = false;
    3864                 :            : 
    3865                 :            :     CState     *state;          /* status of clients */
    3866                 :            :     TState     *threads;        /* array of thread */
    3867                 :            : 
    3868                 :            :     instr_time  start_time;     /* start up time */
    3869                 :            :     instr_time  total_time;
    3870                 :            :     instr_time  conn_total_time;
    3871                 :        174 :     int64       latency_late = 0;
    3872                 :            :     StatsData   stats;
    3873                 :            :     int         weight;
    3874                 :            : 
    3875                 :            :     int         i;
    3876                 :            :     int         nclients_dealt;
    3877                 :            : 
    3878                 :            : #ifdef HAVE_GETRLIMIT
    3879                 :            :     struct rlimit rlim;
    3880                 :            : #endif
    3881                 :            : 
    3882                 :            :     PGconn     *con;
    3883                 :            :     PGresult   *res;
    3884                 :            :     char       *env;
    3885                 :            : 
    3886                 :        174 :     progname = get_progname(argv[0]);
    3887                 :            : 
    3888         [ +  - ]:        174 :     if (argc > 1)
    3889                 :            :     {
    3890 [ +  + ][ -  + ]:        174 :         if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-?") == 0)
    3891                 :            :         {
    3892                 :          2 :             usage();
    3893                 :          2 :             exit(0);
    3894                 :            :         }
    3895 [ +  - ][ +  + ]:        172 :         if (strcmp(argv[1], "--version") == 0 || strcmp(argv[1], "-V") == 0)
    3896                 :            :         {
    3897                 :          2 :             puts("pgbench (PostgreSQL) " PG_VERSION);
    3898                 :          2 :             exit(0);
    3899                 :            :         }
    3900                 :            :     }
    3901                 :            : 
    3902                 :            : #ifdef WIN32
    3903                 :            :     /* stderr is buffered on Win32. */
    3904                 :            :     setvbuf(stderr, NULL, _IONBF, 0);
    3905                 :            : #endif
    3906                 :            : 
    3907 [ +  + ][ +  - ]:        170 :     if ((env = getenv("PGHOST")) != NULL && *env != '\0')
    3908                 :        100 :         pghost = env;
    3909 [ +  + ][ +  - ]:        170 :     if ((env = getenv("PGPORT")) != NULL && *env != '\0')
    3910                 :         98 :         pgport = env;
    3911 [ -  + ][ #  # ]:         72 :     else if ((env = getenv("PGUSER")) != NULL && *env != '\0')
    3912                 :          0 :         login = env;
    3913                 :            : 
    3914                 :        170 :     state = (CState *) pg_malloc(sizeof(CState));
    3915                 :        170 :     memset(state, 0, sizeof(CState));
    3916                 :            : 
    3917         [ +  + ]:        976 :     while ((c = getopt_long(argc, argv, "iI:h:nvp:dqb:SNc:j:Crs:t:T:U:lf:D:F:M:P:R:L:", long_options, &optindex)) != -1)
    3918                 :            :     {
    3919                 :            :         char       *script;
    3920                 :            : 
    3921   [ +  +  +  +  :        878 :         switch (c)
          +  +  +  +  +  
          +  +  +  +  +  
          +  +  +  +  +  
          +  +  +  +  +  
          +  +  +  +  +  
          +  +  +  +  +  
                   +  + ]
    3922                 :            :         {
    3923                 :            :             case 'i':
    3924                 :         12 :                 is_init_mode = true;
    3925                 :         12 :                 break;
    3926                 :            :             case 'I':
    3927         [ -  + ]:          8 :                 if (initialize_steps)
    3928                 :          0 :                     pg_free(initialize_steps);
    3929                 :          8 :                 initialize_steps = pg_strdup(optarg);
    3930                 :          8 :                 checkInitSteps(initialize_steps);
    3931                 :          6 :                 initialization_option_set = true;
    3932                 :          6 :                 break;
    3933                 :            :             case 'h':
    3934                 :          2 :                 pghost = pg_strdup(optarg);
    3935                 :          2 :                 break;
    3936                 :            :             case 'n':
    3937                 :         86 :                 is_no_vacuum = true;
    3938                 :         86 :                 break;
    3939                 :            :             case 'v':
    3940                 :          2 :                 benchmarking_option_set = true;
    3941                 :          2 :                 do_vacuum_accounts = true;
    3942                 :          2 :                 break;
    3943                 :            :             case 'p':
    3944                 :          2 :                 pgport = pg_strdup(optarg);
    3945                 :          2 :                 break;
    3946                 :            :             case 'd':
    3947                 :          4 :                 debug++;
    3948                 :          4 :                 break;
    3949                 :            :             case 'c':
    3950                 :         20 :                 benchmarking_option_set = true;
    3951                 :         20 :                 nclients = atoi(optarg);
    3952 [ +  + ][ -  + ]:         20 :                 if (nclients <= 0 || nclients > MAXCLIENTS)
    3953                 :            :                 {
    3954                 :          2 :                     fprintf(stderr, "invalid number of clients: \"%s\"\n",
    3955                 :            :                             optarg);
    3956                 :          2 :                     exit(1);
    3957                 :            :                 }
    3958                 :            : #ifdef HAVE_GETRLIMIT
    3959                 :            : #ifdef RLIMIT_NOFILE            /* most platforms use RLIMIT_NOFILE */
    3960         [ -  + ]:         18 :                 if (getrlimit(RLIMIT_NOFILE, &rlim) == -1)
    3961                 :            : #else                           /* but BSD doesn't ... */
    3962                 :            :                 if (getrlimit(RLIMIT_OFILE, &rlim) == -1)
    3963                 :            : #endif                          /* RLIMIT_NOFILE */
    3964                 :            :                 {
    3965                 :          0 :                     fprintf(stderr, "getrlimit failed: %s\n", strerror(errno));
    3966                 :          0 :                     exit(1);
    3967                 :            :                 }
    3968         [ -  + ]:         18 :                 if (rlim.rlim_cur < nclients + 3)
    3969                 :            :                 {
    3970                 :          0 :                     fprintf(stderr, "need at least %d open files, but system limit is %ld\n",
    3971                 :          0 :                             nclients + 3, (long) rlim.rlim_cur);
    3972                 :          0 :                     fprintf(stderr, "Reduce number of clients, or use limit/ulimit to increase the system limit.\n");
    3973                 :          0 :                     exit(1);
    3974                 :            :                 }
    3975                 :            : #endif                          /* HAVE_GETRLIMIT */
    3976                 :         18 :                 break;
    3977                 :            :             case 'j':           /* jobs */
    3978                 :          6 :                 benchmarking_option_set = true;
    3979                 :          6 :                 nthreads = atoi(optarg);
    3980         [ +  + ]:          6 :                 if (nthreads <= 0)
    3981                 :            :                 {
    3982                 :          2 :                     fprintf(stderr, "invalid number of threads: \"%s\"\n",
    3983                 :            :                             optarg);
    3984                 :          2 :                     exit(1);
    3985                 :            :                 }
    3986                 :            : #ifndef ENABLE_THREAD_SAFETY
    3987                 :            :                 if (nthreads != 1)
    3988                 :            :                 {
    3989                 :            :                     fprintf(stderr, "threads are not supported on this platform; use -j1\n");
    3990                 :            :                     exit(1);
    3991                 :            :                 }
    3992                 :            : #endif                          /* !ENABLE_THREAD_SAFETY */
    3993                 :          4 :                 break;
    3994                 :            :             case 'C':
    3995                 :          4 :                 benchmarking_option_set = true;
    3996                 :          4 :                 is_connect = true;
    3997                 :          4 :                 break;
    3998                 :            :             case 'r':
    3999                 :          4 :                 benchmarking_option_set = true;
    4000                 :          4 :                 is_latencies = true;
    4001                 :          4 :                 break;
    4002                 :            :             case 's':
    4003                 :          6 :                 scale_given = true;
    4004                 :          6 :                 scale = atoi(optarg);
    4005         [ +  + ]:          6 :                 if (scale <= 0)
    4006                 :            :                 {
    4007                 :          2 :                     fprintf(stderr, "invalid scaling factor: \"%s\"\n", optarg);
    4008                 :          2 :                     exit(1);
    4009                 :            :                 }
    4010                 :          4 :                 break;
    4011                 :            :             case 't':
    4012                 :         96 :                 benchmarking_option_set = true;
    4013                 :         96 :                 nxacts = atoi(optarg);
    4014         [ +  + ]:         96 :                 if (nxacts <= 0)
    4015                 :            :                 {
    4016                 :          2 :                     fprintf(stderr, "invalid number of transactions: \"%s\"\n",
    4017                 :            :                             optarg);
    4018                 :          2 :                     exit(1);
    4019                 :            :                 }
    4020                 :         94 :                 break;
    4021                 :            :             case 'T':
    4022                 :         10 :                 benchmarking_option_set = true;
    4023                 :         10 :                 duration = atoi(optarg);
    4024         [ +  + ]:         10 :                 if (duration <= 0)
    4025                 :            :                 {
    4026                 :          2 :                     fprintf(stderr, "invalid duration: \"%s\"\n", optarg);
    4027                 :          2 :                     exit(1);
    4028                 :            :                 }
    4029                 :          8 :                 break;
    4030                 :            :             case 'U':
    4031                 :          2 :                 login = pg_strdup(optarg);
    4032                 :          2 :                 break;
    4033                 :            :             case 'l':
    4034                 :         14 :                 benchmarking_option_set = true;
    4035                 :         14 :                 use_log = true;
    4036                 :         14 :                 break;
    4037                 :            :             case 'q':
    4038                 :          2 :                 initialization_option_set = true;
    4039                 :          2 :                 use_quiet = true;
    4040                 :          2 :                 break;
    4041                 :            :             case 'b':
    4042         [ +  + ]:         24 :                 if (strcmp(optarg, "list") == 0)
    4043                 :            :                 {
    4044                 :          2 :                     listAvailableScripts();
    4045                 :          2 :                     exit(0);
    4046                 :            :                 }
    4047                 :         22 :                 weight = parseScriptWeight(optarg, &script);
    4048                 :         18 :                 process_builtin(findBuiltin(script), weight);
    4049                 :         14 :                 benchmarking_option_set = true;
    4050                 :         14 :                 internal_script_used = true;
    4051                 :         14 :                 break;
    4052                 :            :             case 'S':
    4053                 :        268 :                 process_builtin(findBuiltin("select-only"), 1);
    4054                 :        266 :                 benchmarking_option_set = true;
    4055                 :        266 :                 internal_script_used = true;
    4056                 :        266 :                 break;
    4057                 :            :             case 'N':
    4058                 :          2 :                 process_builtin(findBuiltin("simple-update"), 1);
    4059                 :          2 :                 benchmarking_option_set = true;
    4060                 :          2 :                 internal_script_used = true;
    4061                 :          2 :                 break;
    4062                 :            :             case 'f':
    4063                 :         80 :                 weight = parseScriptWeight(optarg, &script);
    4064                 :         80 :                 process_file(script, weight);
    4065                 :         52 :                 benchmarking_option_set = true;
    4066                 :         52 :                 break;
    4067                 :            :             case 'D':
    4068                 :            :                 {
    4069                 :            :                     char       *p;
    4070                 :            : 
    4071                 :         74 :                     benchmarking_option_set = true;
    4072                 :            : 
    4073 [ +  + ][ +  - ]:         74 :                     if ((p = strchr(optarg, '=')) == NULL || p == optarg || *(p + 1) == '\0')
                 [ -  + ]
    4074                 :            :                     {
    4075                 :          2 :                         fprintf(stderr, "invalid variable definition: \"%s\"\n",
    4076                 :            :                                 optarg);
    4077                 :          2 :                         exit(1);
    4078                 :            :                     }
    4079                 :            : 
    4080                 :         72 :                     *p++ = '\0';
    4081         [ -  + ]:         72 :                     if (!putVariable(&state[0], "option", optarg, p))
    4082                 :          0 :                         exit(1);
    4083                 :            :                 }
    4084                 :         72 :                 break;
    4085                 :            :             case 'F':
    4086                 :          6 :                 initialization_option_set = true;
    4087                 :          6 :                 fillfactor = atoi(optarg);
    4088 [ +  + ][ -  + ]:          6 :                 if (fillfactor < 10 || fillfactor > 100)
    4089                 :            :                 {
    4090                 :          2 :                     fprintf(stderr, "invalid fillfactor: \"%s\"\n", optarg);
    4091                 :          2 :                     exit(1);
    4092                 :            :                 }
    4093                 :          4 :                 break;
    4094                 :            :             case 'M':
    4095                 :         78 :                 benchmarking_option_set = true;
    4096         [ +  + ]:        224 :                 for (querymode = 0; querymode < NUM_QUERYMODE; querymode++)
    4097         [ +  + ]:        222 :                     if (strcmp(optarg, QUERYMODE[querymode]) == 0)
    4098                 :         76 :                         break;
    4099         [ +  + ]:         78 :                 if (querymode >= NUM_QUERYMODE)
    4100                 :            :                 {
    4101                 :          2 :                     fprintf(stderr, "invalid query mode (-M): \"%s\"\n",
    4102                 :            :                             optarg);
    4103                 :          2 :                     exit(1);
    4104                 :            :                 }
    4105                 :         76 :                 break;
    4106                 :            :             case 'P':
    4107                 :          4 :                 benchmarking_option_set = true;
    4108                 :          4 :                 progress = atoi(optarg);
    4109         [ +  + ]:          4 :                 if (progress <= 0)
    4110                 :            :                 {
    4111                 :          2 :                     fprintf(stderr, "invalid thread progress delay: \"%s\"\n",
    4112                 :            :                             optarg);
    4113                 :          2 :                     exit(1);
    4114                 :            :                 }
    4115                 :          2 :                 break;
    4116                 :            :             case 'R':
    4117                 :            :                 {
    4118                 :            :                     /* get a double from the beginning of option value */
    4119                 :          6 :                     double      throttle_value = atof(optarg);
    4120                 :            : 
    4121                 :          6 :                     benchmarking_option_set = true;
    4122                 :            : 
    4123         [ +  + ]:          6 :                     if (throttle_value <= 0.0)
    4124                 :            :                     {
    4125                 :          2 :                         fprintf(stderr, "invalid rate limit: \"%s\"\n", optarg);
    4126                 :          2 :                         exit(1);
    4127                 :            :                     }
    4128                 :            :                     /* Invert rate limit into a time offset */
    4129                 :          4 :                     throttle_delay = (int64) (1000000.0 / throttle_value);
    4130                 :            :                 }
    4131                 :          4 :                 break;
    4132                 :            :             case 'L':
    4133                 :            :                 {
    4134                 :          6 :                     double      limit_ms = atof(optarg);
    4135                 :            : 
    4136         [ +  + ]:          6 :                     if (limit_ms <= 0.0)
    4137                 :            :                     {
    4138                 :          2 :                         fprintf(stderr, "invalid latency limit: \"%s\"\n",
    4139                 :            :                                 optarg);
    4140                 :          2 :                         exit(1);
    4141                 :            :                     }
    4142                 :          4 :                     benchmarking_option_set = true;
    4143                 :          4 :                     latency_limit = (int64) (limit_ms * 1000);
    4144                 :            :                 }
    4145                 :          4 :                 break;
    4146                 :            :             case 1:             /* unlogged-tables */
    4147                 :          4 :                 initialization_option_set = true;
    4148                 :          4 :                 unlogged_tables = true;
    4149                 :          4 :                 break;
    4150                 :            :             case 2:             /* tablespace */
    4151                 :          2 :                 initialization_option_set = true;
    4152                 :          2 :                 tablespace = pg_strdup(optarg);
    4153                 :          2 :                 break;
    4154                 :            :             case 3:             /* index-tablespace */
    4155                 :          2 :                 initialization_option_set = true;
    4156                 :          2 :                 index_tablespace = pg_strdup(optarg);
    4157                 :          2 :                 break;
    4158                 :            :             case 4:             /* sampling-rate */
    4159                 :         10 :                 benchmarking_option_set = true;
    4160                 :         10 :                 sample_rate = atof(optarg);
    4161 [ +  + ][ -  + ]:         10 :                 if (sample_rate <= 0.0 || sample_rate > 1.0)
    4162                 :            :                 {
    4163                 :          2 :                     fprintf(stderr, "invalid sampling rate: \"%s\"\n", optarg);
    4164                 :          2 :                     exit(1);
    4165                 :            :                 }
    4166                 :          8 :                 break;
    4167                 :            :             case 5:             /* aggregate-interval */
    4168                 :         12 :                 benchmarking_option_set = true;
    4169                 :         12 :                 agg_interval = atoi(optarg);
    4170         [ +  + ]:         12 :                 if (agg_interval <= 0)
    4171                 :            :                 {
    4172                 :          2 :                     fprintf(stderr, "invalid number of seconds for aggregation: \"%s\"\n",
    4173                 :            :                             optarg);
    4174                 :          2 :                     exit(1);
    4175                 :            :                 }
    4176                 :         10 :                 break;
    4177                 :            :             case 6:             /* progress-timestamp */
    4178                 :          4 :                 progress_timestamp = true;
    4179                 :          4 :                 benchmarking_option_set = true;
    4180                 :          4 :                 break;
    4181                 :            :             case 7:             /* log-prefix */
    4182                 :          8 :                 benchmarking_option_set = true;
    4183                 :          8 :                 logfile_prefix = pg_strdup(optarg);
    4184                 :          8 :                 break;
    4185                 :            :             case 8:             /* foreign-keys */
    4186                 :          4 :                 initialization_option_set = true;
    4187                 :          4 :                 foreign_keys = true;
    4188                 :          4 :                 break;
    4189                 :            :             default:
    4190                 :          4 :                 fprintf(stderr, _("Try \"%s --help\" for more information.\n"), progname);
    4191                 :        810 :                 exit(1);
    4192                 :            :                 break;
    4193                 :            :         }
    4194                 :            :     }
    4195                 :            : 
    4196                 :            :     /* set default script if none */
    4197 [ +  + ][ +  + ]:         98 :     if (num_scripts == 0 && !is_init_mode)
    4198                 :            :     {
    4199                 :         20 :         process_builtin(findBuiltin("tpcb-like"), 1);
    4200                 :         20 :         benchmarking_option_set = true;
    4201                 :         20 :         internal_script_used = true;
    4202                 :            :     }
    4203                 :            : 
    4204                 :            :     /* if not simple query mode, parse the script(s) to find parameters */
    4205         [ +  + ]:         98 :     if (querymode != QUERY_SIMPLE)
    4206                 :            :     {
    4207         [ +  + ]:         92 :         for (i = 0; i < num_scripts; i++)
    4208                 :            :         {
    4209                 :         48 :             Command   **commands = sql_script[i].commands;
    4210                 :            :             int         j;
    4211                 :            : 
    4212         [ +  + ]:        128 :             for (j = 0; commands[j] != NULL; j++)
    4213                 :            :             {
    4214         [ +  + ]:         82 :                 if (commands[j]->type != SQL_COMMAND)
    4215                 :         50 :                     continue;
    4216         [ +  + ]:         32 :                 if (!parseQuery(commands[j]))
    4217                 :          2 :                     exit(1);
    4218                 :            :             }
    4219                 :            :         }
    4220                 :            :     }
    4221                 :            : 
    4222                 :            :     /* compute total_weight */
    4223         [ +  + ]:        192 :     for (i = 0; i < num_scripts; i++)
    4224                 :            :         /* cannot overflow: weight is 32b, total_weight 64b */
    4225                 :         96 :         total_weight += sql_script[i].weight;
    4226                 :            : 
    4227 [ +  + ][ +  + ]:         96 :     if (total_weight == 0 && !is_init_mode)
    4228                 :            :     {
    4229                 :          2 :         fprintf(stderr, "total script weight must not be zero\n");
    4230                 :          2 :         exit(1);
    4231                 :            :     }
    4232                 :            : 
    4233                 :            :     /* show per script stats if several scripts are used */
    4234         [ +  + ]:         94 :     if (num_scripts > 1)
    4235                 :          2 :         per_script_stats = true;
    4236                 :            : 
    4237                 :            :     /*
    4238                 :            :      * Don't need more threads than there are clients.  (This is not merely an
    4239                 :            :      * optimization; throttle_delay is calculated incorrectly below if some
    4240                 :            :      * threads have no clients assigned to them.)
    4241                 :            :      */
    4242         [ +  + ]:         94 :     if (nthreads > nclients)
    4243                 :          2 :         nthreads = nclients;
    4244                 :            : 
    4245                 :            :     /* compute a per thread delay */
    4246                 :         94 :     throttle_delay *= nthreads;
    4247                 :            : 
    4248         [ +  + ]:         94 :     if (argc > optind)
    4249                 :          2 :         dbName = argv[optind];
    4250                 :            :     else
    4251                 :            :     {
    4252 [ +  + ][ +  - ]:         92 :         if ((env = getenv("PGDATABASE")) != NULL && *env != '\0')
    4253                 :         68 :             dbName = env;
    4254 [ -  + ][ #  # ]:         24 :         else if (login != NULL && *login != '\0')
    4255                 :          0 :             dbName = login;
    4256                 :            :         else
    4257                 :         24 :             dbName = "";
    4258                 :            :     }
    4259                 :            : 
    4260         [ +  + ]:         94 :     if (is_init_mode)
    4261                 :            :     {
    4262         [ +  + ]:          8 :         if (benchmarking_option_set)
    4263                 :            :         {
    4264                 :          2 :             fprintf(stderr, "some of the specified options cannot be used in initialization (-i) mode\n");
    4265                 :          2 :             exit(1);
    4266                 :            :         }
    4267                 :            : 
    4268         [ +  + ]:          6 :         if (initialize_steps == NULL)
    4269                 :          2 :             initialize_steps = pg_strdup(DEFAULT_INIT_STEPS);
    4270                 :            : 
    4271         [ +  + ]:          6 :         if (is_no_vacuum)
    4272                 :            :         {
    4273                 :            :             /* Remove any vacuum step in initialize_steps */
    4274                 :            :             char       *p;
    4275                 :            : 
    4276         [ +  + ]:          8 :             while ((p = strchr(initialize_steps, 'v')) != NULL)
    4277                 :          6 :                 *p = ' ';
    4278                 :            :         }
    4279                 :            : 
    4280         [ +  + ]:          6 :         if (foreign_keys)
    4281                 :            :         {
    4282                 :            :             /* Add 'f' to end of initialize_steps, if not already there */
    4283         [ +  - ]:          4 :             if (strchr(initialize_steps, 'f') == NULL)
    4284                 :            :             {
    4285                 :          4 :                 initialize_steps = (char *)
    4286                 :          4 :                     pg_realloc(initialize_steps,
    4287                 :          4 :                                strlen(initialize_steps) + 2);
    4288                 :          4 :                 strcat(initialize_steps, "f");
    4289                 :            :             }
    4290                 :            :         }
    4291                 :            : 
    4292                 :          6 :         runInitSteps(initialize_steps);
    4293                 :          6 :         exit(0);
    4294                 :            :     }
    4295                 :            :     else
    4296                 :            :     {
    4297         [ +  + ]:         86 :         if (initialization_option_set)
    4298                 :            :         {
    4299                 :          4 :             fprintf(stderr, "some of the specified options cannot be used in benchmarking mode\n");
    4300                 :          4 :             exit(1);
    4301                 :            :         }
    4302                 :            :     }
    4303                 :            : 
    4304 [ +  + ][ +  + ]:         82 :     if (nxacts > 0 && duration > 0)
    4305                 :            :     {
    4306                 :          4 :         fprintf(stderr, "specify either a number of transactions (-t) or a duration (-T), not both\n");
    4307                 :          4 :         exit(1);
    4308                 :            :     }
    4309                 :            : 
    4310                 :            :     /* Use DEFAULT_NXACTS if neither nxacts nor duration is specified. */
    4311 [ +  + ][ +  + ]:         78 :     if (nxacts <= 0 && duration <= 0)
    4312                 :         12 :         nxacts = DEFAULT_NXACTS;
    4313                 :            : 
    4314                 :            :     /* --sampling-rate may be used only with -l */
    4315 [ +  + ][ +  + ]:         78 :     if (sample_rate > 0.0 && !use_log)
    4316                 :            :     {
    4317                 :          2 :         fprintf(stderr, "log sampling (--sampling-rate) is allowed only when logging transactions (-l)\n");
    4318                 :          2 :         exit(1);
    4319                 :            :     }
    4320                 :            : 
    4321                 :            :     /* --sampling-rate may not be used with --aggregate-interval */
    4322 [ +  + ][ +  + ]:         76 :     if (sample_rate > 0.0 && agg_interval > 0)
    4323                 :            :     {
    4324                 :          2 :         fprintf(stderr, "log sampling (--sampling-rate) and aggregation (--aggregate-interval) cannot be used at the same time\n");
    4325                 :          2 :         exit(1);
    4326                 :            :     }
    4327                 :            : 
    4328 [ +  + ][ +  + ]:         74 :     if (agg_interval > 0 && !use_log)
    4329                 :            :     {
    4330                 :          2 :         fprintf(stderr, "log aggregation is allowed only when actually logging transactions\n");
    4331                 :          2 :         exit(1);
    4332                 :            :     }
    4333                 :            : 
    4334 [ +  + ][ +  + ]:         72 :     if (!use_log && logfile_prefix)
    4335                 :            :     {
    4336                 :          2 :         fprintf(stderr, "log file prefix (--log-prefix) is allowed only when logging transactions (-l)\n");
    4337                 :          2 :         exit(1);
    4338                 :            :     }
    4339                 :            : 
    4340 [ +  + ][ +  + ]:         70 :     if (duration > 0 && agg_interval > duration)
    4341                 :            :     {
    4342                 :          2 :         fprintf(stderr, "number of seconds for aggregation (%d) must not be higher than test duration (%d)\n", agg_interval, duration);
    4343                 :          2 :         exit(1);
    4344                 :            :     }
    4345                 :            : 
    4346 [ +  + ][ +  - ]:         68 :     if (duration > 0 && agg_interval > 0 && duration % agg_interval != 0)
                 [ +  - ]
    4347                 :            :     {
    4348                 :          2 :         fprintf(stderr, "duration (%d) must be a multiple of aggregation interval (%d)\n", duration, agg_interval);
    4349                 :          2 :         exit(1);
    4350                 :            :     }
    4351                 :            : 
    4352 [ +  + ][ +  - ]:         66 :     if (progress_timestamp && progress == 0)
    4353                 :            :     {
    4354                 :          2 :         fprintf(stderr, "--progress-timestamp is allowed only under --progress\n");
    4355                 :          2 :         exit(1);
    4356                 :            :     }
    4357                 :            : 
    4358                 :            :     /*
    4359                 :            :      * save main process id in the global variable because process id will be
    4360                 :            :      * changed after fork.
    4361                 :            :      */
    4362                 :         64 :     main_pid = (int) getpid();
    4363                 :            : 
    4364         [ +  + ]:         64 :     if (nclients > 1)
    4365                 :            :     {
    4366                 :         14 :         state = (CState *) pg_realloc(state, sizeof(CState) * nclients);
    4367                 :         14 :         memset(state + 1, 0, sizeof(CState) * (nclients - 1));
    4368                 :            : 
    4369                 :            :         /* copy any -D switch values to all clients */
    4370         [ +  + ]:         50 :         for (i = 1; i < nclients; i++)
    4371                 :            :         {
    4372                 :            :             int         j;
    4373                 :            : 
    4374                 :         36 :             state[i].id = i;
    4375         [ +  + ]:         38 :             for (j = 0; j < state[0].nvariables; j++)
    4376                 :            :             {
    4377                 :          2 :                 Variable   *var = &state[0].variables[j];
    4378                 :            : 
    4379         [ -  + ]:          2 :                 if (var->is_numeric)
    4380                 :            :                 {
    4381         [ #  # ]:          0 :                     if (!putVariableNumber(&state[i], "startup",
    4382                 :          0 :                                            var->name, &var->num_value))
    4383                 :          0 :                         exit(1);
    4384                 :            :                 }
    4385                 :            :                 else
    4386                 :            :                 {
    4387         [ -  + ]:          2 :                     if (!putVariable(&state[i], "startup",
    4388                 :          2 :                                      var->name, var->value))
    4389                 :          0 :                         exit(1);
    4390                 :            :                 }
    4391                 :            :             }
    4392                 :            :         }
    4393                 :            :     }
    4394                 :            : 
    4395         [ +  + ]:         64 :     if (debug)
    4396                 :            :     {
    4397         [ +  - ]:          2 :         if (duration <= 0)
    4398                 :          2 :             printf("pghost: %s pgport: %s nclients: %d nxacts: %d dbName: %s\n",
    4399                 :            :                    pghost, pgport, nclients, nxacts, dbName);
    4400                 :            :         else
    4401                 :          0 :             printf("pghost: %s pgport: %s nclients: %d duration: %d dbName: %s\n",
    4402                 :            :                    pghost, pgport, nclients, duration, dbName);
    4403                 :            :     }
    4404                 :            : 
    4405                 :            :     /* opening connection... */
    4406                 :         64 :     con = doConnect();
    4407         [ +  + ]:         64 :     if (con == NULL)
    4408                 :          2 :         exit(1);
    4409                 :            : 
    4410         [ -  + ]:         62 :     if (PQstatus(con) == CONNECTION_BAD)
    4411                 :            :     {
    4412                 :          0 :         fprintf(stderr, "connection to database \"%s\" failed\n", dbName);
    4413                 :          0 :         fprintf(stderr, "%s", PQerrorMessage(con));
    4414                 :          0 :         exit(1);
    4415                 :            :     }
    4416                 :            : 
    4417         [ +  + ]:         62 :     if (internal_script_used)
    4418                 :            :     {
    4419                 :            :         /*
    4420                 :            :          * get the scaling factor that should be same as count(*) from
    4421                 :            :          * pgbench_branches if this is not a custom query
    4422                 :            :          */
    4423                 :         14 :         res = PQexec(con, "select count(*) from pgbench_branches");
    4424         [ +  + ]:         14 :         if (PQresultStatus(res) != PGRES_TUPLES_OK)
    4425                 :            :         {
    4426                 :          2 :             char       *sqlState = PQresultErrorField(res, PG_DIAG_SQLSTATE);
    4427                 :            : 
    4428                 :          2 :             fprintf(stderr, "%s", PQerrorMessage(con));
    4429 [ +  - ][ +  - ]:          2 :             if (sqlState && strcmp(sqlState, ERRCODE_UNDEFINED_TABLE) == 0)
    4430                 :            :             {
    4431                 :          2 :                 fprintf(stderr, "Perhaps you need to do initialization (\"pgbench -i\") in database \"%s\"\n", PQdb(con));
    4432                 :            :             }
    4433                 :            : 
    4434                 :          2 :             exit(1);
    4435                 :            :         }
    4436                 :         12 :         scale = atoi(PQgetvalue(res, 0, 0));
    4437         [ -  + ]:         12 :         if (scale < 0)
    4438                 :            :         {
    4439                 :          0 :             fprintf(stderr, "invalid count(*) from pgbench_branches: \"%s\"\n",
    4440                 :            :                     PQgetvalue(res, 0, 0));
    4441                 :          0 :             exit(1);
    4442                 :            :         }
    4443                 :         12 :         PQclear(res);
    4444                 :            : 
    4445                 :            :         /* warn if we override user-given -s switch */
    4446         [ +  + ]:         12 :         if (scale_given)
    4447                 :          2 :             fprintf(stderr,
    4448                 :            :                     "scale option ignored, using count from pgbench_branches table (%d)\n",
    4449                 :            :                     scale);
    4450                 :            :     }
    4451                 :            : 
    4452                 :            :     /*
    4453                 :            :      * :scale variables normally get -s or database scale, but don't override
    4454                 :            :      * an explicit -D switch
    4455                 :            :      */
    4456         [ +  - ]:         60 :     if (lookupVariable(&state[0], "scale") == NULL)
    4457                 :            :     {
    4458         [ +  + ]:        156 :         for (i = 0; i < nclients; i++)
    4459                 :            :         {
    4460         [ -  + ]:         96 :             if (!putVariableInt(&state[i], "startup", "scale", scale))
    4461                 :          0 :                 exit(1);
    4462                 :            :         }
    4463                 :            :     }
    4464                 :            : 
    4465                 :            :     /*
    4466                 :            :      * Define a :client_id variable that is unique per connection. But don't
    4467                 :            :      * override an explicit -D switch.
    4468                 :            :      */
    4469         [ +  - ]:         60 :     if (lookupVariable(&state[0], "client_id") == NULL)
    4470                 :            :     {
    4471         [ +  + ]:        156 :         for (i = 0; i < nclients; i++)
    4472                 :            :         {
    4473         [ -  + ]:         96 :             if (!putVariableInt(&state[i], "startup", "client_id", i))
    4474                 :          0 :                 exit(1);
    4475                 :            :         }
    4476                 :            :     }
    4477                 :            : 
    4478         [ +  + ]:         60 :     if (!is_no_vacuum)
    4479                 :            :     {
    4480                 :          6 :         fprintf(stderr, "starting vacuum...");
    4481                 :          6 :         tryExecuteStatement(con, "vacuum pgbench_branches");
    4482                 :          6 :         tryExecuteStatement(con, "vacuum pgbench_tellers");
    4483                 :          6 :         tryExecuteStatement(con, "truncate pgbench_history");
    4484                 :          6 :         fprintf(stderr, "end.\n");
    4485                 :            : 
    4486         [ -  + ]:          6 :         if (do_vacuum_accounts)
    4487                 :            :         {
    4488                 :          0 :             fprintf(stderr, "starting vacuum pgbench_accounts...");
    4489                 :          0 :             tryExecuteStatement(con, "vacuum analyze pgbench_accounts");
    4490                 :          0 :             fprintf(stderr, "end.\n");
    4491                 :            :         }
    4492                 :            :     }
    4493                 :         60 :     PQfinish(con);
    4494                 :            : 
    4495                 :            :     /* set random seed */
    4496                 :         60 :     INSTR_TIME_SET_CURRENT(start_time);
    4497                 :         60 :     srandom((unsigned int) INSTR_TIME_GET_MICROSEC(start_time));
    4498                 :            : 
    4499                 :            :     /* set up thread data structures */
    4500                 :         60 :     threads = (TState *) pg_malloc(sizeof(TState) * nthreads);
    4501                 :         60 :     nclients_dealt = 0;
    4502                 :            : 
    4503         [ +  + ]:        120 :     for (i = 0; i < nthreads; i++)
    4504                 :            :     {
    4505                 :         60 :         TState     *thread = &threads[i];
    4506                 :            : 
    4507                 :         60 :         thread->tid = i;
    4508                 :         60 :         thread->state = &state[nclients_dealt];
    4509                 :         60 :         thread->nstate =
    4510                 :         60 :             (nclients - nclients_dealt + nthreads - i - 1) / (nthreads - i);
    4511                 :         60 :         thread->random_state[0] = random();
    4512                 :         60 :         thread->random_state[1] = random();
    4513                 :         60 :         thread->random_state[2] = random();
    4514                 :         60 :         thread->logfile = NULL; /* filled in later */
    4515                 :         60 :         thread->latency_late = 0;
    4516                 :         60 :         initStats(&thread->stats, 0);
    4517                 :            : 
    4518                 :         60 :         nclients_dealt += thread->nstate;
    4519                 :            :     }
    4520                 :            : 
    4521                 :            :     /* all clients must be assigned to a thread */
    4522                 :            :     Assert(nclients_dealt == nclients);
    4523                 :            : 
    4524                 :            :     /* get start up time */
    4525                 :         60 :     INSTR_TIME_SET_CURRENT(start_time);
    4526                 :            : 
    4527                 :            :     /* set alarm if duration is specified. */
    4528         [ -  + ]:         60 :     if (duration > 0)
    4529                 :          0 :         setalarm(duration);
    4530                 :            : 
    4531                 :            :     /* start threads */
    4532                 :            : #ifdef ENABLE_THREAD_SAFETY
    4533         [ +  + ]:        120 :     for (i = 0; i < nthreads; i++)
    4534                 :            :     {
    4535                 :         60 :         TState     *thread = &threads[i];
    4536                 :            : 
    4537                 :         60 :         INSTR_TIME_SET_CURRENT(thread->start_time);
    4538                 :            : 
    4539                 :            :         /* compute when to stop */
    4540         [ -  + ]:         60 :         if (duration > 0)
    4541                 :          0 :             end_time = INSTR_TIME_GET_MICROSEC(thread->start_time) +
    4542                 :          0 :                 (int64) 1000000 * duration;
    4543                 :            : 
    4544                 :            :         /* the first thread (i = 0) is executed by main thread */
    4545         [ -  + ]:         60 :         if (i > 0)
    4546                 :            :         {
    4547                 :          0 :             int         err = pthread_create(&thread->thread, NULL, threadRun, thread);
    4548                 :            : 
    4549 [ #  # ][ #  # ]:          0 :             if (err != 0 || thread->thread == INVALID_THREAD)
    4550                 :            :             {
    4551                 :          0 :                 fprintf(stderr, "could not create thread: %s\n", strerror(err));
    4552                 :          0 :                 exit(1);
    4553                 :            :             }
    4554                 :            :         }
    4555                 :            :         else
    4556                 :            :         {
    4557                 :         60 :             thread->thread = INVALID_THREAD;
    4558                 :            :         }
    4559                 :            :     }
    4560                 :            : #else
    4561                 :            :     INSTR_TIME_SET_CURRENT(threads[0].start_time);
    4562                 :            :     /* compute when to stop */
    4563                 :            :     if (duration > 0)
    4564                 :            :         end_time = INSTR_TIME_GET_MICROSEC(threads[0].start_time) +
    4565                 :            :             (int64) 1000000 * duration;
    4566                 :            :     threads[0].thread = INVALID_THREAD;
    4567                 :            : #endif                          /* ENABLE_THREAD_SAFETY */
    4568                 :            : 
    4569                 :            :     /* wait for threads and accumulate results */
    4570                 :         60 :     initStats(&stats, 0);
    4571                 :         60 :     INSTR_TIME_SET_ZERO(conn_total_time);
    4572         [ +  + ]:        120 :     for (i = 0; i < nthreads; i++)
    4573                 :            :     {
    4574                 :         60 :         TState     *thread = &threads[i];
    4575                 :            : 
    4576                 :            : #ifdef ENABLE_THREAD_SAFETY
    4577         [ +  - ]:         60 :         if (threads[i].thread == INVALID_THREAD)
    4578                 :            :             /* actually run this thread directly in the main thread */
    4579                 :         60 :             (void) threadRun(thread);
    4580                 :            :         else
    4581                 :            :             /* wait of other threads. should check that 0 is returned? */
    4582                 :          0 :             pthread_join(thread->thread, NULL);
    4583                 :            : #else
    4584                 :            :         (void) threadRun(thread);
    4585                 :            : #endif                          /* ENABLE_THREAD_SAFETY */
    4586                 :            : 
    4587                 :            :         /* aggregate thread level stats */
    4588                 :         60 :         mergeSimpleStats(&stats.latency, &thread->stats.latency);
    4589                 :         60 :         mergeSimpleStats(&stats.lag, &thread->stats.lag);
    4590                 :         60 :         stats.cnt += thread->stats.cnt;
    4591                 :         60 :         stats.skipped += thread->stats.skipped;
    4592                 :         60 :         latency_late += thread->latency_late;
    4593         [ -  + ]:         60 :         INSTR_TIME_ADD(conn_total_time, thread->conn_time);
    4594                 :            :     }
    4595                 :         60 :     disconnect_all(state, nclients);
    4596                 :            : 
    4597                 :            :     /*
    4598                 :            :      * XXX We compute results as though every client of every thread started
    4599                 :            :      * and finished at the same time.  That model can diverge noticeably from
    4600                 :            :      * reality for a short benchmark run involving relatively many threads.
    4601                 :            :      * The first thread may process notably many transactions before the last
    4602                 :            :      * thread begins.  Improving the model alone would bring limited benefit,
    4603                 :            :      * because performance during those periods of partial thread count can
    4604                 :            :      * easily exceed steady state performance.  This is one of the many ways
    4605                 :            :      * short runs convey deceptive performance figures.
    4606                 :            :      */
    4607                 :         60 :     INSTR_TIME_SET_CURRENT(total_time);
    4608         [ +  + ]:         66 :     INSTR_TIME_SUBTRACT(total_time, start_time);
    4609                 :         60 :     printResults(threads, &stats, total_time, conn_total_time, latency_late);
    4610                 :            : 
    4611                 :         60 :     return 0;
    4612                 :            : }
    4613                 :            : 
    4614                 :            : static void *
    4615                 :         60 : threadRun(void *arg)
    4616                 :            : {
    4617                 :         60 :     TState     *thread = (TState *) arg;
    4618                 :         60 :     CState     *state = thread->state;
    4619                 :            :     instr_time  start,
    4620                 :            :                 end;
    4621                 :         60 :     int         nstate = thread->nstate;
    4622                 :         60 :     int         remains = nstate;   /* number of remaining clients */
    4623                 :            :     int         i;
    4624                 :            : 
    4625                 :            :     /* for reporting progress: */
    4626                 :         60 :     int64       thread_start = INSTR_TIME_GET_MICROSEC(thread->start_time);
    4627                 :         60 :     int64       last_report = thread_start;
    4628                 :         60 :     int64       next_report = last_report + (int64) progress * 1000000;
    4629                 :            :     StatsData   last,
    4630                 :            :                 aggs;
    4631                 :            : 
    4632                 :            :     /*
    4633                 :            :      * Initialize throttling rate target for all of the thread's clients.  It
    4634                 :            :      * might be a little more accurate to reset thread->start_time here too.
    4635                 :            :      * The possible drift seems too small relative to typical throttle delay
    4636                 :            :      * times to worry about it.
    4637                 :            :      */
    4638                 :         60 :     INSTR_TIME_SET_CURRENT(start);
    4639                 :         60 :     thread->throttle_trigger = INSTR_TIME_GET_MICROSEC(start);
    4640                 :            : 
    4641                 :         60 :     INSTR_TIME_SET_ZERO(thread->conn_time);
    4642                 :            : 
    4643                 :         60 :     initStats(&aggs, time(NULL));
    4644                 :         60 :     last = aggs;
    4645                 :            : 
    4646                 :            :     /* open log file if requested */
    4647         [ +  + ]:         60 :     if (use_log)
    4648                 :            :     {
    4649                 :            :         char        logpath[MAXPGPATH];
    4650         [ +  - ]:          4 :         char       *prefix = logfile_prefix ? logfile_prefix : "pgbench_log";
    4651                 :            : 
    4652         [ +  - ]:          4 :         if (thread->tid == 0)
    4653                 :          4 :             snprintf(logpath, sizeof(logpath), "%s.%d", prefix, main_pid);
    4654                 :            :         else
    4655                 :          0 :             snprintf(logpath, sizeof(logpath), "%s.%d.%d", prefix, main_pid, thread->tid);
    4656                 :            : 
    4657                 :          4 :         thread->logfile = fopen(logpath, "w");
    4658                 :            : 
    4659         [ -  + ]:          4 :         if (thread->logfile == NULL)
    4660                 :            :         {
    4661                 :          0 :             fprintf(stderr, "could not open logfile \"%s\": %s\n",
    4662                 :          0 :                     logpath, strerror(errno));
    4663                 :          4 :             goto done;
    4664                 :            :         }
    4665                 :            :     }
    4666                 :            : 
    4667         [ +  + ]:         60 :     if (!is_connect)
    4668                 :            :     {
    4669                 :            :         /* make connections to the database */
    4670         [ +  + ]:        138 :         for (i = 0; i < nstate; i++)
    4671                 :            :         {
    4672         [ -  + ]:         82 :             if ((state[i].con = doConnect()) == NULL)
    4673                 :          0 :                 goto done;
    4674                 :            :         }
    4675                 :            :     }
    4676                 :            : 
    4677                 :            :     /* time after thread and connections set up */
    4678                 :         60 :     INSTR_TIME_SET_CURRENT(thread->conn_time);
    4679         [ +  + ]:         62 :     INSTR_TIME_SUBTRACT(thread->conn_time, thread->start_time);
    4680                 :            : 
    4681                 :            :     /* explicitly initialize the state machines */
    4682         [ +  + ]:        156 :     for (i = 0; i < nstate; i++)
    4683                 :            :     {
    4684                 :         96 :         state[i].state = CSTATE_CHOOSE_SCRIPT;
    4685                 :            :     }
    4686                 :            : 
    4687                 :            :     /* loop till all clients have terminated */
    4688         [ +  + ]:       2932 :     while (remains > 0)
    4689                 :            :     {
    4690                 :            :         fd_set      input_mask;
    4691                 :            :         int         maxsock;    /* max socket number to be waited for */
    4692                 :            :         int64       min_usec;
    4693                 :       2872 :         int64       now_usec = 0;   /* set this only if needed */
    4694                 :            : 
    4695                 :            :         /* identify which client sockets should be checked for input */
    4696                 :       2872 :         FD_ZERO(&input_mask);
    4697                 :       2872 :         maxsock = -1;
    4698                 :       2872 :         min_usec = PG_INT64_MAX;
    4699         [ +  + ]:      13400 :         for (i = 0; i < nstate; i++)
    4700                 :            :         {
    4701                 :      10588 :             CState     *st = &state[i];
    4702                 :            : 
    4703 [ -  + ][ #  # ]:      10588 :             if (st->state == CSTATE_THROTTLE && timer_exceeded)
    4704                 :            :             {
    4705                 :            :                 /* interrupt client that has not started a transaction */
    4706                 :          0 :                 st->state = CSTATE_FINISHED;
    4707                 :          0 :                 PQfinish(st->con);
    4708                 :          0 :                 st->con = NULL;
    4709                 :          0 :                 remains--;
    4710                 :            :             }
    4711 [ +  + ][ -  + ]:      10588 :             else if (st->state == CSTATE_SLEEP || st->state == CSTATE_THROTTLE)
    4712                 :          4 :             {
    4713                 :            :                 /* a nap from the script, or under throttling */
    4714                 :            :                 int64       this_usec;
    4715                 :            : 
    4716                 :            :                 /* get current time if needed */
    4717         [ +  - ]:          4 :                 if (now_usec == 0)
    4718                 :            :                 {
    4719                 :            :                     instr_time  now;
    4720                 :            : 
    4721                 :          4 :                     INSTR_TIME_SET_CURRENT(now);
    4722                 :          4 :                     now_usec = INSTR_TIME_GET_MICROSEC(now);
    4723                 :            :                 }
    4724                 :            : 
    4725                 :            :                 /* min_usec should be the minimum delay across all clients */
    4726                 :          8 :                 this_usec = (st->state == CSTATE_SLEEP ?
    4727         [ +  - ]:          4 :                              st->sleep_until : st->txn_scheduled) - now_usec;
    4728         [ +  - ]:          4 :                 if (min_usec > this_usec)
    4729                 :          4 :                     min_usec = this_usec;
    4730                 :            :             }
    4731         [ +  + ]:      10584 :             else if (st->state == CSTATE_WAIT_RESULT)
    4732                 :            :             {
    4733                 :            :                 /*
    4734                 :            :                  * waiting for result from server - nothing to do unless the
    4735                 :            :                  * socket is readable
    4736                 :            :                  */
    4737                 :       9778 :                 int         sock = PQsocket(st->con);
    4738                 :            : 
    4739         [ -  + ]:       9778 :                 if (sock < 0)
    4740                 :            :                 {
    4741                 :          0 :                     fprintf(stderr, "invalid socket: %s",
    4742                 :          0 :                             PQerrorMessage(st->con));
    4743                 :          0 :                     goto done;
    4744                 :            :                 }
    4745                 :            : 
    4746                 :       9778 :                 FD_SET(sock, &input_mask);
    4747         [ +  + ]:       9778 :                 if (maxsock < sock)
    4748                 :       9778 :                     maxsock = sock;
    4749                 :            :             }
    4750 [ +  - ][ +  + ]:        806 :             else if (st->state != CSTATE_ABORTED &&
    4751                 :        806 :                      st->state != CSTATE_FINISHED)
    4752                 :            :             {
    4753                 :            :                 /*
    4754                 :            :                  * This client thread is ready to do something, so we don't
    4755                 :            :                  * want to wait.  No need to examine additional clients.
    4756                 :            :                  */
    4757                 :         60 :                 min_usec = 0;
    4758                 :         60 :                 break;
    4759                 :            :             }
    4760                 :            :         }
    4761                 :            : 
    4762                 :            :         /* also wake up to print the next progress report on time */
    4763 [ -  + ][ #  # ]:       2872 :         if (progress && min_usec > 0 && thread->tid == 0)
                 [ #  # ]
    4764                 :            :         {
    4765                 :            :             /* get current time if needed */
    4766         [ #  # ]:          0 :             if (now_usec == 0)
    4767                 :            :             {
    4768                 :            :                 instr_time  now;
    4769                 :            : 
    4770                 :          0 :                 INSTR_TIME_SET_CURRENT(now);
    4771                 :          0 :                 now_usec = INSTR_TIME_GET_MICROSEC(now);
    4772                 :            :             }
    4773                 :            : 
    4774         [ #  # ]:          0 :             if (now_usec >= next_report)
    4775                 :          0 :                 min_usec = 0;
    4776         [ #  # ]:          0 :             else if ((next_report - now_usec) < min_usec)
    4777                 :          0 :                 min_usec = next_report - now_usec;
    4778                 :            :         }
    4779                 :            : 
    4780                 :            :         /*
    4781                 :            :          * If no clients are ready to execute actions, sleep until we receive
    4782                 :            :          * data from the server, or a nap-time specified in the script ends,
    4783                 :            :          * or it's time to print a progress report.  Update input_mask to show
    4784                 :            :          * which client(s) received data.
    4785                 :            :          */
    4786         [ +  + ]:       2872 :         if (min_usec > 0)
    4787                 :            :         {
    4788                 :       2812 :             int         nsocks = 0; /* return from select(2) if called */
    4789                 :            : 
    4790         [ +  + ]:       2812 :             if (min_usec != PG_INT64_MAX)
    4791                 :            :             {
    4792         [ -  + ]:          4 :                 if (maxsock != -1)
    4793                 :            :                 {
    4794                 :            :                     struct timeval timeout;
    4795                 :            : 
    4796                 :          0 :                     timeout.tv_sec = min_usec / 1000000;
    4797                 :          0 :                     timeout.tv_usec = min_usec % 1000000;
    4798                 :          0 :                     nsocks = select(maxsock + 1, &input_mask, NULL, NULL, &timeout);
    4799                 :            :                 }
    4800                 :            :                 else            /* nothing active, simple sleep */
    4801                 :            :                 {
    4802                 :          4 :                     pg_usleep(min_usec);
    4803                 :            :                 }
    4804                 :            :             }
    4805                 :            :             else                /* no explicit delay, select without timeout */
    4806                 :            :             {
    4807                 :       2808 :                 nsocks = select(maxsock + 1, &input_mask, NULL, NULL, NULL);
    4808                 :            :             }
    4809                 :            : 
    4810         [ -  + ]:       2812 :             if (nsocks < 0)
    4811                 :            :             {
    4812         [ #  # ]:          0 :                 if (errno == EINTR)
    4813                 :            :                 {
    4814                 :            :                     /* On EINTR, go back to top of loop */
    4815                 :          0 :                     continue;
    4816                 :            :                 }
    4817                 :            :                 /* must be something wrong */
    4818                 :          0 :                 fprintf(stderr, "select() failed: %s\n", strerror(errno));
    4819                 :       2812 :                 goto done;
    4820                 :            :             }
    4821                 :            :         }
    4822                 :            :         else
    4823                 :            :         {
    4824                 :            :             /* min_usec == 0, i.e. something needs to be executed */
    4825                 :            : 
    4826                 :            :             /* If we didn't call select(), don't try to read any data */
    4827                 :         60 :             FD_ZERO(&input_mask);
    4828                 :            :         }
    4829                 :            : 
    4830                 :            :         /* ok, advance the state machine of each connection */
    4831         [ +  + ]:      13496 :         for (i = 0; i < nstate; i++)
    4832                 :            :         {
    4833                 :      10624 :             CState     *st = &state[i];
    4834                 :            : 
    4835         [ +  + ]:      10624 :             if (st->state == CSTATE_WAIT_RESULT)
    4836                 :            :             {
    4837                 :            :                 /* don't call doCustom unless data is available */
    4838                 :       9778 :                 int         sock = PQsocket(st->con);
    4839                 :            : 
    4840         [ -  + ]:       9778 :                 if (sock < 0)
    4841                 :            :                 {
    4842                 :          0 :                     fprintf(stderr, "invalid socket: %s",
    4843                 :          0 :                             PQerrorMessage(st->con));
    4844                 :          0 :                     goto done;
    4845                 :            :                 }
    4846                 :            : 
    4847         [ +  + ]:       9778 :                 if (!FD_ISSET(sock, &input_mask))
    4848                 :       9778 :                     continue;
    4849                 :            :             }
    4850 [ +  + ][ -  + ]:        846 :             else if (st->state == CSTATE_FINISHED ||
    4851                 :        100 :                      st->state == CSTATE_ABORTED)
    4852                 :            :             {
    4853                 :            :                 /* this client is done, no need to consider it anymore */
    4854                 :        746 :                 continue;
    4855                 :            :             }
    4856                 :            : 
    4857                 :       4136 :             doCustom(thread, st, &aggs);
    4858                 :            : 
    4859                 :            :             /* If doCustom changed client to finished state, reduce remains */
    4860 [ +  + ][ +  + ]:       4136 :             if (st->state == CSTATE_FINISHED || st->state == CSTATE_ABORTED)
    4861                 :         96 :                 remains--;
    4862                 :            :         }
    4863                 :            : 
    4864                 :            :         /* progress report is made by thread 0 for all threads */
    4865 [ -  + ][ #  # ]:       2872 :         if (progress && thread->tid == 0)
    4866                 :            :         {
    4867                 :            :             instr_time  now_time;
    4868                 :            :             int64       now;
    4869                 :            : 
    4870                 :          0 :             INSTR_TIME_SET_CURRENT(now_time);
    4871                 :          0 :             now = INSTR_TIME_GET_MICROSEC(now_time);
    4872         [ #  # ]:          0 :             if (now >= next_report)
    4873                 :            :             {
    4874                 :            :                 /* generate and show report */
    4875                 :            :                 StatsData   cur;
    4876                 :          0 :                 int64       run = now - last_report,
    4877                 :            :                             ntx;
    4878                 :            :                 double      tps,
    4879                 :            :                             total_run,
    4880                 :            :                             latency,
    4881                 :            :                             sqlat,
    4882                 :            :                             lag,
    4883                 :            :                             stdev;
    4884                 :            :                 char        tbuf[64];
    4885                 :            : 
    4886                 :            :                 /*
    4887                 :            :                  * Add up the statistics of all threads.
    4888                 :            :                  *
    4889                 :            :                  * XXX: No locking. There is no guarantee that we get an
    4890                 :            :                  * atomic snapshot of the transaction count and latencies, so
    4891                 :            :                  * these figures can well be off by a small amount. The
    4892                 :            :                  * progress report's purpose is to give a quick overview of
    4893                 :            :                  * how the test is going, so that shouldn't matter too much.
    4894                 :            :                  * (If a read from a 64-bit integer is not atomic, you might
    4895                 :            :                  * get a "torn" read and completely bogus latencies though!)
    4896                 :            :                  */
    4897                 :          0 :                 initStats(&cur, 0);
    4898         [ #  # ]:          0 :                 for (i = 0; i < nthreads; i++)
    4899                 :            :                 {
    4900                 :          0 :                     mergeSimpleStats(&cur.latency, &thread[i].stats.latency);
    4901                 :          0 :                     mergeSimpleStats(&cur.lag, &thread[i].stats.lag);
    4902                 :          0 :                     cur.cnt += thread[i].stats.cnt;
    4903                 :          0 :                     cur.skipped += thread[i].stats.skipped;
    4904                 :            :                 }
    4905                 :            : 
    4906                 :            :                 /* we count only actually executed transactions */
    4907                 :          0 :                 ntx = (cur.cnt - cur.skipped) - (last.cnt - last.skipped);
    4908                 :          0 :                 total_run = (now - thread_start) / 1000000.0;
    4909                 :          0 :                 tps = 1000000.0 * ntx / run;
    4910         [ #  # ]:          0 :                 if (ntx > 0)
    4911                 :            :                 {
    4912                 :          0 :                     latency = 0.001 * (cur.latency.sum - last.latency.sum) / ntx;
    4913                 :          0 :                     sqlat = 1.0 * (cur.latency.sum2 - last.latency.sum2) / ntx;
    4914                 :          0 :                     stdev = 0.001 * sqrt(sqlat - 1000000.0 * latency * latency);
    4915                 :          0 :                     lag = 0.001 * (cur.lag.sum - last.lag.sum) / ntx;
    4916                 :            :                 }
    4917                 :            :                 else
    4918                 :            :                 {
    4919                 :          0 :                     latency = sqlat = stdev = lag = 0;
    4920                 :            :                 }
    4921                 :            : 
    4922         [ #  # ]:          0 :                 if (progress_timestamp)
    4923                 :            :                 {
    4924                 :            :                     /*
    4925                 :            :                      * On some platforms the current system timestamp is
    4926                 :            :                      * available in now_time, but rather than get entangled
    4927                 :            :                      * with that, we just eat the cost of an extra syscall in
    4928                 :            :                      * all cases.
    4929                 :            :                      */
    4930                 :            :                     struct timeval tv;
    4931                 :            : 
    4932                 :          0 :                     gettimeofday(&tv, NULL);
    4933                 :          0 :                     snprintf(tbuf, sizeof(tbuf), "%ld.%03ld s",
    4934                 :          0 :                              (long) tv.tv_sec, (long) (tv.tv_usec / 1000));
    4935                 :            :                 }
    4936                 :            :                 else
    4937                 :            :                 {
    4938                 :            :                     /* round seconds are expected, but the thread may be late */
    4939                 :          0 :                     snprintf(tbuf, sizeof(tbuf), "%.1f s", total_run);
    4940                 :            :                 }
    4941                 :            : 
    4942                 :          0 :                 fprintf(stderr,
    4943                 :            :                         "progress: %s, %.1f tps, lat %.3f ms stddev %.3f",
    4944                 :            :                         tbuf, tps, latency, stdev);
    4945                 :            : 
    4946         [ #  # ]:          0 :                 if (throttle_delay)
    4947                 :            :                 {
    4948                 :          0 :                     fprintf(stderr, ", lag %.3f ms", lag);
    4949         [ #  # ]:          0 :                     if (latency_limit)
    4950                 :          0 :                         fprintf(stderr, ", " INT64_FORMAT " skipped",
    4951                 :          0 :                                 cur.skipped - last.skipped);
    4952                 :            :                 }
    4953                 :          0 :                 fprintf(stderr, "\n");
    4954                 :            : 
    4955                 :          0 :                 last = cur;
    4956                 :          0 :                 last_report = now;
    4957                 :            : 
    4958                 :            :                 /*
    4959                 :            :                  * Ensure that the next report is in the future, in case
    4960                 :            :                  * pgbench/postgres got stuck somewhere.
    4961                 :            :                  */
    4962                 :            :                 do
    4963                 :            :                 {
    4964                 :          0 :                     next_report += (int64) progress * 1000000;
    4965         [ #  # ]:       2872 :                 } while (now >= next_report);
    4966                 :            :             }
    4967                 :            :         }
    4968                 :            :     }
    4969                 :            : 
    4970                 :            : done:
    4971                 :         60 :     INSTR_TIME_SET_CURRENT(start);
    4972                 :         60 :     disconnect_all(state, nstate);
    4973                 :         60 :     INSTR_TIME_SET_CURRENT(end);
    4974 [ -  + ][ -  + ]:         60 :     INSTR_TIME_ACCUM_DIFF(thread->conn_time, end, start);
    4975         [ +  + ]:         60 :     if (thread->logfile)
    4976                 :            :     {
    4977         [ -  + ]:          4 :         if (agg_interval > 0)
    4978                 :            :         {
    4979                 :            :             /* log aggregated but not yet reported transactions */
    4980                 :          0 :             doLog(thread, state, &aggs, false, 0, 0);
    4981                 :            :         }
    4982                 :          4 :         fclose(thread->logfile);
    4983                 :          4 :         thread->logfile = NULL;
    4984                 :            :     }
    4985                 :         60 :     return NULL;
    4986                 :            : }
    4987                 :            : 
    4988                 :            : /*
    4989                 :            :  * Support for duration option: set timer_exceeded after so many seconds.
    4990                 :            :  */
    4991                 :            : 
    4992                 :            : #ifndef WIN32
    4993                 :            : 
    4994                 :            : static void
    4995                 :          0 : handle_sig_alarm(SIGNAL_ARGS)
    4996                 :            : {
    4997                 :          0 :     timer_exceeded = true;
    4998                 :          0 : }
    4999                 :            : 
    5000                 :            : static void
    5001                 :          0 : setalarm(int seconds)
    5002                 :            : {
    5003                 :          0 :     pqsignal(SIGALRM, handle_sig_alarm);
    5004                 :          0 :     alarm(seconds);
    5005                 :          0 : }
    5006                 :            : 
    5007                 :            : #else                           /* WIN32 */
    5008                 :            : 
    5009                 :            : static VOID CALLBACK
    5010                 :            : win32_timer_callback(PVOID lpParameter, BOOLEAN TimerOrWaitFired)
    5011                 :            : {
    5012                 :            :     timer_exceeded = true;
    5013                 :            : }
    5014                 :            : 
    5015                 :            : static void
    5016                 :            : setalarm(int seconds)
    5017                 :            : {
    5018                 :            :     HANDLE      queue;
    5019                 :            :     HANDLE      timer;
    5020                 :            : 
    5021                 :            :     /* This function will be called at most once, so we can cheat a bit. */
    5022                 :            :     queue = CreateTimerQueue();
    5023                 :            :     if (seconds > ((DWORD) -1) / 1000 ||
    5024                 :            :         !CreateTimerQueueTimer(&timer, queue,
    5025                 :            :                                win32_timer_callback, NULL, seconds * 1000, 0,
    5026                 :            :                                WT_EXECUTEINTIMERTHREAD | WT_EXECUTEONLYONCE))
    5027                 :            :     {
    5028                 :            :         fprintf(stderr, "failed to set timer\n");
    5029                 :            :         exit(1);
    5030                 :            :     }
    5031                 :            : }
    5032                 :            : 
    5033                 :            : /* partial pthread implementation for Windows */
    5034                 :            : 
    5035                 :            : typedef struct win32_pthread
    5036                 :            : {
    5037                 :            :     HANDLE      handle;
    5038                 :            :     void       *(*routine) (void *);
    5039                 :            :     void       *arg;
    5040                 :            :     void       *result;
    5041                 :            : } win32_pthread;
    5042                 :            : 
    5043                 :            : static unsigned __stdcall
    5044                 :            : win32_pthread_run(void *arg)
    5045                 :            : {
    5046                 :            :     win32_pthread *th = (win32_pthread *) arg;
    5047                 :            : 
    5048                 :            :     th->result = th->routine(th->arg);
    5049                 :            : 
    5050                 :            :     return 0;
    5051                 :            : }
    5052                 :            : 
    5053                 :            : static int
    5054                 :            : pthread_create(pthread_t *thread,
    5055                 :            :                pthread_attr_t *attr,
    5056                 :            :                void *(*start_routine) (void *),
    5057                 :            :                void *arg)
    5058                 :            : {
    5059                 :            :     int         save_errno;
    5060                 :            :     win32_pthread *th;
    5061                 :            : 
    5062                 :            :     th = (win32_pthread *) pg_malloc(sizeof(win32_pthread));
    5063                 :            :     th->routine = start_routine;
    5064                 :            :     th->arg = arg;
    5065                 :            :     th->result = NULL;
    5066                 :            : 
    5067                 :            :     th->handle = (HANDLE) _beginthreadex(NULL, 0, win32_pthread_run, th, 0, NULL);
    5068                 :            :     if (th->handle == NULL)
    5069                 :            :     {
    5070                 :            :         save_errno = errno;
    5071                 :            :         free(th);
    5072                 :            :         return save_errno;
    5073                 :            :     }
    5074                 :            : 
    5075                 :            :     *thread = th;
    5076                 :            :     return 0;
    5077                 :            : }
    5078                 :            : 
    5079                 :            : static int
    5080                 :            : pthread_join(pthread_t th, void **thread_return)
    5081                 :            : {
    5082                 :            :     if (th == NULL || th->handle == NULL)
    5083                 :            :         return errno = EINVAL;
    5084                 :            : 
    5085                 :            :     if (WaitForSingleObject(th->handle, INFINITE) != WAIT_OBJECT_0)
    5086                 :            :     {
    5087                 :            :         _dosmaperr(GetLastError());
    5088                 :            :         return errno;
    5089                 :            :     }
    5090                 :            : 
    5091                 :            :     if (thread_return)
    5092                 :            :         *thread_return = th->result;
    5093                 :            : 
    5094                 :            :     CloseHandle(th->handle);
    5095                 :            :     free(th);
    5096                 :            :     return 0;
    5097                 :            : }
    5098                 :            : 
    5099                 :            : #endif                          /* WIN32 */

Generated by: LCOV version 1.13