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 <map>

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

#include <boost/noncopyable.hpp>

#include <vrfutil/objectCounter.h>

#include <tbb/task_arena.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);


typedef std::map<size_t, std::weak_ptr<DtAsyncJobMapEntry> > DtAsyncJobMap;

class DT_DLL_vrfutil DtAsyncJobServer
{
public:
   DtAsyncJobServer(unsigned maxThreads = 0);

   virtual ~DtAsyncJobServer();

   virtual std::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();

   virtual void waitOnJobs();

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&);

   mutable Mutex myAsyncJobMapMutex;
   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;
   };


   std::unique_ptr<tbb::task_arena> myTaskArena;

   static long long theQuittingIoPeriod;
};