1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
16 /////////////////////////////////////////////////////////////////////////////////////////////////
19 // access class to a DataBase in an AliEn storage //
21 /////////////////////////////////////////////////////////////////////////////////////////////////
25 #include <TGridResult.h>
30 #include <TObjArray.h>
31 #include <TObjString.h>
35 #include "AliCDBEntry.h"
36 #include "AliCDBGrid.h"
37 #include "AliCDBManager.h"
42 //_____________________________________________________________________________
43 AliCDBGrid::AliCDBGrid(const char *gridUrl, const char *user, const char *dbFolder,
44 const char *se, const char* cacheFolder, Bool_t operateDisconnected,
45 Long64_t cacheSize, Long_t cleanupInterval) :
51 fCacheFolder(cacheFolder),
52 fOperateDisconnected(operateDisconnected),
53 fCacheSize(cacheSize),
54 fCleanupInterval(cleanupInterval)
58 // if the same Grid is alreay active, skip connection
59 if (!gGrid || fGridUrl != gGrid->GridUrl()
60 || (( fUser != "" ) && ( fUser != gGrid->GetUser() )) ) {
61 // connection to the Grid
62 AliInfo("Connection to the Grid...");
64 AliInfo(Form("gGrid = %x; fGridUrl = %s; gGrid->GridUrl() = %s",gGrid,fGridUrl.Data(), gGrid->GridUrl()));
65 AliInfo(Form("fUser = %s; gGrid->GetUser() = %s",fUser.Data(), gGrid->GetUser()));
67 AliDebug(2,Form("1. Grid Url = %s",fGridUrl.Data()));
68 if (fGridUrl == "raw://") fGridUrl = "alien://";
69 AliDebug(2,Form("2. Grid Url = %s (should be different from previous in cas eof raw)", fGridUrl.Data()));
70 TGrid::Connect(fGridUrl.Data(),fUser.Data());
74 AliError("Connection failed!");
78 TString initDir(gGrid->Pwd(0));
79 if (fDBFolder[0] != '/') {
80 fDBFolder.Prepend(initDir);
83 // check DBFolder: trying to cd to DBFolder; if it does not exist, create it
84 if(!gGrid->Cd(fDBFolder.Data(),0)){
85 AliDebug(2,Form("Creating new folder <%s> ...",fDBFolder.Data()));
86 TGridResult* res = gGrid->Command(Form("mkdir -p %s",fDBFolder.Data()));
87 TString result = res->GetKey(0,"__result__");
89 AliFatal(Form("Cannot create folder <%s> !",fDBFolder.Data()));
93 AliDebug(2,Form("Folder <%s> found",fDBFolder.Data()));
96 // removes any '/' at the end of path, then append one '/'
97 while(fDBFolder.EndsWith("/")) fDBFolder.Remove(fDBFolder.Last('/'));
101 fBaseFolder = fDBFolder;
105 // Check if local cache folder is already defined
106 TString origCache(TFile::GetCacheFileDir());
107 if(fCacheFolder.Length() > 0) {
108 if(origCache.Length() == 0) {
109 AliInfo(Form("Setting local cache to: %s", fCacheFolder.Data()));
110 } else if(fCacheFolder != origCache) {
111 AliWarning(Form("Local cache folder was already defined, changing it to: %s",
112 fCacheFolder.Data()));
115 // default settings are: operateDisconnected=kTRUE, forceCacheread = kFALSE
116 if(!TFile::SetCacheFileDir(fCacheFolder.Data(), fOperateDisconnected)) {
117 AliError(Form("Could not set cache folder %s !", fCacheFolder.Data()));
120 // reset fCacheFolder because the function may have
121 // slightly changed the folder name (e.g. '/' added)
122 fCacheFolder = TFile::GetCacheFileDir();
125 // default settings are: cacheSize=1GB, cleanupInterval = 0
126 if(!TFile::ShrinkCacheFileDir(fCacheSize, fCleanupInterval)) {
127 AliError(Form("Could not set following values "
128 "to ShrinkCacheFileDir: cacheSize = %d, cleanupInterval = %d !",
129 fCacheSize, fCleanupInterval));
133 // return to the initial directory
134 gGrid->Cd(initDir.Data(),0);
137 //_____________________________________________________________________________
138 AliCDBGrid::~AliCDBGrid()
141 delete gGrid; gGrid=0;
145 //_____________________________________________________________________________
146 Bool_t AliCDBGrid::FilenameToId(TString& filename, AliCDBId& id) {
147 // build AliCDBId from full path filename (fDBFolder/path/Run#x_#y_v#z_s0.root)
149 if(filename.Contains(fDBFolder)){
150 filename = filename(fDBFolder.Length(),filename.Length()-fDBFolder.Length());
153 TString idPath = filename(0,filename.Last('/'));
155 if(!id.IsValid()) return kFALSE;
157 filename=filename(idPath.Length()+1,filename.Length()-idPath.Length());
160 // valid filename: Run#firstRun_#lastRun_v#version_s0.root
161 TRegexp keyPattern("^Run[0-9]+_[0-9]+_v[0-9]+_s0.root$");
162 keyPattern.Index(filename, &mSize);
165 // TODO backward compatibility ... maybe remove later!
167 TRegexp oldKeyPattern("^Run[0-9]+_[0-9]+_v[0-9]+.root$");
168 oldKeyPattern.Index(filename, &oldmSize);
170 AliDebug(2,Form("Bad filename <%s>.", filename.Data()));
173 AliDebug(2,Form("Old filename format <%s>.", filename.Data()));
174 id.SetSubVersion(-11); // TODO trick to ensure backward compatibility
178 id.SetSubVersion(-1); // TODO trick to ensure backward compatibility
181 filename.Resize(filename.Length() - sizeof(".root") + 1);
183 TObjArray* strArray = (TObjArray*) filename.Tokenize("_");
185 TString firstRunString(((TObjString*) strArray->At(0))->GetString());
186 id.SetFirstRun(atoi(firstRunString.Data() + 3));
187 id.SetLastRun(atoi(((TObjString*) strArray->At(1))->GetString()));
189 TString verString(((TObjString*) strArray->At(2))->GetString());
190 id.SetVersion(atoi(verString.Data() + 1));
197 //_____________________________________________________________________________
198 Bool_t AliCDBGrid::IdToFilename(const AliCDBId& id, TString& filename) const {
199 // build file name from AliCDBId (path, run range, version) and fDBFolder
201 if (!id.GetAliCDBRunRange().IsValid()) {
202 AliDebug(2,Form("Invalid run range <%d, %d>.",
203 id.GetFirstRun(), id.GetLastRun()));
207 if (id.GetVersion() < 0) {
208 AliDebug(2,Form("Invalid version <%d>.", id.GetVersion()));
212 filename = Form("Run%d_%d_v%d",
217 if (id.GetSubVersion() != -11) filename += "_s0"; // TODO to ensure backward compatibility
220 filename.Prepend(fDBFolder + id.GetPath() + '/');
225 //_____________________________________________________________________________
226 Bool_t AliCDBGrid::PrepareId(AliCDBId& id) {
227 // prepare id (version) of the object that will be stored (called by PutEntry)
229 TString initDir(gGrid->Pwd(0));
231 TString dirName(fDBFolder);
233 Bool_t dirExist=kFALSE;
237 // go to the path; if directory does not exist, create it
238 for(int i=0;i<3;i++){
239 dirName+=Form("%s/",id.GetPathLevel(i).Data());
240 dirExist=gGrid->Cd(dirName,0);
242 AliDebug(2,Form("Creating new folder <%s> ...",dirName.Data()));
243 if(!gGrid->Mkdir(dirName,"",0)){
244 AliError(Form("Cannot create directory <%s> !",dirName.Data()));
245 gGrid->Cd(initDir.Data());
249 // if folders are new add tags to them
251 // TODO Currently disabled
252 // add short lived tag!
253 // AliInfo("Tagging level 1 folder with \"ShortLived\" tag");
254 // if(!AddTag(dirName,"ShortLived_try")){
255 // AliError(Form("Could not tag folder %s !", dirName.Data()));
256 // if(!gGrid->Rmdir(dirName.Data())){
257 // AliError(Form("Unexpected: could not remove %s directory!", dirName.Data()));
263 AliDebug(2,"Tagging level 2 folder with \"CDB\" and \"CDB_MD\" tag");
264 if(!AddTag(dirName,"CDB")){
265 AliError(Form("Could not tag folder %s !", dirName.Data()));
266 if(!gGrid->Rmdir(dirName.Data())){
267 AliError(Form("Unexpected: could not remove %s directory!", dirName.Data()));
271 if(!AddTag(dirName,"CDB_MD")){
272 AliError(Form("Could not tag folder %s !", dirName.Data()));
273 if(!gGrid->Rmdir(dirName.Data())){
274 AliError(Form("Unexpected: could not remove %s directory!", dirName.Data()));
279 // TODO Currently disabled
280 // add short lived tag!
281 // TString path=id.GetPath();
282 // if(AliCDBManager::Instance()->IsShortLived(path.Data())) {
283 // AliInfo(Form("Tagging %s as short lived", dirName.Data()));
284 // if(!TagShortLived(dirName, kTRUE)){
285 // AliError(Form("Could not tag folder %s !", dirName.Data()));
286 // if(!gGrid->Rmdir(dirName.Data())){
287 // AliError(Form("Unexpected: could not remove %s directory!", dirName.Data()));
292 // AliInfo(Form("Tagging %s as long lived", dirName.Data()));
293 // if(!TagShortLived(dirName, kFALSE)){
294 // AliError(Form("Could not tag folder %s !", dirName.Data()));
295 // if(!gGrid->Rmdir(dirName.Data())){
296 // AliError(Form("Unexpected: could not remove %s directory!", dirName.Data()));
304 gGrid->Cd(initDir,0);
307 AliCDBId anId; // the id got from filename
308 AliCDBRunRange lastRunRange(-1,-1); // highest runRange found
309 Int_t lastVersion=0; // highest version found
311 TGridResult *res = gGrid->Ls(dirName);
313 //loop on the files in the directory, look for highest version
314 for(int i=0; i < res->GetEntries(); i++){
315 filename=res->GetFileNamePath(i);
316 if (!FilenameToId(filename, anId)) continue;
317 if (anId.GetAliCDBRunRange().Overlaps(id.GetAliCDBRunRange()) && anId.GetVersion() > lastVersion) {
318 lastVersion = anId.GetVersion();
319 lastRunRange = anId.GetAliCDBRunRange();
325 if (!id.HasVersion()){
326 id.SetVersion(lastVersion + 1);
330 TString lastStorage = id.GetLastStorage();
331 if(lastStorage.Contains(TString("new"), TString::kIgnoreCase) && id.GetVersion() > 1 ){
332 AliDebug(2, Form("A NEW object is being stored with version %d",
334 AliDebug(2, Form("and it will hide previously stored object with version %d!",
338 if(!lastRunRange.IsAnyRange() && !(lastRunRange.IsEqual(&id.GetAliCDBRunRange())))
339 AliWarning(Form("Run range modified w.r.t. previous version (Run%d_%d_v%d)",
340 lastRunRange.GetFirstRun(), lastRunRange.GetLastRun(), id.GetVersion()));
345 //_____________________________________________________________________________
346 AliCDBId* AliCDBGrid::GetId(const TObjArray& validFileIds, const AliCDBId& query) {
347 // look for the Id that matches query's requests (highest or exact version)
349 if(validFileIds.GetEntriesFast() < 1) {
351 } else if (validFileIds.GetEntriesFast() == 1) {
352 return dynamic_cast<AliCDBId*> (validFileIds.At(0)->Clone());
355 TIter iter(&validFileIds);
360 while((anIdPtr = dynamic_cast<AliCDBId*> (iter.Next()))){
361 if(anIdPtr->GetPath() != query.GetPath()) continue;
363 //if(!CheckVersion(query, anIdPtr, result)) return NULL;
365 if (!query.HasVersion()){ // look for highest version
366 if(result && result->GetVersion() > anIdPtr->GetVersion()) continue;
367 if(result && result->GetVersion() == anIdPtr->GetVersion()) {
368 AliError(Form("More than one object valid for run %d, version %d!",
369 query.GetFirstRun(), anIdPtr->GetVersion()));
373 } else { // look for specified version
374 if(query.GetVersion() != anIdPtr->GetVersion()) continue;
375 if(result && result->GetVersion() == anIdPtr->GetVersion()){
376 AliError(Form("More than one object valid for run %d, version %d!",
377 query.GetFirstRun(), anIdPtr->GetVersion()));
385 if (!result) return NULL;
387 return dynamic_cast<AliCDBId*> (result->Clone());
390 //_____________________________________________________________________________
391 AliCDBId* AliCDBGrid::GetEntryId(const AliCDBId& queryId) {
392 // get AliCDBId from the database
393 // User must delete returned object
397 AliCDBId selectedId(queryId);
398 if (!selectedId.HasVersion()) {
399 // if version is not specified, first check the selection criteria list
400 GetSelection(&selectedId);
403 TObjArray validFileIds;
404 validFileIds.SetOwner(1);
406 // look for file matching query requests (path, runRange, version)
407 if(selectedId.GetFirstRun() == fRun &&
408 fPathFilter.Comprises(selectedId.GetAliCDBPath()) && fVersion < 0 && !fMetaDataFilter){
409 // look into list of valid files previously loaded with AliCDBStorage::FillValidFileIds()
410 AliDebug(2, Form("List of files valid for run %d and for path %s was loaded. Looking there!",
411 selectedId.GetFirstRun(), selectedId.GetPath().Data()));
412 dataId = GetId(fValidFileIds, selectedId);
415 // List of files valid for reqested run was not loaded. Looking directly into CDB
416 AliDebug(2, Form("List of files valid for run %d and for path %s was not loaded. Looking directly into CDB!",
417 selectedId.GetFirstRun(), selectedId.GetPath().Data()));
420 MakeQueryFilter(selectedId.GetFirstRun(), selectedId.GetLastRun(), 0, filter);
422 TString pattern = Form("%s/Run*", selectedId.GetPath().Data());
423 if(selectedId.GetVersion() >= 0) pattern += Form("_v%d*",selectedId.GetVersion());
425 AliDebug(2,Form("pattern: %s", pattern.Data()));
427 TGridResult *res = gGrid->Query(fDBFolder, pattern, filter, "");
429 AliCDBId validFileId;
430 for(int i=0; i<res->GetEntries(); i++){
431 TString filename = res->GetKey(i, "lfn");
432 if(filename == "") continue;
433 if(FilenameToId(filename, validFileId))
434 validFileIds.AddLast(validFileId.Clone());
438 dataId = GetId(validFileIds, selectedId);
444 //_____________________________________________________________________________
445 AliCDBEntry* AliCDBGrid::GetEntry(const AliCDBId& queryId) {
446 // get AliCDBEntry from the database
448 AliCDBId* dataId = GetEntryId(queryId);
450 if (!dataId) return NULL;
453 if (!IdToFilename(*dataId, filename)) {
454 AliDebug(2,Form("Bad data ID encountered! Subnormal error!"));
459 AliCDBEntry* anEntry = GetEntryFromFile(filename, dataId);
465 //_____________________________________________________________________________
466 AliCDBEntry* AliCDBGrid::GetEntryFromFile(TString& filename, AliCDBId* dataId){
467 // Get AliCBEntry object from file "filename"
469 AliDebug(2,Form("Opening file: %s",filename.Data()));
471 filename.Prepend("/alien");
473 // if option="CACHEREAD" TFile will use the local caching facility!
474 TString option="READ";
475 if(fCacheFolder != ""){
477 // Check if local cache folder was changed in the meanwhile
478 TString origCache(TFile::GetCacheFileDir());
479 if(fCacheFolder != origCache) {
480 AliWarning(Form("Local cache folder has been overwritten!! fCacheFolder = %s origCache = %s",
481 fCacheFolder.Data(), origCache.Data()));
482 TFile::SetCacheFileDir(fCacheFolder.Data(), fOperateDisconnected);
483 TFile::ShrinkCacheFileDir(fCacheSize, fCleanupInterval);
486 option.Prepend("CACHE");
489 AliDebug(2, Form("Option: %s", option.Data()));
491 TFile *file = TFile::Open(filename, option);
493 AliDebug(2,Form("Can't open file <%s>!", filename.Data()));
497 // get the only AliCDBEntry object from the file
498 // the object in the file is an AliCDBEntry entry named "AliCDBEntry"
500 AliCDBEntry* anEntry = dynamic_cast<AliCDBEntry*> (file->Get("AliCDBEntry"));
503 AliDebug(2,Form("Bad storage data: file does not contain an AliCDBEntry object!"));
508 // The object's Id is not reset during storage
509 // If object's Id runRange or version do not match with filename,
510 // it means that someone renamed file by hand. In this case a warning msg is issued.
513 AliCDBId entryId = anEntry->GetId();
514 Int_t tmpSubVersion = dataId->GetSubVersion();
515 dataId->SetSubVersion(entryId.GetSubVersion()); // otherwise filename and id may mismatch
516 if(!entryId.IsEqual(dataId)){
517 AliWarning(Form("Mismatch between file name and object's Id!"));
518 AliWarning(Form("File name: %s", dataId->ToString().Data()));
519 AliWarning(Form("Object's Id: %s", entryId.ToString().Data()));
521 dataId->SetSubVersion(tmpSubVersion);
524 anEntry->SetLastStorage("grid");
526 // Check whether entry contains a TTree. In case load the tree in memory!
527 LoadTreeFromFile(anEntry);
529 // close file, return retieved entry
530 file->Close(); delete file; file=0;
535 //_____________________________________________________________________________
536 TList* AliCDBGrid::GetEntries(const AliCDBId& queryId) {
537 // multiple request (AliCDBStorage::GetAll)
539 TList* result = new TList();
542 TObjArray validFileIds;
543 validFileIds.SetOwner(1);
545 Bool_t alreadyLoaded = kFALSE;
547 // look for file matching query requests (path, runRange)
548 if(queryId.GetFirstRun() == fRun &&
549 fPathFilter.Comprises(queryId.GetAliCDBPath()) && fVersion < 0 && !fMetaDataFilter){
550 // look into list of valid files previously loaded with AliCDBStorage::FillValidFileIds()
551 AliDebug(2,Form("List of files valid for run %d and for path %s was loaded. Looking there!",
552 queryId.GetFirstRun(), queryId.GetPath().Data()));
554 alreadyLoaded = kTRUE;
557 // List of files valid for reqested run was not loaded. Looking directly into CDB
558 AliDebug(2,Form("List of files valid for run %d and for path %s was not loaded. Looking directly into CDB!",
559 queryId.GetFirstRun(), queryId.GetPath().Data()));
562 MakeQueryFilter(queryId.GetFirstRun(), queryId.GetLastRun(), 0, filter);
564 TString pattern = Form("%s/Run*.root", queryId.GetPath().Data());
565 AliDebug(2,Form("pattern: %s", pattern.Data()));
567 TGridResult *res = gGrid->Query(fDBFolder, pattern, filter, "");
569 AliCDBId validFileId;
570 for(int i=0; i<res->GetEntries(); i++){
571 TString filename = res->GetKey(i, "lfn");
572 if(filename == "") continue;
573 if(FilenameToId(filename, validFileId))
574 validFileIds.AddLast(validFileId.Clone());
581 iter = new TIter(&fValidFileIds);
583 iter = new TIter(&validFileIds);
586 TObjArray selectedIds;
587 selectedIds.SetOwner(1);
589 // loop on list of valid Ids to select the right version to get.
590 // According to query and to the selection criteria list, version can be the highest or exact
594 AliCDBPath queryPath = queryId.GetAliCDBPath();
595 while((anIdPtr = dynamic_cast<AliCDBId*> (iter->Next()))){
596 AliCDBPath thisCDBPath = anIdPtr->GetAliCDBPath();
597 if(!(queryPath.Comprises(thisCDBPath)) || pathCopy.GetPath() == thisCDBPath.GetPath()) continue;
598 pathCopy = thisCDBPath;
600 // check the selection criteria list for this query
601 AliCDBId thisId(*anIdPtr);
602 thisId.SetVersion(queryId.GetVersion());
603 if(!thisId.HasVersion()) GetSelection(&thisId);
606 dataId = GetId(fValidFileIds, thisId);
608 dataId = GetId(validFileIds, thisId);
610 if(dataId) selectedIds.Add(dataId);
615 // selectedIds contains the Ids of the files matching all requests of query!
616 // All the objects are now ready to be retrieved
617 iter = new TIter(&selectedIds);
618 while((anIdPtr = dynamic_cast<AliCDBId*> (iter->Next()))){
620 if (!IdToFilename(*anIdPtr, filename)) {
621 AliDebug(2,Form("Bad data ID encountered! Subnormal error!"));
625 AliCDBEntry* anEntry = GetEntryFromFile(filename, anIdPtr);
627 if(anEntry) result->Add(anEntry);
635 //_____________________________________________________________________________
636 Bool_t AliCDBGrid::PutEntry(AliCDBEntry* entry) {
637 // put an AliCDBEntry object into the database
639 AliCDBId& id = entry->GetId();
641 // set version for the entry to be stored
642 if (!PrepareId(id)) return kFALSE;
644 // build filename from entry's id
646 if (!IdToFilename(id, filename)) {
647 AliError("Bad ID encountered! Subnormal error!");
651 TString folderToTag = Form("%s%s",
653 id.GetPath().Data());
655 TDirectory* saveDir = gDirectory;
657 TString fullFilename = Form("/alien%s", filename.Data());
658 // specify SE to filename
659 if (fSE != "default") fullFilename += Form("?se=%s",fSE.Data());
662 TFile *file = TFile::Open(fullFilename,"CREATE");
663 if(!file || !file->IsWritable()){
664 AliError(Form("Can't open file <%s>!", filename.Data()));
665 if(file && !file->IsWritable()) file->Close(); delete file; file=0;
671 //SetTreeToFile(entry, file);
673 entry->SetVersion(id.GetVersion());
675 // write object (key name: "AliCDBEntry")
676 Bool_t result = (file->WriteTObject(entry, "AliCDBEntry") != 0);
677 if (!result) AliError(Form("Can't write entry to file <%s>!", filename.Data()));
680 if (saveDir) saveDir->cd(); else gROOT->cd();
681 file->Close(); delete file; file=0;
685 if(!TagFileId(filename, &id)){
686 AliInfo(Form("CDB tagging failed. Deleting file %s!",filename.Data()));
687 if(!gGrid->Rm(filename.Data()))
688 AliError("Can't delete file!");
692 TagFileMetaData(filename, entry->GetMetaData());
695 AliInfo(Form("CDB object stored into file %s", filename.Data()));
696 AliInfo(Form("Storage Element: %s", fSE.Data()));
699 //_____________________________________________________________________________
700 Bool_t AliCDBGrid::AddTag(TString& folderToTag, const char* tagname){
701 // add "tagname" tag (CDB or CDB_MD) to folder where object will be stored
703 Bool_t result = kTRUE;
704 AliDebug(2, Form("adding %s tag to folder %s", tagname, folderToTag.Data()));
705 TString addTag = Form("addTag %s %s", folderToTag.Data(), tagname);
706 TGridResult *gridres = gGrid->Command(addTag.Data());
707 const char* resCode = gridres->GetKey(0,"__result__"); // '1' if success
708 if(resCode[0] != '1') {
709 AliError(Form("Couldn't add %s tags to folder %s !",
710 tagname, folderToTag.Data()));
717 //_____________________________________________________________________________
718 Bool_t AliCDBGrid::TagFileId(TString& filename, const AliCDBId* id){
719 // tag stored object in CDB table using object Id's parameters
721 TString addTagValue1 = Form("addTagValue %s CDB ", filename.Data());
722 TString addTagValue2 = Form("first_run=%d last_run=%d version=%d ",
726 TString addTagValue3 = Form("path_level_0=\"%s\" path_level_1=\"%s\" path_level_2=\"%s\"",
727 id->GetPathLevel(0).Data(),
728 id->GetPathLevel(1).Data(),
729 id->GetPathLevel(2).Data());
730 TString addTagValue = Form("%s%s%s",
733 addTagValue3.Data());
735 Bool_t result = kFALSE;
736 AliDebug(2, Form("Tagging file. Tag command: %s", addTagValue.Data()));
737 TGridResult* res = gGrid->Command(addTagValue.Data());
738 const char* resCode = res->GetKey(0,"__result__"); // '1' if success
739 if(resCode[0] != '1') {
740 AliError(Form("Couldn't add CDB tag value to file %s !",
744 AliDebug(2, "Object successfully tagged.");
752 //_____________________________________________________________________________
753 Bool_t AliCDBGrid::TagShortLived(TString& filename, Bool_t value){
754 // tag folder with ShortLived tag
756 TString addTagValue = Form("addTagValue %s ShortLived_try value=%d", filename.Data(), value);
758 Bool_t result = kFALSE;
759 AliDebug(2, Form("Tagging file. Tag command: %s", addTagValue.Data()));
760 TGridResult* res = gGrid->Command(addTagValue.Data());
761 const char* resCode = res->GetKey(0,"__result__"); // '1' if success
762 if(resCode[0] != '1') {
763 AliError(Form("Couldn't add ShortLived tag value to file %s !", filename.Data()));
766 AliDebug(2,"Object successfully tagged.");
774 //_____________________________________________________________________________
775 Bool_t AliCDBGrid::TagFileMetaData(TString& filename, const AliCDBMetaData* md){
776 // tag stored object in CDB table using object Id's parameters
778 TString addTagValue1 = Form("addTagValue %s CDB_MD ", filename.Data());
779 TString addTagValue2 = Form("object_classname=\"%s\" responsible=\"%s\" beam_period=%d ",
780 md->GetObjectClassName(),
781 md->GetResponsible(),
782 md->GetBeamPeriod());
783 TString addTagValue3 = Form("aliroot_version=\"%s\" comment=\"%s\"",
784 md->GetAliRootVersion(),
786 TString addTagValue = Form("%s%s%s",
789 addTagValue3.Data());
791 Bool_t result = kFALSE;
792 AliDebug(2, Form("Tagging file. Tag command: %s", addTagValue.Data()));
793 TGridResult* res = gGrid->Command(addTagValue.Data());
794 const char* resCode = res->GetKey(0,"__result__"); // '1' if success
795 if(resCode[0] != '1') {
796 AliWarning(Form("Couldn't add CDB_MD tag value to file %s !",
800 AliDebug(2,"Object successfully tagged.");
806 //_____________________________________________________________________________
807 TList* AliCDBGrid::GetIdListFromFile(const char* fileName){
809 TString turl(fileName);
810 turl.Prepend("/alien" + fDBFolder);
811 turl += "?se="; turl += fSE.Data();
812 TFile *file = TFile::Open(turl);
814 AliError(Form("Can't open selection file <%s>!", turl.Data()));
818 TList *list = new TList();
826 keycycle = "AliCDBId;";
829 id = (AliCDBId*) file->Get(keycycle);
833 file->Close(); delete file; file=0;
840 //_____________________________________________________________________________
841 Bool_t AliCDBGrid::Contains(const char* path) const{
842 // check for path in storage's DBFolder
844 TString initDir(gGrid->Pwd(0));
845 TString dirName(fDBFolder);
846 dirName += path; // dirName = fDBFolder/path
847 Bool_t result=kFALSE;
848 if (gGrid->Cd(dirName,0)) result=kTRUE;
849 gGrid->Cd(initDir.Data(),0);
853 //_____________________________________________________________________________
854 void AliCDBGrid::QueryValidFiles()
856 // Query the CDB for files valid for AliCDBStorage::fRun
857 // fills list fValidFileIds with AliCDBId objects created from file name
860 MakeQueryFilter(fRun, fRun, fMetaDataFilter, filter);
862 TString pattern = Form("%s/Run*", fPathFilter.GetPath().Data());
863 if(fVersion >= 0) pattern += Form("_v%d*", fVersion);
865 AliDebug(2,Form("pattern: %s", pattern.Data()));
867 TGridResult *res = gGrid->Query(fDBFolder, pattern, filter, "");
870 AliError("Grid query failed");
874 AliCDBId validFileId;
875 for(int i=0; i<res->GetEntries(); i++){
876 TString filename = res->GetKey(i, "lfn");
877 if(filename == "") continue;
878 AliDebug(2,Form("Found valid file: %s", filename.Data()));
879 Bool_t result = FilenameToId(filename, validFileId);
881 fValidFileIds.AddLast(validFileId.Clone());
888 //_____________________________________________________________________________
889 void AliCDBGrid::MakeQueryFilter(Int_t firstRun, Int_t lastRun,
890 const AliCDBMetaData* md, TString& result) const
892 // create filter for file query
894 result = Form("CDB:first_run<=%d and CDB:last_run>=%d", firstRun, lastRun);
896 // if(version >= 0) {
897 // result += Form(" and CDB:version=%d", version);
899 // if(pathFilter.GetLevel0() != "*") {
900 // result += Form(" and CDB:path_level_0=\"%s\"", pathFilter.GetLevel0().Data());
902 // if(pathFilter.GetLevel1() != "*") {
903 // result += Form(" and CDB:path_level_1=\"%s\"", pathFilter.GetLevel1().Data());
905 // if(pathFilter.GetLevel2() != "*") {
906 // result += Form(" and CDB:path_level_2=\"%s\"", pathFilter.GetLevel2().Data());
910 if(md->GetObjectClassName()[0] != '\0') {
911 result += Form(" and CDB_MD:object_classname=\"%s\"", md->GetObjectClassName());
913 if(md->GetResponsible()[0] != '\0') {
914 result += Form(" and CDB_MD:responsible=\"%s\"", md->GetResponsible());
916 if(md->GetBeamPeriod() != 0) {
917 result += Form(" and CDB_MD:beam_period=%d", md->GetBeamPeriod());
919 if(md->GetAliRootVersion()[0] != '\0') {
920 result += Form(" and CDB_MD:aliroot_version=\"%s\"", md->GetAliRootVersion());
922 if(md->GetComment()[0] != '\0') {
923 result += Form(" and CDB_MD:comment=\"%s\"", md->GetComment());
926 AliDebug(2, Form("filter: %s",result.Data()));
930 //_____________________________________________________________________________
931 Int_t AliCDBGrid::GetLatestVersion(const char* path, Int_t run){
932 // get last version found in the database valid for run and path
934 AliCDBPath aCDBPath(path);
935 if(!aCDBPath.IsValid() || aCDBPath.IsWildcard()) {
936 AliError(Form("Invalid path in request: %s", path));
939 AliCDBId query(path, run, run, -1, -1);
940 AliCDBId* dataId = 0;
942 // look for file matching query requests (path, runRange, version)
943 if(run == fRun && fPathFilter.Comprises(aCDBPath) && fVersion < 0){
944 // look into list of valid files previously loaded with AliCDBStorage::FillValidFileIds()
945 AliDebug(2, Form("List of files valid for run %d and for path %s was loaded. Looking there!",
947 dataId = GetId(fValidFileIds, query);
948 if (!dataId) return -1;
949 Int_t version = dataId->GetVersion();
954 // List of files valid for reqested run was not loaded. Looking directly into CDB
955 AliDebug(2, Form("List of files valid for run %d and for path %s was not loaded. Looking directly into CDB!",
958 TObjArray validFileIds;
959 validFileIds.SetOwner(1);
962 MakeQueryFilter(run, run, 0, filter);
964 TString pattern = Form("%s/Run*.root", path);
965 AliDebug(2,Form("pattern: %s", pattern.Data()));
967 TGridResult *res = gGrid->Query(fDBFolder, pattern, filter, "");
968 AliCDBId validFileId;
969 for(int i=0; i<res->GetEntries(); i++){
970 TString filename = res->GetKey(i, "lfn");
971 if(filename == "") continue;
972 if(FilenameToId(filename, validFileId))
973 validFileIds.AddLast(validFileId.Clone());
977 dataId = GetId(validFileIds, query);
978 if (!dataId) return -1;
980 Int_t version = dataId->GetVersion();
986 //_____________________________________________________________________________
987 Int_t AliCDBGrid::GetLatestSubVersion(const char* /*path*/, Int_t /*run*/, Int_t /*version*/){
988 // get last subversion found in the database valid for run and path
989 AliError("Objects in GRID storage have no sub version!");
994 /////////////////////////////////////////////////////////////////////////////////////////////////
996 // AliCDBGrid factory //
998 /////////////////////////////////////////////////////////////////////////////////////////////////
1000 ClassImp(AliCDBGridFactory)
1002 //_____________________________________________________________________________
1003 Bool_t AliCDBGridFactory::Validate(const char* gridString) {
1004 // check if the string is valid Grid URI
1006 TRegexp gridPattern("^alien://.+$");
1007 TRegexp gridRawPattern("^raw://.+$");
1009 return (TString(gridString).Contains(gridPattern)||TString(gridString).Contains(gridRawPattern));
1012 //_____________________________________________________________________________
1013 AliCDBParam* AliCDBGridFactory::CreateParameter(const char* gridString) {
1014 // create AliCDBGridParam class from the URI string
1016 if (!Validate(gridString)) {
1020 TString buffer(gridString);
1021 Bool_t rawFlag = kFALSE; // flag to say whether we are in the "raw://" case
1023 TString gridUrl = "alien://";
1025 TString dbFolder = "";
1026 TString se = "default";
1027 TString cacheFolder = "";
1028 Bool_t operateDisconnected = kTRUE;
1029 Long64_t cacheSize = (UInt_t) 1024*1024*1024; // 1GB
1030 Long_t cleanupInterval = 0;
1032 TObjArray *arr = buffer.Tokenize('?');
1034 TObjString *str = 0;
1036 while((str = (TObjString*) iter.Next())){
1037 TString entry(str->String());
1038 Int_t indeq = entry.Index('=');
1040 if(entry.BeginsWith("alien://")) { // maybe it's a gridUrl!
1043 } else if(entry.BeginsWith("raw://")) { // maybe it's a gridRawUrl!
1048 AliError(Form("Invalid entry! %s",entry.Data()));
1053 TString key = entry(0,indeq);
1054 TString value = entry(indeq+1,entry.Length()-indeq);
1056 if(key.Contains("grid",TString::kIgnoreCase)) {
1059 else if (key.Contains("user",TString::kIgnoreCase)){
1062 else if (key.Contains("se",TString::kIgnoreCase)){
1065 else if (key.Contains("cacheF",TString::kIgnoreCase)){
1066 cacheFolder = value;
1067 if (!cacheFolder.EndsWith("/"))
1070 else if (key.Contains("folder",TString::kIgnoreCase)){
1073 else if (key.Contains("operateDisc",TString::kIgnoreCase)){
1074 if(value == "kTRUE") {
1075 operateDisconnected = kTRUE;
1076 } else if (value == "kFALSE") {
1077 operateDisconnected = kFALSE;
1078 } else if (value == "0" || value == "1") {
1079 operateDisconnected = (Bool_t) value.Atoi();
1081 AliError(Form("Invalid entry! %s",entry.Data()));
1085 else if (key.Contains("cacheS",TString::kIgnoreCase)){
1086 if(value.IsDigit()) {
1087 cacheSize = value.Atoi();
1089 AliError(Form("Invalid entry! %s",entry.Data()));
1093 else if (key.Contains("cleanupInt",TString::kIgnoreCase)){
1094 if(value.IsDigit()) {
1095 cleanupInterval = value.Atoi();
1097 AliError(Form("Invalid entry! %s",entry.Data()));
1102 AliError(Form("Invalid entry! %s",entry.Data()));
1108 AliDebug(2, Form("gridUrl: %s", gridUrl.Data()));
1109 AliDebug(2, Form("user: %s", user.Data()));
1110 AliDebug(2, Form("dbFolder: %s", dbFolder.Data()));
1111 AliDebug(2, Form("s.e.: %s", se.Data()));
1112 AliDebug(2, Form("local cache folder: %s", cacheFolder.Data()));
1113 AliDebug(2, Form("local cache operate disconnected: %d", operateDisconnected));
1114 AliDebug(2, Form("local cache size: %d", cacheSize));
1115 AliDebug(2, Form("local cache cleanup interval: %d", cleanupInterval));
1117 if(dbFolder == "" && !rawFlag){
1118 AliError("Base folder must be specified!");
1122 return new AliCDBGridParam(gridUrl.Data(), user.Data(),
1123 dbFolder.Data(), se.Data(), cacheFolder.Data(),
1124 operateDisconnected, cacheSize, cleanupInterval);
1127 //_____________________________________________________________________________
1128 AliCDBStorage* AliCDBGridFactory::Create(const AliCDBParam* param) {
1129 // create AliCDBGrid storage instance from parameters
1131 AliCDBGrid *grid = 0;
1132 if (AliCDBGridParam::Class() == param->IsA()) {
1134 const AliCDBGridParam* gridParam = (const AliCDBGridParam*) param;
1135 grid = new AliCDBGrid(gridParam->GridUrl().Data(),
1136 gridParam->GetUser().Data(),
1137 gridParam->GetDBFolder().Data(),
1138 gridParam->GetSE().Data(),
1139 gridParam->GetCacheFolder().Data(),
1140 gridParam->GetOperateDisconnected(),
1141 gridParam->GetCacheSize(),
1142 gridParam->GetCleanupInterval());
1146 if(!gGrid && grid) {
1147 delete grid; grid=0;
1153 /////////////////////////////////////////////////////////////////////////////////////////////////
1155 // AliCDBGrid Parameter class // //
1157 /////////////////////////////////////////////////////////////////////////////////////////////////
1159 ClassImp(AliCDBGridParam)
1161 //_____________________________________________________________________________
1162 AliCDBGridParam::AliCDBGridParam():
1169 fOperateDisconnected(),
1174 // default constructor
1178 //_____________________________________________________________________________
1179 AliCDBGridParam::AliCDBGridParam(const char* gridUrl, const char* user, const char* dbFolder,
1180 const char* se, const char* cacheFolder, Bool_t operateDisconnected,
1181 Long64_t cacheSize, Long_t cleanupInterval):
1185 fDBFolder(dbFolder),
1187 fCacheFolder(cacheFolder),
1188 fOperateDisconnected(operateDisconnected),
1189 fCacheSize(cacheSize),
1190 fCleanupInterval(cleanupInterval)
1196 TString uri = Form("%s?User=%s?DBFolder=%s?SE=%s?CacheFolder=%s"
1197 "?OperateDisconnected=%d?CacheSize=%d?CleanupInterval=%d",
1198 fGridUrl.Data(), fUser.Data(),
1199 fDBFolder.Data(), fSE.Data(), fCacheFolder.Data(),
1200 fOperateDisconnected, fCacheSize, fCleanupInterval);
1202 AliDebug(2,Form("uri = %s", uri.Data()));
1206 //_____________________________________________________________________________
1207 AliCDBGridParam::~AliCDBGridParam() {
1212 //_____________________________________________________________________________
1213 AliCDBParam* AliCDBGridParam::CloneParam() const {
1216 return new AliCDBGridParam(fGridUrl.Data(), fUser.Data(),
1217 fDBFolder.Data(), fSE.Data(), fCacheFolder.Data(),
1218 fOperateDisconnected, fCacheSize, fCleanupInterval);
1221 //_____________________________________________________________________________
1222 ULong_t AliCDBGridParam::Hash() const {
1223 // return Hash function
1225 return fGridUrl.Hash()+fUser.Hash()+fDBFolder.Hash()+fSE.Hash()+fCacheFolder.Hash();
1228 //_____________________________________________________________________________
1229 Bool_t AliCDBGridParam::IsEqual(const TObject* obj) const {
1230 // check if this object is equal to AliCDBParam obj
1236 if (AliCDBGridParam::Class() != obj->IsA()) {
1240 AliCDBGridParam* other = (AliCDBGridParam*) obj;
1242 if(fGridUrl != other->fGridUrl) return kFALSE;
1243 if(fUser != other->fUser) return kFALSE;
1244 if(fDBFolder != other->fDBFolder) return kFALSE;
1245 if(fSE != other->fSE) return kFALSE;
1246 if(fCacheFolder != other->fCacheFolder) return kFALSE;
1247 if(fOperateDisconnected != other->fOperateDisconnected) return kFALSE;
1248 if(fCacheSize != other->fCacheSize) return kFALSE;
1249 if(fCleanupInterval != other->fCleanupInterval) return kFALSE;