mirror of https://github.com/grafana/grafana
Alerting: Export of alert rules in HCL format (#73166)
* import hashicopr/hcl/v2 * add hcl package and export to HCL * annotate export structs --------- Co-authored-by: Konrad Lalik <konrad.lalik@grafana.com>pull/74405/head^2
Yuri Tseretyan
2 years ago
committed by
GitHub
parent
5d89c15851
commit
dce492642a
@ -0,0 +1,30 @@ |
||||
package hcl |
||||
|
||||
import ( |
||||
"fmt" |
||||
|
||||
"github.com/hashicorp/hcl/v2/gohcl" |
||||
"github.com/hashicorp/hcl/v2/hclwrite" |
||||
) |
||||
|
||||
type Resource struct { |
||||
Type string `hcl:"type,label"` |
||||
Name string `hcl:"name,label"` |
||||
Body interface{} `hcl:",block"` |
||||
} |
||||
|
||||
func Encode(resources ...Resource) (data []byte, err error) { |
||||
defer func() { |
||||
if r := recover(); r != nil { |
||||
err = fmt.Errorf("failed to encode struct to HCL: %v", r) |
||||
} |
||||
}() |
||||
f := hclwrite.NewEmptyFile() |
||||
|
||||
for _, resource := range resources { |
||||
blk := gohcl.EncodeAsBlock(resource.Body, "resource") |
||||
blk.SetLabels([]string{resource.Type, resource.Name}) |
||||
f.Body().AppendBlock(blk) |
||||
} |
||||
return f.Bytes(), nil |
||||
} |
||||
@ -0,0 +1,75 @@ |
||||
package hcl |
||||
|
||||
import ( |
||||
"testing" |
||||
|
||||
"github.com/stretchr/testify/require" |
||||
) |
||||
|
||||
func TestEncode(t *testing.T) { |
||||
type data struct { |
||||
Name string `hcl:"name"` |
||||
Number float64 `hcl:"number"` |
||||
NumberRef *float64 `hcl:"numberRef"` |
||||
Bool bool `hcl:"bul"` |
||||
BoolRef *bool `hcl:"bulRef"` |
||||
Ignored string |
||||
Blocks []data `hcl:"blocks,block"` |
||||
SubData *data `hcl:"sub,block"` |
||||
} |
||||
|
||||
encoded, err := Encode(Resource{ |
||||
Type: "grafana_test", |
||||
Name: "test-01", |
||||
Body: &data{ |
||||
Name: "test", |
||||
Number: 123, |
||||
NumberRef: func(f float64) *float64 { return &f }(1333), |
||||
Bool: false, |
||||
BoolRef: func(f bool) *bool { return &f }(true), |
||||
Ignored: "Ignore me", |
||||
Blocks: []data{ |
||||
{ |
||||
Name: "el-0", |
||||
Number: 1, |
||||
}, |
||||
{ |
||||
Name: "el-1", |
||||
Number: 2, |
||||
Bool: true, |
||||
}, |
||||
}, |
||||
SubData: &data{ |
||||
Name: "sub-data", |
||||
Number: 123123, |
||||
}, |
||||
}, |
||||
}) |
||||
require.NoError(t, err) |
||||
t.Log(string(encoded)) |
||||
require.Equal(t, `resource "grafana_test" "test-01" { |
||||
name = "test" |
||||
number = 123 |
||||
numberRef = 1333 |
||||
bul = false |
||||
bulRef = true |
||||
|
||||
blocks { |
||||
name = "el-0" |
||||
number = 1 |
||||
bul = false |
||||
} |
||||
blocks { |
||||
name = "el-1" |
||||
number = 2 |
||||
bul = true |
||||
} |
||||
|
||||
sub { |
||||
name = "sub-data" |
||||
number = 123123 |
||||
bul = false |
||||
} |
||||
} |
||||
`, string(encoded)) |
||||
} |
||||
Loading…
Reference in new issue