core/sqf/monitor/linux/commaccept.h

///////////////////////////////////////////////////////////////////////////////
//
// @@@ START COPYRIGHT @@@
//
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.
//
// @@@ END COPYRIGHT @@@
//
///////////////////////////////////////////////////////////////////////////////

#ifndef COMMACCEPT_H
#define COMMACCEPT_H

#include <string>
#include <pthread.h>
#include <mpi.h>
#include "lock.h"
#include "comm.h"

class CCommAccept : public CLock
                  , public CComm
{
public:

    CCommAccept();
    virtual ~CCommAccept();

    void commAcceptor( void );
    inline const char *getCommPort( void ) { return commPort_.c_str(); }
    inline int         getCommSocketPort( void ) { return( commSocketPort_ ); }
    bool isAccepting( void ) { CAutoLock lock(getLocker()); return( accepting_ ); }
    void processNewComm( MPI_Comm interComm );
    void processNewSock( int sockFd );
    inline void setCommPort( char *commPort) { commPort_ = commPort; }
    inline void setCommSocketPort( int commSocketPort) { commSocketPort_ = commSocketPort; }
    void startAccepting( void );
    void stopAccepting( void );
    void start( void );
    void shutdownWork( void );

private:
    struct message_def *Notice( const char *msgText );

    void commAcceptorIB( void );
    void commAcceptorSock( void );
    void connectToCommSelf( void );
    bool sendNodeInfoMPI( MPI_Comm interComm );
    bool sendNodeInfoSock( int sockFd );

    bool        accepting_;
    bool        shutdown_;
    int         ioWaitTimeout_;
    int         ioRetryCount_;
    int         commSock_;
    int         commSocketPort_;    // Node Re-Integration port
    string      commPort_;          // Node Re-Integration port ip address
    pthread_t   thread_id_;         // commAccept thread's id
};

#endif