]> git.sur5r.net Git - bacula/bacula/blob - bacula/src/cats/sql_update.c
Fix #1335 about postgresql error message during copy session
[bacula/bacula] / bacula / src / cats / sql_update.c
1 /*
2    Bacula® - The Network Backup Solution
3
4    Copyright (C) 2000-2008 Free Software Foundation Europe e.V.
5
6    The main author of Bacula is Kern Sibbald, with contributions from
7    many others, a complete list can be found in the file AUTHORS.
8    This program is Free Software; you can redistribute it and/or
9    modify it under the terms of version two of the GNU General Public
10    License as published by the Free Software Foundation and included
11    in the file LICENSE.
12
13    This program is distributed in the hope that it will be useful, but
14    WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16    General Public License for more details.
17
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
21    02110-1301, USA.
22
23    Bacula® is a registered trademark of Kern Sibbald.
24    The licensor of Bacula is the Free Software Foundation Europe
25    (FSFE), Fiduciary Program, Sumatrastrasse 25, 8006 Zürich,
26    Switzerland, email:ftf@fsfeurope.org.
27 */
28 /*
29  * Bacula Catalog Database Update record interface routines
30  *
31  *    Kern Sibbald, March 2000
32  *
33  *    Version $Id$
34  */
35
36 /* The following is necessary so that we do not include
37  * the dummy external definition of DB.
38  */
39 #define __SQL_C                       /* indicate that this is sql.c */
40
41 #include "bacula.h"
42 #include "cats.h"
43
44 #if    HAVE_SQLITE3 || HAVE_MYSQL || HAVE_SQLITE || HAVE_POSTGRESQL || HAVE_DBI
45
46 /* -----------------------------------------------------------------------
47  *
48  *   Generic Routines (or almost generic)
49  *
50  * -----------------------------------------------------------------------
51  */
52
53 /* -----------------------------------------------------------------------
54  *
55  *   Generic Routines (or almost generic)
56  *
57  * -----------------------------------------------------------------------
58  */
59 /* Update the attributes record by adding the file digest */
60 int
61 db_add_digest_to_file_record(JCR *jcr, B_DB *mdb, FileId_t FileId, char *digest,
62                           int type)
63 {
64    int stat;
65    char ed1[50];
66
67    db_lock(mdb);
68    Mmsg(mdb->cmd, "UPDATE File SET MD5='%s' WHERE FileId=%s", digest, 
69       edit_int64(FileId, ed1));
70    stat = UPDATE_DB(jcr, mdb, mdb->cmd);
71    db_unlock(mdb);
72    return stat;
73 }
74
75 /* Mark the file record as being visited during database
76  * verify compare. Stuff JobId into the MarkId field
77  */
78 int db_mark_file_record(JCR *jcr, B_DB *mdb, FileId_t FileId, JobId_t JobId)
79 {
80    int stat;
81    char ed1[50], ed2[50];
82
83    db_lock(mdb);
84    Mmsg(mdb->cmd, "UPDATE File SET MarkId=%s WHERE FileId=%s", 
85       edit_int64(JobId, ed1), edit_int64(FileId, ed2));
86    stat = UPDATE_DB(jcr, mdb, mdb->cmd);
87    db_unlock(mdb);
88    return stat;
89 }
90
91 /*
92  * Update the Job record at start of Job
93  *
94  *  Returns: false on failure
95  *           true  on success
96  */
97 bool
98 db_update_job_start_record(JCR *jcr, B_DB *mdb, JOB_DBR *jr)
99 {
100    char dt[MAX_TIME_LENGTH];
101    time_t stime;
102    struct tm tm;
103    btime_t JobTDate;
104    int stat;
105    char ed1[50], ed2[50], ed3[50], ed4[50], ed5[50];
106
107    stime = jr->StartTime;
108    (void)localtime_r(&stime, &tm);
109    strftime(dt, sizeof(dt), "%Y-%m-%d %H:%M:%S", &tm);
110    JobTDate = (btime_t)stime;
111
112    db_lock(mdb);
113    Mmsg(mdb->cmd, "UPDATE Job SET JobStatus='%c',Level='%c',StartTime='%s',"
114 "ClientId=%s,JobTDate=%s,PoolId=%s,FileSetId=%s WHERE JobId=%s",
115       (char)(jcr->JobStatus),
116       (char)(jr->JobLevel), dt, 
117       edit_int64(jr->ClientId, ed1),
118       edit_uint64(JobTDate, ed2), 
119       edit_int64(jr->PoolId, ed3),
120       edit_int64(jr->FileSetId, ed4),
121       edit_int64(jr->JobId, ed5));
122
123    stat = UPDATE_DB(jcr, mdb, mdb->cmd);
124    mdb->changes = 0;
125    db_unlock(mdb);
126    return stat;
127 }
128
129 /*
130  * Update Long term statistics with all jobs that were run before
131  * age seconds
132  */
133 int
134 db_update_stats(JCR *jcr, B_DB *mdb, utime_t age)
135 {
136    char ed1[30];
137    utime_t now = (utime_t)time(NULL);
138    edit_uint64(now - age, ed1);
139
140    Mmsg(mdb->cmd, fill_jobhisto, ed1);
141    QUERY_DB(jcr, mdb, mdb->cmd); /* TODO: get a message ? */
142    return sql_affected_rows(mdb);
143 }
144
145 /*
146  * Update the Job record at end of Job
147  *
148  *  Returns: 0 on failure
149  *           1 on success
150  */
151 int
152 db_update_job_end_record(JCR *jcr, B_DB *mdb, JOB_DBR *jr)
153 {
154    char dt[MAX_TIME_LENGTH];
155    char rdt[MAX_TIME_LENGTH];
156    time_t ttime;
157    struct tm tm;
158    int stat;
159    char ed1[30], ed2[30], ed3[50], ed4[50];
160    btime_t JobTDate;
161    char PriorJobId[50];
162
163    if (jr->PriorJobId) {
164       bstrncpy(PriorJobId, edit_int64(jr->PriorJobId, ed1), sizeof(PriorJobId));
165    } else {
166       bstrncpy(PriorJobId, "0", sizeof(PriorJobId));
167    }
168
169    ttime = jr->EndTime;
170    (void)localtime_r(&ttime, &tm);
171    strftime(dt, sizeof(dt), "%Y-%m-%d %H:%M:%S", &tm);
172
173    if (jr->RealEndTime == 0) {
174       jr->RealEndTime = jr->EndTime;
175    }
176    ttime = jr->RealEndTime;
177    (void)localtime_r(&ttime, &tm);
178    strftime(rdt, sizeof(rdt), "%Y-%m-%d %H:%M:%S", &tm);
179
180    JobTDate = ttime;
181
182    db_lock(mdb);
183    Mmsg(mdb->cmd,
184       "UPDATE Job SET JobStatus='%c',EndTime='%s',"
185 "ClientId=%u,JobBytes=%s,ReadBytes=%s,JobFiles=%u,JobErrors=%u,VolSessionId=%u,"
186 "VolSessionTime=%u,PoolId=%u,FileSetId=%u,JobTDate=%s,"
187 "RealEndTime='%s',PriorJobId=%s WHERE JobId=%s",
188       (char)(jr->JobStatus), dt, jr->ClientId, edit_uint64(jr->JobBytes, ed1),
189       edit_uint64(jr->ReadBytes, ed4),
190       jr->JobFiles, jr->JobErrors, jr->VolSessionId, jr->VolSessionTime,
191       jr->PoolId, jr->FileSetId, edit_uint64(JobTDate, ed2), 
192       rdt,
193       PriorJobId,
194       edit_int64(jr->JobId, ed3));
195
196    stat = UPDATE_DB(jcr, mdb, mdb->cmd);
197
198    db_unlock(mdb);
199    return stat;
200 }
201
202 /*
203  * Update Client record
204  *   Returns: 0 on failure
205  *            1 on success
206  */
207 int
208 db_update_client_record(JCR *jcr, B_DB *mdb, CLIENT_DBR *cr)
209 {
210    int stat;
211    char ed1[50], ed2[50];
212    CLIENT_DBR tcr;
213
214    db_lock(mdb);
215    memcpy(&tcr, cr, sizeof(tcr));
216    if (!db_create_client_record(jcr, mdb, &tcr)) {
217       db_unlock(mdb);
218       return 0;
219    }
220
221    Mmsg(mdb->cmd,
222 "UPDATE Client SET AutoPrune=%d,FileRetention=%s,JobRetention=%s,"
223 "Uname='%s' WHERE Name='%s'",
224       cr->AutoPrune,
225       edit_uint64(cr->FileRetention, ed1),
226       edit_uint64(cr->JobRetention, ed2),
227       cr->Uname, cr->Name);
228
229    stat = UPDATE_DB(jcr, mdb, mdb->cmd);
230    db_unlock(mdb);
231    return stat;
232 }
233
234
235 /*
236  * Update Counters record
237  *   Returns: 0 on failure
238  *            1 on success
239  */
240 int db_update_counter_record(JCR *jcr, B_DB *mdb, COUNTER_DBR *cr)
241 {
242    db_lock(mdb);
243
244    Mmsg(mdb->cmd,
245 "UPDATE Counters SET MinValue=%d,MaxValue=%d,CurrentValue=%d,"
246 "WrapCounter='%s' WHERE Counter='%s'",
247       cr->MinValue, cr->MaxValue, cr->CurrentValue,
248       cr->WrapCounter, cr->Counter);
249
250    int stat = UPDATE_DB(jcr, mdb, mdb->cmd);
251    db_unlock(mdb);
252    return stat;
253 }
254
255
256 int db_update_pool_record(JCR *jcr, B_DB *mdb, POOL_DBR *pr)
257 {
258    int stat;
259    char ed1[50], ed2[50], ed3[50], ed4[50], ed5[50], ed6[50];
260
261    db_lock(mdb);
262    Mmsg(mdb->cmd, "SELECT count(*) from Media WHERE PoolId=%s",
263       edit_int64(pr->PoolId, ed4));
264    pr->NumVols = get_sql_record_max(jcr, mdb);
265    Dmsg1(400, "NumVols=%d\n", pr->NumVols);
266
267    Mmsg(mdb->cmd,
268 "UPDATE Pool SET NumVols=%u,MaxVols=%u,UseOnce=%d,UseCatalog=%d,"
269 "AcceptAnyVolume=%d,VolRetention='%s',VolUseDuration='%s',"
270 "MaxVolJobs=%u,MaxVolFiles=%u,MaxVolBytes=%s,Recycle=%d,"
271 "AutoPrune=%d,LabelType=%d,LabelFormat='%s',RecyclePoolId=%s,"
272 "ScratchPoolId=%s WHERE PoolId=%s",
273       pr->NumVols, pr->MaxVols, pr->UseOnce, pr->UseCatalog,
274       pr->AcceptAnyVolume, edit_uint64(pr->VolRetention, ed1),
275       edit_uint64(pr->VolUseDuration, ed2),
276       pr->MaxVolJobs, pr->MaxVolFiles,
277       edit_uint64(pr->MaxVolBytes, ed3),
278       pr->Recycle, pr->AutoPrune, pr->LabelType,
279       pr->LabelFormat, edit_int64(pr->RecyclePoolId,ed5),
280       edit_int64(pr->ScratchPoolId,ed6),ed4);
281    stat = UPDATE_DB(jcr, mdb, mdb->cmd);
282    db_unlock(mdb);
283    return stat;
284 }
285
286 bool
287 db_update_storage_record(JCR *jcr, B_DB *mdb, STORAGE_DBR *sr)
288 {
289    int stat;
290    char ed1[50];
291    db_lock(mdb);
292    Mmsg(mdb->cmd, "UPDATE Storage SET AutoChanger=%d WHERE StorageId=%s", 
293       sr->AutoChanger, edit_int64(sr->StorageId, ed1));
294
295    stat = UPDATE_DB(jcr, mdb, mdb->cmd);
296    db_unlock(mdb);
297    return stat;
298 }
299
300
301 /*
302  * Update the Media Record at end of Session
303  *
304  * Returns: 0 on failure
305  *          numrows on success
306  */
307 int
308 db_update_media_record(JCR *jcr, B_DB *mdb, MEDIA_DBR *mr)
309 {
310    char dt[MAX_TIME_LENGTH];
311    time_t ttime;
312    struct tm tm;
313    int stat;
314    char ed1[50], ed2[50],  ed3[50],  ed4[50]; 
315    char ed5[50], ed6[50],  ed7[50],  ed8[50];
316    char ed9[50], ed10[50], ed11[50];
317
318
319    Dmsg1(100, "update_media: FirstWritten=%d\n", mr->FirstWritten);
320    db_lock(mdb);
321    if (mr->set_first_written) {
322       Dmsg1(400, "Set FirstWritten Vol=%s\n", mr->VolumeName);
323       ttime = mr->FirstWritten;
324       (void)localtime_r(&ttime, &tm);
325       strftime(dt, sizeof(dt), "%Y-%m-%d %H:%M:%S", &tm);
326       Mmsg(mdb->cmd, "UPDATE Media SET FirstWritten='%s'"
327            " WHERE VolumeName='%s'", dt, mr->VolumeName);
328       stat = UPDATE_DB(jcr, mdb, mdb->cmd);
329       Dmsg1(400, "Firstwritten=%d\n", mr->FirstWritten);
330    }
331
332    /* Label just done? */
333    if (mr->set_label_date) {
334       ttime = mr->LabelDate;
335       if (ttime == 0) {
336          ttime = time(NULL);
337       }
338       (void)localtime_r(&ttime, &tm);
339       strftime(dt, sizeof(dt), "%Y-%m-%d %H:%M:%S", &tm);
340       Mmsg(mdb->cmd, "UPDATE Media SET LabelDate='%s' "
341            "WHERE VolumeName='%s'", dt, mr->VolumeName);
342       UPDATE_DB(jcr, mdb, mdb->cmd);
343    }
344
345    if (mr->LastWritten != 0) {
346       ttime = mr->LastWritten;
347       (void)localtime_r(&ttime, &tm);
348       strftime(dt, sizeof(dt), "%Y-%m-%d %H:%M:%S", &tm);
349       Mmsg(mdb->cmd, "UPDATE Media Set LastWritten='%s' "
350            "WHERE VolumeName='%s'", dt, mr->VolumeName);
351       UPDATE_DB(jcr, mdb, mdb->cmd);
352    }
353
354    /* sanity checks for #1066 */
355    if (mr->VolReadTime < 0) {
356       mr->VolReadTime = 0;
357    }
358    if (mr->VolWriteTime < 0) {
359       mr->VolWriteTime = 0;
360    }
361
362    Mmsg(mdb->cmd, "UPDATE Media SET VolJobs=%u,"
363         "VolFiles=%u,VolBlocks=%u,VolBytes=%s,VolMounts=%u,VolErrors=%u,"
364         "VolWrites=%u,MaxVolBytes=%s,VolStatus='%s',"
365         "Slot=%d,InChanger=%d,VolReadTime=%s,VolWriteTime=%s,VolParts=%d,"
366         "LabelType=%d,StorageId=%s,PoolId=%s,VolRetention=%s,VolUseDuration=%s,"
367         "MaxVolJobs=%d,MaxVolFiles=%d,Enabled=%d,LocationId=%s,"
368         "ScratchPoolId=%s,RecyclePoolId=%s,RecycleCount=%d,Recycle=%d"
369         " WHERE VolumeName='%s'",
370         mr->VolJobs, mr->VolFiles, mr->VolBlocks, edit_uint64(mr->VolBytes, ed1),
371         mr->VolMounts, mr->VolErrors, mr->VolWrites,
372         edit_uint64(mr->MaxVolBytes, ed2),
373         mr->VolStatus, mr->Slot, mr->InChanger,
374         edit_int64(mr->VolReadTime, ed3),
375         edit_int64(mr->VolWriteTime, ed4),
376         mr->VolParts,
377         mr->LabelType,
378         edit_int64(mr->StorageId, ed5),
379         edit_int64(mr->PoolId, ed6),
380         edit_uint64(mr->VolRetention, ed7),
381         edit_uint64(mr->VolUseDuration, ed8),
382         mr->MaxVolJobs, mr->MaxVolFiles,
383         mr->Enabled, edit_uint64(mr->LocationId, ed9),
384         edit_uint64(mr->ScratchPoolId, ed10),
385         edit_uint64(mr->RecyclePoolId, ed11),
386         mr->RecycleCount,mr->Recycle,
387         mr->VolumeName);
388
389    Dmsg1(400, "%s\n", mdb->cmd);
390
391    stat = UPDATE_DB(jcr, mdb, mdb->cmd);
392
393    /* Make sure InChanger is 0 for any record having the same Slot */
394    db_make_inchanger_unique(jcr, mdb, mr);
395
396    db_unlock(mdb);
397    return stat;
398 }
399
400 /*
401  * Update the Media Record Default values from Pool
402  *
403  * Returns: 0 on failure
404  *          numrows on success
405  */
406 int
407 db_update_media_defaults(JCR *jcr, B_DB *mdb, MEDIA_DBR *mr)
408 {
409    int stat;
410    char ed1[50], ed2[50], ed3[50], ed4[50], ed5[50];
411
412
413    db_lock(mdb);
414    if (mr->VolumeName[0]) {
415       Mmsg(mdb->cmd, "UPDATE Media SET "
416            "Recycle=%d,VolRetention=%s,VolUseDuration=%s,"
417            "MaxVolJobs=%u,MaxVolFiles=%u,MaxVolBytes=%s,RecyclePoolId=%s"
418            " WHERE VolumeName='%s'",
419            mr->Recycle,edit_uint64(mr->VolRetention, ed1),
420            edit_uint64(mr->VolUseDuration, ed2),
421            mr->MaxVolJobs, mr->MaxVolFiles,
422            edit_uint64(mr->MaxVolBytes, ed3),
423            edit_uint64(mr->RecyclePoolId, ed4),
424            mr->VolumeName);
425    } else {
426       Mmsg(mdb->cmd, "UPDATE Media SET "
427            "Recycle=%d,VolRetention=%s,VolUseDuration=%s,"
428            "MaxVolJobs=%u,MaxVolFiles=%u,MaxVolBytes=%s,RecyclePoolId=%s"
429            " WHERE PoolId=%s",
430            mr->Recycle,edit_uint64(mr->VolRetention, ed1),
431            edit_uint64(mr->VolUseDuration, ed2),
432            mr->MaxVolJobs, mr->MaxVolFiles,
433            edit_uint64(mr->MaxVolBytes, ed3),
434            edit_int64(mr->RecyclePoolId, ed4),
435            edit_int64(mr->PoolId, ed5));
436    }
437
438    Dmsg1(400, "%s\n", mdb->cmd);
439
440    stat = UPDATE_DB(jcr, mdb, mdb->cmd);
441
442    db_unlock(mdb);
443    return stat;
444 }
445
446
447 /*
448  * If we have a non-zero InChanger, ensure that no other Media
449  *  record has InChanger set on the same Slot.
450  *
451  * This routine assumes the database is already locked.
452  */
453 void
454 db_make_inchanger_unique(JCR *jcr, B_DB *mdb, MEDIA_DBR *mr)
455 {
456    char ed1[50], ed2[50];
457    if (mr->InChanger != 0 && mr->Slot != 0 && mr->StorageId != 0) {
458
459        if (mr->MediaId != 0) {
460           Mmsg(mdb->cmd, "UPDATE Media SET InChanger=0, Slot=0 WHERE "
461                "Slot=%d AND StorageId=%s AND MediaId!=%s",
462                mr->Slot, 
463                edit_int64(mr->StorageId, ed1), edit_int64(mr->MediaId, ed2));
464
465        } else if (*mr->VolumeName) {
466           Mmsg(mdb->cmd, "UPDATE Media SET InChanger=0, Slot=0 WHERE "
467                "Slot=%d AND StorageId=%s AND VolumeName!='%s'",
468                mr->Slot, 
469                edit_int64(mr->StorageId, ed1), mr->VolumeName);
470
471        } else {  /* used by ua_label to reset all volume with this slot */
472           Mmsg(mdb->cmd, "UPDATE Media SET InChanger=0, Slot=0 WHERE "
473                "Slot=%d AND StorageId=%s",
474                mr->Slot, 
475                edit_int64(mr->StorageId, ed1), mr->VolumeName);          
476        }
477        Dmsg1(100, "%s\n", mdb->cmd);
478        UPDATE_DB(jcr, mdb, mdb->cmd);
479    }
480 }
481
482 #endif /* HAVE_SQLITE3 || HAVE_MYSQL || HAVE_SQLITE || HAVE_POSTGRESQL*/