Question

我正在尝试使用text/template即时生成类似ini的配置，其中原始数据以yaml格式提供。

我希望结果输出根据请求来自何处而有所不同。

请考虑以下损坏的代码：

package main

import (
    "fmt"
    "gopkg.in/yaml.v3"
    "os"
    "text/template"
)

var yamlData = `
# comment
---
States:
- StateName: California
  DateEstablished: September 9, 1850
  Cities:
  - CityName: Los Angeles
    Population: 4 Million
  - CityName: Santa Barbara
    Population: 92 Thousand
  - CityName: San Jose
    Population: 1 Million
- StateName: Washington
  DateEstablished: November 11, 1889
  Cities:
  - CityName: Seattle
    Population: 724 Thousand
    Climate: wet
  - CityName: Spokane
    Population: 217 Thousand
    Climate: dry
  - CityName: Scappoose
    Population: 7
`

const reportTemplate string = `{{ range . }}
# {{ if .CityName == requestingCityName }}
# {{ .CityName }}
[RequestingCity]
Population = {{ .Population }}
{{ if .Climate }}Climate = {{ .Climate }}{{ end }}
{{ end }}
# {{ if .CityName != requestingCityName }}
# {{ .CityName }}
[City]
Population = {{ .Population }}
{{ if .Climate }}Climate = {{ .Climate }}{{ end }}
{{ end }}
{{ end }}
`

type dataStruct struct {
    States []struct {
        StateName       string `yaml:"StateName"`
        DateEstablished string `yaml:"DateEstablished"`
        Cities          []struct {
            CityName   string `yaml:"CityName"`
            Population string `yaml:"Population"`
            Climate    string `yaml:"Climate,omitempty"`
        } `yaml:"Cities"`
    } `yaml:"States"`
}

func (d *dataStruct) readData(data []byte) *dataStruct {
    yaml.Unmarshal(data, d)
    return d
}

func main() {

    var report dataStruct
    report.readData([]byte(yamlData))
    t := template.Must(template.New("new").Parse(reportTemplate))

    requestingStateName := "Washington"
    requestingCityName := "Seattle"

    for i := range report.States {
        if report.States[i].StateName == requestingStateName {
            x := report.States[i].Cities
            fmt.Println(t.Execute(os.Stdout, x))

        }
    }
}

此代码中的大多数代码都可以按照我期望的方式“工作”，但是我遇到的问题是如何制作模板。

我希望能够更改requestingCityName的值，以便输出更改为：

如果requestingCityName == "Scappoose" 那么输出将是这样的：


# Scappoose
[RequestingCity]
Population = 7

# Spokane
[City]
Population = 217 Thousand
Climate = dry

# Seattle
[City]
Population = 724 Thousand
Climate = wet

或者如果requestingCityName == "Seattle" 那么输出将是这样的：


# Seattle
[RequestingCity]
Population = 724 Thousand
Climate = wet

# Spokane
[City]
Population = 217 Thousand
Climate = dry

# Scappoose
[City]
Population = 7

如何制作模板以实现所需的行为？

Answer 1

一个简单的解决方案是将一个不同的数据对象传递给Template.Execute。像这样：

type templateData struct {
    requestingCityName string
    cities             []citiesStruct // or whatever you name the struct
}
...
fmt.Println(t.Execute(os.Stdout, templateData{requestingCityName, x}))

此解决方案将要求您更新模板以与新的上下文结构一起使用（即，旧的城市数组现在为.cities，而不是.），但是它使您可以访问{{ 1}}。

Answer 2

您可以使用eq在模板内执行if条件，请参见Compare strings in templates。否定是否可以写if not <a> eq <b>。

我认为您不关心订单。如果您愿意的话，可以事先对其进行排序，因此顶部是请求的城市，下面是所有其他内容。否则，您必须将request-city-printing拆分并输出到请求范围之外，然后在所有城市范围内过滤掉请求。

最干净的方法可能是在保存请求城市的数据结构中添加一个单独的属性，并事先将其中一个城市过滤掉。

哦，如果您还没有将请求的城市添加到数据结构中，请事先进行操作。通过将yaml数据结构嵌入到名为DataRendering的结构（也具有请求的city属性）中来嵌入yaml数据结构，可能是最简单的方法。

基于变量的Golang格式文本模板

2 个答案: