static void dlr_mongodb_shutdown()
{
dbpool_destroy(pool);
dlr_db_fields_destroy(fields);
mongodb_database = NULL;
mongodb_table = NULL;
if (mongodb_namespace) {
gw_free(mongodb_namespace);
mongodb_namespace = NULL;
}
}
static void dlr_sdb_shutdown()
{
dbpool_destroy(pool);
dlr_db_fields_destroy(fields);
}
static void dlr_shutdown_sqlite3()
{
dbpool_destroy(pool);
dlr_db_fields_destroy(fields);
}
void mysql_leave()
{
dbpool_destroy(pool);
}
int main(int argc, char **argv)
{
DBPool *pool;
DBConf *conf = NULL; /* for compiler please */
unsigned int num_threads = 1;
unsigned long i;
int opt;
time_t start = 0, end = 0;
double run_time;
Octstr *user, *pass, *db, *host, *db_type;
int j, bail_out;
user = pass = db = host = db_type = NULL;
gwlib_init();
sql = octstr_imm("SHOW STATUS");
while ((opt = getopt(argc, argv, "v:h:u:p:d:s:q:t:S:T:")) != EOF) {
switch (opt) {
case 'v':
log_set_output_level(atoi(optarg));
break;
case 'h':
host = octstr_create(optarg);
break;
case 'u':
user = octstr_create(optarg);
break;
case 'p':
pass = octstr_create(optarg);
break;
case 'd':
db = octstr_create(optarg);
break;
case 'S':
octstr_destroy(sql);
sql = octstr_create(optarg);
break;
case 's':
pool_size = atoi(optarg);
break;
case 'q':
queries = atoi(optarg);
break;
case 't':
num_threads = atoi(optarg);
break;
case 'T':
db_type = octstr_create(optarg);
break;
case '?':
default:
error(0, "Invalid option %c", opt);
help();
panic(0, "Stopping.");
}
}
if (!optind) {
help();
exit(0);
}
if (!db_type) {
info(0, "No database type given assuming MySQL.");
}
else if (octstr_case_compare(db_type, octstr_imm("mysql")) == 0) {
info(0, "Do tests for mysql database.");
database_type = DBPOOL_MYSQL;
}
else if (octstr_case_compare(db_type, octstr_imm("oracle")) == 0) {
info(0, "Do tests for oracle database.");
database_type = DBPOOL_ORACLE;
}
else if (octstr_case_compare(db_type, octstr_imm("sqlite")) == 0) {
info(0, "Do tests for sqlite database.");
database_type = DBPOOL_SQLITE;
}
else if (octstr_case_compare(db_type, octstr_imm("sqlite3")) == 0) {
info(0, "Do tests for sqlite3 database.");
database_type = DBPOOL_SQLITE3;
}
else {
panic(0, "Unknown database type '%s'", octstr_get_cstr(db_type));
}
/* check if we have the database connection details */
switch (database_type) {
case DBPOOL_ORACLE:
bail_out = (!user || !pass || !db) ? 1 : 0;
break;
case DBPOOL_SQLITE:
case DBPOOL_SQLITE3:
bail_out = (!db) ? 1 : 0;
break;
default:
bail_out = (!host || !user || !pass || !db) ? 1 : 0;
break;
}
if (bail_out) {
help();
panic(0, "Database connection details are not fully provided!");
}
for (j = 0; j < 1; j++) {
/* create DBConf */
switch (database_type) {
#ifdef HAVE_MYSQL
case DBPOOL_MYSQL:
conf = mysql_create_conf(user,pass,db,host);
client_thread = mysql_client_thread;
break;
#endif
#ifdef HAVE_ORACLE
case DBPOOL_ORACLE:
conf = oracle_create_conf(user, pass, db);
client_thread = oracle_client_thread;
break;
#endif
#ifdef HAVE_SQLITE
case DBPOOL_SQLITE:
conf = sqlite_create_conf(db);
client_thread = sqlite_client_thread;
break;
#endif
#ifdef HAVE_SQLITE3
case DBPOOL_SQLITE3:
conf = sqlite3_create_conf(db);
client_thread = sqlite3_client_thread;
break;
#endif
default:
panic(0, "ooops ....");
};
/* create */
info(0,"Creating database pool to `%s' with %d connections type '%s'.",
(host ? octstr_get_cstr(host) : octstr_get_cstr(db)), pool_size, octstr_get_cstr(db_type));
pool = dbpool_create(database_type, conf, pool_size);
debug("",0,"Connections within pool: %ld", dbpool_conn_count(pool));
for (i = 0; i < num_threads; ++i) {
if (gwthread_create(inc_dec_thread, pool) == -1)
panic(0, "Could not create thread %ld", i);
}
gwthread_join_all();
info(0, "Connections within pool: %ld", dbpool_conn_count(pool));
info(0, "Checked pool, %d connections still active and ok", dbpool_check(pool));
/* queries */
info(0,"SQL query is `%s'", octstr_get_cstr(sql));
time(&start);
for (i = 0; i < num_threads; ++i) {
#if 0
if (gwthread_create(inc_dec_thread, pool) == -1)
panic(0, "Couldnot create thread %ld", i);
#endif
if (gwthread_create(client_thread, pool) == -1)
panic(0, "Couldnot create thread %ld", i);
}
gwthread_join_all();
time(&end);
run_time = difftime(end, start);
info(0, "%ld requests in %.2f seconds, %.2f requests/s.",
(queries * num_threads), run_time, (float) (queries * num_threads) / (run_time==0?1:run_time));
/* check all active connections */
debug("",0,"Connections within pool: %ld", dbpool_conn_count(pool));
info(0,"Checked pool, %d connections still active and ok", dbpool_check(pool));
info(0,"Destroying pool");
dbpool_destroy(pool);
} /* for loop */
octstr_destroy(sql);
octstr_destroy(db_type);
octstr_destroy(user);
octstr_destroy(pass);
octstr_destroy(db);
octstr_destroy(host);
gwlib_shutdown();
return 0;
}
void redis_leave()
{
dbpool_destroy(pool);
}
void oracle_leave()
{
dbpool_destroy(pool);
}
