X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/3dbbbd49b1889d832c295a4bb69dd360abf614fa..91ef36eeea827569ec3745696e233aad0f8063aa:/lib/controller/federation/generate.go

diff --git a/lib/controller/federation/generate.go b/lib/controller/federation/generate.go
index 1f37df858c..ab5d9966a4 100644
--- a/lib/controller/federation/generate.go
+++ b/lib/controller/federation/generate.go
@@ -31,7 +31,7 @@ func main() {
 	if err != nil {
 		panic(err)
 	}
-	orig := regexp.MustCompile(`(?ms)\nfunc [^\n]*CollectionList\(.*?\n}\n`).Find(buf)
+	orig := regexp.MustCompile(`(?ms)\nfunc [^\n]*generated_CollectionList\(.*?\n}\n`).Find(buf)
 	if len(orig) == 0 {
 		panic("can't find CollectionList func")
 	}
@@ -49,18 +49,15 @@ func main() {
 		panic(err)
 	}
 	go func() {
+		defer out.Close()
 		out.Write(regexp.MustCompile(`(?ms)^.*package .*?import.*?\n\)\n`).Find(buf))
 		io.WriteString(out, "//\n// -- this file is auto-generated -- do not edit -- edit list.go and run \"go generate\" instead --\n//\n\n")
-		for _, t := range []string{"Container", "Specimen"} {
+		for _, t := range []string{"Container", "Specimen", "User"} {
 			_, err := out.Write(bytes.ReplaceAll(orig, []byte("Collection"), []byte(t)))
 			if err != nil {
 				panic(err)
 			}
 		}
-		err = out.Close()
-		if err != nil {
-			panic(err)
-		}
 	}()
 	err = gofmt.Run()
 	if err != nil {