XrdClTaskManager.hh

Go to the documentation of this file.

//------------------------------------------------------------------------------
// Copyright (c) 2011-2012 by European Organization for Nuclear Research (CERN)
// Author: Lukasz Janyst <ljanyst@cern.ch>
//------------------------------------------------------------------------------
// 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/>.
//------------------------------------------------------------------------------
 
#ifndef __XRD_CL_TASK_MANAGER_HH__
#define __XRD_CL_TASK_MANAGER_HH__
 
#include <ctime>
#include <set>
#include <list>
#include <string>
#include <cstdint>
#include <pthread.h>
#include "XrdSys/XrdSysPthread.hh"
 
namespace XrdCl
{
  //----------------------------------------------------------------------------
  //----------------------------------------------------------------------------
  class Task
  {
    public:
      virtual ~Task() {};
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      virtual time_t Run( time_t now ) = 0;
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      const std::string &GetName() const
      {
        return pName;
      }
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void SetName( const std::string &name )
      {
        pName = name;
      }
 
    private:
      std::string pName;
  };
 
  //----------------------------------------------------------------------------
  //----------------------------------------------------------------------------
  class TaskManager
  {
    public:
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      TaskManager();
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      ~TaskManager();
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      bool Start();
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      bool Stop();
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void RegisterTask( Task *task, time_t time, bool own = true );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void UnregisterTask( Task *task );
 
      //------------------------------------------------------------------------
      //------------------------------------------------------------------------
      void RunTasks();
 
    private:
 
      //------------------------------------------------------------------------
      // Task set helpers
      //------------------------------------------------------------------------
      struct TaskHelper
      {
        TaskHelper( Task *tsk, time_t tme, bool ow = true ):
          task(tsk), execTime(tme), own(ow) {}
        Task   *task;
        time_t  execTime;
        bool    own;
      };
 
      struct TaskHelperCmp
      {
        bool operator () ( const TaskHelper &th1, const TaskHelper &th2 ) const
        {
          return th1.execTime < th2.execTime;
        }
      };
 
      typedef std::multiset<TaskHelper, TaskHelperCmp> TaskSet;
      typedef std::list<Task*>                         TaskList;
 
      //------------------------------------------------------------------------
      // Private variables
      //------------------------------------------------------------------------
      uint16_t    pResolution;
      TaskSet     pTasks;
      TaskList    pToBeUnregistered;
      pthread_t   pRunnerThread;
      bool        pRunning;
      XrdSysMutex pMutex;
      XrdSysMutex pOpMutex;
  };
}
 
#endif // __XRD_CL_TASK_MANAGER_HH__