Use correct waiting methods for shutdown futures.

Author: Brandon Dahler

README.md

@@ -65,7 +65,7 @@ logger.putProperty("RequestId", "422b1569-16f6-4a03-b8f0-fe3fd9b100f8");
 logger.flush();
 
 // flush the sink, waiting up to 10s before giving up
-environment.getSink().shutdown().orTimeout(10_000L, TimeUnit.MILLISECONDS);
+environment.getSink().shutdown().get(10_000L, TimeUnit.MILLISECONDS);
 ```
 
 ## API

examples/agent/src/main/java/agent/App.java

@@ -7,16 +7,18 @@
 import software.amazon.cloudwatchlogs.emf.model.DimensionSet;
 import software.amazon.cloudwatchlogs.emf.model.Unit;
 
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeoutException;
 import java.util.concurrent.TimeUnit;
 
 public class App {
 
-    public static void main(String[] args) {
+    public static void main(String[] args) throws ExecutionException, InterruptedException, TimeoutException {
         DefaultEnvironment environment = new DefaultEnvironment(EnvironmentConfigurationProvider.getConfig());
         emitMetric(environment);
         emitMetric(environment);
         emitMetric(environment);
-        environment.getSink().shutdown().orTimeout(360_000L, TimeUnit.MILLISECONDS);
+        environment.getSink().shutdown().get(360_000L, TimeUnit.MILLISECONDS);
     }
 
     private static void emitMetric(Environment environment) {

examples/ecs-firelens/src/main/java/App.java

@@ -28,6 +28,8 @@
 import java.io.IOException;
 import java.io.OutputStream;
 import java.net.InetSocketAddress;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeoutException;
 import java.util.concurrent.TimeUnit;
 
 public class App {
@@ -48,7 +50,16 @@ public static void main(String[] args) throws Exception {
     private static void registerShutdownHook() {
         // https://aws.amazon.com/blogs/containers/graceful-shutdowns-with-ecs/
         Signal.handle(new Signal("TERM"), sig -> {
-            env.getSink().shutdown().orTimeout(1_000L, TimeUnit.MILLISECONDS);
+            try {
+                env.getSink().shutdown().get(1_000L, TimeUnit.MILLISECONDS);
+            } catch (InterruptedException | ExecutionException | TimeoutException e) {
+                if (e instanceof InterruptedException) {
+                    Thread.currentThread().interrupt();
+                }
+
+                e.printStackTrace();
+            }
+
             System.exit(0);
         });
     }