| /* |
| * 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 org.jetbrains.jps.incremental.storage; |
| |
| import com.intellij.openapi.util.io.FileUtil; |
| import com.intellij.util.io.DataExternalizer; |
| import com.intellij.util.io.IOUtil; |
| import gnu.trove.THashSet; |
| import org.jetbrains.annotations.NotNull; |
| import org.jetbrains.annotations.Nullable; |
| |
| import java.io.*; |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.Collections; |
| import java.util.Set; |
| |
| /** |
| * @author Eugene Zhuravlev |
| * Date: 10/11/12 |
| */ |
| public class OneToManyPathsMapping extends AbstractStateStorage<String, Collection<String>> { |
| public OneToManyPathsMapping(File storePath) throws IOException { |
| super(storePath, new PathStringDescriptor(), new PathCollectionExternalizer()); |
| } |
| |
| @Override |
| public void update(@NotNull String keyPath, @NotNull Collection<String> boundPaths) throws IOException { |
| super.update(FileUtil.toSystemIndependentName(keyPath), normalizePaths(boundPaths)); |
| } |
| |
| public final void update(@NotNull String keyPath, @NotNull String boundPath) throws IOException { |
| super.update(FileUtil.toSystemIndependentName(keyPath), Collections.singleton(FileUtil.toSystemIndependentName(boundPath))); |
| } |
| |
| public final void appendData(@NotNull String keyPath, @NotNull String boundPath) throws IOException { |
| super.appendData(FileUtil.toSystemIndependentName(keyPath), Collections.singleton(FileUtil.toSystemIndependentName(boundPath))); |
| } |
| |
| @Override |
| public void appendData(@NotNull String keyPath, @NotNull Collection<String> boundPaths) throws IOException { |
| super.appendData(FileUtil.toSystemIndependentName(keyPath), normalizePaths(boundPaths)); |
| } |
| |
| @Nullable |
| @Override |
| public Collection<String> getState(@NotNull String keyPath) throws IOException { |
| return super.getState(FileUtil.toSystemIndependentName(keyPath)); |
| } |
| |
| @Override |
| public void remove(@NotNull String keyPath) throws IOException { |
| super.remove(FileUtil.toSystemIndependentName(keyPath)); |
| } |
| |
| public final void removeData(@NotNull String keyPath, @NotNull String boundPath) throws IOException { |
| final Collection<String> outputPaths = getState(FileUtil.toSystemIndependentName(keyPath)); |
| if (outputPaths != null) { |
| final boolean removed = outputPaths.remove(FileUtil.toSystemIndependentName(boundPath)); |
| if (outputPaths.isEmpty()) { |
| remove(keyPath); |
| } |
| else { |
| if (removed) { |
| update(keyPath, outputPaths); |
| } |
| } |
| } |
| } |
| |
| private static class PathCollectionExternalizer implements DataExternalizer<Collection<String>> { |
| public void save(@NotNull DataOutput out, Collection<String> value) throws IOException { |
| for (String str : value) { |
| IOUtil.writeUTF(out, str); |
| } |
| } |
| |
| public Collection<String> read(@NotNull DataInput in) throws IOException { |
| final Set<String> result = new THashSet<String>(FileUtil.PATH_HASHING_STRATEGY); |
| final DataInputStream stream = (DataInputStream)in; |
| while (stream.available() > 0) { |
| final String str = IOUtil.readUTF(stream); |
| result.add(str); |
| } |
| return result; |
| } |
| } |
| |
| private static Collection<String> normalizePaths(Collection<String> outputs) { |
| Collection<String> normalized = new ArrayList<String>(outputs.size()); |
| for (String out : outputs) { |
| normalized.add(FileUtil.toSystemIndependentName(out)); |
| } |
| return normalized; |
| } |
| |
| } |