[vcs-log] minor: remove unused imports
[idea/community.git] / platform / vcs-log / impl / src / com / intellij / vcs / log / data / index / VcsLogMessagesTrigramIndex.java
1 /*
2  * Copyright 2000-2016 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.intellij.vcs.log.data.index;
17
18 import com.intellij.openapi.Disposable;
19 import com.intellij.openapi.util.text.TrigramBuilder;
20 import com.intellij.util.indexing.DataIndexer;
21 import com.intellij.util.indexing.ScalarIndexExtension;
22 import com.intellij.util.indexing.StorageException;
23 import com.intellij.util.indexing.ValueContainer;
24 import com.intellij.vcs.log.VcsFullCommitDetails;
25 import com.intellij.vcs.log.impl.FatalErrorHandler;
26 import gnu.trove.THashMap;
27 import org.jetbrains.annotations.NotNull;
28 import org.jetbrains.annotations.Nullable;
29
30 import java.io.IOException;
31 import java.util.Map;
32
33 import static com.intellij.vcs.log.data.index.VcsLogPersistentIndex.getVersion;
34
35 public class VcsLogMessagesTrigramIndex extends VcsLogFullDetailsIndex<Void> {
36   public static final String TRIGRAMS = "trigrams";
37
38   public VcsLogMessagesTrigramIndex(@NotNull String logId,
39                                     @NotNull FatalErrorHandler fatalErrorHandler,
40                                     @NotNull Disposable disposableParent) throws IOException {
41     super(logId, TRIGRAMS, getVersion(), new TrigramMessageIndexer(), ScalarIndexExtension.VOID_DATA_EXTERNALIZER,
42           fatalErrorHandler, disposableParent);
43   }
44
45   @Nullable
46   public ValueContainer.IntIterator getCommitsForSubstring(@NotNull String string) throws StorageException {
47     MyTrigramProcessor trigramProcessor = new MyTrigramProcessor();
48     TrigramBuilder.processTrigrams(string, trigramProcessor);
49
50     if (trigramProcessor.map.isEmpty()) return null;
51
52     return getCommitsWithAllKeys(trigramProcessor.map.keySet());
53   }
54
55   public static class TrigramMessageIndexer implements DataIndexer<Integer, Void, VcsFullCommitDetails> {
56     @NotNull
57     @Override
58     public Map<Integer, Void> map(@NotNull VcsFullCommitDetails inputData) {
59       MyTrigramProcessor trigramProcessor = new MyTrigramProcessor();
60       TrigramBuilder.processTrigrams(inputData.getFullMessage(), trigramProcessor);
61
62       return trigramProcessor.map;
63     }
64   }
65
66   private static class MyTrigramProcessor extends TrigramBuilder.TrigramProcessor {
67     Map<Integer, Void> map;
68
69     @Override
70     public boolean consumeTrigramsCount(int count) {
71       map = new THashMap<>(count);
72       return true;
73     }
74
75     @Override
76     public boolean execute(int value) {
77       map.put(value, null);
78       return true;
79     }
80   }
81 }