DiskReadThreadPool.cpp 8.49 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
/******************************************************************************
 *
 * This file is part of the Castor project.
 * See http://castor.web.cern.ch/castor
 *
 * Copyright (C) 2003  CERN
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * This program 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 General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 *
 * 
 *
 * @author Castor Dev team, castor-dev@cern.ch
 *****************************************************************************/

#include "castor/tape/tapeserver/daemon/DiskReadThreadPool.hpp"
25
#include "castor/tape/tapeserver/daemon/MigrationTaskInjector.hpp"
26
#include "castor/tape/tapeserver/daemon/MigrationReportPacker.hpp"
27
#include "castor/common/CastorConfiguration.hpp"
28
29
#include <memory>
#include <sstream>
30

31
32
33
34
namespace castor {
namespace tape {
namespace tapeserver {
namespace daemon {
35
36
37
38

//------------------------------------------------------------------------------
// DiskReadThreadPool constructor
//------------------------------------------------------------------------------
39
DiskReadThreadPool::DiskReadThreadPool(int nbThread, uint64_t maxFilesReq,uint64_t maxBytesReq, 
40
    castor::log::LogContext lc, const std::string & remoteFileProtocol) : 
41
42
43
44
    m_diskFileFactory(remoteFileProtocol,
      castor::common::CastorConfiguration::getConfig().
        getConfEntString("XROOT", "PrivateKey", "/opt/xrootd/keys/key.pem")),
    m_lc(lc),m_maxFilesReq(maxFilesReq),
45
    m_maxBytesReq(maxBytesReq), m_nbActiveThread(0) {
46
47
48
49
50
  for(int i=0; i<nbThread; i++) {
    DiskReadWorkerThread * thr = new DiskReadWorkerThread(*this);
    m_threads.push_back(thr);
    m_lc.pushOrReplace(log::Param("threadID",i));
    m_lc.log(LOG_INFO, "DiskReadWorkerThread created");
51
  }
52
53
54
55
56
57
58
59
60
}

//------------------------------------------------------------------------------
// DiskReadThreadPool destructor
//------------------------------------------------------------------------------
DiskReadThreadPool::~DiskReadThreadPool() { 
  while (m_threads.size()) {
    delete m_threads.back();
    m_threads.pop_back();
61
  }
62
  logWithStat(LOG_INFO, "All the DiskReadWorkerThreads have been destroyed");
63
64
65
66
67
68
69
70
71
}

//------------------------------------------------------------------------------
// DiskReadThreadPool::startThreads
//------------------------------------------------------------------------------
void DiskReadThreadPool::startThreads() {
  for (std::vector<DiskReadWorkerThread *>::iterator i=m_threads.begin();
          i != m_threads.end(); i++) {
    (*i)->start();
72
  }
73
74
75
76
77
78
79
80
81
82
  m_lc.log(LOG_INFO, "All the DiskReadWorkerThreads are started");
}

//------------------------------------------------------------------------------
// DiskReadThreadPool::waitThreads
//------------------------------------------------------------------------------
void DiskReadThreadPool::waitThreads() {
  for (std::vector<DiskReadWorkerThread *>::iterator i=m_threads.begin();
          i != m_threads.end(); i++) {
    (*i)->wait();
83
  }
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
}

//------------------------------------------------------------------------------
// DiskReadThreadPool::push
//------------------------------------------------------------------------------
void DiskReadThreadPool::push(DiskReadTask *t) { 
  m_tasks.push(t); 
  m_lc.log(LOG_INFO, "Push a task into the DiskReadThreadPool");
}

//------------------------------------------------------------------------------
// DiskReadThreadPool::finish
//------------------------------------------------------------------------------
void DiskReadThreadPool::finish() {
  /* Insert one endOfSession per thread */
  for (size_t i=0; i<m_threads.size(); i++) {
    m_tasks.push(NULL);
101
  }
102
103
104
105
106
107
}

//------------------------------------------------------------------------------
// DiskReadThreadPool::popAndRequestMore
//------------------------------------------------------------------------------
DiskReadTask* DiskReadThreadPool::popAndRequestMore(castor::log::LogContext &lc){
108
  castor::server::BlockingQueue<DiskReadTask*>::valueRemainingPair 
109
110
111
112
113
114
115
116
117
118
  vrp = m_tasks.popGetSize();
  log::LogContext::ScopedParam sp(lc, log::Param("m_maxFilesReq", m_maxFilesReq));
  log::LogContext::ScopedParam sp0(lc, log::Param("m_maxBytesReq", m_maxBytesReq));

  if(0==vrp.remaining){
    m_injector->requestInjection(true);
    lc.log(LOG_DEBUG, "Requested injection from MigrationTaskInjector (with last call)");
  }else if(vrp.remaining + 1 ==  m_maxFilesReq/2){
    m_injector->requestInjection(false);
    lc.log(LOG_DEBUG, "Requested injection from MigrationTaskInjector (without last call)");
119
  }
120
121
  return vrp.value;
}
122
123
124
//------------------------------------------------------------------------------
//addThreadStats
//------------------------------------------------------------------------------
125
void DiskReadThreadPool::addThreadStats(const DiskStats& other){
126
  castor::server::MutexLocker lock(&m_statAddingProtection);
127
128
  m_pooldStat+=other;
}
129
//------------------------------------------------------------------------------
130
131
132
133
//logWithStat
//------------------------------------------------------------------------------
void DiskReadThreadPool::logWithStat(int level, const std::string& message){
  log::ScopedParamContainer params(m_lc);
134
135
136
137
     params.addTiming("poolTransferTime", m_pooldStat.transferTime)
           .addTiming("poolWaitFreeMemoryTime",m_pooldStat.waitFreeMemoryTime)
           .addTiming("poolCheckingErrorTime",m_pooldStat.checkingErrorTime)
           .addTiming("poolOpeningTime",m_pooldStat.openingTime)
138
           .add("poolFileCount",m_pooldStat.filesCount)
139
140
141
           .add("poolDataVolumeInMB", 1.0*m_pooldStat.dataVolume/1000/1000)
           .add("AveragePoolPayloadTransferSpeedMBps",
                   1.0*m_pooldStat.dataVolume/1000/1000/m_pooldStat.transferTime);
142
143
144
    m_lc.log(level,message);
}
//------------------------------------------------------------------------------
145
// DiskReadWorkerThread::run
146
147
148
149
//------------------------------------------------------------------------------
void DiskReadThreadPool::DiskReadWorkerThread::run() {
  m_lc.pushOrReplace(log::Param("thread", "DiskRead"));
  m_lc.pushOrReplace(log::Param("threadID",m_threadID));
150
  
151
  m_lc.log(LOG_DEBUG, "DiskReadWorkerThread Running");
152
  
153
  std::auto_ptr<DiskReadTask> task;
154
155
  utils::Timer localTime;
  
156
157
  while(1) {
    task.reset( m_parent.popAndRequestMore(m_lc));
158
    m_threadStat.waitInstructionsTime += localTime.secs(utils::Timer::resetCounter);
159
    if (NULL!=task.get()) {
160
      task->execute(m_lc, m_parent.m_diskFileFactory);
161
      m_threadStat += task->getTaskStats();
162
    }
163
164
    else {
      break;
165
    }
166
167
168
  } //end of while(1)
  // We now acknowledge to the task injector that read reached the end. There
  // will hence be no more requests for more. (last thread turns off the light)
169
170
  int remainingThreads = --m_parent.m_nbActiveThread;
  if (!remainingThreads) {
171
172
    m_parent.m_injector->finish();
    m_lc.log(LOG_INFO, "Signaled to task injector the end of disk read threads");
173
174
175
176
  } else {
    castor::log::ScopedParamContainer params(m_lc);
    params.add("remainingThreads", remainingThreads);
    m_lc.log(LOG_DEBUG, "Will not signal the end to task injector yet");
177
  }
178
179
180
181
182
183
184
185
186
187
  m_parent.addThreadStats(m_threadStat);
  logWithStat(LOG_INFO, "Finishing of DiskReadWorkerThread");
}

//------------------------------------------------------------------------------
// DiskReadWorkerThread::logWithStat
//------------------------------------------------------------------------------
void DiskReadThreadPool::DiskReadWorkerThread::
logWithStat(int level, const std::string& message){
  log::ScopedParamContainer params(m_lc);
188
189
190
191
     params.addTiming("threadTransferTime", m_threadStat.transferTime)
           .addTiming("threadWaitFreeMemoryTime",m_threadStat.waitFreeMemoryTime)
           .addTiming("threadCheckingErrorTime",m_threadStat.checkingErrorTime)
           .addTiming("threadOpeningTime",m_threadStat.openingTime)
192
193
194
           .add("threaDataVolumeInMB", 1.0*m_threadStat.dataVolume/1000/1000)
           .add("threadPayloadTransferSpeedMBps",
                   1.0*m_threadStat.dataVolume/1000/1000/m_threadStat.transferTime);
195
    m_lc.log(level,message);
196
}
197
}}}}
198