goctl added

tools/goctl/api/gogen/genetc.go

@@ -0,0 +1,56 @@
+package gogen
+
+import (
+	"bytes"
+	"fmt"
+	"strconv"
+	"text/template"
+
+	"zero/tools/goctl/api/spec"
+	"zero/tools/goctl/api/util"
+)
+
+const (
+	defaultPort = 8888
+	etcDir      = "etc"
+	etcTemplate = `{
+    "Name": "{{.serviceName}}",
+    "Host": "{{.host}}",
+    "Port": {{.port}}
+}`
+)
+
+func genEtc(dir string, api *spec.ApiSpec) error {
+	fp, created, err := util.MaybeCreateFile(dir, etcDir, fmt.Sprintf("%s.json", api.Service.Name))
+	if err != nil {
+		return err
+	}
+	if !created {
+		return nil
+	}
+	defer fp.Close()
+
+	service := api.Service
+	host, ok := util.GetAnnotationValue(service.Annotations, "server", "host")
+	if !ok {
+		host = "0.0.0.0"
+	}
+	port, ok := util.GetAnnotationValue(service.Annotations, "server", "port")
+	if !ok {
+		port = strconv.Itoa(defaultPort)
+	}
+
+	t := template.Must(template.New("etcTemplate").Parse(etcTemplate))
+	buffer := new(bytes.Buffer)
+	err = t.Execute(buffer, map[string]string{
+		"serviceName": service.Name,
+		"host":        host,
+		"port":        port,
+	})
+	if err != nil {
+		return err
+	}
+	formatCode := formatCode(buffer.String())
+	_, err = fp.WriteString(formatCode)
+	return err
+}