trying to get DT object

trinodb · Apr 10, 2024 · 59beb06 · 59beb06
1 parent 7b7d7b8
commit 59beb06
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/.../main/java/io/trino/plugin/hive/metastore/thrift/TokenFetchingMetastoreClientFactory.java b/.../main/java/io/trino/plugin/hive/metastore/thrift/TokenFetchingMetastoreClientFactory.java
@@ -21,6 +21,7 @@
 import dev.failsafe.Failsafe;
 import dev.failsafe.FailsafeExecutor;
 import dev.failsafe.RetryPolicy;
+import dev.failsafe.function.CheckedSupplier;
 import io.trino.cache.NonEvictableLoadingCache;
 import io.trino.plugin.base.security.UserNameProvider;
 import io.trino.plugin.hive.ForHiveMetastore;
@@ -30,6 +31,7 @@
 
 import java.time.Duration;
 import java.util.Optional;
+import java.util.concurrent.Callable;
 
 import static com.google.common.base.Throwables.throwIfInstanceOf;
 import static com.google.common.base.Throwables.throwIfUnchecked;
@@ -123,11 +125,13 @@ private DelegationToken loadDelegationToken(String username)
     {
         try {
             // added retry and stats for the thrift delegation token
-            return (DelegationToken) failSafeExecutor().get(() -> stats.getThriftDelegationToken().wrap(() -> {
+            return (DelegationToken) failSafeExecutor().get((CheckedSupplier <DelegationToken>)() ->
+                            stats.getThriftDelegationToken().wrap(() -> {
                 try (ThriftMetastoreClient client = createMetastoreClient()) {
                     return new DelegationToken(System.nanoTime(), client.getDelegationToken(username));
                 }
-            }));
+            }).call()
+            );
         }
         catch (Exception e) {
             if (e instanceof InterruptedException) {