XrdClXCpCtx.hh

Go to the documentation of this file.

//------------------------------------------------------------------------------
// Copyright (c) 2011-2017 by European Organization for Nuclear Research (CERN)
// Author: Michal Simon <michal.simon@cern.ch>
//------------------------------------------------------------------------------
// This file is part of the XRootD software suite.
//
// XRootD is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// XRootD is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with XRootD.  If not, see <http://www.gnu.org/licenses/>.
//
// In applying this licence, CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
//------------------------------------------------------------------------------
 
#ifndef SRC_XRDCL_XRDCLXCPCTX_HH_
#define SRC_XRDCL_XRDCLXCPCTX_HH_
 
#include "XrdCl/XrdClSyncQueue.hh"
#include "XrdCl/XrdClXRootDResponses.hh"
#include "XrdSys/XrdSysPthread.hh"
 
#include <cstdint>
#include <iostream>
 
namespace XrdCl
{
 
class XCpSrc;
 
class XCpCtx
{
  public:
 
    XCpCtx( const std::vector<std::string> &urls, uint64_t blockSize, uint8_t parallelSrc, uint64_t chunkSize, uint64_t parallelChunks, int64_t fileSize );
 
    void Delete()
    {
      XrdSysMutexHelper lckmtx( pMtx );
      --pRefCount;
      if( !pRefCount )
      {
        lckmtx.UnLock();
        delete this;
        return;
      }
      lckmtx.UnLock();
 
      XrdSysCondVarHelper lckcv( pDoneCV );
      pDone = true;
      pDoneCV.Broadcast();
      lckcv.UnLock();
 
      lckcv.Lock( &pDeleteCV );
      while( !pDelete ) pDeleteCV.Wait();
      lckcv.UnLock();
      delete this;
    }
 
    void Release()
    {
      XrdSysMutexHelper lck( pMtx );
      --pRefCount;
      if( !pRefCount )
      {
        XrdSysCondVarHelper lckcv( pDeleteCV );
        pDelete = true;
        pDeleteCV.Broadcast();
      }
    }
 
    XCpCtx* Self()
    {
      XrdSysMutexHelper lck( pMtx );
      ++pRefCount;
      return this;
    }
 
    bool GetNextUrl( std::string & url );
 
    XCpSrc* WeakestLink( XCpSrc *exclude );
 
    void PutChunk( PageInfo* chunk );
 
    std::pair<uint64_t, uint64_t> GetBlock();
 
    void SetFileSize( int64_t size );
 
    int64_t GetSize()
    {
      XrdSysCondVarHelper lck( pFileSizeCV );
      while( pFileSize < 0 && GetRunning() > 0 ) pFileSizeCV.Wait();
      return pFileSize;
    }
 
    XRootDStatus Initialize();
 
    XRootDStatus GetChunk( XrdCl::PageInfo &ci );
 
    void RemoveSrc( XCpSrc *src )
    {
      XrdSysMutexHelper lck( pMtx );
      pSources.remove( src );
    }
 
    void NotifyIdleSrc();
 
    bool AllDone();
 
    void NotifyInitExpectant()
    {
      pFileSizeCV.Broadcast();
    }
 
 
  private:
 
    size_t GetRunning();
 
    virtual ~XCpCtx();
 
    std::queue<std::string>    pUrls;
 
    uint64_t                   pBlockSize;
 
    uint8_t                    pParallelSrc;
 
    uint32_t                   pChunkSize;
 
    uint8_t                    pParallelChunks;
 
    uint64_t                   pOffset;
 
    int64_t                    pFileSize;
 
    XrdSysCondVar              pFileSizeCV;
 
    std::list<XCpSrc*>         pSources;
 
    SyncQueue<PageInfo*>       pSink;
 
    uint64_t                   pDataReceived;
 
    bool                       pDone;
 
    XrdSysCondVar              pDoneCV;
 
    XrdSysMutex                pMtx;
 
    size_t                     pRefCount;
 
    XrdSysCondVar              pDeleteCV;
 
    bool                       pDelete;
};
 
} /* namespace XrdCl */
 
#endif /* SRC_XRDCL_XRDCLXCPCTX_HH_ */