From b3635b37f2bd0f6d62082ceca83bc2a4725da1d0 Mon Sep 17 00:00:00 2001 From: Mathias Fredriksson Date: Tue, 18 Jul 2023 15:21:10 +0000 Subject: [PATCH] chore(scripts/dbgen): only add arg validation for dbfake --- scripts/dbgen/main.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scripts/dbgen/main.go b/scripts/dbgen/main.go index 0eeda09ba966d..8a7976e32154b 100644 --- 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 {