Best way to remove element from Set about golang-set

[deckarep]

So, it's typically always a bad idea to iterate a data-structure and mutate it during iteration as in your first example. The reason is because it pretty much invalidates the iterator.

Your second example is actually the proper way to do it, especially when dealing with sets. You start off with your original set called: mySet which has some data. You create a different set that is empty. This represents the stuff "taken" out of the original set. But instead of actually taking anything out, you add what is to be removed into diffSet.

Then you just take the difference.

In fact the example works for me without hanging at all.

package main

import (
	"bufio"
	"fmt"
	"os"

	mapset "github.com/deckarep/golang-set"
)

func main() {
	mySet := mapset.NewSet()
	mySet.Add("a")
	mySet.Add("b")
	mySet.Add("c")

	diffSet := mapset.NewSet()
	//fill set with stuff
	myIt := mySet.Iterator()
	scanner := bufio.NewScanner(os.Stdin)
	for stuff := range myIt.C {
		fmt.Println("Found " + stuff.(string) + " remove it?[Y/n]")
		scanner.Scan()
		text := scanner.Text()
		if text == "y" || text == "Y" {
			fmt.Println("removing")
			diffSet.Add(stuff) //this will hang too
			fmt.Println("done")
		}
	}
	mySet = mySet.Difference(diffSet)
	fmt.Printf("result: %+v\n", mySet)
}

[deckarep]

Here is the output of the run above:

Found a remove it?[Y/n]
y
removing
done
Found b remove it?[Y/n]
n
Found c remove it?[Y/n]
n
result: Set{b, c}

[Nhoya]

Yea seems it's working, no idea why it hanged... well if that's the only way i will use it, thanks