prefetch_impl.hpp

ncbi源码

/*
 * ===========================================================================
 * PRODUCTION $Log: prefetch_impl.hpp,v $
 * PRODUCTION Revision 1000.2  2004/06/01 19:22:04  gouriano
 * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.4
 * PRODUCTION
 * ===========================================================================
 */

#ifndef PREFETCH_IMPL__HPP
#define PREFETCH_IMPL__HPP

/*  $Id: prefetch_impl.hpp,v 1000.2 2004/06/01 19:22:04 gouriano Exp $
* ===========================================================================
*
*                            PUBLIC DOMAIN NOTICE
*               National Center for Biotechnology Information
*
*  This software/database is a "United States Government Work" under the
*  terms of the United States Copyright Act.  It was written as part of
*  the author's official duties as a United States Government employee and
*  thus cannot be copyrighted.  This software/database is freely available
*  to the public for use. The National Library of Medicine and the U.S.
*  Government have not placed any restriction on its use or reproduction.
*
*  Although all reasonable efforts have been taken to ensure the accuracy
*  and reliability of the software and data, the NLM and the U.S.
*  Government do not and cannot warrant the performance or results that
*  may be obtained by using this software or data. The NLM and the U.S.
*  Government disclaim all warranties, express or implied, including
*  warranties of performance, merchantability or fitness for any particular
*  purpose.
*
*  Please cite the author in any work or product based on this material.
*
* ===========================================================================
*
* Author: Aleksey Grichenko, Eugene Vasilchenko
*
* File Description:
*   Prefetch implementation
*
*/

#include <corelib/ncbistd.hpp>
#include <corelib/ncbiobj.hpp>
#include <corelib/ncbithr.hpp>
#include <objmgr/data_loader.hpp>
#include <util/thread_pool.hpp>
#include <vector>
#include <set>

BEGIN_NCBI_SCOPE
BEGIN_SCOPE(objects)

class CScope;
class CSeq_id;
class CSeq_id_Handle;
class CBioseq_Handle;
class CPrefetchThread;
class CDataSource;

class NCBI_XOBJMGR_EXPORT CPrefetchToken_Impl : public CObject
{
public:
    typedef vector<CSeq_id_Handle> TIds;
    typedef CDataLoader::TTSE_Lock TTSE_Lock;

    ~CPrefetchToken_Impl(void);

private:
    friend class CPrefetchToken;
    friend class CPrefetchThread;

    CPrefetchToken_Impl(const TIds& ids, unsigned int depth);

    void x_InitPrefetch(CScope& scope);
    void x_SetNon_locking(void);

    // Hide copy methods
    CPrefetchToken_Impl(const CPrefetchToken_Impl&);
    CPrefetchToken_Impl& operator=(const CPrefetchToken_Impl&);

    operator bool(void) const;
    CBioseq_Handle NextBioseqHandle(CScope& scope);

    void AddTokenReference(void);
    void RemoveTokenReference(void);

    // Called by fetching function when id is loaded
    void AddResolvedId(size_t id_idx, TTSE_Lock tse);

    // Checked by CPrefetchThread before processing next id
    bool IsEmpty(void) const;

    typedef vector<TTSE_Lock>   TFetchedTSEs;
    typedef map<TTSE_Lock, int> TTSE_Map;

    int            m_TokenCount;    // Number of tokens referencing this impl
    TIds           m_Ids;           // requested ids in the original order
    size_t         m_CurrentId;     // next id to return
    TFetchedTSEs   m_TSEs;          // loaded TSEs
    TTSE_Map       m_TSEMap;        // Map TSE to number of related IDs
    int            m_PrefetchDepth; // Max. number of TSEs to prefetch
    CSemaphore     m_TSESemaphore;  // Signal to fetch next TSE
    bool           m_Non_locking;   // Do not lock TSEs (used for caching)
    mutable CFastMutex m_Lock;
};


class CPrefetchThread : public CThread
{
public:
    CPrefetchThread(CDataSource& data_source);

    // Add request to the fetcher queue
    void AddRequest(CPrefetchToken_Impl& token);

    // Stop the thread (since Exit() can not be called from
    // data loader's destructor).
    void Terminate(void);

protected:
    virtual void* Main(void);
    // protected destructor as required by CThread
    ~CPrefetchThread(void);

private:
    // Using list to be able to delete elements
    typedef CBlockingQueue<CRef<CPrefetchToken_Impl> > TPrefetchQueue;

    CDataSource&   m_DataSource;
    TPrefetchQueue m_Queue;
    CFastMutex     m_Lock;
    bool           m_Stop;      // used to stop the thread
};


END_SCOPE(objects)
END_NCBI_SCOPE

/*
* ---------------------------------------------------------------------------
* $Log: prefetch_impl.hpp,v $
* Revision 1000.2  2004/06/01 19:22:04  gouriano
* PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.4
*
* Revision 1.4  2004/05/07 13:47:34  grichenk
* Removed single-id constructors.
* Added non-locking prefetch mode.
*
* Revision 1.3  2004/04/21 15:35:06  gorelenk
* Added export prefix to class CPrefetchToken_Impl .
*
* Revision 1.2  2004/04/19 14:52:29  grichenk
* Added prefetch depth limit, redesigned prefetch queue.
*
* Revision 1.1  2004/04/16 13:30:34  grichenk
* Initial revision
*
*
* ===========================================================================
*/

#endif  // PREFETCH_IMPL__HPP