1 /***********************************************************************
2  * Copyright (c) 2013-2024 Commonwealth Computer Research, Inc.
3  * All rights reserved. This program and the accompanying materials
4  * are made available under the terms of the Apache License, Version 2.0
5  * which accompanies this distribution and is available at
6  * http://www.opensource.org/licenses/apache2.0.php.
7  ***********************************************************************/
8 
9 package org.locationtech.geomesa.utils
10 
11 import com.typesafe.scalalogging.LazyLogging
12 
13 import scala.util.{Failure, Success}
14 
15 package object io {
16 
17   /**
18    * Closes and logs any exceptions
19    */
20   object CloseWithLogging extends SafeClose with LazyLogging {
21     override def apply[C : IsCloseable](c: C): Option[Throwable] = {
22       implicitly[IsCloseable[C]].close(c) match {
23         case _: Success[Unit] => None
24         case Failure(e) => logger.warn(s"Error calling close on '$c': ", e); Some(e)
25       }
26     }
27   }
28 
29   /**
30    * Closes and catches any exceptions
31    */
32   object CloseQuietly extends SafeClose {
33     override def apply[C : IsCloseable](c: C): Option[Throwable] =
34       implicitly[IsCloseable[C]].close(c).failed.toOption
35   }
36 
37   /**
38    * Flushes and logs any exceptions
39    */
40   object FlushWithLogging extends SafeFlush with LazyLogging {
41     override def apply[F : IsFlushable](f: F): Option[Throwable] = {
42       implicitly[IsFlushable[F]].flush(f) match {
43         case _: Success[Unit] => None
44         case Failure(e) => logger.warn(s"Error calling flush on '$f': ", e); Some(e)
45       }
46     }
47   }
48 
49   /**
50    * Flushes and catches any exceptions
51    */
52   object FlushQuietly extends SafeFlush {
53     override def apply[F : IsFlushable](f: F): Option[Throwable] =
54       implicitly[IsFlushable[F]].flush(f).failed.toOption
55   }
56 
57   /**
58    * Similar to java's try-with-resources, allows for using an object then closing in a finally block
59    */
60   object WithClose {
61 
62     def apply[C : IsCloseable, T](c: C)(fn: C => T): T = {
63       val ev = implicitly[IsCloseable[C]]
64       try { fn(c) } finally { if (c != null) { ev.close(c) }}
65     }
66 
67     def apply[C1 : IsCloseable, C2 : IsCloseable, T](c1: C1, c2: => C2)(fn: (C1, C2) => T): T =
68       apply(c1) { c1 => apply(c2) { c2 => fn(c1, c2) } }
69   }
70 }