@@ -299,18 +299,18 @@ func filterUnusedStructs(enums []Enum, structs []Struct, queries []Query) ([]Enu
299
299
300
300
for _ , query := range queries {
301
301
if ! query .Arg .isEmpty () {
302
- keepTypes [strings . TrimPrefix ( query .Arg .Type (), "Null" )] = struct {}{}
302
+ keepTypes [query .Arg .Type ()] = struct {}{}
303
303
if query .Arg .IsStruct () {
304
304
for _ , field := range query .Arg .Struct .Fields {
305
- keepTypes [strings . TrimPrefix ( field .Type , "Null" ) ] = struct {}{}
305
+ keepTypes [field .Type ] = struct {}{}
306
306
}
307
307
}
308
308
}
309
309
if query .hasRetType () {
310
- keepTypes [strings . TrimPrefix ( query .Ret .Type (), "Null" )] = struct {}{}
310
+ keepTypes [query .Ret .Type ()] = struct {}{}
311
311
if query .Ret .IsStruct () {
312
312
for _ , field := range query .Ret .Struct .Fields {
313
- keepTypes [strings . TrimPrefix ( field .Type , "Null" ) ] = struct {}{}
313
+ keepTypes [field .Type ] = struct {}{}
314
314
}
315
315
}
316
316
}
@@ -321,6 +321,9 @@ func filterUnusedStructs(enums []Enum, structs []Struct, queries []Query) ([]Enu
321
321
if _ , ok := keepTypes [enum .Name ]; ok {
322
322
keepEnums = append (keepEnums , enum )
323
323
}
324
+ if _ , ok := keepTypes ["Null" + enum .Name ]; ok {
325
+ keepEnums = append (keepEnums , enum )
326
+ }
324
327
}
325
328
326
329
keepStructs := make ([]Struct , 0 , len (structs ))
0 commit comments