From b0eb20f0f7a8665fc62f38ec8996c368a7de463f Mon Sep 17 00:00:00 2001 From: Weilence Date: Wed, 9 Nov 2022 00:32:40 +0800 Subject: [PATCH] entc/gen: enable `BuildFlags` in *.go (#3075) Co-authored-by: Weilence --- entc/load/load.go | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/entc/load/load.go b/entc/load/load.go index 0261189b0..d0fac7c5c 100644 --- a/entc/load/load.go +++ b/entc/load/load.go @@ -87,7 +87,7 @@ func (c *Config) Load() (*SchemaSpec, error) { return nil, fmt.Errorf("entc/load: write file %s: %w", target, err) } defer os.RemoveAll(".entc") - out, err := run(target) + out, err := run(target, c.BuildFlags) if err != nil { return nil, err } @@ -201,8 +201,11 @@ func filename(pkg string) string { } // run 'go run' command and return its output. -func run(target string) (string, error) { - cmd := exec.Command("go", "run", target) +func run(target string, buildFlags []string) (string, error) { + args := []string{"run"} + args = append(args, buildFlags...) + args = append(args, target) + cmd := exec.Command("go", args...) stderr := bytes.NewBuffer(nil) stdout := bytes.NewBuffer(nil) cmd.Stderr = stderr