Java: Ich habe ein Set von benutzerdefinierten Klassen. Wie kann man ein Element des Sets entfernen, modifizieren und wieder einfügen?

Question

Ich habe ein HashSet von benutzerdefinierten Objekten. Ich möchte ein Element aus dem Set entfernen, es modifizieren und es zurückstellen. Gibt es eine Möglichkeit, dies zu tun? Das Problem bei der Verwendung von Set.remove(Object o) ist, dass das von mir entfernte Element nicht zurückgegeben wird. Hier

ist ein Beispiel dafür, was ich versuche zu tun:

import java.util.HashSet; 
import java.util.Set; 
import org.apache.commons.lang3.builder.EqualsBuilder; 
import org.apache.commons.lang3.builder.HashCodeBuilder; 

public class Planet { 
    private String name; 
    private Set<String> moons = new HashSet<String>(); 

    public void addMoon(String name) { 
     moons.add(name); 
    } 

    @Override 
    public boolean equals(Object obj) 
    { 
     if (obj == null) return false; 
     if (obj == this) return true; 
     if (obj.getClass() != getClass()) return false; 
     Planet rhs = (Planet) obj; 
     return new EqualsBuilder() 
       .append(getName(), rhs.getName()) 
       .isEquals(); 
    } 

    @Override 
    public int hashCode() 
    { 
     return new HashCodeBuilder(17,31). 
       append(getName()) 
       .toHashCode(); 
    } 

    public String getName() { return name; } 

    public void setName(String name) { this.name = name; } 

    public Set<String> getMoons() { return moons; } 


} 

Set<Planet> solarSystem = new HashSet<Planet>(); 

Planet addThisPlanet = new Planet(); 
addThisPlanet.setName("Jupiter"); 
addThisPlanet.addMoon("Titan"); 

if (solarSystem.contains(addThisPlanet)) { 
    Planet currentPlanet = // element of solarSystem that equals(addThisPlanet) 
    // remove currentPlanet from solarSystem 
    for(String moon: addThisPlanet.getMoons()) { 
     currentPlanet.addMoon(moon); 
    } 
    solarSystem.add(currentPlanet); 
} else { 
    solarSystem.add(addThisPlanet); 
} 

Hier ist, was ich tue, ist das Hinzufügen von Monden eins nach dem anderen (wie sie „entdeckt“ werden), um bestehende (oder neu) Planeten. Wenn der Planet nicht bereits in der Menge ist, wird er mit seiner aktuellen Menge bekannter Monde hinzugefügt. Wenn der Planet bereits in der Menge ist, möchte ich ihn entfernen, die neu entdeckten Monde zu seinen bekannten Monden hinzufügen und dann den Planeten wieder zu der Menge hinzufügen.

Answer 1

Sie mit diesem unter Verwendung eines Map anstelle eines Set umgehen konnte - hinzufügen Planeten auf die Karte, wenn sie nicht bereits vorhanden sind; Füge Monde mit dem Planetennamen hinzu. Die Monde selbst bleiben ein Set, so dass Versuche, einen Mond hinzuzufügen, es nicht wiederholen.

ich etwas in dieser Richtung denke ...

public class Planet { 
    private final String name; 
    private final Set<String> moons = new HashSet<String>(); 

    // notice I set the name in the Constructor instead of a setter 
    public Planet(String name) { 
     this.name = name; 
    } 

    public void addMoon(String name) { 
     moons.add(name); 
    } 

    @Override 
    public boolean equals(Object obj) { ...same as yours... } 

    @Override 
    public int hashCode() { ...same as yours... } 

    public String getName() { return name; } 

    public Set<String> getMoons() { return moons; } 
} 


// ... elsewhere ... 

final Map<String,Planet> solarSystem = new HashMap<>(); 

public void addMoons(final String planet, final String... moons) 
{ 
    Planet p = solarSystem.get(planet); 
    if (p == null) { 
     p = new Planet(planet); 
     // add the Planet mapped by its name 
     solarSystem.put(p.getName(), p); 
    } 
    for (final String moon : moons) { 
     p.addMoon(moon); 
    } 
} 

Man könnte noch weiter gehen und machen das Solarsystem ein Objekt selbst, anstatt eine Karte oder ein Set.

Answer 2

Das Objekt, das Sie aus dem Set zurückholen möchten, ist das Objekt, das Sie bereits haben, vorausgesetzt, Sie haben es nicht geändert, seit es hinzugefügt wurde. Die Methode contains von Set verwendet die Methode equals des Objekts, um festzustellen, ob sie vorhanden ist.

Siehe API doco für contains: https://docs.oracle.com/javase/7/docs/api/java/util/Set.html#contains(java.lang.Object)

Answer 3

Sie brauchen nicht und es zu entfernen, um Planeten-Set zurückstellen. Sie können es einfach tun, wie diese

if(solarSystem.contains(addThisPlanet)) { 
    for(Planet currentPlanet : solarSystem){ 
    if(currentPlanet.equals(addThisPlanet)) 
{ 
for(String moon: addThisPlanet.getMoons()) 
{ currentPlanet .addMoon(moon); } 

    break; 
    } 
    }else { solarSystem.add(addThisPlanet); }