[fix][sdk] add utilService to Services

wt0530 · Jul 8, 2024 · 058ef42 · 058ef42
1 parent be6e1c8
commit 058ef42
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/java/dingo-sdk/src/main/java/io/dingodb/sdk/service/Services.java b/java/dingo-sdk/src/main/java/io/dingodb/sdk/service/Services.java
@@ -93,6 +93,18 @@ public CoordinatorChannelProvider load(Set<Location> key) {
             }
         });
 
+    private static final LoadingCache<Set<Location>, CoordinatorChannelProvider> utilCache = CacheBuilder
+        .newBuilder()
+        .expireAfterAccess(60, TimeUnit.MINUTES)
+        .expireAfterWrite(60, TimeUnit.MINUTES)
+        .maximumSize(8)
+        .build(new CacheLoader<Set<Location>, CoordinatorChannelProvider>() {
+           @Override
+           public CoordinatorChannelProvider load(Set<Location> key) throws Exception {
+               return coordinatorServiceChannelProvider(key);
+           }
+        });
+
     private static final LoadingCache<Set<Location>, LoadingCache<Long, RegionChannelProvider>> regionCache =
         CacheBuilder.newBuilder()
             .expireAfterAccess(30, TimeUnit.MINUTES)
@@ -204,6 +216,14 @@ public static VersionService versionService(Set<Location> locations) {
         ).getService();
     }
 
+    @SneakyThrows
+    public static UtilService utilService(Set<Location> locations) {
+        Parameters.notEmpty(locations, "locations");
+        return new ServiceCaller<>(
+            utilCache.get(locations),  DEFAULT_RETRY_TIMES, DEFAULT, UtilService.Impl::new
+        ).getService();
+    }
+
     @SneakyThrows
     public static ChannelProvider regionChannelProvider(
         Set<Location> locations, DingoCommonId tableId, DingoCommonId regionId