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_ */