7a82cf80ce
* simplify code, format makefile * simplify code * some optimize by kevwan and benying (#240) Co-authored-by: 杨志泉 <zhiquan.yang@yiducloud.cn> * optimization (#241) * optimize docker file generation, make docker build faster * support k8s deployment yaml generation Co-authored-by: benying <31179034+benyingY@users.noreply.github.com> Co-authored-by: 杨志泉 <zhiquan.yang@yiducloud.cn> Co-authored-by: bittoy <bittoy@qq.com>
105 lines
2.4 KiB
Go
105 lines
2.4 KiB
Go
package kube
|
|
|
|
import (
|
|
"errors"
|
|
"text/template"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/util"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
const (
|
|
category = "kube"
|
|
deployTemplateFile = "deployment.tpl"
|
|
jobTemplateFile = "job.tpl"
|
|
basePort = 30000
|
|
portLimit = 32767
|
|
)
|
|
|
|
var errUnknownServiceType = errors.New("unknown service type")
|
|
|
|
type (
|
|
ServiceType string
|
|
|
|
KubeRequest struct {
|
|
Env string
|
|
ServiceName string
|
|
ServiceType ServiceType
|
|
Namespace string
|
|
Schedule string
|
|
Replicas int
|
|
RevisionHistoryLimit int
|
|
Port int
|
|
LimitCpu int
|
|
LimitMem int
|
|
RequestCpu int
|
|
RequestMem int
|
|
SuccessfulJobsHistoryLimit int
|
|
HpaMinReplicas int
|
|
HpaMaxReplicas int
|
|
}
|
|
|
|
Deployment struct {
|
|
Name string
|
|
Namespace string
|
|
Image string
|
|
Secret string
|
|
Replicas int
|
|
Revisions int
|
|
Port int
|
|
NodePort int
|
|
UseNodePort bool
|
|
RequestCpu int
|
|
RequestMem int
|
|
LimitCpu int
|
|
LimitMem int
|
|
MinReplicas int
|
|
MaxReplicas int
|
|
}
|
|
)
|
|
|
|
func DeploymentCommand(c *cli.Context) error {
|
|
nodePort := c.Int("nodePort")
|
|
// 0 to disable the nodePort type
|
|
if nodePort != 0 && (nodePort < basePort || nodePort > portLimit) {
|
|
return errors.New("nodePort should be between 30000 and 32767")
|
|
}
|
|
|
|
text, err := util.LoadTemplate(category, deployTemplateFile, deploymentTemplate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
out, err := util.CreateIfNotExist(c.String("o"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer out.Close()
|
|
|
|
t := template.Must(template.New("deploymentTemplate").Parse(text))
|
|
return t.Execute(out, Deployment{
|
|
Name: c.String("name"),
|
|
Namespace: c.String("namespace"),
|
|
Image: c.String("image"),
|
|
Secret: c.String("secret"),
|
|
Replicas: c.Int("replicas"),
|
|
Revisions: c.Int("revisions"),
|
|
Port: c.Int("port"),
|
|
NodePort: nodePort,
|
|
UseNodePort: nodePort > 0,
|
|
RequestCpu: c.Int("requestCpu"),
|
|
RequestMem: c.Int("requestMem"),
|
|
LimitCpu: c.Int("limitCpu"),
|
|
LimitMem: c.Int("limitMem"),
|
|
MinReplicas: c.Int("minReplicas"),
|
|
MaxReplicas: c.Int("maxReplicas"),
|
|
})
|
|
}
|
|
|
|
func GenTemplates(_ *cli.Context) error {
|
|
return util.InitTemplates(category, map[string]string{
|
|
deployTemplateFile: deploymentTemplate,
|
|
jobTemplateFile: jobTmeplate,
|
|
})
|
|
}
|