/*
Bacula® - The Network Backup Solution
- Copyright (C) 2000-2010 Free Software Foundation Europe e.V.
+ Copyright (C) 2000-2011 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
+ modify it under the terms of version three of the GNU Affero General Public
License as published by the Free Software Foundation and included
in the file LICENSE.
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
+ You should have received a copy of the GNU Affero 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.
*
* Kern Sibbald, March 2000
*
+ * Major rewrite by Marco van Wieringen, January 2010 for catalog refactoring.
*/
-
-/* The following is necessary so that we do not include
- * the dummy external definition of DB.
- */
-#define __SQL_C /* indicate that this is sql.c */
-
#include "bacula.h"
-#include "cats.h"
#ifdef HAVE_MYSQL
+#include "cats.h"
+#include "bdb_priv.h"
+#include <mysql.h>
+#include <bdb_mysql.h>
+
/* -----------------------------------------------------------------------
*
* MySQL dependent defines and subroutines
* -----------------------------------------------------------------------
*/
-/* List of open databases */
+/*
+ * List of open databases
+ */
static dlist *db_list = NULL;
static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
-/*
- * Retrieve database type
- */
-const char *
-db_get_type(void)
-{
- return "MySQL";
-}
-
-/*
- * Initialize database data structure. In principal this should
- * never have errors, or it is really fatal.
- */
-B_DB *
-db_init_database(JCR *jcr, const char *db_name, const char *db_user, const char *db_password,
- const char *db_address, int db_port, const char *db_socket,
- int mult_db_connections)
+B_DB_MYSQL::B_DB_MYSQL(JCR *jcr,
+ const char *db_driver,
+ const char *db_name,
+ const char *db_user,
+ const char *db_password,
+ const char *db_address,
+ int db_port,
+ const char *db_socket,
+ bool mult_db_connections,
+ bool disable_batch_insert)
{
- B_DB *mdb = NULL;
-
- if (!db_user) {
- Jmsg(jcr, M_FATAL, 0, _("A user name for MySQL must be supplied.\n"));
- return NULL;
- }
- P(mutex); /* lock DB queue */
- if (db_list == NULL) {
- db_list = New(dlist(mdb, &mdb->link));
- }
- /* Look to see if DB already open */
- if (!mult_db_connections) {
- foreach_dlist(mdb, db_list) {
- if (bstrcmp(mdb->db_name, db_name) &&
- bstrcmp(mdb->db_address, db_address) &&
- mdb->db_port == db_port) {
- Dmsg2(100, "DB REopen %d %s\n", mdb->ref_count, db_name);
- mdb->ref_count++;
- V(mutex);
- Dmsg3(100, "initdb ref=%d connected=%d db=%p\n", mdb->ref_count,
- mdb->connected, mdb->db);
- return mdb; /* already open */
- }
- }
- }
- db_check_backend_thread_safe();
- Dmsg0(100, "db_open first time\n");
- mdb = (B_DB *)malloc(sizeof(B_DB));
- memset(mdb, 0, sizeof(B_DB));
- mdb->db_name = bstrdup(db_name);
- mdb->db_user = bstrdup(db_user);
+ /*
+ * Initialize the parent class members.
+ */
+ m_db_interface_type = SQL_INTERFACE_TYPE_MYSQL;
+ m_db_type = SQL_TYPE_MYSQL;
+ m_db_driver = bstrdup("MySQL");
+ m_db_name = bstrdup(db_name);
+ m_db_user = bstrdup(db_user);
if (db_password) {
- mdb->db_password = bstrdup(db_password);
+ m_db_password = bstrdup(db_password);
}
if (db_address) {
- mdb->db_address = bstrdup(db_address);
+ m_db_address = bstrdup(db_address);
}
if (db_socket) {
- mdb->db_socket = bstrdup(db_socket);
+ m_db_socket = bstrdup(db_socket);
}
- mdb->db_port = db_port;
- mdb->errmsg = get_pool_memory(PM_EMSG); /* get error message buffer */
- *mdb->errmsg = 0;
- mdb->cmd = get_pool_memory(PM_EMSG); /* get command buffer */
- mdb->cached_path = get_pool_memory(PM_FNAME);
- mdb->cached_path_id = 0;
- mdb->ref_count = 1;
- mdb->fname = get_pool_memory(PM_FNAME);
- mdb->path = get_pool_memory(PM_FNAME);
- mdb->esc_name = get_pool_memory(PM_FNAME);
- mdb->esc_path = get_pool_memory(PM_FNAME);
- mdb->allow_transactions = mult_db_connections;
- db_list->append(mdb); /* put db in list */
- Dmsg3(100, "initdb ref=%d connected=%d db=%p\n", mdb->ref_count,
- mdb->connected, mdb->db);
- V(mutex);
- return mdb;
+ m_db_port = db_port;
+
+ if (disable_batch_insert) {
+ m_disabled_batch_insert = true;
+ m_have_batch_insert = false;
+ } else {
+ m_disabled_batch_insert = false;
+#if defined(USE_BATCH_FILE_INSERT)
+# if defined(HAVE_MYSQL_THREAD_SAFE)
+ m_have_batch_insert = mysql_thread_safe();
+# else
+ m_have_batch_insert = false;
+# endif /* HAVE_MYSQL_THREAD_SAFE */
+#else
+ m_have_batch_insert = false;
+#endif /* USE_BATCH_FILE_INSERT */
+ }
+ errmsg = get_pool_memory(PM_EMSG); /* get error message buffer */
+ *errmsg = 0;
+ cmd = get_pool_memory(PM_EMSG); /* get command buffer */
+ cached_path = get_pool_memory(PM_FNAME);
+ cached_path_id = 0;
+ m_ref_count = 1;
+ fname = get_pool_memory(PM_FNAME);
+ path = get_pool_memory(PM_FNAME);
+ esc_name = get_pool_memory(PM_FNAME);
+ esc_path = get_pool_memory(PM_FNAME);
+ esc_obj = get_pool_memory(PM_FNAME);
+ m_allow_transactions = mult_db_connections;
+
+ /* At this time, when mult_db_connections == true, this is for
+ * specific console command such as bvfs or batch mode, and we don't
+ * want to share a batch mode or bvfs. In the future, we can change
+ * the creation function to add this parameter.
+ */
+ m_dedicated = mult_db_connections;
+
+ /*
+ * Initialize the private members.
+ */
+ m_db_handle = NULL;
+ m_result = NULL;
+
+ /*
+ * Put the db in the list.
+ */
+ if (db_list == NULL) {
+ db_list = New(dlist(this, &this->m_link));
+ }
+ db_list->append(this);
+}
+
+B_DB_MYSQL::~B_DB_MYSQL()
+{
}
/*
* Now actually open the database. This can generate errors,
* which are returned in the errmsg
*
- * DO NOT close the database or free(mdb) here !!!!
+ * DO NOT close the database or delete mdb here !!!!
*/
-int
-db_open_database(JCR *jcr, B_DB *mdb)
+bool B_DB_MYSQL::db_open_database(JCR *jcr)
{
+ bool retval = false;
int errstat;
P(mutex);
- if (mdb->connected) {
- V(mutex);
- return 1;
+ if (m_connected) {
+ retval = true;
+ goto bail_out;
}
- if ((errstat=rwl_init(&mdb->lock)) != 0) {
+ if ((errstat=rwl_init(&m_lock)) != 0) {
berrno be;
- Mmsg1(&mdb->errmsg, _("Unable to initialize DB lock. ERR=%s\n"),
+ Mmsg1(&errmsg, _("Unable to initialize DB lock. ERR=%s\n"),
be.bstrerror(errstat));
- V(mutex);
- return 0;
+ goto bail_out;
}
- /* connect to the database */
+ /*
+ * Connect to the database
+ */
#ifdef xHAVE_EMBEDDED_MYSQL
// mysql_server_init(0, NULL, NULL);
#endif
- mysql_init(&mdb->mysql);
+ mysql_init(&m_instance);
Dmsg0(50, "mysql_init done\n");
- /* If connection fails, try at 5 sec intervals for 30 seconds. */
+ /*
+ * If connection fails, try at 5 sec intervals for 30 seconds.
+ */
for (int retry=0; retry < 6; retry++) {
- mdb->db = mysql_real_connect(
- &(mdb->mysql), /* db */
- mdb->db_address, /* default = localhost */
- mdb->db_user, /* login name */
- mdb->db_password, /* password */
- mdb->db_name, /* database name */
- mdb->db_port, /* default port */
- mdb->db_socket, /* default = socket */
- CLIENT_FOUND_ROWS); /* flags */
-
- /* If no connect, try once more in case it is a timing problem */
- if (mdb->db != NULL) {
+ m_db_handle = mysql_real_connect(
+ &(m_instance), /* db */
+ m_db_address, /* default = localhost */
+ m_db_user, /* login name */
+ m_db_password, /* password */
+ m_db_name, /* database name */
+ m_db_port, /* default port */
+ m_db_socket, /* default = socket */
+ CLIENT_FOUND_ROWS); /* flags */
+
+ /*
+ * If no connect, try once more in case it is a timing problem
+ */
+ if (m_db_handle != NULL) {
break;
}
bmicrosleep(5,0);
}
- mdb->mysql.reconnect = 1; /* so connection does not timeout */
+ m_instance.reconnect = 1; /* so connection does not timeout */
Dmsg0(50, "mysql_real_connect done\n");
- Dmsg3(50, "db_user=%s db_name=%s db_password=%s\n", mdb->db_user, mdb->db_name,
- mdb->db_password==NULL?"(NULL)":mdb->db_password);
+ Dmsg3(50, "db_user=%s db_name=%s db_password=%s\n", m_db_user, m_db_name,
+ (m_db_password == NULL) ? "(NULL)" : m_db_password);
- if (mdb->db == NULL) {
- Mmsg2(&mdb->errmsg, _("Unable to connect to MySQL server.\n"
+ if (m_db_handle == NULL) {
+ Mmsg2(&errmsg, _("Unable to connect to MySQL server.\n"
"Database=%s User=%s\n"
"MySQL connect failed either server not running or your authorization is incorrect.\n"),
- mdb->db_name, mdb->db_user);
+ m_db_name, m_db_user);
#if MYSQL_VERSION_ID >= 40101
Dmsg3(50, "Error %u (%s): %s\n",
- mysql_errno(&(mdb->mysql)), mysql_sqlstate(&(mdb->mysql)),
- mysql_error(&(mdb->mysql)));
+ mysql_errno(&(m_instance)), mysql_sqlstate(&(m_instance)),
+ mysql_error(&(m_instance)));
#else
Dmsg2(50, "Error %u: %s\n",
- mysql_errno(&(mdb->mysql)), mysql_error(&(mdb->mysql)));
+ mysql_errno(&(m_instance)), mysql_error(&(m_instance)));
#endif
- V(mutex);
- return 0;
+ goto bail_out;
}
- mdb->connected = true;
- if (!check_tables_version(jcr, mdb)) {
- V(mutex);
- return 0;
+ m_connected = true;
+ if (!check_tables_version(jcr, this)) {
+ goto bail_out;
}
- Dmsg3(100, "opendb ref=%d connected=%d db=%p\n", mdb->ref_count,
- mdb->connected, mdb->db);
+ Dmsg3(100, "opendb ref=%d connected=%d db=%p\n", m_ref_count, m_connected, m_db_handle);
+
+ /*
+ * Set connection timeout to 8 days specialy for batch mode
+ */
+ sql_query("SET wait_timeout=691200");
+ sql_query("SET interactive_timeout=691200");
- /* Set connection timeout to 8 days specialy for batch mode */
- sql_query(mdb, "SET wait_timeout=691200");
- sql_query(mdb, "SET interactive_timeout=691200");
+ retval = true;
+bail_out:
V(mutex);
- return 1;
+ return retval;
}
-void
-db_close_database(JCR *jcr, B_DB *mdb)
+void B_DB_MYSQL::db_close_database(JCR *jcr)
{
- if (!mdb) {
- return;
- }
- db_end_transaction(jcr, mdb);
+ db_end_transaction(jcr);
P(mutex);
- sql_free_result(mdb);
- mdb->ref_count--;
- Dmsg3(100, "closedb ref=%d connected=%d db=%p\n", mdb->ref_count,
- mdb->connected, mdb->db);
- if (mdb->ref_count == 0) {
- db_list->remove(mdb);
- if (mdb->connected) {
- Dmsg1(100, "close db=%p\n", mdb->db);
- mysql_close(&mdb->mysql);
+ m_ref_count--;
+ Dmsg3(100, "closedb ref=%d connected=%d db=%p\n", m_ref_count, m_connected, m_db_handle);
+ if (m_ref_count == 0) {
+ sql_free_result();
+ db_list->remove(this);
+ if (m_connected) {
+ Dmsg1(100, "close db=%p\n", m_db_handle);
+ mysql_close(&m_instance);
#ifdef xHAVE_EMBEDDED_MYSQL
// mysql_server_end();
#endif
}
- rwl_destroy(&mdb->lock);
- free_pool_memory(mdb->errmsg);
- free_pool_memory(mdb->cmd);
- free_pool_memory(mdb->cached_path);
- free_pool_memory(mdb->fname);
- free_pool_memory(mdb->path);
- free_pool_memory(mdb->esc_name);
- free_pool_memory(mdb->esc_path);
- if (mdb->db_name) {
- free(mdb->db_name);
+ rwl_destroy(&m_lock);
+ free_pool_memory(errmsg);
+ free_pool_memory(cmd);
+ free_pool_memory(cached_path);
+ free_pool_memory(fname);
+ free_pool_memory(path);
+ free_pool_memory(esc_name);
+ free_pool_memory(esc_path);
+ free_pool_memory(esc_obj);
+ if (m_db_driver) {
+ free(m_db_driver);
+ }
+ if (m_db_name) {
+ free(m_db_name);
}
- if (mdb->db_user) {
- free(mdb->db_user);
+ if (m_db_user) {
+ free(m_db_user);
}
- if (mdb->db_password) {
- free(mdb->db_password);
+ if (m_db_password) {
+ free(m_db_password);
}
- if (mdb->db_address) {
- free(mdb->db_address);
+ if (m_db_address) {
+ free(m_db_address);
}
- if (mdb->db_socket) {
- free(mdb->db_socket);
+ if (m_db_socket) {
+ free(m_db_socket);
}
- free(mdb);
+ delete this;
if (db_list->size() == 0) {
delete db_list;
db_list = NULL;
V(mutex);
}
-void db_check_backend_thread_safe()
-{
-#ifdef HAVE_BATCH_FILE_INSERT
- if (!mysql_thread_safe()) {
- Emsg0(M_ABORT, 0, _("MySQL client library must be thread-safe "
- "when using BatchMode.\n"));
- }
-#endif
-}
-
/*
* This call is needed because the message channel thread
* opens a database on behalf of a jcr that was created in
* closes the database. Thus the msgchan must call here
* to cleanup any thread specific data that it created.
*/
-void db_thread_cleanup()
+void B_DB_MYSQL::db_thread_cleanup(void)
{
#ifndef HAVE_WIN32
- my_thread_end();
+ mysql_thread_end();
#endif
}
/*
- * Return the next unique index (auto-increment) for
- * the given table. Return NULL on error.
+ * Escape strings so that MySQL is happy
*
- * For MySQL, NULL causes the auto-increment value
- * to be updated.
+ * NOTE! len is the length of the old string. Your new
+ * string must be long enough (max 2*old+1) to hold
+ * the escaped output.
*/
-int db_next_index(JCR *jcr, B_DB *mdb, char *table, char *index)
+void B_DB_MYSQL::db_escape_string(JCR *jcr, char *snew, char *old, int len)
{
- strcpy(index, "NULL");
- return 1;
+ mysql_real_escape_string(m_db_handle, snew, old, len);
}
+/*
+ * Escape binary object so that MySQL is happy
+ * Memory is stored in B_DB struct, no need to free it
+ */
+char *B_DB_MYSQL::db_escape_object(JCR *jcr, char *old, int len)
+{
+ esc_obj = check_pool_memory_size(esc_obj, len*2+1);
+ mysql_real_escape_string(m_db_handle, esc_obj, old, len);
+ return esc_obj;
+}
/*
- * Escape strings so that MySQL is happy
- *
- * NOTE! len is the length of the old string. Your new
- * string must be long enough (max 2*old+1) to hold
- * the escaped output.
+ * Unescape binary object so that MySQL is happy
*/
-void
-db_escape_string(JCR *jcr, B_DB *mdb, char *snew, char *old, int len)
+void B_DB_MYSQL::db_unescape_object(JCR *jcr, char *from, int32_t expected_len,
+ POOLMEM **dest, int32_t *dest_len)
+{
+ if (!from) {
+ *dest[0] = 0;
+ *dest_len = 0;
+ return;
+ }
+ *dest = check_pool_memory_size(*dest, expected_len+1);
+ *dest_len = expected_len;
+ memcpy(*dest, from, expected_len);
+ (*dest)[expected_len]=0;
+}
+
+void B_DB_MYSQL::db_start_transaction(JCR *jcr)
+{
+ if (!jcr->attr) {
+ jcr->attr = get_pool_memory(PM_FNAME);
+ }
+ if (!jcr->ar) {
+ jcr->ar = (ATTR_DBR *)malloc(sizeof(ATTR_DBR));
+ }
+}
+
+void B_DB_MYSQL::db_end_transaction(JCR *jcr)
{
- mysql_real_escape_string(mdb->db, snew, old, len);
+ if (jcr && jcr->cached_attribute) {
+ Dmsg0(400, "Flush last cached attribute.\n");
+ if (!db_create_attributes_record(jcr, this, jcr->ar)) {
+ Jmsg1(jcr, M_FATAL, 0, _("Attribute create error. %s"), db_strerror(jcr->db));
+ }
+ jcr->cached_attribute = false;
+ }
}
/*
* Submit a general SQL command (cmd), and for each row returned,
- * the sqlite_handler is called with the ctx.
+ * the result_handler is called with the ctx.
*/
-bool db_sql_query(B_DB *mdb, const char *query, DB_RESULT_HANDLER *result_handler, void *ctx)
+bool B_DB_MYSQL::db_sql_query(const char *query, DB_RESULT_HANDLER *result_handler, void *ctx)
{
+ int ret;
SQL_ROW row;
bool send = true;
+ bool retval = false;
- db_lock(mdb);
- if (sql_query(mdb, query) != 0) {
- Mmsg(mdb->errmsg, _("Query failed: %s: ERR=%s\n"), query, sql_strerror(mdb));
- db_unlock(mdb);
- return false;
+ Dmsg1(500, "db_sql_query starts with %s\n", query);
+
+ db_lock(this);
+ ret = mysql_query(m_db_handle, query);
+ if (ret != 0) {
+ Mmsg(errmsg, _("Query failed: %s: ERR=%s\n"), query, sql_strerror());
+ Dmsg0(500, "db_sql_query failed\n");
+ goto bail_out;
}
+
+ Dmsg0(500, "db_sql_query succeeded. checking handler\n");
+
if (result_handler != NULL) {
- if ((mdb->result = sql_use_result(mdb)) != NULL) {
- int num_fields = sql_num_fields(mdb);
+ if ((m_result = mysql_use_result(m_db_handle)) != NULL) {
+ m_num_fields = mysql_num_fields(m_result);
- /* We *must* fetch all rows */
- while ((row = sql_fetch_row(mdb)) != NULL) {
+ /*
+ * We *must* fetch all rows
+ */
+ while ((row = mysql_fetch_row(m_result)) != NULL) {
if (send) {
/* the result handler returns 1 when it has
* seen all the data it wants. However, we
* loop to the end of the data.
*/
- if (result_handler(ctx, num_fields, row)) {
+ if (result_handler(ctx, m_num_fields, row)) {
send = false;
}
}
}
+ sql_free_result();
+ }
+ }
+
+ Dmsg0(500, "db_sql_query finished\n");
+ retval = true;
+
+bail_out:
+ db_unlock(this);
+ return retval;
+}
- sql_free_result(mdb);
+bool B_DB_MYSQL::sql_query(const char *query, int flags)
+{
+ int ret;
+ bool retval = true;
+
+ Dmsg1(500, "sql_query starts with '%s'\n", query);
+ /*
+ * We are starting a new query. reset everything.
+ */
+ m_num_rows = -1;
+ m_row_number = -1;
+ m_field_number = -1;
+
+ if (m_result) {
+ mysql_free_result(m_result);
+ m_result = NULL;
+ }
+
+ ret = mysql_query(m_db_handle, query);
+ if (ret == 0) {
+ Dmsg0(500, "we have a result\n");
+ if (flags & QF_STORE_RESULT) {
+ m_result = mysql_store_result(m_db_handle);
+ if (m_result != NULL) {
+ m_num_fields = mysql_num_fields(m_result);
+ Dmsg1(500, "we have %d fields\n", m_num_fields);
+ m_num_rows = mysql_num_rows(m_result);
+ Dmsg1(500, "we have %d rows\n", m_num_rows);
+ } else {
+ m_num_fields = 0;
+ m_num_rows = mysql_affected_rows(m_db_handle);
+ Dmsg1(500, "we have %d rows\n", m_num_rows);
+ }
+ } else {
+ m_num_fields = 0;
+ m_num_rows = mysql_affected_rows(m_db_handle);
+ Dmsg1(500, "we have %d rows\n", m_num_rows);
}
+ } else {
+ Dmsg0(500, "we failed\n");
+ m_status = 1; /* failed */
+ retval = false;
}
- db_unlock(mdb);
- return true;
+ return retval;
+}
+void B_DB_MYSQL::sql_free_result(void)
+{
+ db_lock(this);
+ if (m_result) {
+ mysql_free_result(m_result);
+ m_result = NULL;
+ }
+ if (m_fields) {
+ free(m_fields);
+ m_fields = NULL;
+ }
+ m_num_rows = m_num_fields = 0;
+ db_unlock(this);
}
-void my_mysql_free_result(B_DB *mdb)
+SQL_ROW B_DB_MYSQL::sql_fetch_row(void)
{
- db_lock(mdb);
- if (mdb->result) {
- mysql_free_result(mdb->result);
- mdb->result = NULL;
+ if (!m_result) {
+ return NULL;
+ } else {
+ return mysql_fetch_row(m_result);
}
- db_unlock(mdb);
}
-int my_mysql_sql_insert_id(B_DB *mdb, const char *query, const char *table_name)
+const char *B_DB_MYSQL::sql_strerror(void)
+{
+ return mysql_error(m_db_handle);
+}
+
+void B_DB_MYSQL::sql_data_seek(int row)
+{
+ return mysql_data_seek(m_result, row);
+}
+
+int B_DB_MYSQL::sql_affected_rows(void)
+{
+ return mysql_affected_rows(m_db_handle);
+}
+
+uint64_t B_DB_MYSQL::sql_insert_autokey_record(const char *query, const char *table_name)
{
/*
* First execute the insert query and then retrieve the currval.
*/
- if (mysql_query(mdb->db, query)) {
+ if (mysql_query(m_db_handle, query) != 0) {
return 0;
}
- mdb->num_rows = sql_affected_rows(mdb);
- if (mdb->num_rows != 1) {
+ m_num_rows = mysql_affected_rows(m_db_handle);
+ if (m_num_rows != 1) {
return 0;
}
- mdb->changes++;
+ changes++;
- return mysql_insert_id(mdb->db);
+ return mysql_insert_id(m_db_handle);
}
+SQL_FIELD *B_DB_MYSQL::sql_fetch_field(void)
+{
+ int i;
+ MYSQL_FIELD *field;
-#ifdef HAVE_BATCH_FILE_INSERT
-const char *my_mysql_batch_lock_path_query =
- "LOCK TABLES Path write, batch write, Path as p write";
+ if (!m_fields || m_fields_size < m_num_fields) {
+ if (m_fields) {
+ free(m_fields);
+ m_fields = NULL;
+ }
+ Dmsg1(500, "allocating space for %d fields\n", m_num_fields);
+ m_fields = (SQL_FIELD *)malloc(sizeof(SQL_FIELD) * m_num_fields);
+ m_fields_size = m_num_fields;
+
+ for (i = 0; i < m_num_fields; i++) {
+ Dmsg1(500, "filling field %d\n", i);
+ if ((field = mysql_fetch_field(m_result)) != NULL) {
+ m_fields[i].name = field->name;
+ m_fields[i].max_length = field->max_length;
+ m_fields[i].type = field->type;
+ m_fields[i].flags = field->flags;
+
+ Dmsg4(500, "sql_fetch_field finds field '%s' has length='%d' type='%d' and IsNull=%d\n",
+ m_fields[i].name, m_fields[i].max_length, m_fields[i].type, m_fields[i].flags);
+ }
+ }
+ }
+ /*
+ * Increment field number for the next time around
+ */
+ return &m_fields[m_field_number++];
+}
-const char *my_mysql_batch_lock_filename_query =
- "LOCK TABLES Filename write, batch write, Filename as f write";
+bool B_DB_MYSQL::sql_field_is_not_null(int field_type)
+{
+ return IS_NOT_NULL(field_type);
+}
-const char *my_mysql_batch_unlock_tables_query = "UNLOCK TABLES";
+bool B_DB_MYSQL::sql_field_is_numeric(int field_type)
+{
+ return IS_NUM(field_type);
+}
+
+/*
+ * Returns true if OK
+ * false if failed
+ */
+bool B_DB_MYSQL::sql_batch_start(JCR *jcr)
+{
+ bool retval;
+
+ db_lock(this);
+ retval = sql_query("CREATE TEMPORARY TABLE batch ("
+ "FileIndex integer,"
+ "JobId integer,"
+ "Path blob,"
+ "Name blob,"
+ "LStat tinyblob,"
+ "MD5 tinyblob,"
+ "DeltaSeq integer)");
+ db_unlock(this);
+
+ return retval;
+}
+
+/* set error to something to abort operation */
+/*
+ * Returns true if OK
+ * false if failed
+ */
+bool B_DB_MYSQL::sql_batch_end(JCR *jcr, const char *error)
+{
+ m_status = 0;
+
+ return true;
+}
+
+/*
+ * Returns true if OK
+ * false if failed
+ */
+bool B_DB_MYSQL::sql_batch_insert(JCR *jcr, ATTR_DBR *ar)
+{
+ const char *digest;
+ char ed1[50];
+
+ esc_name = check_pool_memory_size(esc_name, fnl*2+1);
+ db_escape_string(jcr, esc_name, fname, fnl);
-const char *my_mysql_batch_fill_path_query =
- "INSERT INTO Path (Path) "
- "SELECT a.Path FROM "
- "(SELECT DISTINCT Path FROM batch) AS a WHERE NOT EXISTS "
- "(SELECT Path FROM Path AS p WHERE p.Path = a.Path)";
+ esc_path = check_pool_memory_size(esc_path, pnl*2+1);
+ db_escape_string(jcr, esc_path, path, pnl);
-const char *my_mysql_batch_fill_filename_query =
- "INSERT INTO Filename (Name) "
- "SELECT a.Name FROM "
- "(SELECT DISTINCT Name FROM batch) AS a WHERE NOT EXISTS "
- "(SELECT Name FROM Filename AS f WHERE f.Name = a.Name)";
-#endif /* HAVE_BATCH_FILE_INSERT */
+ if (ar->Digest == NULL || ar->Digest[0] == 0) {
+ digest = "0";
+ } else {
+ digest = ar->Digest;
+ }
+
+ Mmsg(cmd, "INSERT INTO batch VALUES "
+ "(%u,%s,'%s','%s','%s','%s',%u)",
+ ar->FileIndex, edit_int64(ar->JobId,ed1), esc_path,
+ esc_name, ar->attr, digest, ar->DeltaSeq);
+
+ return sql_query(cmd);
+}
+
+/*
+ * Initialize database data structure. In principal this should
+ * never have errors, or it is really fatal.
+ */
+B_DB *db_init_database(JCR *jcr, const char *db_driver, const char *db_name, const char *db_user,
+ const char *db_password, const char *db_address, int db_port, const char *db_socket,
+ bool mult_db_connections, bool disable_batch_insert)
+{
+ B_DB_MYSQL *mdb = NULL;
+
+ if (!db_user) {
+ Jmsg(jcr, M_FATAL, 0, _("A user name for MySQL must be supplied.\n"));
+ return NULL;
+ }
+ P(mutex); /* lock DB queue */
+
+ /*
+ * Look to see if DB already open
+ */
+ if (db_list && !mult_db_connections) {
+ foreach_dlist(mdb, db_list) {
+ if (mdb->db_match_database(db_driver, db_name, db_address, db_port)) {
+ Dmsg1(100, "DB REopen %s\n", db_name);
+ mdb->increment_refcount();
+ goto bail_out;
+ }
+ }
+ }
+ Dmsg0(100, "db_init_database first time\n");
+ mdb = New(B_DB_MYSQL(jcr, db_driver, db_name, db_user, db_password, db_address,
+ db_port, db_socket, mult_db_connections, disable_batch_insert));
+
+bail_out:
+ V(mutex);
+ return mdb;
+}
#endif /* HAVE_MYSQL */