* Bacula Catalog Database Update record interface routines
*
* Kern Sibbald, March 2000
+ *
+ * Version $Id$
*/
/*
- Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
+ Copyright (C) 2000-2004 Kern Sibbald and John Walker
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License as
*/
-/* *****FIXME**** fix fixed length of select_cmd[] and insert_cmd[] */
-
/* The following is necessary so that we do not include
* the dummy external definition of DB.
*/
#include "bacula.h"
#include "cats.h"
-#if HAVE_MYSQL || HAVE_SQLITE
+#if HAVE_MYSQL || HAVE_SQLITE || HAVE_POSTGRESQL
/* -----------------------------------------------------------------------
*
/* Imported subroutines */
extern void print_result(B_DB *mdb);
-extern int UpdateDB(char *file, int line, B_DB *db, char *update_cmd);
-
-static int do_update(B_DB *mdb, char *cmd)
-{
- int stat;
-
- stat = UPDATE_DB(mdb, cmd);
- return stat;
-}
+extern int UpdateDB(const char *file, int line, JCR *jcr, B_DB *db, char *update_cmd);
/* -----------------------------------------------------------------------
*
*/
/* Update the attributes record by adding the MD5 signature */
int
-db_add_MD5_to_file_record(B_DB *mdb, FileId_t FileId, char *MD5)
+db_add_SIG_to_file_record(JCR *jcr, B_DB *mdb, FileId_t FileId, char *SIG,
+ int type)
{
int stat;
- P(mdb->mutex);
- Mmsg(&mdb->cmd, "UPDATE File SET MD5=\"%s\" WHERE FileId=%d", MD5, FileId);
- stat = UPDATE_DB(mdb, mdb->cmd);
- V(mdb->mutex);
+ db_lock(mdb);
+ Mmsg(mdb->cmd, "UPDATE File SET MD5='%s' WHERE FileId=%u", SIG, FileId);
+ stat = UPDATE_DB(jcr, mdb, mdb->cmd);
+ db_unlock(mdb);
return stat;
}
/* Mark the file record as being visited during database
- * verify compare. Stuff JobId into FileIndex field
+ * verify compare. Stuff JobId into MarkedId field
*/
-int db_mark_file_record(B_DB *mdb, FileId_t FileId, int JobId)
+int db_mark_file_record(JCR *jcr, B_DB *mdb, FileId_t FileId, JobId_t JobId)
{
int stat;
- P(mdb->mutex);
- Mmsg(&mdb->cmd, "UPDATE File SET FileIndex=%d WHERE FileId=%d", JobId, FileId);
- stat = UPDATE_DB(mdb, mdb->cmd);
- V(mdb->mutex);
+ db_lock(mdb);
+ Mmsg(mdb->cmd, "UPDATE File SET MarkId=%u WHERE FileId=%u", JobId, FileId);
+ stat = UPDATE_DB(jcr, mdb, mdb->cmd);
+ db_unlock(mdb);
return stat;
}
/*
- * Update the Job record at end of Job
+ * Update the Job record at start of Job
*
* Returns: 0 on failure
* 1 on success
*/
-int
-db_update_job_start_record(B_DB *mdb, JOB_DBR *jr)
+bool
+db_update_job_start_record(JCR *jcr, B_DB *mdb, JOB_DBR *jr)
{
char dt[MAX_TIME_LENGTH];
time_t stime;
struct tm tm;
- int32_t StartDay;
+ btime_t JobTDate;
int stat;
+ char ed1[30];
stime = jr->StartTime;
localtime_r(&stime, &tm);
strftime(dt, sizeof(dt), "%Y-%m-%d %T", &tm);
- StartDay = (int32_t)(date_encode(tm.tm_year+1900, tm.tm_mon+1, tm.tm_mday) -
- date_encode(2000, 1, 1));
-
- P(mdb->mutex);
- Mmsg(&mdb->cmd, "UPDATE Job SET Level='%c', StartTime=\"%s\", \
-ClientId=%d, StartDay=%d WHERE JobId=%d",
- (char)(jr->Level), dt, jr->ClientId, StartDay, jr->JobId);
- stat = UPDATE_DB(mdb, mdb->cmd);
- V(mdb->mutex);
+ JobTDate = (btime_t)stime;
+
+ db_lock(mdb);
+ Mmsg(mdb->cmd, "UPDATE Job SET JobStatus='%c',Level='%c',StartTime='%s',"
+"ClientId=%u,JobTDate=%s WHERE JobId=%u",
+ (char)(jcr->JobStatus),
+ (char)(jr->JobLevel), dt, jr->ClientId, edit_uint64(JobTDate, ed1), jr->JobId);
+
+ stat = UPDATE_DB(jcr, mdb, mdb->cmd);
+ mdb->changes = 0;
+ db_unlock(mdb);
return stat;
}
+/*
+ * Given an incoming integer, set the string buffer to either NULL or the value
+ *
+ *
+ */
+void edit_num_or_null(char *s, size_t n, uint32_t id) {
+ bsnprintf(s, n, id ? "%u" : "NULL", id);
+}
/*
* 1 on success
*/
int
-db_update_job_end_record(B_DB *mdb, JOB_DBR *jr)
+db_update_job_end_record(JCR *jcr, B_DB *mdb, JOB_DBR *jr)
{
char dt[MAX_TIME_LENGTH];
time_t ttime;
struct tm tm;
int stat;
+ char ed1[30], ed2[30];
+ btime_t JobTDate;
+ char PoolId [50];
+ char FileSetId [50];
+ char ClientId [50];
+
+
+ /* some values are set to zero, which translates to NULL in SQL */
+ edit_num_or_null(PoolId, sizeof(PoolId), jr->PoolId);
+ edit_num_or_null(FileSetId, sizeof(FileSetId), jr->FileSetId);
+ edit_num_or_null(ClientId, sizeof(ClientId), jr->ClientId);
ttime = jr->EndTime;
localtime_r(&ttime, &tm);
strftime(dt, sizeof(dt), "%Y-%m-%d %T", &tm);
+ JobTDate = ttime;
- P(mdb->mutex);
- Mmsg(&mdb->cmd,
+ db_lock(mdb);
+ Mmsg(mdb->cmd,
"UPDATE Job SET JobStatus='%c', EndTime='%s', \
-ClientId=%d, JobBytes=%" lld ", JobFiles=%d, JobErrors=%d, VolSessionId=%d, \
-VolSessionTime=%d, PoolId=%d, FileSetId=%d WHERE JobId=%d",
- (char)(jr->JobStatus), dt, jr->ClientId, jr->JobBytes, jr->JobFiles,
- jr->JobErrors, jr->VolSessionId, jr->VolSessionTime,
- jr->PoolId, jr->FileSetId, jr->JobId);
-
- stat = UPDATE_DB(mdb, mdb->cmd);
- V(mdb->mutex);
+ClientId=%s, JobBytes=%s, JobFiles=%u, JobErrors=%u, VolSessionId=%u, \
+VolSessionTime=%u, PoolId=%s, FileSetId=%s, JobTDate=%s WHERE JobId=%u",
+ (char)(jr->JobStatus), dt, ClientId, edit_uint64(jr->JobBytes, ed1),
+ jr->JobFiles, jr->JobErrors, jr->VolSessionId, jr->VolSessionTime,
+ PoolId, FileSetId, edit_uint64(JobTDate, ed2), jr->JobId);
+
+ stat = UPDATE_DB(jcr, mdb, mdb->cmd);
+ db_unlock(mdb);
return stat;
}
+/*
+ * Update Client record
+ * Returns: 0 on failure
+ * 1 on success
+ */
int
-db_update_pool_record(B_DB *mdb, POOL_DBR *pr)
+db_update_client_record(JCR *jcr, B_DB *mdb, CLIENT_DBR *cr)
{
int stat;
+ char ed1[50], ed2[50];
+ CLIENT_DBR tcr;
+
+ db_lock(mdb);
+ memcpy(&tcr, cr, sizeof(tcr));
+ if (!db_create_client_record(jcr, mdb, &tcr)) {
+ db_unlock(mdb);
+ return 0;
+ }
+
+ Mmsg(mdb->cmd,
+"UPDATE Client SET AutoPrune=%d,FileRetention=%s,JobRetention=%s,"
+"Uname='%s' WHERE Name='%s'",
+ cr->AutoPrune,
+ edit_uint64(cr->FileRetention, ed1),
+ edit_uint64(cr->JobRetention, ed2),
+ cr->Uname, cr->Name);
+
+ stat = UPDATE_DB(jcr, mdb, mdb->cmd);
+ db_unlock(mdb);
+ return stat;
+}
+
+
+/*
+ * Update Counters record
+ * Returns: 0 on failure
+ * 1 on success
+ */
+int db_update_counter_record(JCR *jcr, B_DB *mdb, COUNTER_DBR *cr)
+{
+ db_lock(mdb);
+
+ Mmsg(mdb->cmd,
+"UPDATE Counters SET MinValue=%d,MaxValue=%d,CurrentValue=%d,"
+"WrapCounter='%s' WHERE Counter='%s'",
+ cr->MinValue, cr->MaxValue, cr->CurrentValue,
+ cr->WrapCounter, cr->Counter);
+
+ int stat = UPDATE_DB(jcr, mdb, mdb->cmd);
+ db_unlock(mdb);
+ return stat;
+}
- P(mdb->mutex);
- Mmsg(&mdb->cmd,
-"UPDATE Pool SET NumVols=%d, MaxVols=%d, UseOnce=%d, UseCatalog=%d, \
-AcceptAnyVolume=%d, LabelFormat=\"%s\" WHERE PoolId=%d",
- pr->NumVols, pr->MaxVols, pr->UseOnce, pr->UseCatalog,
- pr->AcceptAnyVolume, pr->LabelFormat, pr->PoolId);
- stat = UPDATE_DB(mdb, mdb->cmd);
- V(mdb->mutex);
+int
+db_update_pool_record(JCR *jcr, B_DB *mdb, POOL_DBR *pr)
+{
+ int stat;
+ char ed1[50], ed2[50], ed3[50];
+
+ db_lock(mdb);
+ Mmsg(mdb->cmd,
+"UPDATE Pool SET NumVols=%u,MaxVols=%u,UseOnce=%d,UseCatalog=%d,"
+"AcceptAnyVolume=%d,VolRetention='%s',VolUseDuration='%s',"
+"MaxVolJobs=%u,MaxVolFiles=%u,MaxVolBytes=%s,Recycle=%d,"
+"AutoPrune=%d,LabelFormat='%s' WHERE PoolId=%u",
+ pr->NumVols, pr->MaxVols, pr->UseOnce, pr->UseCatalog,
+ pr->AcceptAnyVolume, edit_uint64(pr->VolRetention, ed1),
+ edit_uint64(pr->VolUseDuration, ed2),
+ pr->MaxVolJobs, pr->MaxVolFiles,
+ edit_uint64(pr->MaxVolBytes, ed3),
+ pr->Recycle, pr->AutoPrune,
+ pr->LabelFormat, pr->PoolId);
+
+ stat = UPDATE_DB(jcr, mdb, mdb->cmd);
+ db_unlock(mdb);
return stat;
}
* numrows on success
*/
int
-db_update_media_record(B_DB *mdb, MEDIA_DBR *mr)
+db_update_media_record(JCR *jcr, B_DB *mdb, MEDIA_DBR *mr)
{
char dt[MAX_TIME_LENGTH];
time_t ttime;
struct tm tm;
int stat;
+ char ed1[30], ed2[30], ed3[30], ed4[30];
- ttime = mr->LastWritten;
- localtime_r(&ttime, &tm);
- strftime(dt, sizeof(dt), "%Y-%m-%d %T", &tm);
- P(mdb->mutex);
- if (mr->VolMounts == 1) {
- Mmsg(&mdb->cmd, "UPDATE Media SET FirstWritten=\"%s\" WHERE \
- VolumeName=\"%s\"", dt, mr->VolumeName);
- if (do_update(mdb, mdb->cmd) == 0) {
- V(mdb->mutex);
- return 0;
+ Dmsg1(100, "update_media: FirstWritten=%d\n", mr->FirstWritten);
+ db_lock(mdb);
+ if (mr->VolJobs == 1) {
+ Dmsg1(400, "Set FirstWritten Vol=%s\n", mr->VolumeName);
+ ttime = mr->FirstWritten;
+ localtime_r(&ttime, &tm);
+ strftime(dt, sizeof(dt), "%Y-%m-%d %T", &tm);
+ Mmsg(mdb->cmd, "UPDATE Media SET FirstWritten='%s'\
+ WHERE VolumeName='%s'", dt, mr->VolumeName);
+ stat = UPDATE_DB(jcr, mdb, mdb->cmd);
+ Dmsg1(400, "Firstwritten stat=%d\n", stat);
+ }
+
+ /* Label just done? */
+ if (mr->VolBytes == 1) {
+ ttime = mr->LabelDate;
+ if (ttime == 0) {
+ ttime = time(NULL);
}
+ localtime_r(&ttime, &tm);
+ strftime(dt, sizeof(dt), "%Y-%m-%d %T", &tm);
+ Mmsg(mdb->cmd, "UPDATE Media SET LabelDate='%s' "
+ "WHERE VolumeName='%s'", dt, mr->VolumeName);
+ stat = UPDATE_DB(jcr, mdb, mdb->cmd);
+ }
+
+ if (mr->LastWritten != 0) {
+ ttime = mr->LastWritten;
+ localtime_r(&ttime, &tm);
+ strftime(dt, sizeof(dt), "%Y-%m-%d %T", &tm);
+
+ Mmsg(mdb->cmd, "UPDATE Media SET VolJobs=%u,"
+ "VolFiles=%u,VolBlocks=%u,VolBytes=%s,VolMounts=%u,VolErrors=%u,"
+ "VolWrites=%u,MaxVolBytes=%s,LastWritten='%s',VolStatus='%s',"
+ "Slot=%d,InChanger=%d,VolReadTime=%s,VolWriteTime=%s "
+ " WHERE VolumeName='%s'",
+ mr->VolJobs, mr->VolFiles, mr->VolBlocks, edit_uint64(mr->VolBytes, ed1),
+ mr->VolMounts, mr->VolErrors, mr->VolWrites,
+ edit_uint64(mr->MaxVolBytes, ed2), dt,
+ mr->VolStatus, mr->Slot, mr->InChanger,
+ edit_uint64(mr->VolReadTime, ed3),
+ edit_uint64(mr->VolWriteTime, ed4),
+ mr->VolumeName);
+ } else {
+ Mmsg(mdb->cmd, "UPDATE Media SET VolJobs=%u,"
+ "VolFiles=%u,VolBlocks=%u,VolBytes=%s,VolMounts=%u,VolErrors=%u,"
+ "VolWrites=%u,MaxVolBytes=%s,VolStatus='%s',"
+ "Slot=%d,InChanger=%d,VolReadTime=%s,VolWriteTime=%s "
+ " WHERE VolumeName='%s'",
+ mr->VolJobs, mr->VolFiles, mr->VolBlocks, edit_uint64(mr->VolBytes, ed1),
+ mr->VolMounts, mr->VolErrors, mr->VolWrites,
+ edit_uint64(mr->MaxVolBytes, ed2),
+ mr->VolStatus, mr->Slot, mr->InChanger,
+ edit_uint64(mr->VolReadTime, ed3),
+ edit_uint64(mr->VolWriteTime, ed4),
+ mr->VolumeName);
+ }
+
+ Dmsg1(400, "%s\n", mdb->cmd);
+
+ stat = UPDATE_DB(jcr, mdb, mdb->cmd);
+
+ /* Make sure InChanger is 0 for any record having the same Slot */
+ db_make_inchanger_unique(jcr, mdb, mr);
+
+ db_unlock(mdb);
+ return stat;
+}
+
+/*
+ * Update the Media Record Default values from Pool
+ *
+ * Returns: 0 on failure
+ * numrows on success
+ */
+int
+db_update_media_defaults(JCR *jcr, B_DB *mdb, MEDIA_DBR *mr)
+{
+ int stat;
+ char ed1[30], ed2[30], ed3[30];
+
+
+ db_lock(mdb);
+ if (mr->VolumeName[0]) {
+ Mmsg(mdb->cmd, "UPDATE Media SET "
+ "Recycle=%d,VolRetention=%s,VolUseDuration=%s,"
+ "MaxVolJobs=%u,MaxVolFiles=%u,MaxVolBytes=%s"
+ " WHERE VolumeName='%s'",
+ mr->Recycle,edit_uint64(mr->VolRetention, ed1),
+ edit_uint64(mr->VolUseDuration, ed2),
+ mr->MaxVolJobs, mr->MaxVolFiles,
+ edit_uint64(mr->VolBytes, ed3),
+ mr->VolumeName);
+ } else {
+ Mmsg(mdb->cmd, "UPDATE Media SET "
+ "Recycle=%d,VolRetention=%s,VolUseDuration=%s,"
+ "MaxVolJobs=%u,MaxVolFiles=%u,MaxVolBytes=%s"
+ " WHERE PoolId=%u",
+ mr->Recycle,edit_uint64(mr->VolRetention, ed1),
+ edit_uint64(mr->VolUseDuration, ed2),
+ mr->MaxVolJobs, mr->MaxVolFiles,
+ edit_uint64(mr->VolBytes, ed3),
+ mr->PoolId);
}
- Mmsg(&mdb->cmd, "UPDATE Media SET VolJobs=%d,\
- VolFiles=%d, VolBlocks=%d, VolBytes=%" lld ", VolMounts=%d, VolErrors=%d,\
- VolWrites=%d, VolMaxBytes=%" lld ", LastWritten=\"%s\", VolStatus=\"%s\" \
- WHERE VolumeName=\"%s\"",
- mr->VolJobs, mr->VolFiles, mr->VolBlocks, mr->VolBytes, mr->VolMounts,
- mr->VolErrors, mr->VolWrites, mr->VolMaxBytes, dt, mr->VolStatus,
- mr->VolumeName);
+ Dmsg1(400, "%s\n", mdb->cmd);
- stat = UPDATE_DB(mdb, mdb->cmd);
- V(mdb->mutex);
+ stat = UPDATE_DB(jcr, mdb, mdb->cmd);
+
+ db_unlock(mdb);
return stat;
}
-#endif /* HAVE_MYSQL || HAVE_SQLITE */
+
+/*
+ * If we have a non-zero InChanger, ensure that no other Media
+ * record has InChanger set on the same Slot.
+ *
+ * This routine assumes the database is already locked.
+ */
+void
+db_make_inchanger_unique(JCR *jcr, B_DB *mdb, MEDIA_DBR *mr)
+{
+ if (mr->InChanger != 0 && mr->Slot != 0) {
+ Mmsg(mdb->cmd, "UPDATE Media SET InChanger=0 WHERE "
+ "Slot=%d AND MediaId!=%u", mr->Slot, mr->MediaId);
+ Dmsg1(400, "%s\n", mdb->cmd);
+ UPDATE_DB(jcr, mdb, mdb->cmd);
+ }
+}
+
+#endif /* HAVE_MYSQL || HAVE_SQLITE || HAVE_POSTGRESQL*/