LCOV - code coverage report
Current view: top level - src/bin/pgbench - pgbench.c (source / functions) Hit Total Coverage
Test: PostgreSQL 11devel Lines: 1758 2017 87.2 %
Date: 2018-02-20 09:20:13 Functions: 84 87 96.6 %
Legend: Lines: hit not hit | Branches: + taken - not taken # not executed Branches: 1042 1343 77.6 %

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

Generated by: LCOV version 1.13