mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-10 09:11:55 +00:00
Merge pull request #25 from c9s/feature/go-compile-os-arch
feature: support go build with custom os and arch
This commit is contained in:
commit
9f416579ec
|
@ -7,6 +7,7 @@ import (
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
"runtime"
|
||||||
"syscall"
|
"syscall"
|
||||||
"text/template"
|
"text/template"
|
||||||
|
|
||||||
|
@ -30,6 +31,9 @@ var errSlackTokenUndefined = errors.New("slack token is not defined.")
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
RunCmd.Flags().Bool("no-compile", false, "do not compile wrapper binary")
|
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("config", "config/bbgo.yaml", "strategy config file")
|
||||||
RunCmd.Flags().String("since", "", "pnl since time")
|
RunCmd.Flags().String("since", "", "pnl since time")
|
||||||
RootCmd.AddCommand(RunCmd)
|
RootCmd.AddCommand(RunCmd)
|
||||||
|
@ -136,14 +140,37 @@ var RunCmd = &cobra.Command{
|
||||||
return err
|
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 {
|
if err := runConfig(ctx, userConfig); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
cmdutil.WaitForSignal(ctx, syscall.SIGINT, syscall.SIGTERM)
|
cmdutil.WaitForSignal(ctx, syscall.SIGINT, syscall.SIGTERM)
|
||||||
return nil
|
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.Stat(buildDir); os.IsNotExist(err) {
|
||||||
if err := os.MkdirAll(buildDir, 0777); err != nil {
|
if err := os.MkdirAll(buildDir, 0777); err != nil {
|
||||||
return errors.Wrapf(err, "can not create build directory: %s", buildDir)
|
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")
|
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()
|
cwd, err := os.Getwd()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
|
@ -165,28 +201,22 @@ var RunCmd = &cobra.Command{
|
||||||
log.Infof("building binary from %s...", buildTarget)
|
log.Infof("building binary from %s...", buildTarget)
|
||||||
|
|
||||||
buildCmd := exec.CommandContext(ctx, "go", "build", "-tags", "wrapper", "-o", "bbgow", buildTarget)
|
buildCmd := exec.CommandContext(ctx, "go", "build", "-tags", "wrapper", "-o", "bbgow", buildTarget)
|
||||||
|
buildCmd.Env = append(os.Environ(), buildEnvs...)
|
||||||
|
|
||||||
buildCmd.Stdout = os.Stdout
|
buildCmd.Stdout = os.Stdout
|
||||||
buildCmd.Stderr = os.Stderr
|
buildCmd.Stderr = os.Stderr
|
||||||
if err := buildCmd.Run(); err != nil {
|
if err := buildCmd.Run(); err != nil {
|
||||||
return err
|
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")
|
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.Stdout = os.Stdout
|
||||||
runCmd.Stderr = os.Stderr
|
runCmd.Stderr = os.Stderr
|
||||||
if err := runCmd.Run(); err != nil {
|
if err := runCmd.Run(); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
},
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user