move EndMsg to the right place

It's somewhat confusing in the finishing block, when it should actually happen only after all the FileChunkMsg's are sent.
darrengarvey · Mar 20, 2014 · 555b81f · 555b81f
1 parent 8bdc34b
commit 555b81f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/daemon/serve.cpp b/daemon/serve.cpp
@@ -205,6 +205,10 @@ int handle_connection(const string &basedir, CompileJob *job,
                 }
 
                 if (!bytes) {
+                    if( !client->send_msg(EndMsg())) {
+                        log_info() << "write of obj end failed " << endl;
+                        throw myexception(EXIT_DISTCC_FAILED);
+                    }
                     break;
                 }
 
@@ -220,10 +224,6 @@ int handle_connection(const string &basedir, CompileJob *job,
         throw myexception(rmsg.status);
 
     } catch (myexception e) {
-        if (client && e.exitcode() == 0) {
-            client->send_msg(EndMsg());
-        }
-
         delete client;
         client = 0;