FCcmd.cxx File Reference

#include "PoolSvc/FileCatalogUtils.h"
#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/ISvcLocator.h"
#include "StorageSvc/SimpleUtilityBase.h"
#include <exception>
#include <memory>

Include dependency graph for FCcmd.cxx:

Go to the source code of this file.

Classes
class	Options

Functions
void	printUsage ()
int	main (int argc, char **argv)

Function Documentation

main()

int main	(	int	argc,
		char **	argv )

Definition at line 98 of file FCcmd.cxx.

{
    SimpleUtilityBase::initGaudi();
 
    Options options(argc, argv);
    // possible options:
    // const char* opts[] = {"n","l","p","g","u","h",0};
    if( options.exists('h') ) {
        printUsage();
        return 0;
    }
    if( argc < 3 ){
        cerr << "ERROR! insufficient number of arguments" << endl;
        printUsage();
        return 1;
    }
    string  catName;
    // In case none of the options below work, the default FC name is specified in FileCatalogUtils::addCatalog()
    if( options.exists('u') ){
        catName=options.getOptByName('u');
    }else{
        catName=getEnvStr("POOL_CATALOG");
    }
    string lfn    = options.getOptByName('l');
    string pfn    = options.getOptByName('p');
    string newpfn = options.getOptByName('n');
    string guid   = options.getOptByName('g');
    string action = options.getAction();
    string item   = options.getItem();
    
    try{
        SmartIF<Gaudi::IFileCatalogMgr> catalogMgr = Gaudi::svcLocator()->service<Gaudi::IFileCatalogMgr>( "Gaudi::MultiFileCatalog" );
        SmartIF<Gaudi::IFileCatalog> mycatalog( catalogMgr );
        FileCatalogUtils::addCatalog( *catalogMgr, catName, true );
        mycatalog->init();
 
        if( action == "delete" ) {
            if( !lfn.empty() ){
                mycatalog->deleteFID( mycatalog->lookupLFN( lfn ) );
            }else if( !pfn.empty() ) {
                mycatalog->deleteFID( mycatalog->lookupPFN( pfn ) );
            }else {
                cerr << "ERROR! Use PFN or LFN to specify the entry for deletion: " << endl;
                printUsage();
                return 2;
            }
 
        } else if( action == "list" ) {
            if( item == "lfn" ) {
                Gaudi::IFileCatalog::Strings fids;
                if( !pfn.empty() ) {
                    fids.push_back( mycatalog->lookupPFN( pfn ) );
                }else{
                    mycatalog->getFID( fids );
                }
                for( const auto& fid: fids ) {
                    Gaudi::IFileCatalog::Files files;
                    mycatalog->getLFN( fid, files );
                    for( const auto& file: files ) {
                        cout << file.first << " ,   " << file.second << endl;
                    }
                }
            } 
            else if( item == "pfn" ) {
                Gaudi::IFileCatalog::Strings fids;
                if( !lfn.empty() ) {
                    fids.push_back( mycatalog->lookupLFN( lfn ) );
                } else if( !guid.empty() ) {
                    fids.emplace_back( std::move(guid) );
                } else {
                    // go through all FIDs in the catalog
                    mycatalog->getFID( fids );
                }
                for( const auto& fid: fids ) {
                    Gaudi::IFileCatalog::Files files;
                    mycatalog->getPFN( fid, files );
                    for( const auto& file: files ) {
                        string pf = file.first;
                        string filetype = (file.second.empty()? string("NULL") : file.second);
                        cout<<pf<<"    "<<filetype<<endl;
                    }
                }
            }
            else if( item == "guid" ) {
                Gaudi::IFileCatalog::Strings fids;
                if( !pfn.empty() ){
                    fids.push_back( mycatalog->lookupPFN( pfn ) );
                } else if( !lfn.empty() ){
                    fids.push_back( mycatalog->lookupLFN( lfn ) );
                } else {
                    mycatalog->getFID( fids );
                }
                for( const auto& fid: fids ) {
                    cout << fid << endl;
                }
            }
            else {
                cerr<< "ERROR! Unsupported catalog item type for listing: "<< item << endl;
                printUsage();
                return 2;
            }
            
        } else if( action == "register" ) {
            if( item == "lfn" ) {
                if( pfn.empty() || lfn.empty() ){
                    cerr<<"ERROR! You must specify PFName using -p and LFName using -l" << endl;
                    return 3;
                }
                mycatalog->registerLFN( mycatalog->lookupPFN(pfn), lfn );
            }
            else if( item == "pfn" ) {
                if( pfn.empty() or guid.empty() ) {
                    cerr<<"ERROR! You must specify PFName using -p and GUID using -g options" << endl;
                    return 3;
                }
                FileCatalogUtils::registerPFN(*mycatalog, pfn, "ROOT_All", guid);
            } 
            else {
                cerr << "ERROR! Unsupported catalog item type for registration: "<< item << endl;
                printUsage();
                return 3;
            }
 
        } else if( action == "rename" ) {
            if( pfn.empty() || newpfn.empty() ){
                cerr<<"ERROR! You must specify the current PFName using -p and the new PFName using -n option" << endl;
                return 4;
            }
            mycatalog->renamePFN(pfn, newpfn);
        }
        else {
            cerr << "ERROR! Unsupported action: " << action << endl;
            printUsage();
            return 10;
        }
 
        mycatalog->commit();
 
    }catch (const std::runtime_error& er){
        cerr<<er.what()<<endl;
        return 1;
    }catch (const std::exception& er){
        cerr<<er.what()<<endl;
        return 1;
    }
}

printUsage()

void printUsage

(

)

Definition at line 21 of file FCcmd.cxx.

                 {
  cout <<
  "Usage: " << endl <<
  "FCcmd <action> <item> [-p PFName -l LFName -n newPFName -g GUID -u catalogName -h]" << endl <<
  "  supported actions are:" <<endl <<
  "     register : <item> = lfn|pfn  (registering a PFN requires a GUID)" <<endl <<
  "     list     : <item> = guid|lfn|pfn" <<endl <<
  "     delete   : deletes an entry using LFN or PFN" <<endl <<
  "     rename   : rename PFName to a new PFName" <<endl <<
  " example:  FCcmd list pfn -l myLogicalFileName" << endl;
}