source: XIOS/dev/dev_olga/src/server.cpp @ 1168

Last change on this file since 1168 was 1168, checked in by oabramkina, 7 years ago

Two server levels: merging registries of secondary server pools into one and then writing the merged one.

Tested on curie with toy_cmip6.

  • Property copyright set to
    Software name : XIOS (Xml I/O Server)
    http://forge.ipsl.jussieu.fr/ioserver
    Creation date : January 2009
    Licence : CeCCIL version2
    see license file in root directory : Licence_CeCILL_V2-en.txt
    or http://www.cecill.info/licences/Licence_CeCILL_V2-en.html
    Holder : CEA/LSCE (Laboratoire des Sciences du CLimat et de l'Environnement)
    CNRS/IPSL (Institut Pierre Simon Laplace)
    Project Manager : Yann Meurdesoif
    yann.meurdesoif@cea.fr
  • Property svn:eol-style set to native
File size: 24.1 KB
Line 
1#include "globalScopeData.hpp"
2#include "xios_spl.hpp"
3#include "cxios.hpp"
4#include "server.hpp"
5#include "client.hpp"
6#include "type.hpp"
7#include "context.hpp"
8#include "object_template.hpp"
9#include "oasis_cinterface.hpp"
10#include <boost/functional/hash.hpp>
11#include <boost/algorithm/string.hpp>
12#include "mpi.hpp"
13#include "tracer.hpp"
14#include "timer.hpp"
15#include "event_scheduler.hpp"
16
17namespace xios
18{
19    MPI_Comm CServer::intraComm ;
20    list<MPI_Comm> CServer::interCommLeft ;
21    list<MPI_Comm> CServer::interCommRight ;
22    std::list<MPI_Comm> CServer::contextInterComms;
23    std::list<MPI_Comm> CServer::contextIntraComms;
24    int CServer::serverLevel = 0 ;
25    int CServer::nbContexts = 0;
26    bool CServer::isRoot = false ;
27    int CServer::rank_ = INVALID_RANK;
28    StdOFStream CServer::m_infoStream;
29    StdOFStream CServer::m_errorStream;
30    map<string,CContext*> CServer::contextList ;
31    vector<int> CServer::sndServerGlobalRanks;
32    bool CServer::finished=false ;
33    bool CServer::is_MPI_Initialized ;
34    CEventScheduler* CServer::eventScheduler = 0;
35
36//---------------------------------------------------------------
37/*!
38 * \fn void CServer::initialize(void)
39 * Creates intraComm for each possible type of servers (classical, primary or secondary).
40 * In case of secondary servers intraComm is created for each secondary server pool.
41 * (For now the assumption is that there is one proc per pool.)
42 * Creates interComm and stores them into the following lists:
43 *   classical server -- interCommLeft
44 *   primary server -- interCommLeft and interCommRight
45 *   secondary server -- interCommLeft for each pool.
46 */
47    void CServer::initialize(void)
48    {
49      int initialized ;
50      MPI_Initialized(&initialized) ;
51      if (initialized) is_MPI_Initialized=true ;
52      else is_MPI_Initialized=false ;
53      int rank ;
54
55      // Not using OASIS
56      if (!CXios::usingOasis)
57      {
58
59        if (!is_MPI_Initialized)
60        {
61          MPI_Init(NULL, NULL);
62        }
63        CTimer::get("XIOS").resume() ;
64
65        boost::hash<string> hashString ;
66        unsigned long hashServer = hashString(CXios::xiosCodeId);
67
68        unsigned long* hashAll ;
69        unsigned long* srvLevelAll ;
70
71        int size ;
72        int myColor ;
73        int i,c ;
74        MPI_Comm newComm;
75
76        MPI_Comm_size(CXios::globalComm, &size) ;
77        MPI_Comm_rank(CXios::globalComm, &rank_);
78
79        hashAll=new unsigned long[size] ;
80        MPI_Allgather(&hashServer, 1, MPI_LONG, hashAll, 1, MPI_LONG, CXios::globalComm) ;
81
82        map<unsigned long, int> colors ;
83        map<unsigned long, int> leaders ;
84        map<unsigned long, int>::iterator it ;
85
86        int srvNodeSize = 1, srvNodeLeader = 0;
87
88        // (1) Establish client leaders, distribute processes between two server levels
89        for(i=0,c=0;i<size;i++)
90        {
91          if (colors.find(hashAll[i])==colors.end())
92          {
93            colors[hashAll[i]]=c ;
94            leaders[hashAll[i]]=i ;
95            c++ ;
96          }
97          if (CXios::usingServer2)
98          {
99            if (hashAll[i] == hashServer)
100            {
101              if (hashAll[i-1] != hashServer || i == 0)
102              {
103                srvNodeLeader = i;
104              }
105              if (hashAll[i+1] == hashServer)
106              {
107                ++srvNodeSize;
108              }
109              else
110              {
111                if ( (rank_-srvNodeLeader) >= 0 &&
112                     (rank_-srvNodeLeader) < (1.- CXios::ratioServer2/100.)*srvNodeSize )
113                {
114                  serverLevel = 1;
115                }
116                if ( (rank_-srvNodeLeader) >= (1.- CXios::ratioServer2/100.)*srvNodeSize &&
117                     (rank_ - srvNodeLeader) < srvNodeSize )
118                {
119                  serverLevel = 2;
120                }
121                srvNodeSize = 1;
122              }
123            }
124          }
125        }
126
127        // (2) Create intraComm
128        myColor = (serverLevel == 2) ? rank_ : colors[hashServer];
129        MPI_Comm_split(CXios::globalComm, myColor, rank_, &intraComm) ;
130
131        // (3) Create interComm
132        if (CXios::usingServer2)
133        {
134          MPI_Allgather(&serverLevel, 1, MPI_LONG, hashAll, 1, MPI_LONG, CXios::globalComm) ;
135
136          for (i=0; i<size; i++)
137            if (hashAll[i] == 2)
138              sndServerGlobalRanks.push_back(i);
139        }
140
141        if (serverLevel == 0)
142        {
143          int clientLeader;
144          for(it=leaders.begin();it!=leaders.end();it++)
145          {
146            if (it->first!=hashServer)
147            {
148              clientLeader=it->second ;
149              int intraCommSize, intraCommRank ;
150              MPI_Comm_size(intraComm,&intraCommSize) ;
151              MPI_Comm_rank(intraComm,&intraCommRank) ;
152              info(50)<<"intercommCreate::server (classical mode) "<<rank_<<" intraCommSize : "<<intraCommSize
153                       <<" intraCommRank :"<<intraCommRank<<"  clientLeader "<< clientLeader<<endl ;
154
155              MPI_Intercomm_create(intraComm, 0, CXios::globalComm, clientLeader, 0, &newComm) ;
156               interCommLeft.push_back(newComm) ;
157            }
158          }
159        }
160        else if (serverLevel == 1)
161        {
162          int clientLeader, srvSndLeader;
163          int srvPrmLeader ;
164
165          for (it=leaders.begin();it!=leaders.end();it++)
166          {
167            if (it->first != hashServer)
168            {
169              clientLeader=it->second ;
170              int intraCommSize, intraCommRank ;
171              MPI_Comm_size(intraComm, &intraCommSize) ;
172              MPI_Comm_rank(intraComm, &intraCommRank) ;
173              info(50)<<"intercommCreate::server (server level 1) "<<rank_<<" intraCommSize : "<<intraCommSize
174                       <<" intraCommRank :"<<intraCommRank<<"  clientLeader "<< clientLeader<<endl ;
175              MPI_Intercomm_create(intraComm, 0, CXios::globalComm, clientLeader, 0, &newComm) ;
176              interCommLeft.push_back(newComm) ;
177            }
178          }
179
180          for (int i = 0; i < sndServerGlobalRanks.size(); ++i)
181          {
182            int intraCommSize, intraCommRank ;
183            MPI_Comm_size(intraComm, &intraCommSize) ;
184            MPI_Comm_rank(intraComm, &intraCommRank) ;
185            info(50)<<"intercommCreate::client (server level 1) "<<rank_<<" intraCommSize : "<<intraCommSize
186                <<" intraCommRank :"<<intraCommRank<<"  clientLeader "<< sndServerGlobalRanks[i]<<endl ;
187            MPI_Intercomm_create(intraComm, 0, CXios::globalComm, sndServerGlobalRanks[i], 1, &newComm) ;
188            interCommRight.push_back(newComm) ;
189          }
190        }
191        else
192        {
193          int clientLeader;
194          clientLeader = leaders[hashString(CXios::xiosCodeId)];
195          int intraCommSize, intraCommRank ;
196          MPI_Comm_size(intraComm, &intraCommSize) ;
197          MPI_Comm_rank(intraComm, &intraCommRank) ;
198          info(50)<<"intercommCreate::server (server level 2) "<<rank_<<" intraCommSize : "<<intraCommSize
199                   <<" intraCommRank :"<<intraCommRank<<"  clientLeader "<< clientLeader<<endl ;
200
201          MPI_Intercomm_create(intraComm, 0, CXios::globalComm, clientLeader, 1, &newComm) ;
202          interCommLeft.push_back(newComm) ;
203        }
204
205        delete [] hashAll ;
206
207      }
208      // using OASIS
209      else
210      {
211        int size;
212        int myColor;
213        if (!is_MPI_Initialized) oasis_init(CXios::xiosCodeId);
214
215        CTimer::get("XIOS").resume() ;
216        MPI_Comm localComm;
217        oasis_get_localcomm(localComm);
218
219        // Create server intraComm
220        if (!CXios::usingServer2)
221        {
222          MPI_Comm_dup(localComm, &intraComm);
223          MPI_Comm_rank(localComm,&rank_) ;
224        }
225        else
226        {
227          MPI_Comm_rank(localComm,&rank_) ;
228          MPI_Comm_size(localComm,&size) ;
229
230          for (int i=size*CXios::ratioServer2/100; i<size; i++)
231            sndServerGlobalRanks.push_back(i);
232
233          if ( rank_ < (size - sndServerGlobalRanks.size()) )
234          {
235            serverLevel = 1;
236            myColor = 0;
237          }
238          else
239          {
240            serverLevel = 2;
241            myColor = rank_;
242          }
243          MPI_Comm_split(localComm, myColor, rank_, &intraComm) ;
244
245        }
246        MPI_Comm_size(intraComm,&size) ;
247        string codesId=CXios::getin<string>("oasis_codes_id") ;
248
249        vector<string> splitted ;
250        boost::split( splitted, codesId, boost::is_any_of(","), boost::token_compress_on ) ;
251        vector<string>::iterator it ;
252
253        MPI_Comm newComm ;
254        int globalRank ;
255        MPI_Comm_rank(CXios::globalComm,&globalRank);
256
257        for(it=splitted.begin();it!=splitted.end();it++)
258        {
259          oasis_get_intercomm(newComm,*it) ;
260//        interComm.push_back(newComm) ;
261          if ( !CXios::usingServer2)
262          {
263            interCommLeft.push_back(newComm) ;
264            if (rank_==0) MPI_Send(&globalRank,1,MPI_INT,0,0,newComm) ;
265          }
266          else
267          {
268            if (serverLevel == 1)
269            {
270              interCommLeft.push_back(newComm) ;
271              if (rank_==0) MPI_Send(&globalRank,1,MPI_INT,0,0,newComm) ;
272              for (int i = 0; i < sndServerGlobalRanks.size(); ++i)
273              {
274                int srvSndLeader = sndServerGlobalRanks[i];
275                info(50)<<"intercommCreate::client (server level 1) "<<globalRank<<" intraCommSize : "<<size
276                    <<" intraCommRank :"<<rank_<<"  clientLeader "<< srvSndLeader<<endl ;
277                MPI_Intercomm_create(intraComm, 0, CXios::globalComm, srvSndLeader, 0, &newComm) ;
278                interCommRight.push_back(newComm) ;
279              }
280            }
281            else if (serverLevel == 2)
282            {
283              info(50)<<"intercommCreate::server "<<rank_<<" intraCommSize : "<<size
284                       <<" intraCommRank :"<<rank_<<"  clientLeader "<< 0<<endl ;
285              MPI_Intercomm_create(intraComm, 0, localComm, 0, 0, &newComm) ;
286              interCommLeft.push_back(newComm) ;
287            }
288          }
289//          MPI_Comm_remote_size(newComm,&size);
290        }
291              oasis_enddef() ;
292      }
293
294      MPI_Comm_rank(intraComm, &rank) ;
295      if (rank==0) isRoot=true;
296      else isRoot=false;
297     
298      eventScheduler = new CEventScheduler(intraComm) ;
299    }
300
301    void CServer::finalize(void)
302    {
303      CTimer::get("XIOS").suspend() ;
304     
305      delete eventScheduler ;
306
307      for (std::list<MPI_Comm>::iterator it = contextInterComms.begin(); it != contextInterComms.end(); it++)
308        MPI_Comm_free(&(*it));
309
310      for (std::list<MPI_Comm>::iterator it = contextIntraComms.begin(); it != contextIntraComms.end(); it++)
311        MPI_Comm_free(&(*it));
312
313//      for (std::list<MPI_Comm>::iterator it = interComm.begin(); it != interComm.end(); it++)
314//        MPI_Comm_free(&(*it));
315
316//        for (std::list<MPI_Comm>::iterator it = interCommLeft.begin(); it != interCommLeft.end(); it++)
317//          MPI_Comm_free(&(*it));
318
319        for (std::list<MPI_Comm>::iterator it = interCommRight.begin(); it != interCommRight.end(); it++)
320          MPI_Comm_free(&(*it));
321
322      MPI_Comm_free(&intraComm);
323
324      if (!is_MPI_Initialized)
325      {
326        if (CXios::usingOasis) oasis_finalize();
327        else MPI_Finalize() ;
328      }
329      report(0)<<"Performance report : Time spent for XIOS : "<<CTimer::get("XIOS server").getCumulatedTime()<<endl  ;
330      report(0)<<"Performance report : Time spent in processing events : "<<CTimer::get("Process events").getCumulatedTime()<<endl  ;
331      report(0)<<"Performance report : Ratio : "<<CTimer::get("Process events").getCumulatedTime()/CTimer::get("XIOS server").getCumulatedTime()*100.<<"%"<<endl  ;
332      report(100)<<CTimer::getAllCumulatedTime()<<endl ;
333    }
334
335     void CServer::eventLoop(void)
336     {
337       bool stop=false ;
338
339       CTimer::get("XIOS server").resume() ;
340       while(!stop)
341       {
342         if (isRoot)
343         {
344           listenContext();
345           listenRootContext();
346           if (!finished) listenFinalize() ;
347         }
348         else
349         {
350           listenRootContext();
351           if (!finished) listenRootFinalize() ;
352         }
353
354         contextEventLoop() ;
355         if (finished && contextList.empty()) stop=true ;
356         eventScheduler->checkEvent() ;
357       }
358       CTimer::get("XIOS server").suspend() ;
359     }
360
361     void CServer::listenFinalize(void)
362     {
363        list<MPI_Comm>::iterator it, itr;
364        int msg ;
365        int flag ;
366
367        for(it=interCommLeft.begin();it!=interCommLeft.end();it++)
368        {
369           MPI_Status status ;
370           traceOff() ;
371           MPI_Iprobe(0,0,*it,&flag,&status) ;
372           traceOn() ;
373           if (flag==true)
374           {
375              MPI_Recv(&msg,1,MPI_INT,0,0,*it,&status) ;
376              info(20)<<" CServer : Receive client finalize"<<endl ;
377              // Sending server finalize message to secondary servers (if any)
378              for(itr=interCommRight.begin();itr!=interCommRight.end();itr++)
379              {
380                MPI_Send(&msg,1,MPI_INT,0,0,*itr) ;
381              }
382              MPI_Comm_free(&(*it));
383              interCommLeft.erase(it) ;
384              break ;
385            }
386         }
387
388         if (interCommLeft.empty())
389         {
390           int i,size ;
391           MPI_Comm_size(intraComm,&size) ;
392           MPI_Request* requests= new MPI_Request[size-1] ;
393           MPI_Status* status= new MPI_Status[size-1] ;
394
395           for(int i=1;i<size;i++) MPI_Isend(&msg,1,MPI_INT,i,4,intraComm,&requests[i-1]) ;
396           MPI_Waitall(size-1,requests,status) ;
397
398           finished=true ;
399           delete [] requests ;
400           delete [] status ;
401         }
402     }
403
404
405     void CServer::listenRootFinalize()
406     {
407        int flag ;
408        MPI_Status status ;
409        int msg ;
410
411        traceOff() ;
412        MPI_Iprobe(0,4,intraComm, &flag, &status) ;
413        traceOn() ;
414        if (flag==true)
415        {
416           MPI_Recv(&msg,1,MPI_INT,0,4,intraComm,&status) ;
417           finished=true ;
418        }
419      }
420
421     void CServer::listenContext(void)
422     {
423
424       MPI_Status status ;
425       int flag ;
426       static char* buffer ;
427       static MPI_Request request ;
428       static bool recept=false ;
429       int rank ;
430       int count ;
431
432       if (recept==false)
433       {
434         traceOff() ;
435         MPI_Iprobe(MPI_ANY_SOURCE,1,CXios::globalComm, &flag, &status) ;
436         traceOn() ;
437         if (flag==true)
438         {
439           rank=status.MPI_SOURCE ;
440           MPI_Get_count(&status,MPI_CHAR,&count) ;
441           buffer=new char[count] ;
442           MPI_Irecv((void*)buffer,count,MPI_CHAR,rank,1,CXios::globalComm,&request) ;
443           recept=true ;
444         }
445       }
446       else
447       {
448         traceOff() ;
449         MPI_Test(&request,&flag,&status) ;
450         traceOn() ;
451         if (flag==true)
452         {
453           rank=status.MPI_SOURCE ;
454           MPI_Get_count(&status,MPI_CHAR,&count) ;
455           recvContextMessage((void*)buffer,count) ;
456           delete [] buffer ;
457           recept=false ;
458         }
459       }
460     }
461
462     void CServer::recvContextMessage(void* buff,int count)
463     {
464       static map<string,contextMessage> recvContextId;
465       map<string,contextMessage>::iterator it ;
466       CBufferIn buffer(buff,count) ;
467       string id ;
468       int clientLeader ;
469       int nbMessage ;
470
471       buffer>>id>>nbMessage>>clientLeader ;
472
473       it=recvContextId.find(id) ;
474       if (it==recvContextId.end())
475       {
476         contextMessage msg={0,0} ;
477         pair<map<string,contextMessage>::iterator,bool> ret ;
478         ret=recvContextId.insert(pair<string,contextMessage>(id,msg)) ;
479         it=ret.first ;
480       }
481       it->second.nbRecv+=1 ;
482       it->second.leaderRank+=clientLeader ;
483
484       if (it->second.nbRecv==nbMessage)
485       {
486         int size ;
487         MPI_Comm_size(intraComm,&size) ;
488//         MPI_Request* requests= new MPI_Request[size-1] ;
489//         MPI_Status* status= new MPI_Status[size-1] ;
490         MPI_Request* requests= new MPI_Request[size] ;
491         MPI_Status* status= new MPI_Status[size] ;
492
493         CMessage msg ;
494         msg<<id<<it->second.leaderRank;
495         int messageSize=msg.size() ;
496         void * sendBuff = new char[messageSize] ;
497         CBufferOut sendBuffer(sendBuff,messageSize) ;
498         sendBuffer<<msg ;
499
500         // Include root itself in order not to have a divergence
501         for(int i=0; i<size; i++)
502         {
503           MPI_Isend(sendBuff,count,MPI_CHAR,i,2,intraComm,&requests[i]) ;
504         }
505
506//         for(int i=1;i<size;i++)
507//         {
508//            MPI_Isend(buff,count,MPI_CHAR,i,2,intraComm,&requests[i-1]) ;
509//         }
510//         MPI_Waitall(size-1,requests,status) ;
511//         registerContext(buff,count,it->second.leaderRank) ;
512
513         recvContextId.erase(it) ;
514         delete [] requests ;
515         delete [] status ;
516
517       }
518     }
519
520     void CServer::listenRootContext(void)
521     {
522       MPI_Status status ;
523       int flag ;
524       static void* buffer ;
525       static MPI_Request request ;
526       static bool recept=false ;
527       int rank ;
528//       int count ;
529       static int count ;
530       const int root=0 ;
531       boost::hash<string> hashString;
532       size_t hashId = hashString("RegisterContext");
533
534       // (1) Receive context id from the root
535       if (recept==false)
536       {
537         traceOff() ;
538         MPI_Iprobe(root,2,intraComm, &flag, &status) ;
539         traceOn() ;
540         if (flag==true)
541         {
542           MPI_Get_count(&status,MPI_CHAR,&count) ;
543           buffer=new char[count] ;
544           MPI_Irecv((void*)buffer,count,MPI_CHAR,root,2,intraComm,&request) ;
545           recept=true ;
546         }
547       }
548       // (2) If context id is received, save it into a buffer and register an event
549       else
550       {
551         MPI_Test(&request,&flag,&status) ;
552         if (flag==true)
553         {
554           MPI_Get_count(&status,MPI_CHAR,&count) ;
555           eventScheduler->registerEvent(nbContexts,hashId);
556//           registerContext((void*)buffer,count) ;
557//           delete [] buffer ;
558           recept=false ;
559         }
560       }
561       // (3) If event has been scheduled, call register context
562       if (eventScheduler->queryEvent(nbContexts,hashId))
563       {
564         registerContext(buffer,count) ;
565         ++nbContexts;
566         delete [] buffer ;
567       }
568     }
569
570     void CServer::registerContext(void* buff, int count, int leaderRank)
571     {
572       string contextId;
573       CBufferIn buffer(buff, count);
574//       buffer >> contextId;
575       buffer >> contextId>>leaderRank;
576       CContext* context;
577
578       info(20) << "CServer : Register new Context : " << contextId << endl;
579
580       if (contextList.find(contextId) != contextList.end())
581         ERROR("void CServer::registerContext(void* buff, int count, int leaderRank)",
582               << "Context '" << contextId << "' has already been registred");
583
584       context=CContext::create(contextId);
585       contextList[contextId]=context;
586
587       // Primary or classical server: create communication channel with a client
588       // (1) create interComm (with a client)
589       // (2) initialize client and server (contextClient and contextServer)
590       MPI_Comm inter;
591       if (serverLevel < 2)
592       {
593         MPI_Comm contextInterComm;
594         MPI_Intercomm_create(intraComm, 0, CXios::globalComm, leaderRank, 10+leaderRank, &contextInterComm);
595         MPI_Intercomm_merge(contextInterComm,1,&inter);
596         MPI_Barrier(inter);
597         MPI_Comm_free(&inter);
598         context->initServer(intraComm,contextInterComm);
599         contextInterComms.push_back(contextInterComm);
600
601       }
602       // Secondary server: create communication channel with a primary server
603       // (1) duplicate interComm with a primary server
604       // (2) initialize client and server (contextClient and contextServer)
605       // Remark: in the case of the secondary server there is no need to create an interComm calling MPI_Intercomm_create,
606       //         because interComm of CContext is defined on the same processes as the interComm of CServer.
607       //         So just duplicate it.
608       else if (serverLevel == 2)
609       {
610         MPI_Comm_dup(interCommLeft.front(), &inter);
611         contextInterComms.push_back(inter);
612         context->initServer(intraComm, contextInterComms.back());
613       }
614
615       // Primary server:
616       // (1) send create context message to secondary servers
617       // (2) initialize communication channels with secondary servers (create contextClient and contextServer)
618       if (serverLevel == 1)
619       {
620         int i = 0, size;
621         CMessage msg;
622         int messageSize;
623         MPI_Comm_size(intraComm, &size) ;
624         for (std::list<MPI_Comm>::iterator it = interCommRight.begin(); it != interCommRight.end(); it++, ++i)
625         {
626           StdString str = contextId +"_server_" + boost::lexical_cast<string>(i);
627           msg<<str<<size<<rank_ ;
628           messageSize = msg.size() ;
629           buff = new char[messageSize] ;
630           CBufferOut buffer(buff,messageSize) ;
631           buffer<<msg ;
632           MPI_Send(buff, buffer.count(), MPI_CHAR, sndServerGlobalRanks[i], 1, CXios::globalComm) ;
633           MPI_Comm_dup(*it, &inter);
634           contextInterComms.push_back(inter);
635           MPI_Comm_dup(intraComm, &inter);
636           contextIntraComms.push_back(inter);
637           context->initClient(contextIntraComms.back(), contextInterComms.back()) ;
638           delete [] buff ;
639         }
640       }
641     }
642
643     void CServer::contextEventLoop(void)
644     {
645       bool isFinalized ;
646       map<string,CContext*>::iterator it ;
647
648       for(it=contextList.begin();it!=contextList.end();it++)
649       {
650         isFinalized=it->second->isFinalized();
651         if (isFinalized)
652         {
653           contextList.erase(it) ;
654           break ;
655         }
656         else
657           it->second->checkBuffersAndListen();
658       }
659     }
660
661     //! Get rank of the current process in the intraComm
662     int CServer::getRank()
663     {
664       int rank;
665       MPI_Comm_rank(intraComm,&rank);
666       return rank;
667     }
668
669     vector<int>& CServer::getSecondaryServerGlobalRanks()
670     {
671       return sndServerGlobalRanks;
672     }
673
674    /*!
675    * Open a file specified by a suffix and an extension and use it for the given file buffer.
676    * The file name will be suffix+rank+extension.
677    *
678    * \param fileName[in] protype file name
679    * \param ext [in] extension of the file
680    * \param fb [in/out] the file buffer
681    */
682    void CServer::openStream(const StdString& fileName, const StdString& ext, std::filebuf* fb)
683    {
684      StdStringStream fileNameClient;
685      int numDigit = 0;
686      int size = 0;
687      int id;
688      MPI_Comm_size(CXios::globalComm, &size);
689      while (size)
690      {
691        size /= 10;
692        ++numDigit;
693      }
694      id = rank_; //getRank();
695
696      fileNameClient << fileName << "_" << std::setfill('0') << std::setw(numDigit) << id << ext;
697      fb->open(fileNameClient.str().c_str(), std::ios::out);
698      if (!fb->is_open())
699        ERROR("void CServer::openStream(const StdString& fileName, const StdString& ext, std::filebuf* fb)",
700              << std::endl << "Can not open <" << fileNameClient << "> file to write the server log(s).");
701    }
702
703    /*!
704    * \brief Open a file stream to write the info logs
705    * Open a file stream with a specific file name suffix+rank
706    * to write the info logs.
707    * \param fileName [in] protype file name
708    */
709    void CServer::openInfoStream(const StdString& fileName)
710    {
711      std::filebuf* fb = m_infoStream.rdbuf();
712      openStream(fileName, ".out", fb);
713
714      info.write2File(fb);
715      report.write2File(fb);
716    }
717
718    //! Write the info logs to standard output
719    void CServer::openInfoStream()
720    {
721      info.write2StdOut();
722      report.write2StdOut();
723    }
724
725    //! Close the info logs file if it opens
726    void CServer::closeInfoStream()
727    {
728      if (m_infoStream.is_open()) m_infoStream.close();
729    }
730
731    /*!
732    * \brief Open a file stream to write the error log
733    * Open a file stream with a specific file name suffix+rank
734    * to write the error log.
735    * \param fileName [in] protype file name
736    */
737    void CServer::openErrorStream(const StdString& fileName)
738    {
739      std::filebuf* fb = m_errorStream.rdbuf();
740      openStream(fileName, ".err", fb);
741
742      error.write2File(fb);
743    }
744
745    //! Write the error log to standard error output
746    void CServer::openErrorStream()
747    {
748      error.write2StdErr();
749    }
750
751    //! Close the error log file if it opens
752    void CServer::closeErrorStream()
753    {
754      if (m_errorStream.is_open()) m_errorStream.close();
755    }
756}
Note: See TracBrowser for help on using the repository browser.