chore(scripts/dbgen): only add arg validation for dbfake (#8578)

mafredri · web-flow · commit 2fae9b0a6946 · 2023-07-18T17:26:03.000Z
diff --git a/scripts/dbgen/main.go b/scripts/dbgen/main.go
@@ -220,7 +220,9 @@ func orderAndStubDatabaseFunctions(filePath, receiver, structName string, stub f
 
 	for _, fn := range funcs {
 		var bodyStmts []dst.Stmt
-		if len(fn.Func.Params.List) == 2 && fn.Func.Params.List[1].Names[0].Name == "arg" {
+
+		// Add input validation, only relevant for dbfake.
+		if strings.Contains(filePath, "dbfake") && len(fn.Func.Params.List) == 2 && fn.Func.Params.List[1].Names[0].Name == "arg" {
 			/*
 				err := validateDatabaseType(arg)
 				if err != nil {
