go-zero/tools/goctl/kube/kube.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,
	})
}