diff --git a/private/dbGen/generator-functions.go b/private/dbGen/generator-functions.go index f71a278..5c5def7 100755 --- a/private/dbGen/generator-functions.go +++ b/private/dbGen/generator-functions.go @@ -2,6 +2,7 @@ package dbGen import ( "github.com/keenmate/db-gen/private/helpers" + "strings" "text/template" ) @@ -19,5 +20,8 @@ func getTemplateFunctions() template.FuncMap { "snakeCased": func(s string) string { return helpers.ToSnakeCase(s) }, + "trimPrefix": func(p string, s string) string { + return strings.TrimPrefix(s, p) + }, } }