Changeset 67 for osgVisual/src/cluster

Timestamp:

Jul 11, 2010, 1:31:29 PM (14 years ago)

Author:

Torben Dannhauer

Message:

 

Location:

osgVisual/src/cluster

Files:

• dataIO_clusterENet.cpp (modified) (7 diffs)
• dataIO_clusterENet_implementation.cpp (modified) (2 diffs)

osgVisual/src/cluster/dataIO_clusterENet.cpp

@@ -22 +22 @@
 {
         OSG_NOTIFY( osg::ALWAYS ) << "clusterENet constructed" << std::endl;
+        serverToConnect = "unknown";
 }
+
 
 dataIO_clusterENet::~dataIO_clusterENet(void)
@@ -29 +31 @@
 }
 
+
 void dataIO_clusterENet::init( osg::ArgumentParser& arguments_, clustermode clusterMode_, osgVisual::dataIO_transportContainer* sendContainer_, bool compressionEnabled_, bool asAscii_ )
 {
         OSG_NOTIFY( osg::ALWAYS ) << "clusterENet init();" << std::endl;
         
+        // Configure the clustermode
+        clusterMode = clusterMode_;
+
+        // store sendContainer
         sendContainer = sendContainer_;
         
         // create ENet implementation object.
         enet_impl = new osgVisual::dataIO_clusterENet_implementation();
+
+        // initialize ENet implementation
+        if(clusterMode == MASTER)
+        {
+                enet_impl->init(dataIO_clusterENet_implementation::SERVER, port);
+
+                initialized = true;
+        }
+        if(clusterMode == SLAVE)
+        {
+                // Get the server IP
+                if(!arguments_.read("--server",serverToConnect, port))
+                {
+                        // try server discovery
+                        //discoverServer(serverToConnect,port);
+                        /* todo : implement a udp server discovery based on ASIO */
+                }
+
+                // Init ENet
+                enet_impl->init(dataIO_clusterENet_implementation::CLIENT, port);
+
+                // Connect to server with 5 retries:
+                bool connected = false;
+                for(int i=0; i<5; i++)
+                {
+                        std::cout << "Try to connect to server " << serverToConnect << std::endl;
+                        if( enet_impl->connectTo( serverToConnect.c_str(), 5000 ) )
+                        {
+                                // Connect successful.
+                                initialized = true;
+                                connected = true;
+                                break;
+                        }
+                }       // For END
+                if(!connected)
+                {
+                        initialized = false;
+                        std::cout << "Failed to establish connection to server " << serverToConnect << std::endl;
+                }
+        }       // IF SLAVE END
 }
+
 
 void dataIO_clusterENet::shutdown()
@@ -43 +91 @@
         OSG_NOTIFY( osg::ALWAYS ) << "clusterENet shutdown();" << std::endl;
 }
+
 
 bool dataIO_clusterENet::sendTO_OBJvaluesToSlaves()
@@ -51 +100 @@
 }
 
+
 bool dataIO_clusterENet::readTO_OBJvaluesFromMaster()
 {
@@ -57 +107 @@
         return true;
 }
+
 
 void dataIO_clusterENet::reportAsReadyToSwap()
@@ -70 +121 @@
 }
 
+
 bool dataIO_clusterENet::waitForAllReadyToSwap()
 {
@@ -77 +129 @@
 }
 
+
 bool dataIO_clusterENet::sendSwapCommand()
 {

osgVisual/src/cluster/dataIO_clusterENet_implementation.cpp

@@ -9 +9 @@
         std::cout << "Instantiated server class# "<< activeENetInstances << std::endl;
         serverInitialized = false;
+        host = NULL;
         // Start ENet
         if (activeENetInstances++ == 0)
@@ -25 +26 @@
 dataIO_clusterENet_implementation::~dataIO_clusterENet_implementation()
 {
-        // Delete Host Object
-        enet_host_destroy(host);
+        // Delete Host Object if instantiated
+        if(host)
+                enet_host_destroy(host);
 
         // Stop ENet if it is the last element.