fixed small bug in scala example

mbknor · May 22, 2011 · 573b46f · 573b46f
1 parent 0a66e99
commit 573b46f
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/examples/ddsl-scala-examples/src/main/scala/ddslexamples/DdslServiceProvider.scala b/examples/ddsl-scala-examples/src/main/scala/ddslexamples/DdslServiceProvider.scala
@@ -54,16 +54,22 @@ object DdslServiceProvider{
         val out = new BufferedWriter( new OutputStreamWriter(s.getOutputStream ) )
         val in = new BufferedReader( new InputStreamReader( s.getInputStream))
 
-        while( true ){
+        var continue = true
+        while( continue ){
           //wait for command from client
           val command = in.readLine
-          println("Received: " + command)
-
-          //send greeting
-          val msg = "Hello from server at port " + port + " ("+System.currentTimeMillis+")"
-          println("Sending: " + msg)
-          out.write(msg + "\n");
-          out.flush
+          if (command == null) {
+              s.close();
+              continue = false
+          } else {
+              println("Received: " + command)
+
+              //send greeting
+              val msg = "Hello from server at port " + port + " ("+System.currentTimeMillis+")"
+              println("Sending: " + msg)
+              out.write(msg + "\n");
+              out.flush
+          }
         }
       }catch{
         case _ => println("Lost connection")