mariadb/ndb/include/util/SocketServer.hpp
stewart@mysql.com 1edca0ac8c Impl 3 of WL2278 - Dynamic port allocation of cluster nodes
- Change configuration handling so that no default ports are given, but instead 0 is set (port to be dynamic)
- TransporterRegistry::start_service now only tries to connect when fetched port number > 0
2005-01-04 15:20:27 +11:00

133 lines
3.2 KiB
C++

/* Copyright (C) 2003 MySQL AB
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 */
#ifndef SOCKET_SERVER_HPP
#define SOCKET_SERVER_HPP
#include <NdbTCP.h>
#include <NdbMutex.h>
#include <NdbThread.h>
#include <Vector.hpp>
extern "C" void* sessionThread_C(void*);
extern "C" void* socketServerThread_C(void*);
/**
* Socket Server
*/
class SocketServer {
public:
/**
* A Session
*/
class Session {
public:
virtual ~Session() {}
virtual void runSession(){}
virtual void stopSession(){ m_stop = true; }
protected:
friend class SocketServer;
friend void* sessionThread_C(void*);
Session(NDB_SOCKET_TYPE sock): m_socket(sock){ m_stop = m_stopped = false;}
bool m_stop; // Has the session been ordered to stop?
bool m_stopped; // Has the session stopped?
NDB_SOCKET_TYPE m_socket;
};
/**
* A service i.e. a session factory
*/
class Service {
public:
virtual ~Service(){}
/**
* Returned Session will be ran in own thread
*
* To manage threads self, just return NULL
*/
virtual Session * newSession(NDB_SOCKET_TYPE theSock) = 0;
virtual void stopSessions(){}
};
/**
* Constructor / Destructor
*/
SocketServer(int maxSessions = 32);
~SocketServer();
/**
* Setup socket and bind it
* then close the socket
* Returns true if succeding in binding
*/
static bool tryBind(unsigned short port, const char * intface = 0);
/**
* Setup socket
* bind & listen
* Returns false if no success
*/
bool setup(Service *, unsigned short *port, const char * pinterface = 0);
/**
* start/stop the server
*/
void startServer();
void stopServer();
/**
* stop sessions
*
* Note: Implies stopServer
*/
void stopSessions(bool wait = false);
void foreachSession(void (*f)(Session*, void*), void *data);
private:
struct SessionInstance {
Service * m_service;
Session * m_session;
NdbThread * m_thread;
};
struct ServiceInstance {
Service * m_service;
NDB_SOCKET_TYPE m_socket;
};
MutexVector<SessionInstance> m_sessions;
MutexVector<ServiceInstance> m_services;
unsigned m_maxSessions;
void doAccept();
void checkSessions();
void startSession(SessionInstance &);
/**
* Note, this thread is only used when running interactive
*
*/
bool m_stopThread;
struct NdbThread * m_thread;
NdbLockable m_threadLock;
void doRun();
friend void* socketServerThread_C(void*);
friend void* sessionThread_C(void*);
};
#endif