Add stringset and fix bind example.

examples/function/bind.go

@@ -1,7 +1,10 @@
 package function
 
-func InsertFunc(m MultiSet) func(val string) {
+import "github.com/jaeyeom/gogo/generate/stringset"
+
+func InsertFunc(m stringset.StringSet) func(val string) {
 	return func(val string) {
-		Insert(m, val)
+		m.Insert(val)
 	}
+	// Or it could be simply: return m.Insert
 }

examples/function/bind_test.go

@@ -1,6 +1,24 @@
 package function
 
-func ExampleInsertFunc() {
-	m := NewMultiSet()
+import (
+	"bytes"
+	"reflect"
+	"testing"
+
+	"github.com/jaeyeom/gogo/generate/stringset"
+)
+
+func TestInsertFunc(t *testing.T) {
+	m := stringset.NewStringSet()
+	r := bytes.NewBufferString("a\nb\nc\nc\n")
 	ReadFrom(r, InsertFunc(m))
+	expected := stringset.StringSet(map[string]int{
+		"a": 1,
+		"b": 1,
+		"c": 2,
+	})
+	if !reflect.DeepEqual(expected, m) {
+		t.Errorf("%d", len(m))
+		t.Errorf("%v != %v", expected, m)
+	}
 }

generate/intset/intset.go

@@ -1,5 +1,5 @@
 // Generated by multisetgen. DO NOT EDIT!
-package main
+package intset
 
 import "fmt"
 

generate/intset/intset_main.go → generate/intset/intset_test.go

@@ -1,10 +1,10 @@
-package main
+package intset
 
 import "fmt"
 
-//go:generate multisetgen -multiset_typename=IntSet -element_typename=int
+//go:generate multisetgen -multiset_typename=IntSet -element_typename=int -package_name=intset
 
-func main() {
+func Example() {
 	m := NewIntSet()
 	fmt.Println(m.String())
 	fmt.Println(m.Count(3))

generate/stringset/stringset.go

@@ -0,0 +1,38 @@
+// Generated by multisetgen. DO NOT EDIT!
+package stringset
+
+import "fmt"
+
+type StringSet map[string]int
+
+func NewStringSet() StringSet {
+	return StringSet{}
+}
+
+func (m StringSet) Insert(val string) {
+	m[val]++
+}
+
+func (m StringSet) Erase(val string) {
+	if _, exists := m[val]; !exists {
+		return
+	}
+	m[val]--
+	if m[val] <= 0 {
+		delete(m, val)
+	}
+}
+
+func (m StringSet) Count(val string) int {
+	return m[val]
+}
+
+func (m StringSet) String() string {
+	vals := ""
+	for val, count := range m {
+		for i := 0; i < count; i++ {
+			vals += fmt.Sprint(val) + " "
+		}
+	}
+	return "{ " + vals + "}"
+}

generate/stringset/stringset_test.go

@@ -0,0 +1,27 @@
+package stringset
+
+import "fmt"
+
+//go:generate multisetgen -multiset_typename=StringSet -element_typename=string -package_name=stringset
+
+func Example() {
+	m := NewStringSet()
+	fmt.Println(m.String())
+	fmt.Println(m.Count("3"))
+	m.Insert("3")
+	m.Insert("3")
+	m.Insert("3")
+	m.Insert("3")
+	fmt.Println(m.String())
+	fmt.Println(m.Count("3"))
+	m.Insert("1")
+	m.Insert("2")
+	m.Insert("5")
+	m.Insert("7")
+	m.Erase("3")
+	m.Erase("5")
+	fmt.Println(m.Count("3"))
+	fmt.Println(m.Count("1"))
+	fmt.Println(m.Count("2"))
+	fmt.Println(m.Count("5"))
+}