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 }