improve compile function for goos and goarch

pkg/cmd/run.go

@@ -7,6 +7,7 @@ import (
 	"os"
 	"os/exec"
 	"path/filepath"
+	"runtime"
 	"syscall"
 	"text/template"
 
@@ -30,6 +31,9 @@ var errSlackTokenUndefined = errors.New("slack token is not defined.")
 
 func init() {
 	RunCmd.Flags().Bool("no-compile", false, "do not compile wrapper binary")
+	RunCmd.Flags().String("os", runtime.GOOS, "GOOS")
+	RunCmd.Flags().String("arch", runtime.GOARCH, "GOARCH")
+
 	RunCmd.Flags().String("config", "config/bbgo.yaml", "strategy config file")
 	RunCmd.Flags().String("since", "", "pnl since time")
 	RootCmd.AddCommand(RunCmd)
@@ -136,14 +140,37 @@ var RunCmd = &cobra.Command{
 			return err
 		}
 
-		if noCompile {
+		// if there is no custom imports, we don't have to compile
+		if noCompile || len(userConfig.Imports) == 0 {
 			if err := runConfig(ctx, userConfig); err != nil {
 				return err
 			}
 			cmdutil.WaitForSignal(ctx, syscall.SIGINT, syscall.SIGTERM)
 			return nil
-		} else {
+		}
-			buildDir := filepath.Join("build", "bbgow")
+
+		var flagsArgs = []string{"run", "--no-compile"}
+		cmd.Flags().Visit(func(flag *flag.Flag) {
+			flagsArgs = append(flagsArgs, flag.Name, flag.Value.String())
+		})
+		flagsArgs = append(flagsArgs, args...)
+
+		goOS, err := cmd.Flags().GetString("os")
+		if err != nil {
+			return err
+		}
+
+		goArch, err := cmd.Flags().GetString("arch")
+		if err != nil {
+			return err
+		}
+
+		var buildEnvs = []string{"GOOS=" + goOS, "GOARCH=" + goArch}
+		return compileAndRun(ctx, userConfig, buildEnvs, flagsArgs...)
+	},
+}
+
+func compile(buildDir string, userConfig *config.Config) error {
 	if _, err := os.Stat(buildDir); os.IsNotExist(err) {
 		if err := os.MkdirAll(buildDir, 0777); err != nil {
 			return errors.Wrapf(err, "can not create build directory: %s", buildDir)
@@ -155,7 +182,16 @@ var RunCmd = &cobra.Command{
 		return errors.Wrap(err, "compile error")
 	}
 
-			// TODO: use "\" for Windows
+	return nil
+}
+
+func compileAndRun(ctx context.Context, userConfig *config.Config, buildEnvs []string, args ...string) error {
+	buildDir := filepath.Join("build", "bbgow")
+
+	if err := compile(buildDir, userConfig); err != nil {
+		return err
+	}
+
 	cwd, err := os.Getwd()
 	if err != nil {
 		return err
@@ -165,28 +201,22 @@ var RunCmd = &cobra.Command{
 	log.Infof("building binary from %s...", buildTarget)
 
 	buildCmd := exec.CommandContext(ctx, "go", "build", "-tags", "wrapper", "-o", "bbgow", buildTarget)
+	buildCmd.Env = append(os.Environ(), buildEnvs...)
+
 	buildCmd.Stdout = os.Stdout
 	buildCmd.Stderr = os.Stderr
 	if err := buildCmd.Run(); err != nil {
 		return err
 	}
 
-			var flagsArgs = []string{"run", "--no-compile"}
-			cmd.Flags().Visit(func(flag *flag.Flag) {
-				flagsArgs = append(flagsArgs, flag.Name, flag.Value.String())
-			})
-			flagsArgs = append(flagsArgs, args...)
-
 	executePath := filepath.Join(cwd, "bbgow")
-			runCmd := exec.CommandContext(ctx, executePath, flagsArgs...)
+
+	log.Infof("running wrapper binary, args: %v", args)
+	runCmd := exec.CommandContext(ctx, executePath, args...)
 	runCmd.Stdout = os.Stdout
 	runCmd.Stderr = os.Stderr
 	if err := runCmd.Run(); err != nil {
 		return err
 	}
-
-		}
-
 	return nil
-	},
 }