summaryrefslogtreecommitdiffhomepage
path: root/tools/go_generics/generics.go
diff options
context:
space:
mode:
Diffstat (limited to 'tools/go_generics/generics.go')
-rw-r--r--tools/go_generics/generics.go8
1 files changed, 6 insertions, 2 deletions
diff --git a/tools/go_generics/generics.go b/tools/go_generics/generics.go
index ca414d8cb..22c714c13 100644
--- a/tools/go_generics/generics.go
+++ b/tools/go_generics/generics.go
@@ -98,7 +98,7 @@ import (
"regexp"
"strings"
- "gvisor.googlesource.com/gvisor/tools/go_generics/globals"
+ "gvisor.dev/gvisor/tools/go_generics/globals"
)
var (
@@ -222,7 +222,11 @@ func main() {
// Modify the state tag appropriately.
if m := stateTagRegexp.FindStringSubmatch(ident.Name); m != nil {
if t := identifierRegexp.FindStringSubmatch(m[2]); t != nil {
- ident.Name = m[1] + `state:".(` + t[1] + *prefix + t[2] + *suffix + t[3] + `)"` + m[3]
+ typeName := *prefix + t[2] + *suffix
+ if n, ok := types[t[2]]; ok {
+ typeName = n
+ }
+ ident.Name = m[1] + `state:".(` + t[1] + typeName + t[3] + `)"` + m[3]
}
}
}