Replace custom ConcurrentHashMap cache with Caffeine via CacheManager

Co-authored-by: nixel2007 <[email protected]>

Author: Copilot

src/main/java/com/github/_1c_syntax/bsl/languageserver/providers/SemanticTokensProvider.java

@@ -26,13 +26,16 @@
 import com.github._1c_syntax.bsl.languageserver.context.events.ServerContextDocumentRemovedEvent;
 import com.github._1c_syntax.bsl.languageserver.semantictokens.SemanticTokenEntry;
 import com.github._1c_syntax.bsl.languageserver.semantictokens.SemanticTokensSupplier;
+import com.github.benmanes.caffeine.cache.Cache;
 import lombok.RequiredArgsConstructor;
 import org.eclipse.lsp4j.SemanticTokens;
 import org.eclipse.lsp4j.SemanticTokensDelta;
 import org.eclipse.lsp4j.SemanticTokensDeltaParams;
 import org.eclipse.lsp4j.SemanticTokensEdit;
 import org.eclipse.lsp4j.SemanticTokensParams;
 import org.eclipse.lsp4j.jsonrpc.messages.Either;
+import org.springframework.cache.CacheManager;
+import org.springframework.cache.annotation.CacheConfig;
 import org.springframework.context.event.EventListener;
 import org.springframework.stereotype.Component;
 
@@ -42,10 +45,8 @@
 import java.util.Comparator;
 import java.util.HashSet;
 import java.util.List;
-import java.util.Map;
 import java.util.Set;
 import java.util.UUID;
-import java.util.concurrent.ConcurrentHashMap;
 
 /**
  * Провайдер для предоставления семантических токенов.
@@ -56,15 +57,16 @@
  */
 @Component
 @RequiredArgsConstructor
+@CacheConfig(cacheNames = SemanticTokensProvider.CACHE_NAME)
 public class SemanticTokensProvider {
 
-  private final List<SemanticTokensSupplier> suppliers;
-
   /**
-   * Cache for storing previous token data by resultId.
-   * Key: resultId, Value: token data list
+   * Имя кэша для хранения данных семантических токенов.
    */
-  private final Map<String, CachedTokenData> tokenCache = new ConcurrentHashMap<>();
+  public static final String CACHE_NAME = "semanticTokens";
+
+  private final List<SemanticTokensSupplier> suppliers;
+  private final CacheManager cacheManager;
 
   /**
    * Cached semantic token data associated with a document.
@@ -114,7 +116,7 @@ public Either<SemanticTokens, SemanticTokensDelta> getSemanticTokensFullDelta(
     SemanticTokensDeltaParams params
   ) {
     String previousResultId = params.getPreviousResultId();
-    CachedTokenData previousData = tokenCache.get(previousResultId);
+    CachedTokenData previousData = getCachedTokenData(previousResultId);
 
     // Calculate current tokens
     List<SemanticTokenEntry> entries = suppliers.stream()
@@ -140,7 +142,7 @@ public Either<SemanticTokens, SemanticTokensDelta> getSemanticTokensFullDelta(
     cacheTokenData(resultId, documentContext.getUri(), currentData);
 
     // Remove the old cached data
-    tokenCache.remove(previousResultId);
+    evictFromCache(previousResultId);
 
     var delta = new SemanticTokensDelta();
     delta.setResultId(resultId);
@@ -178,21 +180,71 @@ public void handleDocumentRemoved(ServerContextDocumentRemovedEvent event) {
    * @param uri URI документа, для которого нужно очистить кэш
    */
   protected void clearCache(URI uri) {
-    tokenCache.entrySet().removeIf(entry -> entry.getValue().uri().equals(uri));
+    var cache = getNativeCache();
+    if (cache != null) {
+      cache.asMap().entrySet().removeIf(entry -> entry.getValue().uri().equals(uri));
+    }
   }
 
   /**
-   * Generate a unique result ID for caching.
+   * Получить нативный Caffeine кэш для семантических токенов.
+   *
+   * @return нативный Caffeine кэш или null, если кэш не найден
    */
-  private static String generateResultId() {
-    return UUID.randomUUID().toString();
+  @SuppressWarnings("unchecked")
+  private Cache<String, CachedTokenData> getNativeCache() {
+    var springCache = cacheManager.getCache(CACHE_NAME);
+    if (springCache != null) {
+      return (Cache<String, CachedTokenData>) springCache.getNativeCache();
+    }
+    return null;
   }
 
   /**
-   * Cache token data with the given resultId.
+   * Получить кэшированные данные токенов по resultId.
+   *
+   * @param resultId идентификатор результата
+   * @return кэшированные данные или null, если не найдены
+   */
+  private CachedTokenData getCachedTokenData(String resultId) {
+    var cache = getNativeCache();
+    if (cache != null) {
+      return cache.getIfPresent(resultId);
+    }
+    return null;
+  }
+
+  /**
+   * Сохранить данные токенов в кэш.
+   *
+   * @param resultId идентификатор результата
+   * @param uri      URI документа
+   * @param data     данные токенов
    */
   private void cacheTokenData(String resultId, URI uri, List<Integer> data) {
-    tokenCache.put(resultId, new CachedTokenData(uri, data));
+    var cache = getNativeCache();
+    if (cache != null) {
+      cache.put(resultId, new CachedTokenData(uri, data));
+    }
+  }
+
+  /**
+   * Удалить данные из кэша по resultId.
+   *
+   * @param resultId идентификатор результата для удаления
+   */
+  private void evictFromCache(String resultId) {
+    var cache = getNativeCache();
+    if (cache != null) {
+      cache.invalidate(resultId);
+    }
+  }
+
+  /**
+   * Generate a unique result ID for caching.
+   */
+  private static String generateResultId() {
+    return UUID.randomUUID().toString();
   }
 
   /**