Hey, that's better. The server now idles the same, but sends data uber-fast!

2 files changed, 18 insertions, 4 deletions

diff --git a/src/client.h b/src/client.h
index 954f134..2cd700b 100644
--- a/src/client.h
+++ b/src/client.h
@@ -64,6 +64,8 @@ namespace Bu
 
                 void onMessage( const Bu::FString &sMsg );
 
+                bool hasOutput() { return !sWriteBuf.isEmpty(); }
+
         private:
                 Bu::Socket *pSocket;
                 Bu::Protocol *pProto;
diff --git a/src/server.cpp b/src/server.cpp
index eb85997..804bbcc 100644
--- a/src/server.cpp
+++ b/src/server.cpp
@@ -59,10 +59,18 @@ void Bu::Server::scan()
         struct timeval xTimeout = { nTimeoutSec, nTimeoutUSec };
         
         fd_set fdRead = fdActive;
-        fd_set fdWrite = fdActive;
+        fd_set fdWrite /* = fdActive*/;
         fd_set fdException = fdActive;
 
-        if( TEMP_FAILURE_RETRY( select( FD_SETSIZE, &fdRead, NULL, &fdException, &xTimeout ) ) < 0 )
+        FD_ZERO( &fdWrite );
+        for( ClientHash::iterator i = hClients.begin(); i != hClients.end(); i++ )
+        {
+                if( (*i)->hasOutput() )
+                        FD_SET( i.getKey(), &fdWrite );
+        }
+
+        if( TEMP_FAILURE_RETRY( select( FD_SETSIZE,
+                        &fdRead, &fdWrite, &fdException, &xTimeout ) ) < 0 )
         {
                 throw ExceptionBase("Error attempting to scan open connections.");
         }
@@ -88,6 +96,11 @@ void Bu::Server::scan()
                                 }
                         }
                 }
+                if( FD_ISSET( j, &fdWrite ) )
+                {
+                        Client *pClient = hClients.get( j );
+                        pClient->processOutput();
+                }
         }
 
         Bu::List<int> lDelete;
@@ -96,8 +109,7 @@ void Bu::Server::scan()
         // wanting to accept writes (using a select).
         for( ClientHash::iterator i = hClients.begin(); i != hClients.end(); i++ )
         {
-                (*i)->processOutput();
-                if( (*i)->wantsDisconnect() )
+                if( (*i)->wantsDisconnect() && !(*i)->hasOutput() )
                 {
                         lDelete.append( i.getKey() );
                 }