-/*
- * Bacula Catalog Database routines specific to PostgreSQL
- * These are PostgreSQL specific routines
- *
- * Dan Langille, December 2003
- * based upon work done by Kern Sibbald, March 2000
- *
- * Version $Id$
- */
/*
Bacula® - The Network Backup Solution
- Copyright (C) 2003-2006 Free Software Foundation Europe e.V.
+ Copyright (C) 2003-2007 Free Software Foundation Europe e.V.
The main author of Bacula is Kern Sibbald, with contributions from
many others, a complete list can be found in the file AUTHORS.
This program is Free Software; you can redistribute it and/or
modify it under the terms of version two of the GNU General Public
- License as published by the Free Software Foundation plus additions
- that are listed in the file LICENSE.
+ License as published by the Free Software Foundation and included
+ in the file LICENSE.
This program is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of
(FSFE), Fiduciary Program, Sumatrastrasse 25, 8006 Zürich,
Switzerland, email:ftf@fsfeurope.org.
*/
+/*
+ * Bacula Catalog Database routines specific to PostgreSQL
+ * These are PostgreSQL specific routines
+ *
+ * Dan Langille, December 2003
+ * based upon work done by Kern Sibbald, March 2000
+ *
+ * Version $Id$
+ */
/* The following is necessary so that we do not include
int errstat;
char buf[10], *port;
+#ifdef xxx
+ if (!PQisthreadsafe()) {
+ Jmsg(jcr, M_ABORT, 0, _("PostgreSQL configuration problem. "
+ "PostgreSQL library is not thread safe. Connot continue.\n"));
+ }
+#endif
P(mutex);
if (mdb->connected) {
V(mutex);
mdb->connected = false;
if ((errstat=rwl_init(&mdb->lock)) != 0) {
+ berrno be;
Mmsg1(&mdb->errmsg, _("Unable to initialize DB lock. ERR=%s\n"),
- strerror(errstat));
+ be.bstrerror(errstat));
V(mutex);
return 0;
}
return 0;
}
+ mdb->connected = true;
+
if (!check_tables_version(jcr, mdb)) {
V(mutex);
return 0;
sql_query(mdb, "SET datestyle TO 'ISO, YMD'");
- mdb->connected = true;
V(mutex);
return 1;
}
}
db_end_transaction(jcr, mdb);
P(mutex);
+ sql_free_result(mdb);
mdb->ref_count--;
if (mdb->ref_count == 0) {
qdchain(&mdb->bq);
if (mdb->db_socket) {
free(mdb->db_socket);
}
- my_postgresql_free_result(mdb);
free(mdb);
}
V(mutex);
}
+void db_thread_cleanup()
+{ }
+
/*
* Return the next unique index (auto-increment) for
* the given table. Return NULL on error.
Dmsg0(500, "my_postgresql_fetch_row start\n");
- if (mdb->row_number == -1 || mdb->row == NULL) {
+ if (!mdb->row || mdb->row_size < mdb->num_fields) {
Dmsg1(500, "we have need space of %d bytes\n", sizeof(char *) * mdb->num_fields);
- if (mdb->row != NULL) {
+ if (mdb->row) {
Dmsg0(500, "my_postgresql_fetch_row freeing space\n");
free(mdb->row);
- mdb->row = NULL;
}
-
mdb->row = (POSTGRESQL_ROW) malloc(sizeof(char *) * mdb->num_fields);
+ mdb->row_size = mdb->num_fields;
// now reset the row_number now that we have the space allocated
mdb->row_number = 0;
int i;
Dmsg0(500, "my_postgresql_fetch_field starts\n");
- if (mdb->fields == NULL) {
+
+ if (!mdb->fields || mdb->fields_size < mdb->num_fields) {
+ if (mdb->fields) {
+ free(mdb->fields);
+ }
Dmsg1(500, "allocating space for %d fields\n", mdb->num_fields);
mdb->fields = (POSTGRESQL_FIELD *)malloc(sizeof(POSTGRESQL_FIELD) * mdb->num_fields);
+ mdb->fields_size = mdb->num_fields;
for (i = 0; i < mdb->num_fields; i++) {
Dmsg1(500, "filling field %d\n", i);
/*
* Note, if this routine returns 1 (failure), Bacula expects
* that no result has been stored.
+ * This is where QUERY_DB comes with Postgresql.
+ *
+ * Returns: 0 on success
+ * 1 on failure
+ *
*/
-int my_postgresql_query(B_DB *mdb, const char *query) {
+int my_postgresql_query(B_DB *mdb, const char *query)
+{
Dmsg0(500, "my_postgresql_query started\n");
// We are starting a new query. reset everything.
mdb->num_rows = -1;
mdb->row_number = -1;
mdb->field_number = -1;
- if (mdb->result != NULL) {
+ if (mdb->result) {
PQclear(mdb->result); /* hmm, someone forgot to free?? */
+ mdb->result = NULL;
}
Dmsg1(500, "my_postgresql_query starts with '%s'\n", query);
- mdb->result = PQexec(mdb->db, query);
+
+ for (int i=0; i < 10; i++) {
+ mdb->result = PQexec(mdb->db, query);
+ if (mdb->result) {
+ break;
+ }
+ bmicrosleep(5, 0);
+ }
+ if (!mdb->result) {
+ Dmsg1(50, "Query failed: %s\n", query);
+ goto bail_out;
+ }
+
mdb->status = PQresultStatus(mdb->result);
if (mdb->status == PGRES_TUPLES_OK || mdb->status == PGRES_COMMAND_OK) {
Dmsg1(500, "we have a result\n", query);
// how many fields in the set?
- mdb->num_fields = (int) PQnfields(mdb->result);
+ mdb->num_fields = (int)PQnfields(mdb->result);
Dmsg1(500, "we have %d fields\n", mdb->num_fields);
- mdb->num_rows = PQntuples(mdb->result);
+ mdb->num_rows = PQntuples(mdb->result);
Dmsg1(500, "we have %d rows\n", mdb->num_rows);
- mdb->status = 0;
+ mdb->status = 0; /* succeed */
} else {
- Dmsg1(500, "we failed\n", query);
- mdb->status = 1;
+ Dmsg1(50, "Result status failed: %s\n", query);
+ goto bail_out;
}
Dmsg0(500, "my_postgresql_query finishing\n");
+ return mdb->status;
+bail_out:
+ Dmsg1(500, "we failed\n", query);
+ PQclear(mdb->result);
+ mdb->result = NULL;
+ mdb->status = 1; /* failed */
return mdb->status;
}
-void my_postgresql_free_result (B_DB *mdb)
+void my_postgresql_free_result(B_DB *mdb)
{
+
+ db_lock(mdb);
if (mdb->result) {
PQclear(mdb->result);
mdb->result = NULL;
free(mdb->fields);
mdb->fields = NULL;
}
+ db_unlock(mdb);
}
int my_postgresql_currval(B_DB *mdb, char *table_name)
bstrncat(sequence, "_seq", sizeof(sequence));
bsnprintf(query, sizeof(query), "SELECT currval('%s')", sequence);
-// Mmsg(query, "SELECT currval('%s')", sequence);
Dmsg1(500, "my_postgresql_currval invoked with '%s'\n", query);
- result = PQexec(mdb->db, query);
+ for (int i=0; i < 10; i++) {
+ result = PQexec(mdb->db, query);
+ if (result) {
+ break;
+ }
+ bmicrosleep(5, 0);
+ }
+ if (!result) {
+ Dmsg1(50, "Query failed: %s\n", query);
+ goto bail_out;
+ }
Dmsg0(500, "exec done");
id = atoi(PQgetvalue(result, 0, 0));
Dmsg2(500, "got value '%s' which became %d\n", PQgetvalue(result, 0, 0), id);
} else {
+ Dmsg1(50, "Result status failed: %s\n", query);
Mmsg1(&mdb->errmsg, _("error fetching currval: %s\n"), PQerrorMessage(mdb->db));
}
+bail_out:
PQclear(result);
return id;
int my_postgresql_batch_start(JCR *jcr, B_DB *mdb)
{
+ char *query = "COPY batch FROM STDIN";
+
Dmsg0(500, "my_postgresql_batch_start started\n");
if (my_postgresql_query(mdb,
- " CREATE TEMPORARY TABLE batch "
- " (fileindex int, "
- " jobid int, "
- " path varchar, "
- " name varchar, "
- " lstat varchar, "
- " md5 varchar)") == 1)
+ " CREATE TEMPORARY TABLE batch "
+ " (fileindex int, "
+ " jobid int, "
+ " path varchar, "
+ " name varchar, "
+ " lstat varchar, "
+ " md5 varchar)") == 1)
{
Dmsg0(500, "my_postgresql_batch_start failed\n");
return 1;
mdb->row_number = -1;
mdb->field_number = -1;
- if (mdb->result != NULL) {
- my_postgresql_free_result(mdb);
+ my_postgresql_free_result(mdb);
+
+ for (int i=0; i < 10; i++) {
+ mdb->result = PQexec(mdb->db, query);
+ if (mdb->result) {
+ break;
+ }
+ bmicrosleep(5, 0);
+ }
+ if (!mdb->result) {
+ Dmsg1(50, "Query failed: %s\n", query);
+ goto bail_out;
}
- mdb->result = PQexec(mdb->db, "COPY batch FROM STDIN");
mdb->status = PQresultStatus(mdb->result);
if (mdb->status == PGRES_COPY_IN) {
// how many fields in the set?
mdb->num_rows = 0;
mdb->status = 1;
} else {
- Dmsg0(500, "we failed\n");
- mdb->status = 0;
+ Dmsg1(50, "Result status failed: %s\n", query);
+ goto bail_out;
}
Dmsg0(500, "my_postgresql_batch_start finishing\n");
return mdb->status;
+
+bail_out:
+ mdb->status = 0;
+ PQclear(mdb->result);
+ mdb->result = NULL;
+ return mdb->status;
}
/* set error to something to abort operation */
int count=30;
Dmsg0(500, "my_postgresql_batch_end started\n");
- if (!mdb) { /* no files ? */
+ if (!mdb) { /* no files ? */
return 0;
}
}
len = Mmsg(mdb->cmd, "%u\t%s\t%s\t%s\t%s\t%s\n",
- ar->FileIndex, edit_int64(ar->JobId, ed1), mdb->esc_path,
- mdb->esc_name, ar->attr, digest);
+ ar->FileIndex, edit_int64(ar->JobId, ed1), mdb->esc_path,
+ mdb->esc_name, ar->attr, digest);
do {
res = PQputCopyData(mdb->db,
- mdb->cmd,
- len);
+ mdb->cmd,
+ len);
} while (res == 0 && --count > 0);
if (res == 1) {
while (len > 0 && *src) {
switch (*src) {
case '\n':
- c = 'n';
- break;
+ c = 'n';
+ break;
case '\\':
- c = '\\';
- break;
+ c = '\\';
+ break;
case '\t':
- c = 't';
- break;
+ c = 't';
+ break;
case '\r':
- c = 'r';
- break;
+ c = 'r';
+ break;
default:
- c = '\0' ;
+ c = '\0' ;
}
if (c) {
- *dest = '\\';
- dest++;
- *dest = c;
+ *dest = '\\';
+ dest++;
+ *dest = c;
} else {
- *dest = *src;
+ *dest = *src;
}
len--;