40 lines
888 B
Go
40 lines
888 B
Go
|
package cmd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"git.qtrade.icu/coin-quant/trade/pkg/ctl"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
)
|
||
|
|
|
||
|
|
// buildCmd represents the build command
|
||
|
|
var buildCmd = &cobra.Command{
|
||
|
|
Use: "build",
|
||
|
|
Short: "build script to go plugin",
|
||
|
|
Long: `"build script to go plugin`,
|
||
|
|
Run: runBuild,
|
||
|
|
}
|
||
|
|
|
||
|
|
var (
|
||
|
|
output string
|
||
|
|
keepTemp bool
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
rootCmd.AddCommand(buildCmd)
|
||
|
|
buildCmd.PersistentFlags().StringVar(&scriptFile, "script", "", "script file to backtest")
|
||
|
|
buildCmd.PersistentFlags().StringVar(&output, "output", "", "plugin output file")
|
||
|
|
buildCmd.PersistentFlags().BoolVarP(&keepTemp, "keep", "k", false, "keep temp dir")
|
||
|
|
}
|
||
|
|
|
||
|
|
func runBuild(cmd *cobra.Command, args []string) {
|
||
|
|
b := ctl.NewBuilder(scriptFile, output)
|
||
|
|
b.SetKeepTemp(keepTemp)
|
||
|
|
err := b.Build()
|
||
|
|
if err != nil {
|
||
|
|
fmt.Println("build failed:", err.Error())
|
||
|
|
return
|
||
|
|
}
|
||
|
|
fmt.Printf("build success: %s\n", output)
|
||
|
|
}
|