37 lines
839 B
Go
37 lines
839 B
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/spf13/cobra"
|
|
"yizhisec.com/hsv2/forge/internal/controller/maker"
|
|
)
|
|
|
|
func makeES() *cobra.Command {
|
|
var (
|
|
makeHelper bool
|
|
memRate int
|
|
storage int
|
|
)
|
|
|
|
_cmd := &cobra.Command{
|
|
Use: "es",
|
|
Aliases: []string{"elastic", "elasticsearch"},
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
mk := maker.NewMaker()
|
|
return mk.Elastic(
|
|
cmd.Context(),
|
|
maker.WithElasticMakeHelper(makeHelper),
|
|
maker.WithElasticMemRate(memRate),
|
|
maker.WithElasticStorageGi(fmt.Sprintf("%dGi", storage)),
|
|
)
|
|
},
|
|
}
|
|
|
|
_cmd.Flags().BoolVar(&makeHelper, "make-helper", false, "重新构建 helper 镜像")
|
|
_cmd.Flags().IntVar(&memRate, "mem-rate", 1, "内存倍率(从 1G 开始)")
|
|
_cmd.Flags().IntVar(&storage, "storage-size", 100, "存储空间(单位: G)")
|
|
|
|
return _cmd
|
|
}
|