platform-impl/src/com/intellij/openapi/components/impl/stores/CompoundSaveSession.java

   1 package com.intellij.openapi.components.impl.stores;
   2
   3 import com.intellij.openapi.components.StateStorage;
   4 import com.intellij.util.io.fs.IFile;
   5
   6 import java.util.*;
   7
   8 /**
   9  * @author mike
  10  */
  11 public class CompoundSaveSession {
  12   private final Map<StateStorage, StateStorage.SaveSession> mySaveSessions = new HashMap<StateStorage, StateStorage.SaveSession>();
  13
  14   public CompoundSaveSession(final CompoundExternalizationSession compoundExternalizationSession) {
  15     final Collection<StateStorage> stateStorages = compoundExternalizationSession.getStateStorages();
  16
  17     for (StateStorage stateStorage : stateStorages) {
  18       mySaveSessions.put(stateStorage, stateStorage.startSave(compoundExternalizationSession.getExternalizationSession(stateStorage)));
  19     }
  20   }
  21
  22   public List<IFile> getAllStorageFilesToSave() throws StateStorage.StateStorageException {
  23     List<IFile> result = new ArrayList<IFile>();
  24
  25     for (StateStorage stateStorage : mySaveSessions.keySet()) {
  26       final StateStorage.SaveSession saveSession = mySaveSessions.get(stateStorage);
  27
  28       result.addAll(saveSession.getStorageFilesToSave());
  29     }
  30
  31     return result;
  32   }
  33
  34   public void save() throws StateStorage.StateStorageException {
  35     for (StateStorage.SaveSession saveSession : mySaveSessions.values()) {
  36       saveSession.save();
  37     }
  38   }
  39
  40   public Set<String> getUsedMacros() {
  41     Set<String> usedMacros = new HashSet<String>();
  42
  43     for (StateStorage.SaveSession saveSession : mySaveSessions.values()) {
  44       usedMacros.addAll(saveSession.getUsedMacros());
  45     }
  46
  47     return usedMacros;
  48   }
  49
  50   public void finishSave() {
  51     RuntimeException re = null;
  52     for (StateStorage stateStorage : mySaveSessions.keySet()) {
  53       final StateStorage.SaveSession saveSession = mySaveSessions.get(stateStorage);
  54       try {
  55         stateStorage.finishSave(saveSession);
  56       } catch(RuntimeException t) {
  57         re = t;
  58       }
  59     }
  60
  61     if (re != null) {
  62       throw re;
  63     }
  64   }
  65
  66   public StateStorage.SaveSession getSaveSession(final StateStorage storage) {
  67     return mySaveSessions.get(storage);
  68   }
  69
  70   public List<IFile> getAllStorageFiles() {
  71     List<IFile> result = new ArrayList<IFile>();
  72     for (StateStorage.SaveSession saveSession : mySaveSessions.values()) {
  73       result.addAll(saveSession.getAllStorageFiles());
  74     }
  75
  76     return result;
  77   }
  78
  79   public Set<StateStorage> getStateStorages() {
  80     return mySaveSessions.keySet();
  81   }
  82 }