ryzom/tools/stats_scan/job_manager.cpp

   1 // Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
   2 // Copyright (C) 2010  Winch Gate Property Limited
   3 //
   4 // This program is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU Affero General Public License as
   6 // published by the Free Software Foundation, either version 3 of the
   7 // License, or (at your option) any later version.
   8 //
   9 // This program is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU Affero General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Affero General Public License
  15 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 #include "job_manager.h"
  18
  19 class CFinishedJob: public CJobManager::IJob
  20 {
  21 public:
  22         bool finished()                                                                 { return true; }
  23         std::string getShortStatus()                                    { return _ShortStatus; }
  24         std::string getStatus()                                                 { return _Status; }
  25         void display(NLMISC::CLog* log=NLMISC::InfoLog) { log->displayNL("%s",_Status.c_str()); }
  26         void update()                                                                   {}
  27
  28         CFinishedJob(CJobManager::IJob* theFinishedJob)
  29         {
  30                 if (theFinishedJob==NULL)
  31                         return;
  32                 _Status=theFinishedJob->getStatus();
  33                 _ShortStatus=theFinishedJob->getShortStatus();
  34         }
  35
  36 private:
  37         std::string _Status;
  38         std::string _ShortStatus;
  39
  40 };
  41
  42 CJobManager* CJobManager::getInstance()
  43 {
  44         static CJobManager* mgr=NULL;
  45         if (mgr==NULL)
  46         {
  47                 mgr=new CJobManager;
  48         }
  49         return mgr;
  50 }
  51
  52 CJobManager::CJobManager()
  53 {
  54         _Paused=false;
  55         _JobUpdatesPerUpdate=1;
  56 }
  57
  58 void CJobManager::serviceUpdate()
  59 {
  60         if (_Paused)
  61                 return;
  62
  63         for (uint32 count=0;count<_JobUpdatesPerUpdate &&!_UnfinishedJobs.empty();++count)
  64         {
  65                 nlassert(_UnfinishedJobs.front()<_Jobs.size());
  66                 NLMISC::CSmartPtr<IJob>& theJob= _Jobs[_UnfinishedJobs.front()];
  67                 if (theJob->finished())
  68                 {
  69                         // delete the job and replace it with a light weight 'finished job' marker
  70                         theJob= new CFinishedJob(theJob);
  71
  72                         // remove this job from the list of unfinished jobs
  73                         _UnfinishedJobs.pop_front();
  74
  75                         // decrement the updates counter to counteract the auto incrment
  76                         --count;
  77                 }
  78                 else
  79                 {
  80                         theJob->update();
  81                 }
  82         }
  83 }
  84
  85
  86 uint32 CJobManager::addJob(NLMISC::CSmartPtr<CJobManager::IJob> job)
  87 {
  88         nlassert(job!=NULL);
  89         uint32 id= (uint32)_Jobs.size();
  90         _UnfinishedJobs.push_back(id);
  91         _Jobs.push_back(job);
  92         return id;
  93 }
  94
  95 void CJobManager::promoteJob(uint32 idx)
  96 {
  97         TUnfinishedJobs::iterator it;
  98         for (it=_UnfinishedJobs.begin(); it!=_UnfinishedJobs.end(); ++it)
  99         {
 100                 if (*it==idx)
 101                 {
 102                         _UnfinishedJobs.erase(it);
 103                         _UnfinishedJobs.push_front(idx);
 104                         return;
 105                 }
 106         }
 107         nlwarning("Failed to promote job with ID %d as not found in unfinished jobs list",idx);
 108 }
 109
 110 void CJobManager::pause()
 111 {
 112         _Paused= true;
 113 }
 114
 115 void CJobManager::resume()
 116 {
 117         _Paused= false;
 118 }
 119
 120 void CJobManager::setJobUpdatesPerUpdate(uint32 count)
 121 {
 122         _JobUpdatesPerUpdate= count;
 123         if (count==0 || count>100)
 124                 nlwarning("Suspicious value of JobUpdatesPerUpdate: %d",count);
 125 }
 126
 127 uint32 CJobManager::getJobUpdatesPerUpdate()
 128 {
 129         return _JobUpdatesPerUpdate;
 130 }
 131
 132 std::string CJobManager::getStatus()
 133 {
 134         std::string result;
 135
 136         if (_Paused) result+="[Paused] ";
 137
 138         if (!_UnfinishedJobs.empty())
 139         {
 140                 uint32 idx=_UnfinishedJobs.front();
 141                 nlassert(idx<_Jobs.size());
 142                 result+=_Jobs[idx]->getStatus();
 143         }
 144
 145         result+=NLMISC::toString(" [Updates per cycle: %d]",_JobUpdatesPerUpdate);
 146
 147         return result;
 148 }
 149
 150 void CJobManager::listJobs(NLMISC::CLog* log)
 151 {
 152         for (uint32 i=0;i< _Jobs.size(); ++i)
 153         {
 154                 if (!_Jobs[i]->finished())
 155                         nlinfo("%4d*: %s",i,_Jobs[i]->getStatus().c_str());
 156         }
 157         nlinfo("%d unfinished jobs (%d  in total)",_UnfinishedJobs.size(),_Jobs.size());
 158 }
 159
 160 void CJobManager::listJobHistory(NLMISC::CLog* log)
 161 {
 162         for (uint32 i=0;i< _Jobs.size(); ++i)
 163         {
 164                 nlinfo("%4d%c: %s",i,_Jobs[i]->finished()? ' ': '*',_Jobs[i]->getStatus().c_str());
 165         }
 166         nlinfo("%d unfinished jobs (%d  in total)",_UnfinishedJobs.size(),_Jobs.size());
 167 }
 168
 169 void CJobManager::displayCurrentJob(NLMISC::CLog* log)
 170 {
 171         if (!_UnfinishedJobs.empty())
 172                 displayJob(_UnfinishedJobs.front(),log);
 173 }
 174
 175 void CJobManager::displayJob(uint32 jobId,NLMISC::CLog* log)
 176 {
 177         nlassert(jobId<_Jobs.size());
 178         _Jobs[jobId]->display(log);
 179 }