1 files changed, 133 insertions, 0 deletions
diff --git a/storage/ndb/include/util/SocketServer.hpp b/storage/ndb/include/util/SocketServer.hpp
new file mode 100644
index 00000000000..2e1afb74945
--- /dev/null
+++ b/storage/ndb/include/util/SocketServer.hpp
@@ -0,0 +1,133 @@
+/* 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