]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/CDB/AliCDBGrid.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
New production macros from Mihaela
[u/mrichter/AliRoot.git] / STEER / CDB / AliCDBGrid.cxx
CommitLineData
9e1ceb13 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15
16/////////////////////////////////////////////////////////////////////////////////////////////////
17// //
18// AliCDBGrid //
19// access class to a DataBase in an AliEn storage //
20// //
21/////////////////////////////////////////////////////////////////////////////////////////////////
22
b09247a2 23#include <cstdlib>
9e1ceb13 24#include <TGrid.h>
25#include <TGridResult.h>
26#include <TFile.h>
6dc56e97 27#include <TKey.h>
9e1ceb13 28#include <TROOT.h>
4667c116 29#include <TList.h>
9e1ceb13 30#include <TObjArray.h>
31#include <TObjString.h>
89004e35 32#include <TMath.h>
9e1ceb13 33#include <TRegexp.h>
34
35#include "AliLog.h"
fdf65bb5 36#include "AliCDBEntry.h"
9e1ceb13 37#include "AliCDBGrid.h"
4667c116 38#include "AliCDBManager.h"
9e1ceb13 39
40
41ClassImp(AliCDBGrid)
42
43//_____________________________________________________________________________
c75fabed 44AliCDBGrid::AliCDBGrid(const char *gridUrl, const char *user, const char *dbFolder,
45 const char *se, const char* cacheFolder, Bool_t operateDisconnected,
46 Long64_t cacheSize, Long_t cleanupInterval) :
9e1ceb13 47AliCDBStorage(),
b05400be 48fGridUrl(gridUrl),
9e1ceb13 49fUser(user),
b05400be 50fDBFolder(dbFolder),
c75fabed 51fSE(se),
c5941d67 52fMirrorSEs(""),
c75fabed 53fCacheFolder(cacheFolder),
54fOperateDisconnected(operateDisconnected),
55fCacheSize(cacheSize),
56fCleanupInterval(cleanupInterval)
9e1ceb13 57{
58// constructor //
59
9e1ceb13 60 // if the same Grid is alreay active, skip connection
4667c116 61 if (!gGrid || fGridUrl != gGrid->GridUrl()
024cf675 62 || (( fUser != "" ) && ( fUser != gGrid->GetUser() )) ) {
9e1ceb13 63 // connection to the Grid
62032124 64 AliInfo("Connection to the Grid...");
024cf675 65 if(gGrid){
6720fa1f 66 AliInfo(Form("gGrid = %p; fGridUrl = %s; gGrid->GridUrl() = %s",gGrid,fGridUrl.Data(), gGrid->GridUrl()));
024cf675 67 AliInfo(Form("fUser = %s; gGrid->GetUser() = %s",fUser.Data(), gGrid->GetUser()));
68 }
b05400be 69 TGrid::Connect(fGridUrl.Data(),fUser.Data());
9e1ceb13 70 }
71
72 if(!gGrid) {
73 AliError("Connection failed!");
74 return;
75 }
76
77 TString initDir(gGrid->Pwd(0));
b05400be 78 if (fDBFolder[0] != '/') {
79 fDBFolder.Prepend(initDir);
9e1ceb13 80 }
81
82 // check DBFolder: trying to cd to DBFolder; if it does not exist, create it
b05400be 83 if(!gGrid->Cd(fDBFolder.Data(),0)){
84 AliDebug(2,Form("Creating new folder <%s> ...",fDBFolder.Data()));
d7744171 85 TGridResult* res = gGrid->Command(Form("mkdir -p %s",fDBFolder.Data()));
86 TString result = res->GetKey(0,"__result__");
87 if(result == "0"){
88 AliFatal(Form("Cannot create folder <%s> !",fDBFolder.Data()));
89 return;
9e1ceb13 90 }
91 } else {
b05400be 92 AliDebug(2,Form("Folder <%s> found",fDBFolder.Data()));
9e1ceb13 93 }
94
95 // removes any '/' at the end of path, then append one '/'
b05400be 96 while(fDBFolder.EndsWith("/")) fDBFolder.Remove(fDBFolder.Last('/'));
97 fDBFolder+="/";
9e1ceb13 98
62032124 99 fType="alien";
100 fBaseFolder = fDBFolder;
101
c75fabed 102 // Setting the cache
103
104 // Check if local cache folder is already defined
105 TString origCache(TFile::GetCacheFileDir());
106 if(fCacheFolder.Length() > 0) {
107 if(origCache.Length() == 0) {
108 AliInfo(Form("Setting local cache to: %s", fCacheFolder.Data()));
109 } else if(fCacheFolder != origCache) {
110 AliWarning(Form("Local cache folder was already defined, changing it to: %s",
111 fCacheFolder.Data()));
112 }
113
114 // default settings are: operateDisconnected=kTRUE, forceCacheread = kFALSE
115 if(!TFile::SetCacheFileDir(fCacheFolder.Data(), fOperateDisconnected)) {
116 AliError(Form("Could not set cache folder %s !", fCacheFolder.Data()));
117 fCacheFolder = "";
118 } else {
119 // reset fCacheFolder because the function may have
120 // slightly changed the folder name (e.g. '/' added)
121 fCacheFolder = TFile::GetCacheFileDir();
122 }
123
124 // default settings are: cacheSize=1GB, cleanupInterval = 0
125 if(!TFile::ShrinkCacheFileDir(fCacheSize, fCleanupInterval)) {
126 AliError(Form("Could not set following values "
d2a0a483 127 "to ShrinkCacheFileDir: cacheSize = %lld, cleanupInterval = %ld !",
c75fabed 128 fCacheSize, fCleanupInterval));
129 }
130 }
131
9e1ceb13 132 // return to the initial directory
133 gGrid->Cd(initDir.Data(),0);
5bfe3965 134
135 fNretry = 3; // default
136 fInitRetrySeconds = 5; // default
9e1ceb13 137}
138
139//_____________________________________________________________________________
140AliCDBGrid::~AliCDBGrid()
141{
142// destructor
62032124 143 delete gGrid; gGrid=0;
9e1ceb13 144
145}
146
147//_____________________________________________________________________________
62032124 148Bool_t AliCDBGrid::FilenameToId(TString& filename, AliCDBId& id) {
4005d0b5 149// build AliCDBId from full path filename (fDBFolder/path/Run#x_#y_v#z_s0.root)
62032124 150
151 if(filename.Contains(fDBFolder)){
152 filename = filename(fDBFolder.Length(),filename.Length()-fDBFolder.Length());
153 }
154
155 TString idPath = filename(0,filename.Last('/'));
156 id.SetPath(idPath);
157 if(!id.IsValid()) return kFALSE;
158
159 filename=filename(idPath.Length()+1,filename.Length()-idPath.Length());
9e1ceb13 160
161 Ssiz_t mSize;
4005d0b5 162 // valid filename: Run#firstRun_#lastRun_v#version_s0.root
163 TRegexp keyPattern("^Run[0-9]+_[0-9]+_v[0-9]+_s0.root$");
9e1ceb13 164 keyPattern.Index(filename, &mSize);
165 if (!mSize) {
4005d0b5 166
167 // TODO backward compatibility ... maybe remove later!
168 Ssiz_t oldmSize;
169 TRegexp oldKeyPattern("^Run[0-9]+_[0-9]+_v[0-9]+.root$");
170 oldKeyPattern.Index(filename, &oldmSize);
171 if(!oldmSize) {
172 AliDebug(2,Form("Bad filename <%s>.", filename.Data()));
173 return kFALSE;
174 } else {
175 AliDebug(2,Form("Old filename format <%s>.", filename.Data()));
176 id.SetSubVersion(-11); // TODO trick to ensure backward compatibility
177 }
178
179 } else {
180 id.SetSubVersion(-1); // TODO trick to ensure backward compatibility
181 }
9e1ceb13 182
62032124 183 filename.Resize(filename.Length() - sizeof(".root") + 1);
9e1ceb13 184
62032124 185 TObjArray* strArray = (TObjArray*) filename.Tokenize("_");
9e1ceb13 186
187 TString firstRunString(((TObjString*) strArray->At(0))->GetString());
62032124 188 id.SetFirstRun(atoi(firstRunString.Data() + 3));
189 id.SetLastRun(atoi(((TObjString*) strArray->At(1))->GetString()));
190
9e1ceb13 191 TString verString(((TObjString*) strArray->At(2))->GetString());
62032124 192 id.SetVersion(atoi(verString.Data() + 1));
9e1ceb13 193
194 delete strArray;
195
196 return kTRUE;
197}
198
199//_____________________________________________________________________________
c3a7b59a 200Bool_t AliCDBGrid::IdToFilename(const AliCDBId& id, TString& filename) const {
62032124 201// build file name from AliCDBId (path, run range, version) and fDBFolder
9e1ceb13 202
62032124 203 if (!id.GetAliCDBRunRange().IsValid()) {
204 AliDebug(2,Form("Invalid run range <%d, %d>.",
205 id.GetFirstRun(), id.GetLastRun()));
9e1ceb13 206 return kFALSE;
207 }
208
62032124 209 if (id.GetVersion() < 0) {
210 AliDebug(2,Form("Invalid version <%d>.", id.GetVersion()));
9e1ceb13 211 return kFALSE;
212 }
62032124 213
4005d0b5 214 filename = Form("Run%d_%d_v%d",
62032124 215 id.GetFirstRun(),
216 id.GetLastRun(),
217 id.GetVersion());
218
4005d0b5 219 if (id.GetSubVersion() != -11) filename += "_s0"; // TODO to ensure backward compatibility
220 filename += ".root";
221
62032124 222 filename.Prepend(fDBFolder + id.GetPath() + '/');
9e1ceb13 223
224 return kTRUE;
225}
226
89004e35 227//_____________________________________________________________________________
5bfe3965 228void AliCDBGrid::SetRetry(Int_t nretry, Int_t initsec) {
89004e35 229
230 // Function to set the exponential retry for putting entries in the OCDB
231
5bfe3965 232 AliWarning("WARNING!!! You are changing the exponential retry times and delay: this function should be used by experts!");
89004e35 233 fNretry = nretry;
5bfe3965 234 fInitRetrySeconds = initsec;
235 AliDebug(2,Form("fNretry = %d, fInitRetrySeconds = %d", fNretry, fInitRetrySeconds));
89004e35 236}
237
238
9e1ceb13 239//_____________________________________________________________________________
240Bool_t AliCDBGrid::PrepareId(AliCDBId& id) {
241// prepare id (version) of the object that will be stored (called by PutEntry)
242
243 TString initDir(gGrid->Pwd(0));
9e1ceb13 244
b05400be 245 TString dirName(fDBFolder);
9e1ceb13 246
247 Bool_t dirExist=kFALSE;
4005d0b5 248
4667c116 249
250
9e1ceb13 251 // go to the path; if directory does not exist, create it
4667c116 252 for(int i=0;i<3;i++){
0e10c8fd 253 //TString cmd("find -d ");
254 //cmd += Form("%s ",dirName);
255 //cmd +=
256 //gGrid->Command(cmd.Data());
4667c116 257 dirName+=Form("%s/",id.GetPathLevel(i).Data());
9e1ceb13 258 dirExist=gGrid->Cd(dirName,0);
259 if (!dirExist) {
b05400be 260 AliDebug(2,Form("Creating new folder <%s> ...",dirName.Data()));
9e1ceb13 261 if(!gGrid->Mkdir(dirName,"",0)){
262 AliError(Form("Cannot create directory <%s> !",dirName.Data()));
263 gGrid->Cd(initDir.Data());
264 return kFALSE;
265 }
4667c116 266
267 // if folders are new add tags to them
268 if(i == 1) {
269 // TODO Currently disabled
c75fabed 270 // add short lived tag!
4667c116 271 // AliInfo("Tagging level 1 folder with \"ShortLived\" tag");
272 // if(!AddTag(dirName,"ShortLived_try")){
273 // AliError(Form("Could not tag folder %s !", dirName.Data()));
274 // if(!gGrid->Rmdir(dirName.Data())){
275 // AliError(Form("Unexpected: could not remove %s directory!", dirName.Data()));
276 // }
277 // return 0;
278 //}
279
280 } else if(i == 2) {
c75fabed 281 AliDebug(2,"Tagging level 2 folder with \"CDB\" and \"CDB_MD\" tag");
4667c116 282 if(!AddTag(dirName,"CDB")){
283 AliError(Form("Could not tag folder %s !", dirName.Data()));
284 if(!gGrid->Rmdir(dirName.Data())){
285 AliError(Form("Unexpected: could not remove %s directory!", dirName.Data()));
286 }
287 return 0;
288 }
289 if(!AddTag(dirName,"CDB_MD")){
290 AliError(Form("Could not tag folder %s !", dirName.Data()));
291 if(!gGrid->Rmdir(dirName.Data())){
292 AliError(Form("Unexpected: could not remove %s directory!", dirName.Data()));
293 }
294 return 0;
295 }
296
4667c116 297 // TODO Currently disabled
c75fabed 298 // add short lived tag!
4667c116 299 // TString path=id.GetPath();
300 // if(AliCDBManager::Instance()->IsShortLived(path.Data())) {
301 // AliInfo(Form("Tagging %s as short lived", dirName.Data()));
302 // if(!TagShortLived(dirName, kTRUE)){
303 // AliError(Form("Could not tag folder %s !", dirName.Data()));
304 // if(!gGrid->Rmdir(dirName.Data())){
305 // AliError(Form("Unexpected: could not remove %s directory!", dirName.Data()));
306 // }
307 // return 0;
308 // }
309 // } else {
310 // AliInfo(Form("Tagging %s as long lived", dirName.Data()));
311 // if(!TagShortLived(dirName, kFALSE)){
312 // AliError(Form("Could not tag folder %s !", dirName.Data()));
313 // if(!gGrid->Rmdir(dirName.Data())){
314 // AliError(Form("Unexpected: could not remove %s directory!", dirName.Data()));
315 // }
316 // return 0;
317 // }
318 // }
319 }
9e1ceb13 320 }
321 }
9e1ceb13 322 gGrid->Cd(initDir,0);
323
62032124 324 TString filename;
325 AliCDBId anId; // the id got from filename
9e1ceb13 326 AliCDBRunRange lastRunRange(-1,-1); // highest runRange found
9e1ceb13 327 Int_t lastVersion=0; // highest version found
328
329 TGridResult *res = gGrid->Ls(dirName);
330
331 //loop on the files in the directory, look for highest version
332 for(int i=0; i < res->GetEntries(); i++){
62032124 333 filename=res->GetFileNamePath(i);
334 if (!FilenameToId(filename, anId)) continue;
335 if (anId.GetAliCDBRunRange().Overlaps(id.GetAliCDBRunRange()) && anId.GetVersion() > lastVersion) {
336 lastVersion = anId.GetVersion();
337 lastRunRange = anId.GetAliCDBRunRange();
9e1ceb13 338 }
339
340 }
4005d0b5 341 delete res;
342
be1b1008 343 // GRP entries with explicitly set version escape default incremental versioning
344 if(id.GetPath().Contains("GRP") && id.HasVersion() && lastVersion!=0)
345 {
346 AliDebug(5,Form("Entry %s won't be put in the destination OCDB", id.ToString().Data()));
347 return kFALSE;
1ac347b2 348 }
05ca3c19 349
350 id.SetVersion(lastVersion + 1);
4005d0b5 351 id.SetSubVersion(0);
9e1ceb13 352
353 TString lastStorage = id.GetLastStorage();
354 if(lastStorage.Contains(TString("new"), TString::kIgnoreCase) && id.GetVersion() > 1 ){
84090f85 355 AliDebug(2, Form("A NEW object is being stored with version %d",
9e1ceb13 356 id.GetVersion()));
84090f85 357 AliDebug(2, Form("and it will hide previously stored object with version %d!",
9e1ceb13 358 id.GetVersion()-1));
359 }
360
4005d0b5 361 if(!lastRunRange.IsAnyRange() && !(lastRunRange.IsEqual(&id.GetAliCDBRunRange())))
9e1ceb13 362 AliWarning(Form("Run range modified w.r.t. previous version (Run%d_%d_v%d)",
363 lastRunRange.GetFirstRun(), lastRunRange.GetLastRun(), id.GetVersion()));
4005d0b5 364
9e1ceb13 365 return kTRUE;
366}
367
368//_____________________________________________________________________________
6dc56e97 369AliCDBId* AliCDBGrid::GetId(const TObjArray& validFileIds, const AliCDBId& query) {
62032124 370// look for the Id that matches query's requests (highest or exact version)
9e1ceb13 371
98873e41 372 if(validFileIds.GetEntriesFast() < 1)
62032124 373 return NULL;
9e1ceb13 374
62032124 375 TIter iter(&validFileIds);
9e1ceb13 376
62032124 377 AliCDBId *anIdPtr=0;
378 AliCDBId* result=0;
9e1ceb13 379
62032124 380 while((anIdPtr = dynamic_cast<AliCDBId*> (iter.Next()))){
62032124 381 if(anIdPtr->GetPath() != query.GetPath()) continue;
382
383 //if(!CheckVersion(query, anIdPtr, result)) return NULL;
9e1ceb13 384
385 if (!query.HasVersion()){ // look for highest version
62032124 386 if(result && result->GetVersion() > anIdPtr->GetVersion()) continue;
387 if(result && result->GetVersion() == anIdPtr->GetVersion()) {
4667c116 388 AliError(Form("More than one object valid for run %d, version %d!",
62032124 389 query.GetFirstRun(), anIdPtr->GetVersion()));
390 return NULL;
9e1ceb13 391 }
62032124 392 result = anIdPtr;
9e1ceb13 393 } else { // look for specified version
62032124 394 if(query.GetVersion() != anIdPtr->GetVersion()) continue;
395 if(result && result->GetVersion() == anIdPtr->GetVersion()){
4667c116 396 AliError(Form("More than one object valid for run %d, version %d!",
62032124 397 query.GetFirstRun(), anIdPtr->GetVersion()));
398 return NULL;
9e1ceb13 399 }
62032124 400 result = anIdPtr;
9e1ceb13 401 }
62032124 402
403 }
c75fabed 404
405 if (!result) return NULL;
62032124 406
4667c116 407 return dynamic_cast<AliCDBId*> (result->Clone());
62032124 408}
409
9e1ceb13 410//_____________________________________________________________________________
4667c116 411AliCDBId* AliCDBGrid::GetEntryId(const AliCDBId& queryId) {
412// get AliCDBId from the database
413// User must delete returned object
9e1ceb13 414
62032124 415 AliCDBId* dataId=0;
416
417 AliCDBId selectedId(queryId);
418 if (!selectedId.HasVersion()) {
9e1ceb13 419 // if version is not specified, first check the selection criteria list
4b5e0dce 420 GetSelection(&selectedId);
62032124 421 }
422
6dc56e97 423 TObjArray validFileIds;
62032124 424 validFileIds.SetOwner(1);
425
426 // look for file matching query requests (path, runRange, version)
9e96e52d 427 if(selectedId.GetFirstRun() == fRun && fPathFilter.Comprises(selectedId.GetAliCDBPath()) &&
428 fVersion == selectedId.GetVersion() && !fMetaDataFilter){
62032124 429 // look into list of valid files previously loaded with AliCDBStorage::FillValidFileIds()
98873e41 430 AliDebug(2, Form("List of files valid for run %d was loaded. Looking there for fileids valid for path %s!",
62032124 431 selectedId.GetFirstRun(), selectedId.GetPath().Data()));
432 dataId = GetId(fValidFileIds, selectedId);
433
9e1ceb13 434 } else {
62032124 435 // List of files valid for reqested run was not loaded. Looking directly into CDB
9e96e52d 436 AliDebug(2, Form("List of files valid for run %d and version %d was not loaded. Looking directly into CDB for fileids valid for path %s!",
437 selectedId.GetFirstRun(), selectedId.GetVersion(), selectedId.GetPath().Data()));
62032124 438
439 TString filter;
6dc56e97 440 MakeQueryFilter(selectedId.GetFirstRun(), selectedId.GetLastRun(), 0, filter);
62032124 441
ca0602ee 442 TString pattern = ".root";
443 TString optionQuery = "-y -m";
0e10c8fd 444 if(selectedId.GetVersion() >= 0) {
61f0b0b6 445 pattern.Prepend(Form("_v%d_s0",selectedId.GetVersion()));
0e10c8fd 446 optionQuery = "";
447 }
448
ca0602ee 449 TString folderCopy(Form("%s%s/Run",fDBFolder.Data(),selectedId.GetPath().Data()));
6dc56e97 450
ca0602ee 451 if (optionQuery.Contains("-y")){
0e10c8fd 452 AliInfo("Only latest version will be returned");
453 }
454
ca0602ee 455 AliDebug(2,Form("** fDBFolder = %s, pattern = %s, filter = %s",folderCopy.Data(), pattern.Data(), filter.Data()));
456 TGridResult *res = gGrid->Query(folderCopy, pattern, filter, optionQuery.Data());
ecae96f8 457 if (res) {
458 AliCDBId validFileId;
459 for(int i=0; i<res->GetEntries(); i++){
460 TString filename = res->GetKey(i, "lfn");
461 if(filename == "") continue;
462 if(FilenameToId(filename, validFileId))
463 validFileIds.AddLast(validFileId.Clone());
464 }
465 delete res;
466 }
62032124 467 dataId = GetId(validFileIds, selectedId);
9e1ceb13 468 }
469
4667c116 470 return dataId;
471}
472
473//_____________________________________________________________________________
474AliCDBEntry* AliCDBGrid::GetEntry(const AliCDBId& queryId) {
475// get AliCDBEntry from the database
476
477 AliCDBId* dataId = GetEntryId(queryId);
478
62032124 479 if (!dataId) return NULL;
9e1ceb13 480
481 TString filename;
62032124 482 if (!IdToFilename(*dataId, filename)) {
024cf675 483 AliDebug(2,Form("Bad data ID encountered! Subnormal error!"));
4667c116 484 delete dataId;
9e1ceb13 485 return NULL;
486 }
487
62032124 488 AliCDBEntry* anEntry = GetEntryFromFile(filename, dataId);
489
4667c116 490 delete dataId;
62032124 491 return anEntry;
492}
493
494//_____________________________________________________________________________
8e245d15 495AliCDBEntry* AliCDBGrid::GetEntryFromFile(TString& filename, AliCDBId* dataId){
62032124 496// Get AliCBEntry object from file "filename"
9e1ceb13 497
024cf675 498 AliDebug(2,Form("Opening file: %s",filename.Data()));
62032124 499
500 filename.Prepend("/alien");
c75fabed 501
502 // if option="CACHEREAD" TFile will use the local caching facility!
503 TString option="READ";
504 if(fCacheFolder != ""){
505
506 // Check if local cache folder was changed in the meanwhile
507 TString origCache(TFile::GetCacheFileDir());
508 if(fCacheFolder != origCache) {
509 AliWarning(Form("Local cache folder has been overwritten!! fCacheFolder = %s origCache = %s",
510 fCacheFolder.Data(), origCache.Data()));
511 TFile::SetCacheFileDir(fCacheFolder.Data(), fOperateDisconnected);
512 TFile::ShrinkCacheFileDir(fCacheSize, fCleanupInterval);
513 }
514
515 option.Prepend("CACHE");
516 }
517
518 AliDebug(2, Form("Option: %s", option.Data()));
519
520 TFile *file = TFile::Open(filename, option);
9e1ceb13 521 if (!file) {
024cf675 522 AliDebug(2,Form("Can't open file <%s>!", filename.Data()));
9e1ceb13 523 return NULL;
524 }
525
526 // get the only AliCDBEntry object from the file
527 // the object in the file is an AliCDBEntry entry named "AliCDBEntry"
528
62032124 529 AliCDBEntry* anEntry = dynamic_cast<AliCDBEntry*> (file->Get("AliCDBEntry"));
9e1ceb13 530
62032124 531 if (!anEntry) {
532 AliDebug(2,Form("Bad storage data: file does not contain an AliCDBEntry object!"));
533 file->Close();
9e1ceb13 534 return NULL;
535 }
536
9e1ceb13 537 // The object's Id is not reset during storage
538 // If object's Id runRange or version do not match with filename,
539 // it means that someone renamed file by hand. In this case a warning msg is issued.
62032124 540
541 if(anEntry){
542 AliCDBId entryId = anEntry->GetId();
4005d0b5 543 Int_t tmpSubVersion = dataId->GetSubVersion();
544 dataId->SetSubVersion(entryId.GetSubVersion()); // otherwise filename and id may mismatch
8e245d15 545 if(!entryId.IsEqual(dataId)){
b8ec52f6 546 AliWarning(Form("Mismatch between file name and object's Id!"));
547 AliWarning(Form("File name: %s", dataId->ToString().Data()));
548 AliWarning(Form("Object's Id: %s", entryId.ToString().Data()));
62032124 549 }
4005d0b5 550 dataId->SetSubVersion(tmpSubVersion);
9e1ceb13 551 }
552
62032124 553 anEntry->SetLastStorage("grid");
554
145f45a2 555 // Check whether entry contains a TTree. In case load the tree in memory!
556 LoadTreeFromFile(anEntry);
557
9e1ceb13 558 // close file, return retieved entry
559 file->Close(); delete file; file=0;
62032124 560
561 return anEntry;
9e1ceb13 562}
563
564//_____________________________________________________________________________
62032124 565TList* AliCDBGrid::GetEntries(const AliCDBId& queryId) {
9e1ceb13 566// multiple request (AliCDBStorage::GetAll)
567
62032124 568 TList* result = new TList();
569 result->SetOwner();
9e1ceb13 570
6dc56e97 571 TObjArray validFileIds;
62032124 572 validFileIds.SetOwner(1);
9e1ceb13 573
62032124 574 Bool_t alreadyLoaded = kFALSE;
9e1ceb13 575
62032124 576 // look for file matching query requests (path, runRange)
577 if(queryId.GetFirstRun() == fRun &&
c3a7b59a 578 fPathFilter.Comprises(queryId.GetAliCDBPath()) && fVersion < 0 && !fMetaDataFilter){
62032124 579 // look into list of valid files previously loaded with AliCDBStorage::FillValidFileIds()
580 AliDebug(2,Form("List of files valid for run %d and for path %s was loaded. Looking there!",
581 queryId.GetFirstRun(), queryId.GetPath().Data()));
9e1ceb13 582
62032124 583 alreadyLoaded = kTRUE;
9e1ceb13 584
62032124 585 } else {
586 // List of files valid for reqested run was not loaded. Looking directly into CDB
587 AliDebug(2,Form("List of files valid for run %d and for path %s was not loaded. Looking directly into CDB!",
588 queryId.GetFirstRun(), queryId.GetPath().Data()));
589
590 TString filter;
6dc56e97 591 MakeQueryFilter(queryId.GetFirstRun(), queryId.GetLastRun(), 0, filter);
592
5ef33267 593 TString path = queryId.GetPath();
6dc56e97 594
5ef33267 595 TString pattern = "Run*.root";
596 TString optionQuery = "-y";
0e10c8fd 597
5ef33267 598 TString addFolder = "";
599 if (!path.Contains("*")){
600 if (!path.BeginsWith("/")) addFolder += "/";
601 addFolder += path;
602 }
603 else{
604 if (path.BeginsWith("/")) path.Remove(0,1);
605 if (path.EndsWith("/")) path.Remove(path.Length()-1,1);
606 TObjArray* tokenArr = path.Tokenize("/");
607 if (tokenArr->GetEntries() != 3) {
608 AliError("Not a 3 level path! Keeping old query...");
609 pattern.Prepend(path+"/");
610 }
611 else{
612 TString str0 = ((TObjString*)tokenArr->At(0))->String();
613 TString str1 = ((TObjString*)tokenArr->At(1))->String();
614 TString str2 = ((TObjString*)tokenArr->At(2))->String();
615 if (str0 != "*" && str1 != "*" && str2 == "*"){
616 // e.g. "ITS/Calib/*"
617 addFolder = "/"+str0+"/"+str1;
618 }
619 else if (str0 != "*" && str1 == "*" && str2 == "*"){
620 // e.g. "ITS/*/*"
621 addFolder = "/"+str0;
622 }
623 else if (str0 == "*" && str1 == "*" && str2 == "*"){
624 // e.g. "*/*/*"
625 // do nothing: addFolder is already an empty string;
626 }
627 else{
628 // e.g. "ITS/*/RecoParam"
629 pattern.Prepend(path+"/");
630 }
631 }
632 delete tokenArr; tokenArr=0;
0e10c8fd 633 }
634
5ef33267 635 TString folderCopy(Form("%s%s",fDBFolder.Data(),addFolder.Data()));
636
637 AliDebug(2,Form("fDBFolder = %s, pattern = %s, filter = %s",folderCopy.Data(), pattern.Data(), filter.Data()));
638
639 TGridResult *res = gGrid->Query(folderCopy, pattern, filter, optionQuery.Data());
640
641 if (!res) {
642 AliError("Grid query failed");
643 return 0;
644 }
62032124 645
62032124 646 AliCDBId validFileId;
647 for(int i=0; i<res->GetEntries(); i++){
648 TString filename = res->GetKey(i, "lfn");
c3a7b59a 649 if(filename == "") continue;
62032124 650 if(FilenameToId(filename, validFileId))
651 validFileIds.AddLast(validFileId.Clone());
652 }
653 delete res;
9e1ceb13 654 }
655
62032124 656 TIter *iter=0;
657 if(alreadyLoaded){
658 iter = new TIter(&fValidFileIds);
659 } else {
660 iter = new TIter(&validFileIds);
661 }
9e1ceb13 662
6dc56e97 663 TObjArray selectedIds;
62032124 664 selectedIds.SetOwner(1);
665
666 // loop on list of valid Ids to select the right version to get.
667 // According to query and to the selection criteria list, version can be the highest or exact
668 AliCDBPath pathCopy;
669 AliCDBId* anIdPtr=0;
670 AliCDBId* dataId=0;
671 AliCDBPath queryPath = queryId.GetAliCDBPath();
672 while((anIdPtr = dynamic_cast<AliCDBId*> (iter->Next()))){
673 AliCDBPath thisCDBPath = anIdPtr->GetAliCDBPath();
674 if(!(queryPath.Comprises(thisCDBPath)) || pathCopy.GetPath() == thisCDBPath.GetPath()) continue;
675 pathCopy = thisCDBPath;
676
677 // check the selection criteria list for this query
678 AliCDBId thisId(*anIdPtr);
679 thisId.SetVersion(queryId.GetVersion());
680 if(!thisId.HasVersion()) GetSelection(&thisId);
681
682 if(alreadyLoaded){
683 dataId = GetId(fValidFileIds, thisId);
684 } else {
685 dataId = GetId(validFileIds, thisId);
9e1ceb13 686 }
4667c116 687 if(dataId) selectedIds.Add(dataId);
9e1ceb13 688 }
9e1ceb13 689
62032124 690 delete iter; iter=0;
9e1ceb13 691
62032124 692 // selectedIds contains the Ids of the files matching all requests of query!
693 // All the objects are now ready to be retrieved
694 iter = new TIter(&selectedIds);
695 while((anIdPtr = dynamic_cast<AliCDBId*> (iter->Next()))){
696 TString filename;
697 if (!IdToFilename(*anIdPtr, filename)) {
698 AliDebug(2,Form("Bad data ID encountered! Subnormal error!"));
699 continue;
700 }
9e1ceb13 701
62032124 702 AliCDBEntry* anEntry = GetEntryFromFile(filename, anIdPtr);
9e1ceb13 703
62032124 704 if(anEntry) result->Add(anEntry);
9e1ceb13 705
62032124 706 }
707 delete iter; iter=0;
708
709 return result;
9e1ceb13 710}
711
712//_____________________________________________________________________________
06eeadea 713Bool_t AliCDBGrid::PutEntry(AliCDBEntry* entry, const char* mirrors) {
703128af 714 // put an AliCDBEntry object into the database
62032124 715
9e1ceb13 716 AliCDBId& id = entry->GetId();
717
718 // set version for the entry to be stored
62032124 719 if (!PrepareId(id)) return kFALSE;
9e1ceb13 720
721 // build filename from entry's id
722 TString filename;
62032124 723 if (!IdToFilename(id, filename)) {
9e1ceb13 724 AliError("Bad ID encountered! Subnormal error!");
725 return kFALSE;
62032124 726 }
727
728 TString folderToTag = Form("%s%s",
703128af 729 fDBFolder.Data(),
730 id.GetPath().Data());
62032124 731
9e1ceb13 732 TDirectory* saveDir = gDirectory;
733
fb33abde 734 TString fullFilename = Form("/alien%s", filename.Data());
06eeadea 735 TString seMirrors(mirrors);
c5941d67 736 if(seMirrors.IsNull() || seMirrors.IsWhitespace()) seMirrors=GetMirrorSEs();
62032124 737 // specify SE to filename
c5941d67 738 // if a list of SEs was passed to this method or set via SetMirrorSEs, set the first as SE for opening the file.
739 // The other SEs will be used in cascade in case of failure in opening the file.
740 // The remaining SEs will be used to create replicas.
06eeadea 741 TObjArray *arraySEs = seMirrors.Tokenize(',');
742 Int_t nSEs = arraySEs->GetEntries();
743 Int_t remainingSEs = 1;
744 if(nSEs == 0){
703128af 745 if (fSE != "default") fullFilename += Form("?se=%s",fSE.Data());
06eeadea 746 }else{
703128af 747 remainingSEs = nSEs;
06eeadea 748 }
62032124 749
9e1ceb13 750 // open file
89004e35 751 TFile *file=0;
5bfe3965 752 AliDebug(2, Form("fNretry = %d, fInitRetrySeconds = %d",fNretry,fInitRetrySeconds));
06eeadea 753 TString targetSE("");
703128af 754
755 Bool_t result = kFALSE;
756 Bool_t reOpenResult = kFALSE;
757 Int_t reOpenAttempts=0;
758 while( !reOpenResult && reOpenAttempts<2){ //loop to check the file after closing it, to catch the unlikely but possible case when the file
759 // is cleaned up by alien just before closing as a consequence of a network disconnection while writing
760
761 while( !file && remainingSEs>0){
762 if(nSEs!=0){
763 TObjString *target = (TObjString*) arraySEs->At(nSEs-remainingSEs);
764 targetSE=target->String();
765 if ( !(targetSE.BeginsWith("ALICE::") && targetSE.CountChar(':')==4) ) {
766 AliError(Form("\"%s\" is an invalid storage element identifier.",targetSE.Data()));
767 continue;
768 }
769 if(fullFilename.Contains('?')) fullFilename.Remove(fullFilename.Last('?'));
770 fullFilename += Form("?se=%s",targetSE.Data());
771 }
772 Int_t remainingAttempts=fNretry;
773 Int_t nsleep = fInitRetrySeconds; // number of seconds between attempts. We let it increase exponentially
774 AliDebug(2, Form("Uploading file into SE #%d: %s",nSEs-remainingSEs+1,targetSE.Data()));
775 while(remainingAttempts > 0) {
776 AliDebug(2, Form("Uploading file into OCDB at %s - Attempt #%d",targetSE.Data(),fNretry-remainingAttempts+1));
777 remainingAttempts--;
778 file = TFile::Open(fullFilename,"CREATE");
779 if(!file || !file->IsWritable()){
780 if(file) file->Close(); delete file; file=0; // file is not writable
781 TString message(TString::Format("Attempt %d failed.",fNretry-remainingAttempts));
782 if(remainingAttempts>0) {
783 message += " Sleeping for "; message += nsleep; message += " seconds";
784 }else{
785 if(remainingSEs>0) message += " Trying to upload at next SE";
786 }
787 AliDebug(2, message.Data());
788 if(remainingAttempts>0) sleep(nsleep);
789 }else{
703128af 790 remainingAttempts=0;
791 }
792 nsleep*=fInitRetrySeconds;
793 }
794 remainingSEs--;
06eeadea 795 }
703128af 796 if(!file){
797 AliError(Form("All %d attempts have failed on all %d SEs. Returning...",fNretry,nSEs));
798 return kFALSE;
89004e35 799 }
9e1ceb13 800
703128af 801 file->cd();
145f45a2 802
703128af 803 //SetTreeToFile(entry, file);
804 entry->SetVersion(id.GetVersion());
9e1ceb13 805
703128af 806 // write object (key name: "AliCDBEntry")
807 result = (file->WriteTObject(entry, "AliCDBEntry") != 0);
808 if (!result) AliError(Form("Can't write entry to file <%s>!", filename.Data()));
809 file->Close();
9e1ceb13 810
703128af 811 if(result)
812 {
813 AliDebug(2, Form("Reopening file %s for checking its correctness",fullFilename.Data()));
814 TFile* ffile = TFile::Open(fullFilename.Data(),"READ");
815 if(!ffile){
816 reOpenResult = kFALSE;
817 AliInfo(Form("The file %s was closed successfully but cannot be reopened. Trying now to regenerate it (regeneration attempt number %d)",
818 fullFilename.Data(),++reOpenAttempts));
819 delete file; file=0;
820 AliDebug(2, Form("Removing file %s", filename.Data()));
821 if(!gGrid->Rm(filename.Data()))
822 AliError("Can't delete file!");
823 remainingSEs++;
824 }else{
825 reOpenResult = kTRUE;
826 ffile->Close();
827 }
828 delete ffile; ffile=0;
829 }
830 }
9e1ceb13 831
832 if (saveDir) saveDir->cd(); else gROOT->cd();
703128af 833 delete file; file=0;
834
835 if(result && reOpenResult) {
c3a7b59a 836
c3a7b59a 837 if(!TagFileId(filename, &id)){
838 AliInfo(Form("CDB tagging failed. Deleting file %s!",filename.Data()));
839 if(!gGrid->Rm(filename.Data()))
840 AliError("Can't delete file!");
841 return kFALSE;
842 }
843
844 TagFileMetaData(filename, entry->GetMetaData());
703128af 845 }else{
846 AliError("The file could not be opend or the object could not be written");
847 if(!gGrid->Rm(filename.Data()))
848 AliError("Can't delete file!");
849 return kFALSE;
9e1ceb13 850 }
62032124 851
c75fabed 852 AliInfo(Form("CDB object stored into file %s", filename.Data()));
06eeadea 853 if(nSEs==0)
703128af 854 AliInfo(Form("Storage Element: %s", fSE.Data()));
06eeadea 855 else
703128af 856 AliInfo(Form("Storage Element: %s", targetSE.Data()));
06eeadea 857
703128af 858 //In case of other SEs specified by the user, mirror the file to the remaining SEs
a4a99395 859 for(Int_t i=0; i<nSEs; i++){
703128af 860 if(i==nSEs-remainingSEs-1) continue; // skip mirroring to the SE where the file was saved
861 TString mirrorCmd("mirror ");
862 mirrorCmd += filename;
863 mirrorCmd += " ";
864 TObjString *target = (TObjString*) arraySEs->At(i);
865 TString mirrorSE(target->String());
866 mirrorCmd += mirrorSE;
867 AliDebug(5,Form("mirror command: \"%s\"",mirrorCmd.Data()));
868 AliInfo(Form("Mirroring to storage element: %s", mirrorSE.Data()));
869 gGrid->Command(mirrorCmd.Data());
06eeadea 870 }
871
703128af 872 return kTRUE;
62032124 873}
c5941d67 874
62032124 875//_____________________________________________________________________________
876Bool_t AliCDBGrid::AddTag(TString& folderToTag, const char* tagname){
877// add "tagname" tag (CDB or CDB_MD) to folder where object will be stored
878
879 Bool_t result = kTRUE;
880 AliDebug(2, Form("adding %s tag to folder %s", tagname, folderToTag.Data()));
881 TString addTag = Form("addTag %s %s", folderToTag.Data(), tagname);
882 TGridResult *gridres = gGrid->Command(addTag.Data());
883 const char* resCode = gridres->GetKey(0,"__result__"); // '1' if success
884 if(resCode[0] != '1') {
885 AliError(Form("Couldn't add %s tags to folder %s !",
886 tagname, folderToTag.Data()));
887 result = kFALSE;
888 }
889 delete gridres;
9e1ceb13 890 return result;
891}
892
62032124 893//_____________________________________________________________________________
c3a7b59a 894Bool_t AliCDBGrid::TagFileId(TString& filename, const AliCDBId* id){
62032124 895// tag stored object in CDB table using object Id's parameters
896
0e10c8fd 897
898 TString dirname(filename);
899 Int_t dirNumber = gGrid->Mkdir(dirname.Remove(dirname.Last('/')),"-d");
900
f430a9ce 901 TString addTagValue1 = Form("addTagValue %s CDB ", filename.Data());
902 TString addTagValue2 = Form("first_run=%d last_run=%d version=%d ",
62032124 903 id->GetFirstRun(),
904 id->GetLastRun(),
905 id->GetVersion());
0e10c8fd 906 TString addTagValue3 = Form("path_level_0=\"%s\" path_level_1=\"%s\" path_level_2=\"%s\" ",
4667c116 907 id->GetPathLevel(0).Data(),
908 id->GetPathLevel(1).Data(),
909 id->GetPathLevel(2).Data());
0e10c8fd 910 //TString addTagValue4 = Form("version_path=\"%s\" dir_number=%d",Form("%d_%s",id->GetVersion(),filename.Data()),dirNumber);
911 TString addTagValue4 = Form("version_path=\"%09d%s\" dir_number=%d",id->GetVersion(),filename.Data(),dirNumber);
912 TString addTagValue = Form("%s%s%s%s",
f430a9ce 913 addTagValue1.Data(),
914 addTagValue2.Data(),
0e10c8fd 915 addTagValue3.Data(),
916 addTagValue4.Data());
62032124 917
c3a7b59a 918 Bool_t result = kFALSE;
62032124 919 AliDebug(2, Form("Tagging file. Tag command: %s", addTagValue.Data()));
920 TGridResult* res = gGrid->Command(addTagValue.Data());
921 const char* resCode = res->GetKey(0,"__result__"); // '1' if success
922 if(resCode[0] != '1') {
c3a7b59a 923 AliError(Form("Couldn't add CDB tag value to file %s !",
62032124 924 filename.Data()));
c3a7b59a 925 result = kFALSE;
62032124 926 } else {
c75fabed 927 AliDebug(2, "Object successfully tagged.");
c3a7b59a 928 result = kTRUE;
62032124 929 }
930 delete res;
c3a7b59a 931 return result;
62032124 932
933}
934
4667c116 935//_____________________________________________________________________________
936Bool_t AliCDBGrid::TagShortLived(TString& filename, Bool_t value){
937// tag folder with ShortLived tag
938
939 TString addTagValue = Form("addTagValue %s ShortLived_try value=%d", filename.Data(), value);
940
941 Bool_t result = kFALSE;
942 AliDebug(2, Form("Tagging file. Tag command: %s", addTagValue.Data()));
943 TGridResult* res = gGrid->Command(addTagValue.Data());
944 const char* resCode = res->GetKey(0,"__result__"); // '1' if success
945 if(resCode[0] != '1') {
946 AliError(Form("Couldn't add ShortLived tag value to file %s !", filename.Data()));
947 result = kFALSE;
948 } else {
c75fabed 949 AliDebug(2,"Object successfully tagged.");
4667c116 950 result = kTRUE;
951 }
952 delete res;
953 return result;
954
955}
956
62032124 957//_____________________________________________________________________________
c3a7b59a 958Bool_t AliCDBGrid::TagFileMetaData(TString& filename, const AliCDBMetaData* md){
62032124 959// tag stored object in CDB table using object Id's parameters
960
f430a9ce 961 TString addTagValue1 = Form("addTagValue %s CDB_MD ", filename.Data());
962 TString addTagValue2 = Form("object_classname=\"%s\" responsible=\"%s\" beam_period=%d ",
62032124 963 md->GetObjectClassName(),
964 md->GetResponsible(),
965 md->GetBeamPeriod());
f430a9ce 966 TString addTagValue3 = Form("aliroot_version=\"%s\" comment=\"%s\"",
62032124 967 md->GetAliRootVersion(),
968 md->GetComment());
969 TString addTagValue = Form("%s%s%s",
f430a9ce 970 addTagValue1.Data(),
971 addTagValue2.Data(),
972 addTagValue3.Data());
62032124 973
c3a7b59a 974 Bool_t result = kFALSE;
62032124 975 AliDebug(2, Form("Tagging file. Tag command: %s", addTagValue.Data()));
976 TGridResult* res = gGrid->Command(addTagValue.Data());
977 const char* resCode = res->GetKey(0,"__result__"); // '1' if success
978 if(resCode[0] != '1') {
979 AliWarning(Form("Couldn't add CDB_MD tag value to file %s !",
980 filename.Data()));
c3a7b59a 981 result = kFALSE;
62032124 982 } else {
c75fabed 983 AliDebug(2,"Object successfully tagged.");
c3a7b59a 984 result = kTRUE;
62032124 985 }
c3a7b59a 986 return result;
62032124 987}
988
b05400be 989//_____________________________________________________________________________
990TList* AliCDBGrid::GetIdListFromFile(const char* fileName){
6dc56e97 991
b05400be 992 TString turl(fileName);
993 turl.Prepend("/alien" + fDBFolder);
62032124 994 turl += "?se="; turl += fSE.Data();
b05400be 995 TFile *file = TFile::Open(turl);
996 if (!file) {
997 AliError(Form("Can't open selection file <%s>!", turl.Data()));
998 return NULL;
999 }
1000
1001 TList *list = new TList();
1002 list->SetOwner();
1003 int i=0;
1004 TString keycycle;
62032124 1005
b05400be 1006 AliCDBId *id;
1007 while(1){
1008 i++;
1009 keycycle = "AliCDBId;";
1010 keycycle+=i;
1011
1012 id = (AliCDBId*) file->Get(keycycle);
1013 if(!id) break;
1014 list->AddFirst(id);
1015 }
1016 file->Close(); delete file; file=0;
1017
1018 return list;
1019
1020
1021}
1022
1023//_____________________________________________________________________________
1024Bool_t AliCDBGrid::Contains(const char* path) const{
1025// check for path in storage's DBFolder
1026
1027 TString initDir(gGrid->Pwd(0));
1028 TString dirName(fDBFolder);
1029 dirName += path; // dirName = fDBFolder/path
1030 Bool_t result=kFALSE;
1031 if (gGrid->Cd(dirName,0)) result=kTRUE;
1032 gGrid->Cd(initDir.Data(),0);
1033 return result;
1034}
1035
62032124 1036//_____________________________________________________________________________
1037void AliCDBGrid::QueryValidFiles()
1038{
1039// Query the CDB for files valid for AliCDBStorage::fRun
1040// fills list fValidFileIds with AliCDBId objects created from file name
1041
1042 TString filter;
6dc56e97 1043 MakeQueryFilter(fRun, fRun, fMetaDataFilter, filter);
1044
61f0b0b6 1045 TString path = fPathFilter.GetPath();
1046
1047 TString pattern = "Run*";
0e10c8fd 1048 TString optionQuery = "-y";
1049 if(fVersion >= 0) {
61f0b0b6 1050 pattern += Form("_v%d_s0", fVersion);
0e10c8fd 1051 optionQuery = "";
1052 }
6dc56e97 1053 pattern += ".root";
1054 AliDebug(2,Form("pattern: %s", pattern.Data()));
1055
61f0b0b6 1056 TString addFolder = "";
1057 if (!path.Contains("*")){
1058 if (!path.BeginsWith("/")) addFolder += "/";
1059 addFolder += path;
1060 }
1061 else{
1062 if (path.BeginsWith("/")) path.Remove(0,1);
1063 if (path.EndsWith("/")) path.Remove(path.Length()-1,1);
1064 TObjArray* tokenArr = path.Tokenize("/");
1065 if (tokenArr->GetEntries() != 3) {
1066 AliError("Not a 3 level path! Keeping old query...");
5ef33267 1067 pattern.Prepend(path+"/");
61f0b0b6 1068 }
1069 else{
1070 TString str0 = ((TObjString*)tokenArr->At(0))->String();
1071 TString str1 = ((TObjString*)tokenArr->At(1))->String();
1072 TString str2 = ((TObjString*)tokenArr->At(2))->String();
1073 if (str0 != "*" && str1 != "*" && str2 == "*"){
1074 // e.g. "ITS/Calib/*"
1075 addFolder = "/"+str0+"/"+str1;
1076 }
1077 else if (str0 != "*" && str1 == "*" && str2 == "*"){
1078 // e.g. "ITS/*/*"
1079 addFolder = "/"+str0;
1080 }
1081 else if (str0 == "*" && str1 == "*" && str2 == "*"){
1082 // e.g. "*/*/*"
1083 // do nothing: addFolder is already an empty string;
1084 }
1085 else{
1086 // e.g. "ITS/*/RecoParam"
5ef33267 1087 pattern.Prepend(path+"/");
61f0b0b6 1088 }
1089 }
1090 delete tokenArr; tokenArr=0;
1091 }
1092
1093 TString folderCopy(Form("%s%s",fDBFolder.Data(),addFolder.Data()));
1094
5ef33267 1095 AliDebug(2,Form("fDBFolder = %s, pattern = %s, filter = %s",folderCopy.Data(), pattern.Data(), filter.Data()));
0e10c8fd 1096
1097 if (optionQuery == "-y"){
1098 AliInfo("Only latest version will be returned");
1099 }
1100
5ef33267 1101 TGridResult *res = gGrid->Query(folderCopy, pattern, filter, optionQuery.Data());
62032124 1102
b7e6419d 1103 if (!res) {
1104 AliError("Grid query failed");
1105 return;
1106 }
1107
b68a24b3 1108 TIter next(res);
1109 TMap *map;
1110 while ((map = (TMap*)next())) {
1111 TObjString *entry;
1112 if ((entry = (TObjString *) ((TMap *)map)->GetValue("lfn"))) {
1113 TString& filename = entry->String();
1114 if(filename.IsNull()) continue;
1115 AliDebug(2,Form("Found valid file: %s", filename.Data()));
1116 AliCDBId *validFileId = new AliCDBId;
1117 Bool_t result = FilenameToId(filename, *validFileId);
1118 if(result) {
1119 fValidFileIds.AddLast(validFileId);
1120 }
1121 else {
1122 delete validFileId;
1123 }
1124 }
62032124 1125 }
1126 delete res;
1127
1128}
1129
1130//_____________________________________________________________________________
c3a7b59a 1131void AliCDBGrid::MakeQueryFilter(Int_t firstRun, Int_t lastRun,
62032124 1132 const AliCDBMetaData* md, TString& result) const
1133{
1134// create filter for file query
1135
c3a7b59a 1136 result = Form("CDB:first_run<=%d and CDB:last_run>=%d", firstRun, lastRun);
62032124 1137
6dc56e97 1138// if(version >= 0) {
1139// result += Form(" and CDB:version=%d", version);
1140// }
1141// if(pathFilter.GetLevel0() != "*") {
1142// result += Form(" and CDB:path_level_0=\"%s\"", pathFilter.GetLevel0().Data());
1143// }
1144// if(pathFilter.GetLevel1() != "*") {
1145// result += Form(" and CDB:path_level_1=\"%s\"", pathFilter.GetLevel1().Data());
1146// }
1147// if(pathFilter.GetLevel2() != "*") {
1148// result += Form(" and CDB:path_level_2=\"%s\"", pathFilter.GetLevel2().Data());
1149// }
62032124 1150
1151 if(md){
1152 if(md->GetObjectClassName()[0] != '\0') {
1153 result += Form(" and CDB_MD:object_classname=\"%s\"", md->GetObjectClassName());
1154 }
1155 if(md->GetResponsible()[0] != '\0') {
1156 result += Form(" and CDB_MD:responsible=\"%s\"", md->GetResponsible());
1157 }
1158 if(md->GetBeamPeriod() != 0) {
1159 result += Form(" and CDB_MD:beam_period=%d", md->GetBeamPeriod());
1160 }
1161 if(md->GetAliRootVersion()[0] != '\0') {
1162 result += Form(" and CDB_MD:aliroot_version=\"%s\"", md->GetAliRootVersion());
1163 }
1164 if(md->GetComment()[0] != '\0') {
1165 result += Form(" and CDB_MD:comment=\"%s\"", md->GetComment());
1166 }
1167 }
1168 AliDebug(2, Form("filter: %s",result.Data()));
1169
1170}
1171
c3a7b59a 1172//_____________________________________________________________________________
1173Int_t AliCDBGrid::GetLatestVersion(const char* path, Int_t run){
1174// get last version found in the database valid for run and path
1175
c3a7b59a 1176 AliCDBPath aCDBPath(path);
1177 if(!aCDBPath.IsValid() || aCDBPath.IsWildcard()) {
1178 AliError(Form("Invalid path in request: %s", path));
1179 return -1;
1180 }
1181 AliCDBId query(path, run, run, -1, -1);
c3a7b59a 1182
0e10c8fd 1183 AliCDBId* dataId = 0;
c3a7b59a 1184 // look for file matching query requests (path, runRange, version)
6dc56e97 1185 if(run == fRun && fPathFilter.Comprises(aCDBPath) && fVersion < 0){
c3a7b59a 1186 // look into list of valid files previously loaded with AliCDBStorage::FillValidFileIds()
1187 AliDebug(2, Form("List of files valid for run %d and for path %s was loaded. Looking there!",
1188 run, path));
1189 dataId = GetId(fValidFileIds, query);
1190 if (!dataId) return -1;
4667c116 1191 Int_t version = dataId->GetVersion();
1192 delete dataId;
1193 return version;
c3a7b59a 1194
1195 }
1196 // List of files valid for reqested run was not loaded. Looking directly into CDB
1197 AliDebug(2, Form("List of files valid for run %d and for path %s was not loaded. Looking directly into CDB!",
1198 run, path));
1199
4667c116 1200 TObjArray validFileIds;
1201 validFileIds.SetOwner(1);
1202
c3a7b59a 1203 TString filter;
6dc56e97 1204 MakeQueryFilter(run, run, 0, filter);
1205
5ef33267 1206 TString pattern = ".root";
1207
1208 TString folderCopy(Form("%s%s/Run",fDBFolder.Data(),path));
1209
1210 AliDebug(2,Form("** fDBFolder = %s, pattern = %s, filter = %s",folderCopy.Data(), pattern.Data(), filter.Data()));
1211 TGridResult *res = gGrid->Query(folderCopy, pattern, filter, "-y -m");
c3a7b59a 1212
c3a7b59a 1213 AliCDBId validFileId;
0e10c8fd 1214 if (res->GetEntries()>1){
1215 AliWarning("Number of found entries >1, even if option -y was used");
1216 for(int i=0; i<res->GetEntries(); i++){
1217 TString filename = res->GetKey(i, "lfn");
1218 if(filename == "") continue;
1219 if(FilenameToId(filename, validFileId))
c3a7b59a 1220 validFileIds.AddLast(validFileId.Clone());
0e10c8fd 1221 }
1222 dataId = GetId(validFileIds, query);
1223 if (!dataId) return -1;
1224
1225 Int_t version = dataId->GetVersion();
1226 delete dataId;
1227 return version;
1228 }
1229 else if (res->GetEntries()==1){
1230 TString filename = res->GetKey(0, "lfn");
1231 if(filename == "") {
1232 AliError("The only entry found has filename empty");
1233 return -1;
1234 }
1235 if(FilenameToId(filename, validFileId)) return validFileId.GetVersion();
1236 else{
1237 AliError("Impossible to get FileId from filename");
1238 return -1;
1239 }
1240 }
1241 else {
1242 AliError("No entries found");
1243 return -1;
c3a7b59a 1244 }
c3a7b59a 1245
0e10c8fd 1246 delete res;
62ea513e 1247
c3a7b59a 1248
1249}
1250
1251//_____________________________________________________________________________
1252Int_t AliCDBGrid::GetLatestSubVersion(const char* /*path*/, Int_t /*run*/, Int_t /*version*/){
1253// get last subversion found in the database valid for run and path
1254 AliError("Objects in GRID storage have no sub version!");
1255 return -1;
1256}
1257
1258
9e1ceb13 1259/////////////////////////////////////////////////////////////////////////////////////////////////
1260// //
1261// AliCDBGrid factory //
1262// //
1263/////////////////////////////////////////////////////////////////////////////////////////////////
1264
1265ClassImp(AliCDBGridFactory)
1266
1267//_____________________________________________________________________________
1268Bool_t AliCDBGridFactory::Validate(const char* gridString) {
1269// check if the string is valid Grid URI
1270
b05400be 1271 TRegexp gridPattern("^alien://.+$");
9e1ceb13 1272
27cac7e3 1273 return TString(gridString).Contains(gridPattern);
9e1ceb13 1274}
1275
1276//_____________________________________________________________________________
1277AliCDBParam* AliCDBGridFactory::CreateParameter(const char* gridString) {
1278// create AliCDBGridParam class from the URI string
1279
1280 if (!Validate(gridString)) {
1281 return NULL;
1282 }
c75fabed 1283
b05400be 1284 TString buffer(gridString);
c3a7b59a 1285
1286 TString gridUrl = "alien://";
b05400be 1287 TString user = "";
4667c116 1288 TString dbFolder = "";
fb33abde 1289 TString se = "default";
c75fabed 1290 TString cacheFolder = "";
1291 Bool_t operateDisconnected = kTRUE;
1292 Long64_t cacheSize = (UInt_t) 1024*1024*1024; // 1GB
1293 Long_t cleanupInterval = 0;
b05400be 1294
1295 TObjArray *arr = buffer.Tokenize('?');
1296 TIter iter(arr);
62032124 1297 TObjString *str = 0;
4667c116 1298
b05400be 1299 while((str = (TObjString*) iter.Next())){
1300 TString entry(str->String());
1301 Int_t indeq = entry.Index('=');
1302 if(indeq == -1) {
1303 if(entry.BeginsWith("alien://")) { // maybe it's a gridUrl!
1304 gridUrl = entry;
1305 continue;
1306 } else {
1307 AliError(Form("Invalid entry! %s",entry.Data()));
1308 continue;
1309 }
1310 }
1311
1312 TString key = entry(0,indeq);
1313 TString value = entry(indeq+1,entry.Length()-indeq);
1314
1315 if(key.Contains("grid",TString::kIgnoreCase)) {
1316 gridUrl += value;
c75fabed 1317 }
b05400be 1318 else if (key.Contains("user",TString::kIgnoreCase)){
1319 user = value;
1320 }
b05400be 1321 else if (key.Contains("se",TString::kIgnoreCase)){
1322 se = value;
1323 }
d7744171 1324 else if (key.Contains("cacheF",TString::kIgnoreCase)){
c75fabed 1325 cacheFolder = value;
f8703115 1326 if (!cacheFolder.IsNull() && !cacheFolder.EndsWith("/"))
c75fabed 1327 cacheFolder += "/";
1328 }
d7744171 1329 else if (key.Contains("folder",TString::kIgnoreCase)){
1330 dbFolder = value;
1331 }
c75fabed 1332 else if (key.Contains("operateDisc",TString::kIgnoreCase)){
1333 if(value == "kTRUE") {
1334 operateDisconnected = kTRUE;
1335 } else if (value == "kFALSE") {
1336 operateDisconnected = kFALSE;
1337 } else if (value == "0" || value == "1") {
1338 operateDisconnected = (Bool_t) value.Atoi();
1339 } else {
1340 AliError(Form("Invalid entry! %s",entry.Data()));
1341 return NULL;
1342 }
1343 }
d7744171 1344 else if (key.Contains("cacheS",TString::kIgnoreCase)){
c75fabed 1345 if(value.IsDigit()) {
1346 cacheSize = value.Atoi();
1347 } else {
1348 AliError(Form("Invalid entry! %s",entry.Data()));
1349 return NULL;
1350 }
1351 }
1352 else if (key.Contains("cleanupInt",TString::kIgnoreCase)){
1353 if(value.IsDigit()) {
1354 cleanupInterval = value.Atoi();
1355 } else {
1356 AliError(Form("Invalid entry! %s",entry.Data()));
1357 return NULL;
1358 }
1359 }
b05400be 1360 else{
1361 AliError(Form("Invalid entry! %s",entry.Data()));
c75fabed 1362 return NULL;
b05400be 1363 }
1364 }
1365 delete arr; arr=0;
c75fabed 1366
1367 AliDebug(2, Form("gridUrl: %s", gridUrl.Data()));
1368 AliDebug(2, Form("user: %s", user.Data()));
1369 AliDebug(2, Form("dbFolder: %s", dbFolder.Data()));
1370 AliDebug(2, Form("s.e.: %s", se.Data()));
1371 AliDebug(2, Form("local cache folder: %s", cacheFolder.Data()));
1372 AliDebug(2, Form("local cache operate disconnected: %d", operateDisconnected));
d2a0a483 1373 AliDebug(2, Form("local cache size: %lld", cacheSize));
6720fa1f 1374 AliDebug(2, Form("local cache cleanup interval: %ld", cleanupInterval));
9e1ceb13 1375
27cac7e3 1376 if(dbFolder == ""){
c75fabed 1377 AliError("Base folder must be specified!");
4667c116 1378 return NULL;
1379 }
1380
c75fabed 1381 return new AliCDBGridParam(gridUrl.Data(), user.Data(),
1382 dbFolder.Data(), se.Data(), cacheFolder.Data(),
1383 operateDisconnected, cacheSize, cleanupInterval);
9e1ceb13 1384}
1385
1386//_____________________________________________________________________________
1387AliCDBStorage* AliCDBGridFactory::Create(const AliCDBParam* param) {
1388// create AliCDBGrid storage instance from parameters
1389
62032124 1390 AliCDBGrid *grid = 0;
9e1ceb13 1391 if (AliCDBGridParam::Class() == param->IsA()) {
62032124 1392
9e1ceb13 1393 const AliCDBGridParam* gridParam = (const AliCDBGridParam*) param;
62032124 1394 grid = new AliCDBGrid(gridParam->GridUrl().Data(),
1395 gridParam->GetUser().Data(),
1396 gridParam->GetDBFolder().Data(),
c75fabed 1397 gridParam->GetSE().Data(),
1398 gridParam->GetCacheFolder().Data(),
1399 gridParam->GetOperateDisconnected(),
1400 gridParam->GetCacheSize(),
1401 gridParam->GetCleanupInterval());
9e1ceb13 1402
9e1ceb13 1403 }
1404
c3a7b59a 1405 if(!gGrid && grid) {
1406 delete grid; grid=0;
1407 }
1408
62032124 1409 return grid;
9e1ceb13 1410}
1411
1412/////////////////////////////////////////////////////////////////////////////////////////////////
1413// //
c75fabed 1414// AliCDBGrid Parameter class // //
9e1ceb13 1415// //
1416/////////////////////////////////////////////////////////////////////////////////////////////////
1417
1418ClassImp(AliCDBGridParam)
1419
1420//_____________________________________________________________________________
62032124 1421AliCDBGridParam::AliCDBGridParam():
1422 AliCDBParam(),
1423 fGridUrl(),
1424 fUser(),
1425 fDBFolder(),
c75fabed 1426 fSE(),
1427 fCacheFolder(),
1428 fOperateDisconnected(),
1429 fCacheSize(),
1430 fCleanupInterval()
1431
62032124 1432 {
9e1ceb13 1433// default constructor
1434
1435}
1436
1437//_____________________________________________________________________________
c75fabed 1438AliCDBGridParam::AliCDBGridParam(const char* gridUrl, const char* user, const char* dbFolder,
1439 const char* se, const char* cacheFolder, Bool_t operateDisconnected,
1440 Long64_t cacheSize, Long_t cleanupInterval):
62032124 1441 AliCDBParam(),
b05400be 1442 fGridUrl(gridUrl),
9e1ceb13 1443 fUser(user),
b05400be 1444 fDBFolder(dbFolder),
c75fabed 1445 fSE(se),
1446 fCacheFolder(cacheFolder),
1447 fOperateDisconnected(operateDisconnected),
1448 fCacheSize(cacheSize),
1449 fCleanupInterval(cleanupInterval)
9e1ceb13 1450{
1451// constructor
c75fabed 1452
9e1ceb13 1453 SetType("alien");
1454
c75fabed 1455 TString uri = Form("%s?User=%s?DBFolder=%s?SE=%s?CacheFolder=%s"
d2a0a483 1456 "?OperateDisconnected=%d?CacheSize=%lld?CleanupInterval=%ld",
62032124 1457 fGridUrl.Data(), fUser.Data(),
c75fabed 1458 fDBFolder.Data(), fSE.Data(), fCacheFolder.Data(),
1459 fOperateDisconnected, fCacheSize, fCleanupInterval);
62032124 1460
1461 SetURI(uri.Data());
9e1ceb13 1462}
1463
1464//_____________________________________________________________________________
1465AliCDBGridParam::~AliCDBGridParam() {
1466// destructor
1467
1468}
1469
1470//_____________________________________________________________________________
1471AliCDBParam* AliCDBGridParam::CloneParam() const {
1472// clone parameter
1473
62032124 1474 return new AliCDBGridParam(fGridUrl.Data(), fUser.Data(),
c75fabed 1475 fDBFolder.Data(), fSE.Data(), fCacheFolder.Data(),
1476 fOperateDisconnected, fCacheSize, fCleanupInterval);
9e1ceb13 1477}
1478
1479//_____________________________________________________________________________
1480ULong_t AliCDBGridParam::Hash() const {
1481// return Hash function
1482
c75fabed 1483 return fGridUrl.Hash()+fUser.Hash()+fDBFolder.Hash()+fSE.Hash()+fCacheFolder.Hash();
9e1ceb13 1484}
1485
1486//_____________________________________________________________________________
1487Bool_t AliCDBGridParam::IsEqual(const TObject* obj) const {
1488// check if this object is equal to AliCDBParam obj
1489
1490 if (this == obj) {
1491 return kTRUE;
1492 }
1493
1494 if (AliCDBGridParam::Class() != obj->IsA()) {
1495 return kFALSE;
1496 }
1497
1498 AliCDBGridParam* other = (AliCDBGridParam*) obj;
1499
b05400be 1500 if(fGridUrl != other->fGridUrl) return kFALSE;
9e1ceb13 1501 if(fUser != other->fUser) return kFALSE;
b05400be 1502 if(fDBFolder != other->fDBFolder) return kFALSE;
9e1ceb13 1503 if(fSE != other->fSE) return kFALSE;
c75fabed 1504 if(fCacheFolder != other->fCacheFolder) return kFALSE;
1505 if(fOperateDisconnected != other->fOperateDisconnected) return kFALSE;
1506 if(fCacheSize != other->fCacheSize) return kFALSE;
1507 if(fCleanupInterval != other->fCleanupInterval) return kFALSE;
9e1ceb13 1508 return kTRUE;
1509}
1510