How to use allocateIdentifier method of main Package

...346	if retString != "" {347		retString = " " + retString348	}349	ia := newIdentifierAllocator(argNames)350	idRecv := ia.allocateIdentifier("m")351	g.p("// %v mocks base method", m.Name)352	g.p("func (%v *%v) %v(%v)%v {", idRecv, mockType, m.Name, argString, retString)353	g.in()354	var callArgs string355	if m.Variadic == nil {356		if len(argNames) > 0 {357			callArgs = ", " + strings.Join(argNames, ", ")358		}359	} else {360		// Non-trivial. The generated code must build a []interface{},361		// but the variadic argument may be any type.362		idVarArgs := ia.allocateIdentifier("varargs")363		idVArg := ia.allocateIdentifier("a")364		g.p("%s := []interface{}{%s}", idVarArgs, strings.Join(argNames[:len(argNames)-1], ", "))365		g.p("for _, %s := range %s {", idVArg, argNames[len(argNames)-1])366		g.in()367		g.p("%s = append(%s, %s)", idVarArgs, idVarArgs, idVArg)368		g.out()369		g.p("}")370		callArgs = ", " + idVarArgs + "..."371	}372	if len(m.Out) == 0 {373		g.p(`%v.ctrl.Call(%v, %q%v)`, idRecv, idRecv, m.Name, callArgs)374	} else {375		idRet := ia.allocateIdentifier("ret")376		g.p(`%v := %v.ctrl.Call(%v, %q%v)`, idRet, idRecv, idRecv, m.Name, callArgs)377		// Go does not allow "naked" type assertions on nil values, so we use the two-value form here.378		// The value of that is either (x.(T), true) or (Z, false), where Z is the zero value for T.379		// Happily, this coincides with the semantics we want here.380		retNames := make([]string, len(rets))381		for i, t := range rets {382			retNames[i] = ia.allocateIdentifier(fmt.Sprintf("ret%d", i))383			g.p("%s, _ := %s[%d].(%s)", retNames[i], idRet, i, t)384		}385		g.p("return " + strings.Join(retNames, ", "))386	}387	g.out()388	g.p("}")389	return nil390}391func (g *generator) GenerateMockRecorderMethod(mockType string, m *model.Method) error {392	argNames := g.getArgNames(m)393	var argString string394	if m.Variadic == nil {395		argString = strings.Join(argNames, ", ")396	} else {397		argString = strings.Join(argNames[:len(argNames)-1], ", ")398	}399	if argString != "" {400		argString += " interface{}"401	}402	if m.Variadic != nil {403		if argString != "" {404			argString += ", "405		}406		argString += fmt.Sprintf("%s ...interface{}", argNames[len(argNames)-1])407	}408	ia := newIdentifierAllocator(argNames)409	idRecv := ia.allocateIdentifier("mr")410	g.p("// %v indicates an expected call of %v", m.Name, m.Name)411	g.p("func (%s *%vMockRecorder) %v(%v) *gomock.Call {", idRecv, mockType, m.Name, argString)412	g.in()413	var callArgs string414	if m.Variadic == nil {415		if len(argNames) > 0 {416			callArgs = ", " + strings.Join(argNames, ", ")417		}418	} else {419		if len(argNames) == 1 {420			// Easy: just use ... to push the arguments through.421			callArgs = ", " + argNames[0] + "..."422		} else {423			// Hard: create a temporary slice.424			idVarArgs := ia.allocateIdentifier("varargs")425			g.p("%s := append([]interface{}{%s}, %s...)",426				idVarArgs,427				strings.Join(argNames[:len(argNames)-1], ", "),428				argNames[len(argNames)-1])429			callArgs = ", " + idVarArgs + "..."430		}431	}432	g.p(`return %s.mock.ctrl.RecordCallWithMethodType(%s.mock, "%s", reflect.TypeOf((*%s)(nil).%s)%s)`, idRecv, idRecv, m.Name, mockType, m.Name, callArgs)433	g.out()434	g.p("}")435	return nil436}437func (g *generator) getArgNames(m *model.Method) []string {438	argNames := make([]string, len(m.In))439	for i, p := range m.In {440		name := p.Name441		if name == "" {442			name = fmt.Sprintf("arg%d", i)443		}444		argNames[i] = name445	}446	if m.Variadic != nil {447		name := m.Variadic.Name448		if name == "" {449			name = fmt.Sprintf("arg%d", len(m.In))450		}451		argNames = append(argNames, name)452	}453	return argNames454}455func (g *generator) getArgTypes(m *model.Method, pkgOverride string) []string {456	argTypes := make([]string, len(m.In))457	for i, p := range m.In {458		argTypes[i] = p.Type.String(g.packageMap, pkgOverride)459	}460	if m.Variadic != nil {461		argTypes = append(argTypes, "..."+m.Variadic.Type.String(g.packageMap, pkgOverride))462	}463	return argTypes464}465type identifierAllocator map[string]struct{}466func newIdentifierAllocator(taken []string) identifierAllocator {467	a := make(identifierAllocator, len(taken))468	for _, s := range taken {469		a[s] = struct{}{}470	}471	return a472}473func (o identifierAllocator) allocateIdentifier(want string) string {474	id := want475	for i := 2; ; i++ {476		if _, ok := o[id]; !ok {477			o[id] = struct{}{}478			return id479		}480		id = want + "_" + strconv.Itoa(i)481	}482}483// Output returns the generator's output, formatted in the standard Go style.484func (g *generator) Output() []byte {485	src, err := format.Source(g.buf.Bytes())486	if err != nil {487		log.Fatalf("Failed to format generated source code: %s\n%s", err, g.buf.String())...

...129		}130	}131	return true132}133func TestIdentifierAllocator_allocateIdentifier(t *testing.T) {134	a := newIdentifierAllocator([]string{"taken"})135	t2 := a.allocateIdentifier("taken_2")136	if t2 != "taken_2" {137		t.Fatalf("expected 'taken_2', got %q", t2)138	}139	expected := []string{"taken", "taken_2"}140	if !allocatorContainsIdentifiers(a, expected) {141		t.Fatalf("allocator doesn't contain the expected items - allocator: %#v, expected items: %#v", a, expected)142	}143	t3 := a.allocateIdentifier("taken")144	if t3 != "taken_3" {145		t.Fatalf("expected 'taken_3', got %q", t3)146	}147	expected = []string{"taken", "taken_2", "taken_3"}148	if !allocatorContainsIdentifiers(a, expected) {149		t.Fatalf("allocator doesn't contain the expected items - allocator: %#v, expected items: %#v", a, expected)150	}151	t4 := a.allocateIdentifier("taken")152	if t4 != "taken_4" {153		t.Fatalf("expected 'taken_4', got %q", t4)154	}155	expected = []string{"taken", "taken_2", "taken_3", "taken_4"}156	if !allocatorContainsIdentifiers(a, expected) {157		t.Fatalf("allocator doesn't contain the expected items - allocator: %#v, expected items: %#v", a, expected)158	}159	id := a.allocateIdentifier("id")160	if id != "id" {161		t.Fatalf("expected 'id', got %q", id)162	}163	expected = []string{"taken", "taken_2", "taken_3", "taken_4", "id"}164	if !allocatorContainsIdentifiers(a, expected) {165		t.Fatalf("allocator doesn't contain the expected items - allocator: %#v, expected items: %#v", a, expected)166	}167}...

1import (2func main() {3    fmt.Println(allocateIdentifier())4}5import (6func main() {7    fmt.Println(allocateIdentifier())8}9import (10func main() {11    fmt.Println(allocateIdentifier())12}13import (14func main() {15    fmt.Println(allocateIdentifier())16}17import (18func main() {19    fmt.Println(allocateIdentifier())20}21import (22func main() {23    fmt.Println(allocateIdentifier())24}25import (26func main() {27    fmt.Println(allocateIdentifier())28}29import (30func main() {31    fmt.Println(allocateIdentifier())32}33import (34func main() {35    fmt.Println(allocateIdentifier())36}37import (38func main() {39    fmt.Println(allocateIdentifier())40}41import (42func main() {43    fmt.Println(allocateIdentifier())44}45import (46func main() {47    fmt.Println(allocateIdentifier())48}49import (50func main() {51    fmt.Println(

1import "fmt"2func main() {3    fmt.Println(allocateIdentifier("A"))4    fmt.Println(allocateIdentifier("A"))5    fmt.Println(allocateIdentifier("B"))6    fmt.Println(allocateIdentifier("B"))7    fmt.Println(allocateIdentifier("A"))8    fmt.Println(allocateIdentifier("A"))9    fmt.Println(allocateIdentifier("B"))10    fmt.Println(allocateIdentifier("B"))11}

1import (2type Person struct {3}4func (p *Person) allocateIdentifier() {5}6func main() {7    p := Person{Name: "John"}8    p.allocateIdentifier()9    fmt.Println(p.Id)10}11import (12type Person struct {13}14func (p *Person) allocateIdentifier() {15}16func main() {17    p := Person{Name: "John"}18    p.allocateIdentifier()19    fmt.Println(p.Id)20}21import (22type Person struct {23}24func (p Person) allocateIdentifier() {

1func main() {2    obj.allocateIdentifier()3}4func (m main) allocateIdentifier() {5    obj1.allocateIdentifier()6}7func (m main) allocateIdentifier() {8    obj1.allocateIdentifier()9}10func main() {11    obj.allocateIdentifier()12}13func main() {14    obj.allocateIdentifier()15}

1import (2func main() {3	fmt.Println(mylib.AllocateIdentifier())4}5import (6func AllocateIdentifier() string {7	rand.Seed(time.Now().UnixNano())8	return fmt.Sprintf("%d", rand.Int())9}10import (11func TestAllocateIdentifier(t *testing.T) {12}13import (14func BenchmarkAllocateIdentifier(b *testing.B) {15}16--- PASS: TestAllocateIdentifier (0.00s)17func TestXxx(*testing.T)18func TestXxx(*testing.T)

1import "fmt"2func main() {3    id = allocateIdentifier()4    fmt.Println(id)5}6func allocateIdentifier() int {7}8import "fmt"9func main() {10    id = allocateIdentifier()11    fmt.Println(id)12}13func allocateIdentifier() int {14}15func main() {16    id = allocateIdentifier()17    fmt.Println(id)18}19import "fmt"20func main() {21    id = allocateIdentifier()22    fmt.Println(id)23}24func allocateIdentifier() int {25}26func allocateIdentifier() int {27}