asyncJobServer.h

Go to the documentation of this file.

/*******************************************************************************
** Copyright 2015 VT MAK
*******************************************************************************/


#pragma once

#include <vrfutil/vrfutilDefines.h>

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

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

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

#include <vrfutil/objectCounter.h>

class DtAsyncJob;
class DtAsyncJobMapEntry;

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

class DT_DLL_vrfutil DtAsyncJobResult
#ifdef DtObjectDebugger_Enable
   : public DtObjectDebugger<DtAsyncJobResult>
#endif
{
public:   
   virtual ~DtAsyncJobResult();
};

DT_DEFINE_OBJECT_DEBUGGER_NAME(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 void cancelAllJobsAndWait();

   virtual bool empty() const;

   virtual bool running() const;

   virtual void printAsyncJobMap();

   virtual void tick();

protected:
   friend class DtAsyncJobMapEntry;
   friend class DtAsyncJob;

   typedef tbb::spin_mutex Mutex;

   virtual size_t getNextId();

   virtual void remove(size_t id);

   virtual void jobDestroyed(DtAsyncJob&);

   DtAsyncJobMap myAsyncJobMap;

   mutable Mutex myIdGenMutex;
   size_t myNextId;

   mutable Mutex myNumTasksMutex;
   size_t myNumTasks;

   typedef std::list<DtAsyncJob*> JobList;
   mutable Mutex myJobListMutex;
   JobList myCurrentJobs;

   class DT_DLL_vrfutil JobListScopeGuard : boost::noncopyable
   {
   public:
      JobListScopeGuard(DtAsyncJobServer& server, DtAsyncJob& job);
      ~JobListScopeGuard();

   private:
      DtAsyncJobServer& server;
      JobList::iterator i;
   };
};