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 <stdint.h>
#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__