Rx3/libdatabase
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
e7119d0a40
libdatabase/utils/testInsert.c
agibert e7119d0a40 Initial creation.
2006-02-28 23:28:21 +00:00

369 lines
11 KiB
C
Raw Blame History

/*---------------------------------------------------------------------------------*/
/* $RCSfile: testInsert.c,v $ */
/*---------------------------------------------------------------------------------*/
/* $Revision: 1.1 $ */
/* $Name: $ */
/* $Date: 2006/02/28 23:28:21 $ */
/* $Author: agibert $ */
/*---------------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------------*/
/* This file is part of LibDataBase */
/* */
/* LibDataBase is free software; you can redistribute it and/or modify */
/* it under the terms of the GNU Lesser General Public Licence as published by */
/* the Free Software Foundation; either version 2.1 of the License, or */
/* (at your option) any later version. */
/* */
/* LibDataBase 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 Lesser General Public License for more details. */
/* */
/* You should have received a copy of the GNU Lesser General Public License */
/* along with LibDataBase; if not, write to the Free Software */
/* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
/*---------------------------------------------------------------------------------*/
#include <stdio.h>
#include <libgen.h>
#include <tool.h>
#include "../lib/database.h"
/* Globals */
#define DEFAULT_REQUEST "INSERT INTO dummy (c1, c2, c3) VALUES (:1, :2, :3)"
#define TRUE 1
#define FALSE 0
const char *USAGE = "Usage: %s [--req <SQL Request>] [--from <n>] [--count <c>] [--db <db>] [--size <s>] [--baseIndex <i>]\n\n\
Executes an INSERT SQL request in a table which must have the following format:\n\
\"CREATE TABLE dummy (c1 NUMBER(35) primary key, c2 VARCHAR2(16), c3 NUMBER(10))\".\n\
Data to insert are taken from an array which elements are generated by the program (see\n\
option --baseIndex).\n\
\n\
Options:\n\
--req:\tExecutes specified request. If not specified, a default request is used:\n\
\t\"INSERT INTO dummy (c1, c2, c3) VALUES (:1, :2, :3)\".\n\
--from:\tStarting row in the array of input data (default: 0).\n\
--count:\tRows count to use in the input array (1 to 65536) (Default: 0).\n\
--size:\tTotal number of rows of the input array generated by the program (default: 1).\n\
--db:\t\tDatabase specification in the format: login/password@oracleSID.\n\
--baseIndex:\tStart index to use to create input data. Ex: if baseIndex = 10, first row\n\
to insert will have columns values: 10, 'A10', 110; second row will have\n\
11, 'A11', 111; etc. (default: 1).\n\
\n\
Examples:\n\
_ '%s --db \"login/pwd@srv\" --size 100 --count 20' will insert 5 times 20 rows in table dummy.\n\
_ '%s --db \"login/pwd@srv\" --size 10 --count 4' will insert 2 times 4 rows then the 2\n\
remaining rows.\n";
extern int strcasecmp(const char *, const char *);
extern char *strdup(const char *);
void parseArgs(int argc, char **argv);
void usage(char **argv);
void say (const char *format, ...);
int setupHostVars(void);
void setupData(int n);
void splitDBSpec(char *spec, const char **login, const char **pwd, const char **server);
int arraySize = 1;
int baseIndex = 0;
int from = 0;
int count = 0;
int debugLevel = 0;
char *req = NULL;
const char *login = "";
const char *pwd = "";
const char *server = "";
DBT_Connection conn;
DBT_Statement st;
DBT_Result res;
DBT_HostVar hv1;
DBT_HostVar hv2;
DBT_HostVar hv3;
int *dynInt1 = NULL;
char *dynBuf = NULL;
int *dynInt3 = NULL;
DBT_Indicator *dynIndic1 = NULL;
DBT_Indicator *dynIndic2 = NULL;
DBT_Indicator *dynIndic3 = NULL;
/**
* Main.
*/
int main(int argc, char **argv) {
int totalRowsProcessed = 0;
DBT_Status rc;
TOOLT_Counter *counter = NULL;
/* Parse arguments */
parseArgs(argc, argv);
if (req == NULL) {
req = strdup(DEFAULT_REQUEST);
}
counter = TOOL_Counter_Alloc(1);
rc = DB_Library_Open(DBD_DEBUG | DBD_ERRMSG);
if (DB_ERROR(rc)) {
printf("Error: Could not open library (%s)\n", DB_Error_Message_Get());
return 1;
}
/* Connect to Database */
rc = DB_Connect(&conn, login, pwd, server);
if (DB_ERROR(rc)) {
printf("Error: Could not connect (%s)\n", DB_Error_Message_Get());
return 1;
}
/* Init data that are going to be inserted */
say("Setup data and HostVars\n");
setupData(arraySize);
if (setupHostVars() == FALSE)
return 1;
/* Init statement */
rc = DB_Statement_Init(&conn, &st);
if (DB_ERROR(rc)) {
printf("Error: Could not init statement (err %d: %s)\n", rc, DB_Error_Message_Get());
return 1;
}
say("Preparing statement for query: '%s'\n", req);
rc = DB_Statement_Prepare(&st, req);
if (DB_ERROR(rc)) {
printf("Error: Could not prepare statement (err %d: %s)\n", rc, DB_Error_Message_Get());
return 1;
}
say("Bind Vars\n");
rc = DB_Statement_BindVars(&st, arraySize, &hv1, &hv2, &hv3, NULL);
if (DB_ERROR(rc)) {
printf("Error: Could not bind vars (%s)\n", DB_Error_Message_Get());
return 1;
}
/* Start transaction */
say("\nStart Transaction\n");
DB_Transaction_Start(&conn);
say("Executing Statement (array size: %d; starting from row %d, %d by %d)...\n", arraySize, from, count, count);
TOOL_Counter_Start(counter, 0);
while (from < arraySize) {
if (from + count > arraySize)
count = arraySize - from;
rc = DB_Statement_Execute(&st, from, count, &res);
totalRowsProcessed += res.rowsProcessed;
if (DB_ERROR(rc))
break;
from += count;
}
/* End transaction */
TOOL_Counter_Stop(counter, 0);
if (DB_ERROR(rc)) {
say("Error: %s", DB_Error_Message_Get());
say("%d rows processed (error at iteration %d)\n", totalRowsProcessed, res.errorIteration);
}
else {
say("OK, %d rows processed in %ld ms\n", totalRowsProcessed, TOOL_Counter_Get(counter, 0));
}
/*DB_Transaction_Rollback(&conn);*/
say("Committing Transaction\n\n");
DB_Transaction_Commit(&conn);
if (req)
free(req);
TOOL_Counter_Free(counter);
DB_Disconnect(&conn);
DB_Library_Close();
return 0;
}
/**
* Affichage sur la sortie standard.
*/
void say(const char *format, ...) {
char buffer[512];
va_list v;
va_start(v, format);
vsnprintf(buffer, sizeof(buffer), format, v);
fprintf (stdout, "%s", buffer);
va_end(v);
}
/**
* Arguments.
*/
void parseArgs(int argc, char **argv) {
int i = 0;
for (i = 1; i < argc; ++i) {
if (! strcasecmp(argv[i], "--debuglevel")) {
if (++i >= argc) {
usage(argv);
exit(1);
}
debugLevel = atoi(argv[i]);
DB_Debug_Level_Set(debugLevel);
}
else if (! strcasecmp(argv[i], "--from")) {
if (++i >= argc) {
usage(argv);
exit(1);
}
from = atoi(argv[i]);
}
else if (! strcasecmp(argv[i], "--count")) {
if (++i >= argc) {
usage(argv);
exit(1);
}
count = atoi(argv[i]);
}
else if (! strcasecmp(argv[i], "--size")) {
if (++i >= argc) {
usage(argv);
exit(1);
}
arraySize = atoi(argv[i]);
}
else if (! strcasecmp(argv[i], "--baseIndex")) {
if (++i >= argc) {
usage(argv);
exit(1);
}
baseIndex = atoi(argv[i]);
}
else if (! strcasecmp(argv[i], "--req")) {
if (++i >= argc) {
usage(argv);
exit(1);
}
req = strdup(argv[i]);
}
else if (! strcasecmp(argv[i], "--db")) {
if (++i >= argc) {
usage(argv);
exit(1);
}
splitDBSpec(argv[i], &login, &pwd, &server);
}
else if (! strcasecmp(argv[i], "--help") || ! strcasecmp(argv[i], "-h") || ! strcasecmp(argv[i], "-?")) {
usage(argv);
exit(1);
}
else {
usage(argv);
exit(1);
}
}
}
/**
* Usage.
*/
void usage(char **argv) {
char *b = basename(argv[0]);
printf(USAGE, b, b, b);
}
/**
* Initializes host variables.
* Associates an 'hostvar' with a type, size, array of values and array of indicators.
*/
int setupHostVars(void) {
if (DB_HostVar_Setup(&hv1, DBD_INTEGER, 0, dynInt1, dynIndic1) != DBS_OK ||
DB_HostVar_Setup(&hv2, DBD_STRING, 20, dynBuf, dynIndic2) != DBS_OK ||
DB_HostVar_Setup(&hv3, DBD_INTEGER, 0, dynInt3, dynIndic3) != DBS_OK) {
say("Error: %s\n", DB_Error_Message_Get());
return FALSE;
}
return TRUE;
}
/**
* Setup arrays of values and indicators.
* (IN) n: Rows count to allocate for each array.
*/
void setupData(int n) {
int i;
dynInt1 = malloc(n * sizeof(int));
dynBuf = malloc((unsigned) n * 20);
dynInt3 = malloc(n * sizeof(int));
dynIndic1 = malloc(n * sizeof(int));
dynIndic2 = malloc(n * sizeof(int));
dynIndic3 = malloc(n * sizeof(int));
for (i = 0; i < n; i++) {
int x = i + baseIndex;
dynIndic1[i] = 0;
dynIndic2[i] = 0;
dynIndic3[i] = 0;
dynInt1[i] = x;
sprintf(dynBuf+20*i, "A%d", x);
dynInt3[i] = 100+x;
}
}
/**
* Splits a database specification in the format "login/password@server" into its
* basic components 'login', 'password' and 'server'.
* Note that the initial 'spec' string may be altered.
*/
void splitDBSpec(char *spec, const char **login_, const char **pwd_, const char **server_) {
char *p = NULL;
*login_ = spec;
p = strchr(spec, '/');
if (p == NULL)
return;
*p = 0;
*pwd_ = ++p;
p = strchr(p, '@');
if (p == NULL)
return;
*p = 0;
*server_ = ++p;
}
Reference in New Issue View Git Blame Copy Permalink