marton-bod commented on a change in pull request #2261: URL: https://github.com/apache/hive/pull/2261#discussion_r634500584
########## File path: iceberg/iceberg-handler/src/main/java/org/apache/iceberg/mr/Catalogs.java ########## @@ -284,4 +334,42 @@ private static String getCatalogType(Configuration conf, String catalogName) { } } } + + public static class TableCache { + private static final Cache<String, Table> tableCache = Caffeine.newBuilder() + .expireAfterAccess(12, TimeUnit.HOURS).build(); + private static final Logger LOG = LoggerFactory.getLogger(TableCache.class); + + public static void removeTables(Configuration conf) { + String queryId = conf.get(HiveConf.ConfVars.HIVEQUERYID.varname); + if (queryId != null && !queryId.isEmpty()) { + Set<String> queryKeys = tableCache.asMap().keySet().stream() + .filter(k -> k.startsWith(queryId)).collect(Collectors.toSet()); + tableCache.invalidateAll(queryKeys); + } else { + LOG.warn("Query id is not present in config, therefore no Iceberg table object is removed " + Review comment: I think what we do expect is that the `hive.query.id` is always populated in the config. If it's absent, that's a warning sign and something which could prevent us from evicting from the cache. This could be a useful log line in case we see that the cache gets overly large and we're looking for the root cause why the iceberg tables are not evicted. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: gitbox-unsubscr...@hive.apache.org For additional commands, e-mail: gitbox-h...@hive.apache.org