asyncJobServer.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright 2010 VT MAK
*******************************************************************************/



#pragma once


#include <stdlib.h>
#include <memory>
#include <list>

#include <tbb/concurrent_hash_map.h>
#ifdef WIN32
   #include <WinSock2.h>
#endif
#include <tbb/spin_mutex.h>

#include <boost/shared_ptr.hpp>
#include <boost/weak_ptr.hpp>

class DtAsyncJob;
class DtAsyncJobMapEntry;

enum DtAsyncJobStatus
{
   queued,
   processing,
   complete,
   canceled,
   error
};


#include <vrfutil/vrfutilDefines.h>
class DT_DLL_vrfutil DtAsyncJobResult 
{
public:
   DtAsyncJobResult() {};
   
   virtual ~DtAsyncJobResult() {};
};

struct MyHashCompare 
{
   static size_t hash( const size_t& x ) 
   {
      return x;
   }
   
   static bool equal( const size_t& x, const size_t& y ) 
   {
      return x==y;
   }
};


typedef tbb::concurrent_hash_map<size_t, boost::weak_ptr<DtAsyncJobMapEntry>, MyHashCompare> DtAsyncJobMap;

class DT_DLL_vrfutil DtAsyncJobServer
{
public:
   DtAsyncJobServer();

   virtual ~DtAsyncJobServer();

   virtual boost::shared_ptr<DtAsyncJobMapEntry> addJob(DtAsyncJob* job);

   virtual void cancelAllJobs();

   virtual bool empty() const;

   virtual bool running() const;

   virtual void printAsyncJobMap();

protected:
   friend class DtAsyncJobMapEntry;
   friend class DtAsyncJob;

   virtual size_t getNextId();

   virtual void remove(size_t id);

   virtual void jobDestroyed(DtAsyncJob&);

   DtAsyncJobMap myAsyncJobMap;

   mutable tbb::spin_mutex myIdGenMutex;
   size_t myNextId;

   mutable tbb::spin_mutex myNumTasksMutex;
   size_t myNumTasks;
};