Improve Counter demo

hassain-saeed28 · Mar 10, 2024 · b2c3a3e · b2c3a3e
1 parent 687387f
commit b2c3a3e
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/src/main/java/bftsmart/demo/counter/CounterClient.java b/src/main/java/bftsmart/demo/counter/CounterClient.java
@@ -37,11 +37,8 @@ public static void main(String[] args) throws IOException {
             System.out.println("       default <number of operations> equals 1000");
             System.exit(-1);
         }
-
-        ServiceProxy counterProxy = new ServiceProxy(Integer.parseInt(args[0]));
 
-        try {
-
+        try (ServiceProxy counterProxy = new ServiceProxy(Integer.parseInt(args[0]))) {
             int inc = Integer.parseInt(args[1]);
             int numberOfOps = (args.length > 2) ? Integer.parseInt(args[2]) : 1000;
 
@@ -63,8 +60,6 @@ public static void main(String[] args) throws IOException {
                     break;
                 }
             }
-        } catch(IOException | NumberFormatException e){
-            counterProxy.close();
         }
     }
 }