]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/cats/sql_update.c
projects / bacula / bacula / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
Optimize tree.c a bit, take Scott's spec & patch
[bacula/bacula] / bacula / src / cats / sql_update.c
diff --git a/bacula/src/cats/sql_update.c b/bacula/src/cats/sql_update.c
index 520c1b553a10faee2074be66eed5ebaf97f5ccb8..4acb2b8f59f69ff2a147fc753b527020061204d2 100644 (file)
--- a/bacula/src/cats/sql_update.c
+++ b/bacula/src/cats/sql_update.c
@@ -7,7 +7,7 @@
  */
 
 /*
-   Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
+   Copyright (C) 2000-2003 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
@@ -45,7 +45,7 @@
 
 /* Imported subroutines */
 extern void print_result(B_DB *mdb);
-extern int UpdateDB(char *file, int line, B_DB *db, char *update_cmd);
+extern int UpdateDB(char *file, int line, JCR *jcr, B_DB *db, char *update_cmd);
 
 /* -----------------------------------------------------------------------
  *
@@ -55,13 +55,14 @@ extern int UpdateDB(char *file, int line, 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;
 
    db_lock(mdb);
-   Mmsg(&mdb->cmd, "UPDATE File SET MD5='%s' WHERE FileId=%u", MD5, FileId);
-   stat = UPDATE_DB(mdb, mdb->cmd);
+   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;
 }
@@ -69,13 +70,13 @@ db_add_MD5_to_file_record(B_DB *mdb, FileId_t FileId, char *MD5)
 /* Mark the file record as being visited during database
  * verify compare. Stuff JobId into MarkedId field
  */
-int db_mark_file_record(B_DB *mdb, FileId_t FileId, JobId_t JobId) 
+int db_mark_file_record(JCR *jcr, B_DB *mdb, FileId_t FileId, JobId_t JobId) 
 {
    int stat;
 
    db_lock(mdb);
    Mmsg(&mdb->cmd, "UPDATE File SET MarkId=%u WHERE FileId=%u", JobId, FileId);
-   stat = UPDATE_DB(mdb, mdb->cmd);
+   stat = UPDATE_DB(jcr, mdb, mdb->cmd);
    db_unlock(mdb);
    return stat;
 }
@@ -87,7 +88,7 @@ int db_mark_file_record(B_DB *mdb, FileId_t FileId, JobId_t JobId)
  *          1 on success
  */
 int
-db_update_job_start_record(B_DB *mdb, JOB_DBR *jr)
+db_update_job_start_record(JCR *jcr, B_DB *mdb, JOB_DBR *jr)
 {
    char dt[MAX_TIME_LENGTH];
    time_t stime;
@@ -102,10 +103,11 @@ db_update_job_start_record(B_DB *mdb, JOB_DBR *jr)
    JobTDate = (btime_t)stime;
 
    db_lock(mdb);
-   Mmsg(&mdb->cmd, "UPDATE Job SET Level='%c', StartTime='%s', \
-ClientId=%u, JobTDate=%s WHERE JobId=%u",
+   Mmsg(&mdb->cmd, "UPDATE Job SET Level='%c', StartTime='%s',"
+"ClientId=%u, JobTDate=%s WHERE JobId=%u",
       (char)(jr->Level), dt, jr->ClientId, edit_uint64(JobTDate, ed1), jr->JobId);
-   stat = UPDATE_DB(mdb, mdb->cmd);
+
+   stat = UPDATE_DB(jcr, mdb, mdb->cmd);
    db_unlock(mdb);
    mdb->changes = 0;
    return stat;
@@ -120,7 +122,7 @@ ClientId=%u, JobTDate=%s WHERE JobId=%u",
  *          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;
@@ -143,25 +145,87 @@ VolSessionTime=%u, PoolId=%u, FileSetId=%u, JobTDate=%s WHERE JobId=%u",
       jr->JobFiles, jr->JobErrors, jr->VolSessionId, jr->VolSessionTime, 
       jr->PoolId, jr->FileSetId, edit_uint64(JobTDate, ed2), jr->JobId);
 
-   stat = UPDATE_DB(mdb, mdb->cmd);
+   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 Pool SET NumVols=%d, MaxVols=%d, UseOnce=%d, UseCatalog=%d, \
-AcceptAnyVolume=%d, LabelFormat='%s' WHERE PoolId=%u",
+"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;
+}
+
+
+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, pr->LabelFormat, pr->PoolId);
+      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(mdb, mdb->cmd);
+   stat = UPDATE_DB(jcr, mdb, mdb->cmd);
    db_unlock(mdb);
    return stat;
 }
@@ -173,7 +237,7 @@ AcceptAnyVolume=%d, LabelFormat='%s' WHERE PoolId=%u",
  *         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;
@@ -181,32 +245,49 @@ db_update_media_record(B_DB *mdb, MEDIA_DBR *mr)
    int stat;
    char ed1[30], ed2[30];
        
-   ttime = mr->LastWritten;
-   localtime_r(&ttime, &tm);
-   strftime(dt, sizeof(dt), "%Y-%m-%d %T", &tm);
 
    Dmsg1(100, "update_media: FirstWritten=%d\n", mr->FirstWritten);
    db_lock(mdb);
-   if (mr->VolMounts == 1 && mr->VolBlocks==1 && mr->VolWrites==1) {
+   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(mdb, mdb->cmd);
+      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);
+   }
+
+   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, VolMaxBytes=%s, LastWritten='%s', VolStatus='%s',\
+ VolFiles=%u,VolBlocks=%u,VolBytes=%s,VolMounts=%u,VolErrors=%u,\
+ VolWrites=%u,MaxVolBytes=%s,LastWritten='%s',VolStatus='%s',\
  Slot=%d WHERE VolumeName='%s'",
    mr->VolJobs, mr->VolFiles, mr->VolBlocks, edit_uint64(mr->VolBytes, ed1),
    mr->VolMounts, mr->VolErrors, mr->VolWrites, 
-   edit_uint64(mr->VolMaxBytes, ed2), dt, 
+   edit_uint64(mr->MaxVolBytes, ed2), dt, 
    mr->VolStatus, mr->Slot, mr->VolumeName);
 
    Dmsg1(400, "%s\n", mdb->cmd);
 
-   stat = UPDATE_DB(mdb, mdb->cmd);
+   stat = UPDATE_DB(jcr, mdb, mdb->cmd);
    db_unlock(mdb);
    return stat;
 }
Cloned from http://git.bacula.org/bacula