對java集合中的每個刪除操作和添加操作

通常我有一個擴展HashSet的類。插入到該集合中的每個對象都具有「所有者」，並且我想要分別計算屬於每個所有者的對象的數量。所以我寫了下面的代碼：

public class Viruses extends HashSet<Virus> { 
    private HashMap<RaceName, Integer> countsPerRace = new HashMap<RaceName, Integer>(); 

    @Override 
    public boolean add(Virus virus) { 
     if(super.add(virus)) { 
      RaceName race = virus.getOwner().getRace().getName(); 
      if(countsPerRace.containsKey(race)) { 
       countsPerRace.put(race, countsPerRace.get(race) + 1); 
      } else { 
       countsPerRace.put(race, 1); 
      } 
      return true; 
     } else { 
      return false; 
     } 
    } 

    @Override 
    public boolean remove(Object virus) { 
     if(super.remove(virus)) { 
      RaceName race = ((Virus)virus).getOwner().getRace().getName(); 
      if(countsPerRace.containsKey(race)) { 
       countsPerRace.put(race, countsPerRace.get(race) - 1); 
      } else { 
       throw new Exception("This should not happen..."); 
      } 
      return true; 
     } else { 
      return false; 
     } 
    } 

    /** 
    * Returns number of viruses of given race. 
    * @param raceId raceName of the viruses, which is equivalent of an owner id as there should never be two owners with the same race 
    * @return number of viruses of given race. 
    */ 
    public int getCount(RaceName raceId) { 
     return countsPerRace.containsKey(raceId) ? countsPerRace.get(raceId) : 0; 
    } 

    // I don't need these, so I thought the best idea will be just to throw an exception here. 
    @Override 
    public boolean removeAll(Collection<?> collection) { 
     throw new EngineRuntimeException("Unsupported operation!"); 
    } 

    @Override 
    public boolean addAll(Collection<? extends Virus> collection) { 
     throw new EngineRuntimeException("Unsupported operation!"); 
    } 
}

的問題是remove方法不叫，如果我使用迭代器刪除對象。每次在Java中添加或從Java集合中刪除對象時，是否都有執行操作的方法？如果不是，我確實需要覆蓋哪些方法或類才能確保我的集合保持一致，而不管以哪種方式刪除或添加內容？

來源

2014-10-29 Arsen

在你的情況下使用組合而不是繼承可能會更容易。 – assylias 2014-10-29 18:36:30

正如您所發現的那樣，不能保證迭代器使用公開的remove方法。

我強烈建議您在這種情況下考慮使用組合而不是繼承。

但是，如果你想與傳承解決方案，繼續前進，你必須做這樣的事情：

@Override 
public Iterator<Virus> iterator() { 
    final Iterator<Virus> delegate = super.iterator(); 
    return new Iterator<Virus>() { 
     @Override 
     public boolean hasNext() { 
      return delegate.hasNext(); 
     } 
     @Override 
     public void remove() { 
      // put your custom remove logic here 
      // ... 
      delegate.remove(); 
     } 
     @Override 
     public Virus next() { 
      return delegate.next(); 
     } 
    }; 
}

來源

2014-10-29 18:38:20 aioobe

感謝您的建議。我決定使用構圖並實現Collection。 – Arsen 2014-10-30 12:09:18

用的組合物。

創建一個類並實現迭代。該類將包裝一個哈希集，並將公開添加和刪除方法。您可以使用與可迭代移除相同的移除方法。

來源

2014-10-29 18:41:59 VGaur

對java集合中的每個刪除操作和添加操作

回答

相關問題