platform/platform-impl/src/com/intellij/openapi/fileTypes/impl/IgnoredFileCache.java

/*
 * Copyright 2000-2014 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.intellij.openapi.fileTypes.impl;

import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.openapi.vfs.VirtualFileManager;
import com.intellij.openapi.vfs.newvfs.BulkFileListener;
import com.intellij.openapi.vfs.newvfs.NewVirtualFile;
import com.intellij.openapi.vfs.newvfs.events.VFileEvent;
import com.intellij.openapi.vfs.newvfs.events.VFilePropertyChangeEvent;
import com.intellij.util.containers.ConcurrentBitSet;
import com.intellij.util.containers.ConcurrentIntObjectMap;
import com.intellij.util.containers.StripedLockIntObjectConcurrentHashMap;
import com.intellij.util.messages.MessageBusConnection;
import org.jetbrains.annotations.NotNull;

import java.util.List;

/**
 * @author peter
 */
class IgnoredFileCache {
  private final ConcurrentBitSet myCheckedIds = new ConcurrentBitSet();
  private final ConcurrentIntObjectMap<Object> myIgnoredIds = new StripedLockIntObjectConcurrentHashMap<Object>();
  private final IgnoredPatternSet myIgnoredPatterns;
  private volatile int myVfsEventNesting = 0;

  IgnoredFileCache(@NotNull IgnoredPatternSet ignoredPatterns) {
    myIgnoredPatterns = ignoredPatterns;
    MessageBusConnection connect = ApplicationManager.getApplication().getMessageBus().connect();
    connect.subscribe(VirtualFileManager.VFS_CHANGES, new BulkFileListener() {
      @Override
      public void before(@NotNull List<? extends VFileEvent> events) {
        // during VFS event processing the system may be in inconsistent state, don't cache it
        myVfsEventNesting++;
        clearCacheForChangedFiles(events);
      }

      @Override
      public void after(@NotNull List<? extends VFileEvent> events) {
        clearCacheForChangedFiles(events);
        myVfsEventNesting--;
      }

      private void clearCacheForChangedFiles(@NotNull List<? extends VFileEvent> events) {
        for (final VFileEvent event : events) {
          VirtualFile file = event.getFile();
          if (file instanceof NewVirtualFile && event instanceof VFilePropertyChangeEvent) {
            int id = ((NewVirtualFile)file).getId();
            if (id >= 0) {
              myCheckedIds.clear(id);
              myIgnoredIds.remove(id);
            }
          }
        }
      }
    });
  }

  void clearCache() {
    myCheckedIds.clear();
    myIgnoredIds.clear();
  }

  boolean isFileIgnored(@NotNull VirtualFile file) {
    if (myVfsEventNesting != 0 || !(file instanceof NewVirtualFile)) {
      return isFileIgnoredNoCache(file);
    }

    int id = ((NewVirtualFile)file).getId();
    if (id < 0) {
      return isFileIgnoredNoCache(file);
    }

    ConcurrentBitSet checkedIds = myCheckedIds;
    if (checkedIds.get(id)) {
      return myIgnoredIds.containsKey(id);
    }

    boolean result = isFileIgnoredNoCache(file);
    if (result) {
      myIgnoredIds.put(id, Boolean.TRUE);
    }
    else {
      myIgnoredIds.remove(id);
    }
    checkedIds.set(id);
    return result;
  }

  private boolean isFileIgnoredNoCache(@NotNull VirtualFile file) {
    return myIgnoredPatterns.isIgnored(file.getName());
  }
}