Anonymous func that checks value of arg type interface for nil

99designs · Jun 7, 2022 · 9a06da1 · 9a06da1
1 parent 65e6810
commit 9a06da1
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/codegen/field.go b/codegen/field.go
@@ -557,7 +557,20 @@ func (f *Field) CallArgs() string {
 	}
 
 	for _, arg := range f.Args {
-		args = append(args, "fc.Args["+strconv.Quote(arg.Name)+"].("+templates.CurrentImports.LookupType(arg.TypeReference.GO)+")")
+		tmp := "fc.Args[" + strconv.Quote(arg.Name) + "].(" + templates.CurrentImports.LookupType(arg.TypeReference.GO) + ")"
+
+		if types.IsInterface(arg.TypeReference.GO) {
+			tmp = fmt.Sprintf(`
+				func () interface{} {
+					if fc.Args["%s"] == nil {
+						return nil
+					}
+					return fc.Args["%s"].(interface{})
+				}()`, arg.Name, arg.Name,
+			)
+		}
+
+		args = append(args, tmp)
 	}
 
 	return strings.Join(args, ", ")