49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
package cmd
|
|
|
|
import (
|
|
"github.com/spf13/pflag"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var (
|
|
target string
|
|
center string
|
|
mode string
|
|
agentPort string
|
|
debugGoc bool
|
|
buildFlags string
|
|
|
|
goRunExecFlag string
|
|
goRunArguments string
|
|
)
|
|
|
|
// addBasicFlags adds a
|
|
func addBasicFlags(cmdset *pflag.FlagSet) {
|
|
cmdset.StringVar(¢er, "center", "http://127.0.0.1:7777", "cover profile host center")
|
|
// bind to viper
|
|
viper.BindPFlags(cmdset)
|
|
}
|
|
|
|
func addCommonFlags(cmdset *pflag.FlagSet) {
|
|
addBasicFlags(cmdset)
|
|
cmdset.StringVar(&mode, "mode", "count", "coverage mode: set, count, atomic")
|
|
cmdset.StringVar(&agentPort, "agentport", "", "specify fixed port for registered service communicate with goc server. if not provided, using a random one")
|
|
cmdset.StringVar(&buildFlags, "buildflags", "", "specify the build flags")
|
|
// bind to viper
|
|
viper.BindPFlags(cmdset)
|
|
}
|
|
|
|
func addBuildFlags(cmdset *pflag.FlagSet) {
|
|
addCommonFlags(cmdset)
|
|
// bind to viper
|
|
viper.BindPFlags(cmdset)
|
|
}
|
|
|
|
func addRunFlags(cmdset *pflag.FlagSet) {
|
|
addBuildFlags(cmdset)
|
|
cmdset.StringVar(&goRunExecFlag, "exec", "", "same as -exec flag in 'go run' command")
|
|
cmdset.StringVar(&goRunArguments, "arguments", "", "same as 'arguments' in 'go run' command")
|
|
// bind to viper
|
|
viper.BindPFlags(cmdset)
|
|
}
|