]> git.uio.no Git - u/mrichter/AliRoot.git/blob - STEER/CDB/AliCDBGrid.cxx
Adding an additional loop wrapping the opening the alien file and writing to it....
[u/mrichter/AliRoot.git] / STEER / CDB / AliCDBGrid.cxx
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
23 #include <cstdlib>
24 #include <TGrid.h>
25 #include <TGridResult.h>
26 #include <TFile.h>
27 #include <TKey.h>
28 #include <TROOT.h>
29 #include <TList.h>
30 #include <TObjArray.h>
31 #include <TObjString.h>
32 #include <TMath.h>
33 #include <TRegexp.h>
34
35 #include "AliLog.h"
36 #include "AliCDBEntry.h"
37 #include "AliCDBGrid.h"
38 #include "AliCDBManager.h"
39
40
41 ClassImp(AliCDBGrid)
42
43 //_____________________________________________________________________________
44 AliCDBGrid::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) :
47 AliCDBStorage(),
48 fGridUrl(gridUrl),
49 fUser(user),
50 fDBFolder(dbFolder),
51 fSE(se),
52 fMirrorSEs(""),
53 fCacheFolder(cacheFolder),
54 fOperateDisconnected(operateDisconnected),
55 fCacheSize(cacheSize),
56 fCleanupInterval(cleanupInterval)
57 {
58 // constructor //
59
60         // if the same Grid is alreay active, skip connection
61         if (!gGrid || fGridUrl != gGrid->GridUrl()
62              || (( fUser != "" ) && ( fUser != gGrid->GetUser() )) ) {
63                 // connection to the Grid
64                 AliInfo("Connection to the Grid...");
65                 if(gGrid){
66                         AliInfo(Form("gGrid = %p; fGridUrl = %s; gGrid->GridUrl() = %s",gGrid,fGridUrl.Data(), gGrid->GridUrl()));
67                         AliInfo(Form("fUser = %s; gGrid->GetUser() = %s",fUser.Data(), gGrid->GetUser()));
68                 }
69                 TGrid::Connect(fGridUrl.Data(),fUser.Data());
70         }
71
72         if(!gGrid) {
73                 AliError("Connection failed!");
74                 return;
75         }
76
77         TString initDir(gGrid->Pwd(0));
78         if (fDBFolder[0] != '/') {
79                 fDBFolder.Prepend(initDir);
80         }
81
82         // check DBFolder: trying to cd to DBFolder; if it does not exist, create it
83         if(!gGrid->Cd(fDBFolder.Data(),0)){
84                 AliDebug(2,Form("Creating new folder <%s> ...",fDBFolder.Data()));
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;
90                 }
91         } else {
92                 AliDebug(2,Form("Folder <%s> found",fDBFolder.Data()));
93         }
94
95         // removes any '/' at the end of path, then append one '/'
96         while(fDBFolder.EndsWith("/")) fDBFolder.Remove(fDBFolder.Last('/')); 
97         fDBFolder+="/";
98
99         fType="alien";
100         fBaseFolder = fDBFolder;
101
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 "
127                                 "to ShrinkCacheFileDir: cacheSize = %lld, cleanupInterval = %ld !",
128                                 fCacheSize, fCleanupInterval));
129                 }
130         }
131
132         // return to the initial directory
133         gGrid->Cd(initDir.Data(),0);
134
135         fNretry = 3;  // default
136         fInitRetrySeconds = 5;   // default
137 }
138
139 //_____________________________________________________________________________
140 AliCDBGrid::~AliCDBGrid()
141 {
142 // destructor
143         delete gGrid; gGrid=0;
144
145 }
146
147 //_____________________________________________________________________________
148 Bool_t AliCDBGrid::FilenameToId(TString& filename, AliCDBId& id) {
149 // build AliCDBId from full path filename (fDBFolder/path/Run#x_#y_v#z_s0.root)
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());
160
161         Ssiz_t mSize;
162         // valid filename: Run#firstRun_#lastRun_v#version_s0.root
163         TRegexp keyPattern("^Run[0-9]+_[0-9]+_v[0-9]+_s0.root$");
164         keyPattern.Index(filename, &mSize);
165         if (!mSize) {
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         }
182
183         filename.Resize(filename.Length() - sizeof(".root") + 1);
184
185         TObjArray* strArray = (TObjArray*) filename.Tokenize("_");
186
187         TString firstRunString(((TObjString*) strArray->At(0))->GetString());
188         id.SetFirstRun(atoi(firstRunString.Data() + 3));
189         id.SetLastRun(atoi(((TObjString*) strArray->At(1))->GetString()));
190
191         TString verString(((TObjString*) strArray->At(2))->GetString());
192         id.SetVersion(atoi(verString.Data() + 1));
193
194         delete strArray;
195
196         return kTRUE;
197 }
198
199 //_____________________________________________________________________________
200 Bool_t AliCDBGrid::IdToFilename(const AliCDBId& id, TString& filename) const {
201 // build file name from AliCDBId (path, run range, version) and fDBFolder
202
203         if (!id.GetAliCDBRunRange().IsValid()) {
204                 AliDebug(2,Form("Invalid run range <%d, %d>.",
205                         id.GetFirstRun(), id.GetLastRun()));
206                 return kFALSE;
207         }
208
209         if (id.GetVersion() < 0) {
210                 AliDebug(2,Form("Invalid version <%d>.", id.GetVersion()));
211                 return kFALSE;
212         }
213
214         filename = Form("Run%d_%d_v%d",
215                                 id.GetFirstRun(),
216                                 id.GetLastRun(),
217                                 id.GetVersion());
218
219         if (id.GetSubVersion() != -11) filename += "_s0"; // TODO to ensure backward compatibility
220         filename += ".root";
221
222         filename.Prepend(fDBFolder + id.GetPath() + '/');
223
224         return kTRUE;
225 }
226
227 //_____________________________________________________________________________
228 void AliCDBGrid::SetRetry(Int_t nretry, Int_t initsec) {
229
230         // Function to set the exponential retry for putting entries in the OCDB
231
232         AliWarning("WARNING!!! You are changing the exponential retry times and delay: this function should be used by experts!"); 
233         fNretry = nretry;
234         fInitRetrySeconds = initsec;
235         AliDebug(2,Form("fNretry = %d, fInitRetrySeconds = %d", fNretry, fInitRetrySeconds));
236
237
238
239 //_____________________________________________________________________________
240 Bool_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));
244
245         TString dirName(fDBFolder);
246
247         Bool_t dirExist=kFALSE;
248
249
250
251         // go to the path; if directory does not exist, create it
252         for(int i=0;i<3;i++){
253                 //TString cmd("find -d ");
254                 //cmd += Form("%s ",dirName);
255                 //cmd += 
256                 //gGrid->Command(cmd.Data());
257                 dirName+=Form("%s/",id.GetPathLevel(i).Data());
258                 dirExist=gGrid->Cd(dirName,0);
259                 if (!dirExist) {
260                         AliDebug(2,Form("Creating new folder <%s> ...",dirName.Data()));
261                         if(!gGrid->Mkdir(dirName,"",0)){
262                                 AliError(Form("Cannot create directory <%s> !",dirName.Data()));
263                                 gGrid->Cd(initDir.Data());
264                         return kFALSE;
265                         }
266
267                         // if folders are new add tags to them
268                         if(i == 1) {
269                                 // TODO Currently disabled
270                                 // add short lived tag!
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) {
281                                 AliDebug(2,"Tagging level 2 folder with \"CDB\" and \"CDB_MD\" tag");
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
297                                 // TODO Currently disabled
298                                 // add short lived tag!
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                         }
320                 }
321         }
322         gGrid->Cd(initDir,0);
323
324         TString filename;
325         AliCDBId anId; // the id got from filename
326         AliCDBRunRange lastRunRange(-1,-1); // highest runRange found
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++){
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();
338                 }
339
340         }
341         delete res;
342
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;
348         }
349
350         id.SetVersion(lastVersion + 1);
351         id.SetSubVersion(0);
352
353         TString lastStorage = id.GetLastStorage();
354         if(lastStorage.Contains(TString("new"), TString::kIgnoreCase) && id.GetVersion() > 1 ){
355                 AliDebug(2, Form("A NEW object is being stored with version %d",
356                                         id.GetVersion()));
357                 AliDebug(2, Form("and it will hide previously stored object with version %d!",
358                                         id.GetVersion()-1));
359         }
360
361         if(!lastRunRange.IsAnyRange() && !(lastRunRange.IsEqual(&id.GetAliCDBRunRange())))
362                 AliWarning(Form("Run range modified w.r.t. previous version (Run%d_%d_v%d)",
363                         lastRunRange.GetFirstRun(), lastRunRange.GetLastRun(), id.GetVersion()));
364
365         return kTRUE;
366 }
367
368 //_____________________________________________________________________________
369 AliCDBId* AliCDBGrid::GetId(const TObjArray& validFileIds, const AliCDBId& query) {
370 // look for the Id that matches query's requests (highest or exact version)
371
372         if(validFileIds.GetEntriesFast() < 1)
373                 return NULL;
374
375         TIter iter(&validFileIds);
376
377         AliCDBId *anIdPtr=0;
378         AliCDBId* result=0;
379
380         while((anIdPtr = dynamic_cast<AliCDBId*> (iter.Next()))){
381                 if(anIdPtr->GetPath() != query.GetPath()) continue;
382
383                 //if(!CheckVersion(query, anIdPtr, result)) return NULL;
384
385                 if (!query.HasVersion()){ // look for highest version
386                         if(result && result->GetVersion() > anIdPtr->GetVersion()) continue;
387                         if(result && result->GetVersion() == anIdPtr->GetVersion()) {
388                                 AliError(Form("More than one object valid for run %d, version %d!",
389                                         query.GetFirstRun(), anIdPtr->GetVersion()));
390                                 return NULL;
391                         }
392                         result = anIdPtr;
393                 } else { // look for specified version
394                         if(query.GetVersion() != anIdPtr->GetVersion()) continue;
395                         if(result && result->GetVersion() == anIdPtr->GetVersion()){
396                                 AliError(Form("More than one object valid for run %d, version %d!",
397                                         query.GetFirstRun(), anIdPtr->GetVersion()));
398                                 return NULL;
399                         }
400                         result = anIdPtr;
401                 }
402
403         }
404         
405         if (!result) return NULL;
406
407         return dynamic_cast<AliCDBId*> (result->Clone());
408 }
409
410 //_____________________________________________________________________________
411 AliCDBId* AliCDBGrid::GetEntryId(const AliCDBId& queryId) {
412 // get AliCDBId from the database
413 // User must delete returned object
414
415         AliCDBId* dataId=0;
416
417         AliCDBId selectedId(queryId);
418         if (!selectedId.HasVersion()) {
419                 // if version is not specified, first check the selection criteria list
420                 GetSelection(&selectedId);
421         }
422
423         TObjArray validFileIds;
424         validFileIds.SetOwner(1);
425
426         // look for file matching query requests (path, runRange, version)
427         if(selectedId.GetFirstRun() == fRun && fPathFilter.Comprises(selectedId.GetAliCDBPath()) &&
428                         fVersion == selectedId.GetVersion() && !fMetaDataFilter){
429                 // look into list of valid files previously loaded with AliCDBStorage::FillValidFileIds()
430                 AliDebug(2, Form("List of files valid for run %d was loaded. Looking there for fileids valid for path %s!",
431                                         selectedId.GetFirstRun(), selectedId.GetPath().Data()));
432                 dataId = GetId(fValidFileIds, selectedId);
433
434         } else {
435                 // List of files valid for reqested run was not loaded. Looking directly into CDB
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()));
438
439                 TString filter;
440                 MakeQueryFilter(selectedId.GetFirstRun(), selectedId.GetLastRun(), 0, filter);
441
442                 TString pattern = ".root";
443                 TString optionQuery = "-y -m";
444                 if(selectedId.GetVersion() >= 0) {
445                         pattern.Prepend(Form("_v%d_s0",selectedId.GetVersion()));
446                         optionQuery = "";
447                 }
448
449                 TString folderCopy(Form("%s%s/Run",fDBFolder.Data(),selectedId.GetPath().Data()));
450
451                 if (optionQuery.Contains("-y")){
452                         AliInfo("Only latest version will be returned");
453                 }
454
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());
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                 }       
467                 dataId = GetId(validFileIds, selectedId);
468         }
469
470         return dataId;
471 }
472
473 //_____________________________________________________________________________
474 AliCDBEntry* AliCDBGrid::GetEntry(const AliCDBId& queryId) {
475 // get AliCDBEntry from the database
476
477         AliCDBId* dataId = GetEntryId(queryId);
478
479         if (!dataId) return NULL;
480
481         TString filename;
482         if (!IdToFilename(*dataId, filename)) {
483                 AliDebug(2,Form("Bad data ID encountered! Subnormal error!"));
484                 delete dataId;
485                 return NULL;
486         }
487
488         AliCDBEntry* anEntry = GetEntryFromFile(filename, dataId);
489
490         delete dataId;
491         return anEntry;
492 }
493
494 //_____________________________________________________________________________
495 AliCDBEntry* AliCDBGrid::GetEntryFromFile(TString& filename, AliCDBId* dataId){
496 // Get AliCBEntry object from file "filename"
497
498         AliDebug(2,Form("Opening file: %s",filename.Data()));
499
500         filename.Prepend("/alien");
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);
521         if (!file) {
522                 AliDebug(2,Form("Can't open file <%s>!", filename.Data()));
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
529         AliCDBEntry* anEntry = dynamic_cast<AliCDBEntry*> (file->Get("AliCDBEntry"));
530
531         if (!anEntry) {
532                 AliDebug(2,Form("Bad storage data: file does not contain an AliCDBEntry object!"));
533                 file->Close();
534                 return NULL;
535         }
536
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.
540
541         if(anEntry){
542                 AliCDBId entryId = anEntry->GetId();
543                 Int_t tmpSubVersion = dataId->GetSubVersion();
544                 dataId->SetSubVersion(entryId.GetSubVersion()); // otherwise filename and id may mismatch
545                 if(!entryId.IsEqual(dataId)){
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()));
549                 }
550                 dataId->SetSubVersion(tmpSubVersion);
551         }
552
553         anEntry->SetLastStorage("grid");
554
555         // Check whether entry contains a TTree. In case load the tree in memory!
556         LoadTreeFromFile(anEntry);
557
558         // close file, return retieved entry
559         file->Close(); delete file; file=0;
560
561         return anEntry;
562 }
563
564 //_____________________________________________________________________________
565 TList* AliCDBGrid::GetEntries(const AliCDBId& queryId) {
566 // multiple request (AliCDBStorage::GetAll)
567
568         TList* result = new TList();
569         result->SetOwner();
570
571         TObjArray validFileIds;
572         validFileIds.SetOwner(1);
573
574         Bool_t alreadyLoaded = kFALSE;
575
576         // look for file matching query requests (path, runRange)
577         if(queryId.GetFirstRun() == fRun &&
578                         fPathFilter.Comprises(queryId.GetAliCDBPath()) && fVersion < 0 && !fMetaDataFilter){
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()));
582
583                 alreadyLoaded = kTRUE;
584
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;
591                 MakeQueryFilter(queryId.GetFirstRun(), queryId.GetLastRun(), 0, filter);
592
593                 TString path = queryId.GetPath();
594
595                 TString pattern = "Run*.root";
596                 TString optionQuery = "-y";
597
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;
633                 }
634
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                 }
645
646                 AliCDBId validFileId;
647                 for(int i=0; i<res->GetEntries(); i++){
648                         TString filename = res->GetKey(i, "lfn");
649                         if(filename == "") continue;
650                         if(FilenameToId(filename, validFileId))
651                                         validFileIds.AddLast(validFileId.Clone());
652                 }
653                 delete res;
654         }
655
656         TIter *iter=0;
657         if(alreadyLoaded){
658                 iter = new TIter(&fValidFileIds);
659         } else {
660                 iter = new TIter(&validFileIds);
661         }
662
663         TObjArray selectedIds;
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);
686                 }
687                 if(dataId) selectedIds.Add(dataId);
688         }
689
690         delete iter; iter=0;
691
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                 }
701
702                 AliCDBEntry* anEntry = GetEntryFromFile(filename, anIdPtr);
703
704                 if(anEntry) result->Add(anEntry);
705
706         }
707         delete iter; iter=0;
708
709         return result;
710 }
711
712 //_____________________________________________________________________________
713 Bool_t AliCDBGrid::PutEntry(AliCDBEntry* entry, const char* mirrors) {
714         // put an AliCDBEntry object into the database
715
716         AliCDBId& id = entry->GetId();
717
718         // set version for the entry to be stored
719         if (!PrepareId(id)) return kFALSE;
720
721         // build filename from entry's id
722         TString filename;
723         if (!IdToFilename(id, filename)) {
724                 AliError("Bad ID encountered! Subnormal error!");
725                 return kFALSE;
726         }
727
728         TString folderToTag = Form("%s%s",
729                         fDBFolder.Data(),
730                         id.GetPath().Data());
731
732         TDirectory* saveDir = gDirectory;
733
734         TString fullFilename = Form("/alien%s", filename.Data());
735         TString seMirrors(mirrors);
736         if(seMirrors.IsNull() || seMirrors.IsWhitespace()) seMirrors=GetMirrorSEs();
737         // specify SE to filename
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.
741         TObjArray *arraySEs = seMirrors.Tokenize(',');
742         Int_t nSEs = arraySEs->GetEntries();
743         Int_t remainingSEs = 1;
744         if(nSEs == 0){
745                 if (fSE != "default") fullFilename += Form("?se=%s",fSE.Data());
746         }else{
747                 remainingSEs = nSEs;
748         }
749
750         // open file
751         TFile *file=0;
752         AliDebug(2, Form("fNretry = %d, fInitRetrySeconds = %d",fNretry,fInitRetrySeconds));
753         TString targetSE("");
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{
790                                         Printf("File opened successfully at attempt n. %d",fNretry-remainingAttempts);
791                                         remainingAttempts=0;
792                                 }
793                                 nsleep*=fInitRetrySeconds;
794                         }
795                         remainingSEs--;
796                 }
797                 if(!file){
798                         AliError(Form("All %d attempts have failed on all %d SEs. Returning...",fNretry,nSEs));
799                         return kFALSE;
800                 }
801
802                 file->cd();
803
804                 //SetTreeToFile(entry, file);
805                 entry->SetVersion(id.GetVersion());
806
807                 // write object (key name: "AliCDBEntry")
808                 result = (file->WriteTObject(entry, "AliCDBEntry") != 0);
809                 if (!result) AliError(Form("Can't write entry to file <%s>!", filename.Data()));
810                 file->Close();
811
812                 if(result)
813                 {
814                         AliDebug(2, Form("Reopening file %s for checking its correctness",fullFilename.Data()));
815                         TFile* ffile = TFile::Open(fullFilename.Data(),"READ");
816                         if(!ffile){
817                                 reOpenResult = kFALSE;
818                                 AliInfo(Form("The file %s was closed successfully but cannot be reopened. Trying now to regenerate it (regeneration attempt number %d)",
819                                                         fullFilename.Data(),++reOpenAttempts));
820                                 delete file; file=0;
821                                 AliDebug(2, Form("Removing file %s", filename.Data()));
822                                 if(!gGrid->Rm(filename.Data()))
823                                         AliError("Can't delete file!");
824                                 remainingSEs++;
825                         }else{
826                                 reOpenResult = kTRUE;
827                                 ffile->Close();
828                         }
829                         delete ffile; ffile=0;
830                 }
831         }
832
833         if (saveDir) saveDir->cd(); else gROOT->cd();
834         delete file; file=0;
835
836         if(result && reOpenResult) {
837
838                 if(!TagFileId(filename, &id)){
839                         AliInfo(Form("CDB tagging failed. Deleting file %s!",filename.Data()));
840                         if(!gGrid->Rm(filename.Data()))
841                                 AliError("Can't delete file!");
842                         return kFALSE;
843                 }
844
845                 TagFileMetaData(filename, entry->GetMetaData());
846         }else{
847                 AliError("The file could not be opend or the object could not be written");
848                 if(!gGrid->Rm(filename.Data()))
849                         AliError("Can't delete file!");
850                 return kFALSE;
851         }
852
853         AliInfo(Form("CDB object stored into file %s", filename.Data()));
854         if(nSEs==0)
855                 AliInfo(Form("Storage Element: %s", fSE.Data()));
856         else
857                 AliInfo(Form("Storage Element: %s", targetSE.Data()));
858
859         //In case of other SEs specified by the user, mirror the file to the remaining SEs
860         for(Int_t i=0; i<nSEs; i++){
861                 if(i==nSEs-remainingSEs-1) continue; // skip mirroring to the SE where the file was saved
862                 TString mirrorCmd("mirror ");
863                 mirrorCmd += filename;
864                 mirrorCmd += " ";
865                 TObjString *target = (TObjString*) arraySEs->At(i);
866                 TString mirrorSE(target->String());
867                 mirrorCmd += mirrorSE;
868                 AliDebug(5,Form("mirror command: \"%s\"",mirrorCmd.Data()));
869                 AliInfo(Form("Mirroring to storage element: %s", mirrorSE.Data()));
870                 gGrid->Command(mirrorCmd.Data());
871         }
872
873         return kTRUE;
874 }
875
876 //_____________________________________________________________________________
877 Bool_t AliCDBGrid::AddTag(TString& folderToTag, const char* tagname){
878 // add "tagname" tag (CDB or CDB_MD) to folder where object will be stored
879
880         Bool_t result = kTRUE;
881         AliDebug(2, Form("adding %s tag to folder %s", tagname, folderToTag.Data()));
882         TString addTag = Form("addTag %s %s", folderToTag.Data(), tagname);
883         TGridResult *gridres = gGrid->Command(addTag.Data());
884         const char* resCode = gridres->GetKey(0,"__result__"); // '1' if success
885         if(resCode[0] != '1') {
886                 AliError(Form("Couldn't add %s tags to folder %s !",
887                                                 tagname, folderToTag.Data()));
888                 result = kFALSE;
889         }
890         delete gridres;
891         return result;
892 }
893
894 //_____________________________________________________________________________
895 Bool_t AliCDBGrid::TagFileId(TString& filename, const AliCDBId* id){
896 // tag stored object in CDB table using object Id's parameters
897
898
899         TString dirname(filename);
900         Int_t dirNumber = gGrid->Mkdir(dirname.Remove(dirname.Last('/')),"-d");
901         
902         TString addTagValue1 = Form("addTagValue %s CDB ", filename.Data());
903         TString addTagValue2 = Form("first_run=%d last_run=%d version=%d ",
904                                         id->GetFirstRun(),
905                                         id->GetLastRun(),
906                                         id->GetVersion());
907         TString addTagValue3 = Form("path_level_0=\"%s\" path_level_1=\"%s\" path_level_2=\"%s\" ",
908                                         id->GetPathLevel(0).Data(),
909                                         id->GetPathLevel(1).Data(),
910                                         id->GetPathLevel(2).Data());
911         //TString addTagValue4 = Form("version_path=\"%s\" dir_number=%d",Form("%d_%s",id->GetVersion(),filename.Data()),dirNumber); 
912         TString addTagValue4 = Form("version_path=\"%09d%s\" dir_number=%d",id->GetVersion(),filename.Data(),dirNumber); 
913         TString addTagValue = Form("%s%s%s%s",
914                                         addTagValue1.Data(),
915                                         addTagValue2.Data(),
916                                         addTagValue3.Data(),
917                                         addTagValue4.Data());
918
919         Bool_t result = kFALSE;
920         AliDebug(2, Form("Tagging file. Tag command: %s", addTagValue.Data()));
921         TGridResult* res = gGrid->Command(addTagValue.Data());
922         const char* resCode = res->GetKey(0,"__result__"); // '1' if success
923         if(resCode[0] != '1') {
924                 AliError(Form("Couldn't add CDB tag value to file %s !",
925                                                 filename.Data()));
926                 result = kFALSE;
927         } else {
928                 AliDebug(2, "Object successfully tagged.");
929                 result = kTRUE;
930         }
931         delete res;
932         return result;
933
934 }
935
936 //_____________________________________________________________________________
937 Bool_t AliCDBGrid::TagShortLived(TString& filename, Bool_t value){
938 // tag folder with ShortLived tag
939
940         TString addTagValue = Form("addTagValue %s ShortLived_try value=%d", filename.Data(), value);
941
942         Bool_t result = kFALSE;
943         AliDebug(2, Form("Tagging file. Tag command: %s", addTagValue.Data()));
944         TGridResult* res = gGrid->Command(addTagValue.Data());
945         const char* resCode = res->GetKey(0,"__result__"); // '1' if success
946         if(resCode[0] != '1') {
947                 AliError(Form("Couldn't add ShortLived tag value to file %s !", filename.Data()));
948                 result = kFALSE;
949         } else {
950                 AliDebug(2,"Object successfully tagged.");
951                 result = kTRUE;
952         }
953         delete res;
954         return result;
955
956 }
957
958 //_____________________________________________________________________________
959 Bool_t AliCDBGrid::TagFileMetaData(TString& filename, const AliCDBMetaData* md){
960 // tag stored object in CDB table using object Id's parameters
961
962         TString addTagValue1 = Form("addTagValue %s CDB_MD ", filename.Data());
963         TString addTagValue2 = Form("object_classname=\"%s\" responsible=\"%s\" beam_period=%d ",
964                                         md->GetObjectClassName(),
965                                         md->GetResponsible(),
966                                         md->GetBeamPeriod());
967         TString addTagValue3 = Form("aliroot_version=\"%s\" comment=\"%s\"",
968                                         md->GetAliRootVersion(),
969                                         md->GetComment());
970         TString addTagValue = Form("%s%s%s",
971                                         addTagValue1.Data(),
972                                         addTagValue2.Data(),
973                                         addTagValue3.Data());
974
975         Bool_t result = kFALSE;
976         AliDebug(2, Form("Tagging file. Tag command: %s", addTagValue.Data()));
977         TGridResult* res = gGrid->Command(addTagValue.Data());
978         const char* resCode = res->GetKey(0,"__result__"); // '1' if success
979         if(resCode[0] != '1') {
980                 AliWarning(Form("Couldn't add CDB_MD tag value to file %s !",
981                                                 filename.Data()));
982                 result = kFALSE;
983         } else {
984                 AliDebug(2,"Object successfully tagged.");
985                 result = kTRUE;
986         }
987         return result;
988 }
989
990 //_____________________________________________________________________________
991 TList* AliCDBGrid::GetIdListFromFile(const char* fileName){
992
993         TString turl(fileName);
994         turl.Prepend("/alien" + fDBFolder);
995         turl += "?se="; turl += fSE.Data();
996         TFile *file = TFile::Open(turl);
997         if (!file) {
998                 AliError(Form("Can't open selection file <%s>!", turl.Data()));
999                 return NULL;
1000         }
1001
1002         TList *list = new TList();
1003         list->SetOwner();
1004         int i=0;
1005         TString keycycle;
1006
1007         AliCDBId *id;
1008         while(1){
1009                 i++;
1010                 keycycle = "AliCDBId;";
1011                 keycycle+=i;
1012                 
1013                 id = (AliCDBId*) file->Get(keycycle);
1014                 if(!id) break;
1015                 list->AddFirst(id);
1016         }
1017         file->Close(); delete file; file=0;
1018         
1019         return list;
1020
1021
1022 }
1023
1024 //_____________________________________________________________________________
1025 Bool_t AliCDBGrid::Contains(const char* path) const{
1026 // check for path in storage's DBFolder
1027
1028         TString initDir(gGrid->Pwd(0));
1029         TString dirName(fDBFolder);
1030         dirName += path; // dirName = fDBFolder/path
1031         Bool_t result=kFALSE;
1032         if (gGrid->Cd(dirName,0)) result=kTRUE;
1033         gGrid->Cd(initDir.Data(),0);
1034         return result;
1035 }
1036
1037 //_____________________________________________________________________________
1038 void AliCDBGrid::QueryValidFiles()
1039 {
1040 // Query the CDB for files valid for AliCDBStorage::fRun
1041 // fills list fValidFileIds with AliCDBId objects created from file name
1042
1043         TString filter;
1044         MakeQueryFilter(fRun, fRun, fMetaDataFilter, filter);
1045
1046         TString path = fPathFilter.GetPath();
1047
1048         TString pattern = "Run*";
1049         TString optionQuery = "-y";
1050         if(fVersion >= 0) {
1051                 pattern += Form("_v%d_s0", fVersion);
1052                 optionQuery = "";
1053         }
1054         pattern += ".root";
1055         AliDebug(2,Form("pattern: %s", pattern.Data()));
1056
1057         TString addFolder = "";
1058         if (!path.Contains("*")){
1059                 if (!path.BeginsWith("/")) addFolder += "/";
1060                 addFolder += path;
1061         }
1062         else{
1063                 if (path.BeginsWith("/")) path.Remove(0,1);
1064                 if (path.EndsWith("/")) path.Remove(path.Length()-1,1); 
1065                 TObjArray* tokenArr = path.Tokenize("/");
1066                 if (tokenArr->GetEntries() != 3) {
1067                         AliError("Not a 3 level path! Keeping old query...");
1068                         pattern.Prepend(path+"/");
1069                 }
1070                 else{
1071                         TString str0 = ((TObjString*)tokenArr->At(0))->String();
1072                         TString str1 = ((TObjString*)tokenArr->At(1))->String();
1073                         TString str2 = ((TObjString*)tokenArr->At(2))->String();
1074                         if (str0 != "*" && str1 != "*" && str2 == "*"){
1075                                 // e.g. "ITS/Calib/*"
1076                                 addFolder = "/"+str0+"/"+str1;
1077                         }
1078                         else if (str0 != "*" && str1 == "*" && str2 == "*"){    
1079                                 // e.g. "ITS/*/*"
1080                                 addFolder = "/"+str0;
1081                         }
1082                         else if (str0 == "*" && str1 == "*" && str2 == "*"){    
1083                                 // e.g. "*/*/*"
1084                                 // do nothing: addFolder is already an empty string;
1085                         }
1086                         else{
1087                                 // e.g. "ITS/*/RecoParam"
1088                                 pattern.Prepend(path+"/");
1089                         }
1090                 }
1091                 delete tokenArr; tokenArr=0;
1092         }
1093
1094         TString folderCopy(Form("%s%s",fDBFolder.Data(),addFolder.Data()));
1095
1096         AliDebug(2,Form("fDBFolder = %s, pattern = %s, filter = %s",folderCopy.Data(), pattern.Data(), filter.Data()));
1097
1098         if (optionQuery == "-y"){
1099                 AliInfo("Only latest version will be returned");
1100         } 
1101
1102         TGridResult *res = gGrid->Query(folderCopy, pattern, filter, optionQuery.Data());  
1103
1104         if (!res) {
1105                 AliError("Grid query failed");
1106                 return;
1107         }
1108
1109         TIter next(res);
1110         TMap *map;
1111         while ((map = (TMap*)next())) {
1112           TObjString *entry;
1113           if ((entry = (TObjString *) ((TMap *)map)->GetValue("lfn"))) {
1114             TString& filename = entry->String();
1115             if(filename.IsNull()) continue;
1116             AliDebug(2,Form("Found valid file: %s", filename.Data()));
1117             AliCDBId *validFileId = new AliCDBId;
1118             Bool_t result = FilenameToId(filename, *validFileId);
1119             if(result) {
1120               fValidFileIds.AddLast(validFileId);
1121             }
1122             else {
1123               delete validFileId;
1124             }
1125           }
1126         }
1127         delete res;
1128
1129 }
1130
1131 //_____________________________________________________________________________
1132 void AliCDBGrid::MakeQueryFilter(Int_t firstRun, Int_t lastRun,
1133                                         const AliCDBMetaData* md, TString& result) const
1134 {
1135 // create filter for file query
1136
1137         result = Form("CDB:first_run<=%d and CDB:last_run>=%d", firstRun, lastRun);
1138
1139 //      if(version >= 0) {
1140 //              result += Form(" and CDB:version=%d", version);
1141 //      }
1142 //      if(pathFilter.GetLevel0() != "*") {
1143 //              result += Form(" and CDB:path_level_0=\"%s\"", pathFilter.GetLevel0().Data());
1144 //      }
1145 //      if(pathFilter.GetLevel1() != "*") {
1146 //              result += Form(" and CDB:path_level_1=\"%s\"", pathFilter.GetLevel1().Data());
1147 //      }
1148 //      if(pathFilter.GetLevel2() != "*") {
1149 //              result += Form(" and CDB:path_level_2=\"%s\"", pathFilter.GetLevel2().Data());
1150 //      }
1151
1152         if(md){
1153                 if(md->GetObjectClassName()[0] != '\0') {
1154                         result += Form(" and CDB_MD:object_classname=\"%s\"", md->GetObjectClassName());
1155                 }
1156                 if(md->GetResponsible()[0] != '\0') {
1157                         result += Form(" and CDB_MD:responsible=\"%s\"", md->GetResponsible());
1158                 }
1159                 if(md->GetBeamPeriod() != 0) {
1160                         result += Form(" and CDB_MD:beam_period=%d", md->GetBeamPeriod());
1161                 }
1162                 if(md->GetAliRootVersion()[0] != '\0') {
1163                         result += Form(" and CDB_MD:aliroot_version=\"%s\"", md->GetAliRootVersion());
1164                 }
1165                 if(md->GetComment()[0] != '\0') {
1166                         result += Form(" and CDB_MD:comment=\"%s\"", md->GetComment());
1167                 }
1168         }
1169         AliDebug(2, Form("filter: %s",result.Data()));
1170
1171 }
1172
1173 //_____________________________________________________________________________
1174 Int_t AliCDBGrid::GetLatestVersion(const char* path, Int_t run){
1175 // get last version found in the database valid for run and path
1176
1177         AliCDBPath aCDBPath(path);
1178         if(!aCDBPath.IsValid() || aCDBPath.IsWildcard()) {
1179                 AliError(Form("Invalid path in request: %s", path));
1180                 return -1;
1181         }
1182         AliCDBId query(path, run, run, -1, -1);
1183
1184         AliCDBId* dataId = 0;
1185         // look for file matching query requests (path, runRange, version)
1186         if(run == fRun && fPathFilter.Comprises(aCDBPath) && fVersion < 0){
1187                 // look into list of valid files previously loaded with AliCDBStorage::FillValidFileIds()
1188                 AliDebug(2, Form("List of files valid for run %d and for path %s was loaded. Looking there!",
1189                                         run, path));
1190                 dataId = GetId(fValidFileIds, query);
1191                 if (!dataId) return -1;
1192                 Int_t version = dataId->GetVersion();
1193                 delete dataId;
1194                 return version;
1195
1196         }
1197         // List of files valid for reqested run was not loaded. Looking directly into CDB
1198         AliDebug(2, Form("List of files valid for run %d and for path %s was not loaded. Looking directly into CDB!",
1199                                 run, path));
1200
1201         TObjArray validFileIds;
1202         validFileIds.SetOwner(1);
1203
1204         TString filter;
1205         MakeQueryFilter(run, run, 0, filter);
1206
1207         TString pattern = ".root";
1208
1209         TString folderCopy(Form("%s%s/Run",fDBFolder.Data(),path));
1210
1211         AliDebug(2,Form("** fDBFolder = %s, pattern = %s, filter = %s",folderCopy.Data(), pattern.Data(), filter.Data()));
1212         TGridResult *res = gGrid->Query(folderCopy, pattern, filter, "-y -m");
1213
1214         AliCDBId validFileId;
1215         if (res->GetEntries()>1){
1216                 AliWarning("Number of found entries >1, even if option -y was used");
1217                 for(int i=0; i<res->GetEntries(); i++){
1218                         TString filename = res->GetKey(i, "lfn");
1219                         if(filename == "") continue;
1220                         if(FilenameToId(filename, validFileId))
1221                                 validFileIds.AddLast(validFileId.Clone());
1222                 }
1223                 dataId = GetId(validFileIds, query);
1224                 if (!dataId) return -1;
1225                 
1226                 Int_t version = dataId->GetVersion();
1227                 delete dataId;
1228                 return version;
1229         }
1230         else if (res->GetEntries()==1){
1231                 TString filename = res->GetKey(0, "lfn");
1232                 if(filename == "") {
1233                         AliError("The only entry found has filename empty");
1234                         return -1;
1235                 }
1236                 if(FilenameToId(filename, validFileId)) return validFileId.GetVersion();
1237                 else{
1238                         AliError("Impossible to get FileId from filename");
1239                         return -1;
1240                 }
1241         }
1242         else {
1243                 AliError("No entries found");
1244                 return -1;
1245         }
1246
1247         delete res;
1248
1249
1250 }
1251
1252 //_____________________________________________________________________________
1253 Int_t AliCDBGrid::GetLatestSubVersion(const char* /*path*/, Int_t /*run*/, Int_t /*version*/){
1254 // get last subversion found in the database valid for run and path
1255         AliError("Objects in GRID storage have no sub version!");
1256         return -1;
1257 }
1258
1259
1260 /////////////////////////////////////////////////////////////////////////////////////////////////
1261 //                                                                                             //
1262 // AliCDBGrid factory                                                                          //
1263 //                                                                                             //
1264 /////////////////////////////////////////////////////////////////////////////////////////////////
1265
1266 ClassImp(AliCDBGridFactory)
1267
1268 //_____________________________________________________________________________
1269 Bool_t AliCDBGridFactory::Validate(const char* gridString) {
1270 // check if the string is valid Grid URI
1271
1272         TRegexp gridPattern("^alien://.+$");
1273
1274         return TString(gridString).Contains(gridPattern);
1275 }
1276
1277 //_____________________________________________________________________________
1278 AliCDBParam* AliCDBGridFactory::CreateParameter(const char* gridString) {
1279 // create AliCDBGridParam class from the URI string
1280
1281         if (!Validate(gridString)) {
1282                 return NULL;
1283         }
1284
1285         TString buffer(gridString);
1286
1287         TString gridUrl         = "alien://";
1288         TString user            = "";
1289         TString dbFolder        = "";
1290         TString se              = "default";
1291         TString cacheFolder     = "";
1292         Bool_t  operateDisconnected = kTRUE;
1293         Long64_t cacheSize          = (UInt_t) 1024*1024*1024; // 1GB
1294         Long_t cleanupInterval      = 0;
1295
1296         TObjArray *arr = buffer.Tokenize('?');
1297         TIter iter(arr);
1298         TObjString *str = 0;
1299
1300         while((str = (TObjString*) iter.Next())){
1301                 TString entry(str->String());
1302                 Int_t indeq = entry.Index('=');
1303                 if(indeq == -1) {
1304                         if(entry.BeginsWith("alien://")) { // maybe it's a gridUrl!
1305                                 gridUrl = entry;
1306                                 continue;
1307                         } else {
1308                                 AliError(Form("Invalid entry! %s",entry.Data()));
1309                                 continue;
1310                         }
1311                 }
1312                 
1313                 TString key = entry(0,indeq);
1314                 TString value = entry(indeq+1,entry.Length()-indeq);
1315
1316                 if(key.Contains("grid",TString::kIgnoreCase)) {
1317                         gridUrl += value;
1318                 }
1319                 else if (key.Contains("user",TString::kIgnoreCase)){
1320                         user = value;
1321                 }
1322                 else if (key.Contains("se",TString::kIgnoreCase)){
1323                         se = value;
1324                 }
1325                 else if (key.Contains("cacheF",TString::kIgnoreCase)){
1326                         cacheFolder = value;
1327                         if (!cacheFolder.IsNull() && !cacheFolder.EndsWith("/"))
1328                                 cacheFolder += "/";
1329                 }
1330                 else if (key.Contains("folder",TString::kIgnoreCase)){
1331                         dbFolder = value;
1332                 }
1333                 else if (key.Contains("operateDisc",TString::kIgnoreCase)){
1334                         if(value == "kTRUE") {
1335                                 operateDisconnected = kTRUE;
1336                         } else if (value == "kFALSE") {
1337                                 operateDisconnected = kFALSE;
1338                         } else if (value == "0" || value == "1") {
1339                                 operateDisconnected = (Bool_t) value.Atoi();
1340                         } else {
1341                                 AliError(Form("Invalid entry! %s",entry.Data()));
1342                                 return NULL;
1343                         }
1344                 }
1345                 else if (key.Contains("cacheS",TString::kIgnoreCase)){
1346                         if(value.IsDigit()) {
1347                                 cacheSize = value.Atoi();
1348                         } else {
1349                                 AliError(Form("Invalid entry! %s",entry.Data()));
1350                                 return NULL;
1351                         }
1352                 }
1353                 else if (key.Contains("cleanupInt",TString::kIgnoreCase)){
1354                         if(value.IsDigit()) {
1355                                 cleanupInterval = value.Atoi();
1356                         } else {
1357                                 AliError(Form("Invalid entry! %s",entry.Data()));
1358                                 return NULL;
1359                         }
1360                 }
1361                 else{
1362                         AliError(Form("Invalid entry! %s",entry.Data()));
1363                         return NULL;
1364                 }
1365         }
1366         delete arr; arr=0;
1367
1368         AliDebug(2, Form("gridUrl:      %s", gridUrl.Data()));
1369         AliDebug(2, Form("user: %s", user.Data()));
1370         AliDebug(2, Form("dbFolder:     %s", dbFolder.Data()));
1371         AliDebug(2, Form("s.e.: %s", se.Data()));
1372         AliDebug(2, Form("local cache folder: %s", cacheFolder.Data()));
1373         AliDebug(2, Form("local cache operate disconnected: %d", operateDisconnected));
1374         AliDebug(2, Form("local cache size: %lld", cacheSize));
1375         AliDebug(2, Form("local cache cleanup interval: %ld", cleanupInterval));
1376
1377         if(dbFolder == ""){
1378                 AliError("Base folder must be specified!");
1379                 return NULL;
1380         }
1381
1382         return new AliCDBGridParam(gridUrl.Data(), user.Data(),
1383                           dbFolder.Data(), se.Data(), cacheFolder.Data(),
1384                           operateDisconnected, cacheSize, cleanupInterval);
1385 }
1386
1387 //_____________________________________________________________________________
1388 AliCDBStorage* AliCDBGridFactory::Create(const AliCDBParam* param) {
1389 // create AliCDBGrid storage instance from parameters
1390         
1391         AliCDBGrid *grid = 0;
1392         if (AliCDBGridParam::Class() == param->IsA()) {
1393
1394                 const AliCDBGridParam* gridParam = (const AliCDBGridParam*) param;
1395                 grid = new AliCDBGrid(gridParam->GridUrl().Data(),
1396                                       gridParam->GetUser().Data(),
1397                                       gridParam->GetDBFolder().Data(),
1398                                       gridParam->GetSE().Data(),
1399                                       gridParam->GetCacheFolder().Data(),
1400                                       gridParam->GetOperateDisconnected(),
1401                                       gridParam->GetCacheSize(),
1402                                       gridParam->GetCleanupInterval());
1403
1404         }
1405
1406         if(!gGrid && grid) {
1407                 delete grid; grid=0;
1408         }
1409
1410         return grid;
1411 }
1412
1413 /////////////////////////////////////////////////////////////////////////////////////////////////
1414 //                                                                                             //
1415 // AliCDBGrid Parameter class                                                                  //                                         //
1416 //                                                                                             //
1417 /////////////////////////////////////////////////////////////////////////////////////////////////
1418
1419 ClassImp(AliCDBGridParam)
1420
1421 //_____________________________________________________________________________
1422 AliCDBGridParam::AliCDBGridParam():
1423  AliCDBParam(),
1424  fGridUrl(),
1425  fUser(),
1426  fDBFolder(),
1427  fSE(),
1428  fCacheFolder(),
1429  fOperateDisconnected(),
1430  fCacheSize(),
1431  fCleanupInterval()
1432
1433  {
1434 // default constructor
1435
1436 }
1437
1438 //_____________________________________________________________________________
1439 AliCDBGridParam::AliCDBGridParam(const char* gridUrl, const char* user, const char* dbFolder,
1440                                 const char* se, const char* cacheFolder, Bool_t operateDisconnected,
1441                                 Long64_t cacheSize, Long_t cleanupInterval):
1442  AliCDBParam(),
1443  fGridUrl(gridUrl),
1444  fUser(user),
1445  fDBFolder(dbFolder),
1446  fSE(se),
1447  fCacheFolder(cacheFolder),
1448  fOperateDisconnected(operateDisconnected),
1449  fCacheSize(cacheSize),
1450  fCleanupInterval(cleanupInterval)
1451 {
1452 // constructor
1453
1454         SetType("alien");
1455
1456         TString uri = Form("%s?User=%s?DBFolder=%s?SE=%s?CacheFolder=%s"
1457                         "?OperateDisconnected=%d?CacheSize=%lld?CleanupInterval=%ld",
1458                         fGridUrl.Data(), fUser.Data(),
1459                         fDBFolder.Data(), fSE.Data(), fCacheFolder.Data(),
1460                         fOperateDisconnected, fCacheSize, fCleanupInterval);
1461
1462         SetURI(uri.Data());
1463 }
1464
1465 //_____________________________________________________________________________
1466 AliCDBGridParam::~AliCDBGridParam() {
1467 // destructor
1468
1469 }
1470
1471 //_____________________________________________________________________________
1472 AliCDBParam* AliCDBGridParam::CloneParam() const {
1473 // clone parameter
1474
1475         return new AliCDBGridParam(fGridUrl.Data(), fUser.Data(),
1476                                         fDBFolder.Data(), fSE.Data(), fCacheFolder.Data(),
1477                                         fOperateDisconnected, fCacheSize, fCleanupInterval);
1478 }
1479
1480 //_____________________________________________________________________________
1481 ULong_t AliCDBGridParam::Hash() const {
1482 // return Hash function
1483
1484         return fGridUrl.Hash()+fUser.Hash()+fDBFolder.Hash()+fSE.Hash()+fCacheFolder.Hash();
1485 }
1486
1487 //_____________________________________________________________________________
1488 Bool_t AliCDBGridParam::IsEqual(const TObject* obj) const {
1489 // check if this object is equal to AliCDBParam obj
1490
1491         if (this == obj) {
1492                 return kTRUE;
1493         }
1494
1495         if (AliCDBGridParam::Class() != obj->IsA()) {
1496                 return kFALSE;
1497         }
1498
1499         AliCDBGridParam* other = (AliCDBGridParam*) obj;
1500
1501         if(fGridUrl != other->fGridUrl) return kFALSE;
1502         if(fUser != other->fUser) return kFALSE;
1503         if(fDBFolder != other->fDBFolder) return kFALSE;
1504         if(fSE != other->fSE) return kFALSE;
1505         if(fCacheFolder != other->fCacheFolder) return kFALSE;
1506         if(fOperateDisconnected != other->fOperateDisconnected) return kFALSE;
1507         if(fCacheSize != other->fCacheSize) return kFALSE;
1508         if(fCleanupInterval != other->fCleanupInterval) return kFALSE;
1509         return kTRUE;
1510 }
1511