228 lines
6.8 KiB
C
228 lines
6.8 KiB
C
/* Copyright (C) 2004 MySQL AB
|
|
Copyright (C) 2004-2016 Alexey Kopytov <akopytov@gmail.com>
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#ifndef SYSBENCH_H
|
|
#define SYSBENCH_H
|
|
|
|
#ifdef STDC_HEADERS
|
|
# include <stdio.h>
|
|
# include <stdlib.h>
|
|
# include <string.h>
|
|
# include <stdbool.h>
|
|
#endif
|
|
#ifdef HAVE_UNISTD_H
|
|
# include <unistd.h>
|
|
# include <sys/types.h>
|
|
#endif
|
|
#ifdef HAVE_PTHREAD_H
|
|
# include <pthread.h>
|
|
#endif
|
|
|
|
#include "sb_list.h"
|
|
#include "sb_options.h"
|
|
#include "sb_timer.h"
|
|
#include "sb_logger.h"
|
|
|
|
#include "tests/sb_cpu.h"
|
|
#include "tests/sb_fileio.h"
|
|
#include "tests/sb_memory.h"
|
|
#include "tests/sb_threads.h"
|
|
#include "tests/sb_mutex.h"
|
|
|
|
/* Macros to control global execution mutex */
|
|
#define SB_THREAD_MUTEX_LOCK() pthread_mutex_lock(&sb_globals.exec_mutex)
|
|
#define SB_THREAD_MUTEX_UNLOCK() pthread_mutex_unlock(&sb_globals.exec_mutex)
|
|
|
|
/* Maximum number of elements in --report-checkpoints list */
|
|
#define MAX_CHECKPOINTS 256
|
|
|
|
/* Sysbench commands */
|
|
typedef enum
|
|
{
|
|
SB_COMMAND_NULL,
|
|
SB_COMMAND_PREPARE,
|
|
SB_COMMAND_RUN,
|
|
SB_COMMAND_CLEANUP,
|
|
SB_COMMAND_HELP,
|
|
SB_COMMAND_VERSION
|
|
} sb_cmd_t;
|
|
|
|
/* Request types definition */
|
|
|
|
typedef enum
|
|
{
|
|
SB_REQ_TYPE_NULL,
|
|
SB_REQ_TYPE_CPU,
|
|
SB_REQ_TYPE_MEMORY,
|
|
SB_REQ_TYPE_FILE,
|
|
SB_REQ_TYPE_SQL,
|
|
SB_REQ_TYPE_THREADS,
|
|
SB_REQ_TYPE_MUTEX,
|
|
SB_REQ_TYPE_SCRIPT
|
|
} sb_event_type_t;
|
|
|
|
/* Request structure definition */
|
|
|
|
struct sb_test; /* Forward declaration */
|
|
|
|
typedef struct
|
|
{
|
|
int type;
|
|
struct sb_test_t *test;
|
|
|
|
/* type-specific data */
|
|
union
|
|
{
|
|
sb_file_request_t file_request;
|
|
sb_mem_request_t mem_request;
|
|
sb_threads_request_t threads_request;
|
|
sb_mutex_request_t mutex_request;
|
|
} u;
|
|
} sb_event_t;
|
|
|
|
typedef enum
|
|
{
|
|
SB_STAT_INTERMEDIATE,
|
|
SB_STAT_CUMULATIVE
|
|
} sb_stat_t;
|
|
|
|
/* Test commands definition */
|
|
|
|
typedef int sb_cmd_help(void);
|
|
typedef int sb_cmd_prepare(void);
|
|
typedef int sb_cmd_run(void);
|
|
typedef int sb_cmd_cleanup(void);
|
|
|
|
|
|
/* Test operations definition */
|
|
|
|
typedef int sb_op_init(void);
|
|
typedef int sb_op_prepare(void);
|
|
typedef int sb_op_thread_init(int);
|
|
typedef int sb_op_thread_run(int);
|
|
typedef void sb_op_print_mode(void);
|
|
typedef sb_event_t sb_op_next_event(int);
|
|
typedef int sb_op_execute_event(sb_event_t *, int);
|
|
typedef void sb_op_print_stats(sb_stat_t);
|
|
typedef int sb_op_thread_done(int);
|
|
typedef int sb_op_cleanup(void);
|
|
typedef int sb_op_done(void);
|
|
|
|
/* Test commands structure definitions */
|
|
|
|
typedef struct
|
|
{
|
|
sb_cmd_help *help; /* print help */
|
|
sb_cmd_prepare *prepare; /* prepare for the test */
|
|
sb_cmd_run *run; /* run the test */
|
|
sb_cmd_cleanup *cleanup; /* cleanup the test database, files, etc. */
|
|
} sb_commands_t;
|
|
|
|
/* Test operations structure definition */
|
|
|
|
typedef struct
|
|
{
|
|
sb_op_init *init; /* initialization function */
|
|
sb_op_prepare *prepare; /* called after timers start, but
|
|
before thread execution */
|
|
sb_op_thread_init *thread_init; /* thread initialization
|
|
(called when each thread starts) */
|
|
sb_op_print_mode *print_mode; /* print mode function */
|
|
sb_op_next_event *next_event; /* event generation function */
|
|
sb_op_execute_event *execute_event; /* event execution function */
|
|
sb_op_print_stats *print_stats; /* stats generation function */
|
|
sb_op_thread_run *thread_run; /* main thread loop */
|
|
sb_op_thread_done *thread_done; /* thread finalize function */
|
|
sb_op_cleanup *cleanup; /* called after exit from thread,
|
|
but before timers stop */
|
|
sb_op_done *done; /* finalize function */
|
|
} sb_operations_t;
|
|
|
|
/* Test structure definition */
|
|
|
|
typedef struct sb_test
|
|
{
|
|
const char *sname;
|
|
const char *lname;
|
|
sb_operations_t ops;
|
|
sb_commands_t cmds;
|
|
sb_arg_t *args;
|
|
|
|
sb_list_item_t listitem;
|
|
} sb_test_t;
|
|
|
|
/* Thread context definition */
|
|
|
|
typedef struct
|
|
{
|
|
sb_test_t *test;
|
|
pthread_t thread;
|
|
unsigned int id;
|
|
} sb_thread_ctxt_t;
|
|
|
|
|
|
/* sysbench global variables */
|
|
|
|
typedef struct
|
|
{
|
|
sb_cmd_t command; /* command passed from command line */
|
|
int error; /* global error - everyone exit */
|
|
pthread_mutex_t exec_mutex; /* execution mutex */
|
|
unsigned int num_threads; /* number of threads to use */
|
|
unsigned int num_running; /* number of threads currently active */
|
|
unsigned int report_interval; /* intermediate reports interval */
|
|
unsigned int percentile; /* percentile rank for latency stats */
|
|
unsigned int histogram; /* show histogram in latency stats */
|
|
/* array of report checkpoints */
|
|
unsigned int checkpoints[MAX_CHECKPOINTS];
|
|
unsigned int n_checkpoints; /* number of checkpoints */
|
|
unsigned int tx_rate; /* target transaction rate */
|
|
unsigned int max_requests; /* maximum number of requests */
|
|
unsigned int max_time; /* total execution time limit */
|
|
unsigned char debug; /* debug flag */
|
|
int force_shutdown; /* whether we must force test shutdown */
|
|
unsigned int timeout; /* forced shutdown timeout */
|
|
unsigned char validate; /* validation flag */
|
|
unsigned char verbosity; /* log verbosity */
|
|
int event_queue_length; /* length of request queue when tx-rate is
|
|
used */
|
|
int concurrency; /* number of concurrent requests when tx-rate is
|
|
used */
|
|
/* 1 when forced shutdown is in progress, 0 otherwise */
|
|
int forced_shutdown_in_progress;
|
|
} sb_globals_t;
|
|
|
|
extern sb_globals_t sb_globals;
|
|
extern pthread_mutex_t event_queue_mutex;
|
|
|
|
/* Global execution timer */
|
|
extern sb_timer_t sb_exec_timer CK_CC_CACHELINE;
|
|
|
|
/* timers for checkpoint reports */
|
|
extern sb_timer_t sb_intermediate_timer;
|
|
extern sb_timer_t sb_checkpoint_timer1;
|
|
extern sb_timer_t sb_checkpoint_timer2;
|
|
|
|
extern TLS int sb_tls_thread_id;
|
|
|
|
sb_event_t sb_next_event(sb_test_t *test, int thread_id);
|
|
void sb_event_start(int thread_id);
|
|
void sb_event_stop(int thread_id);
|
|
|
|
#endif
|