Add a sorting plugin.

This commit is contained in:
Jeffrey Regan
2019-06-07 14:42:51 -07:00
parent 2fce1a6d25
commit 449175e3a6
9 changed files with 240 additions and 41 deletions

View File

@@ -0,0 +1,44 @@
// Code generated by pluginator on PreferredOrderTransformer; DO NOT EDIT.
package builtin
import (
"sigs.k8s.io/kustomize/pkg/resmap"
"sigs.k8s.io/kustomize/pkg/resource"
"sigs.k8s.io/kustomize/pkg/transformers/config"
"sort"
)
// Sort the resources using an ordering defined in the Gvk class.
// This puts cluster-wide basic resources with no
// dependencies (like Namespace, StorageClass, etc.)
// first, and resources with a high number of dependencies
// (like ValidatingWebhookConfiguration) last.
type PreferredOrderTransformerPlugin struct {
Prefix string `json:"prefix,omitempty" yaml:"prefix,omitempty"`
Suffix string `json:"suffix,omitempty" yaml:"suffix,omitempty"`
FieldSpecs []config.FieldSpec `json:"fieldSpecs,omitempty" yaml:"fieldSpecs,omitempty"`
}
func NewPreferredOrderTransformerPlugin() *PreferredOrderTransformerPlugin {
return &PreferredOrderTransformerPlugin{}
}
/*
func (p *PreferredOrderTransformerPlugin) Config(
ldr ifc.Loader, rf *resmap.Factory, c []byte) (err error) {
return nil
}
*/
func (p *PreferredOrderTransformerPlugin) Transform(m resmap.ResMap) error {
resources := make([]*resource.Resource, m.Size())
ids := m.AllIds()
sort.Sort(resmap.IdSlice(ids))
for i, id := range ids {
resources[i] = m.GetById(id)
}
m.Clear()
for i, id := range ids {
m.AppendWithId(id, resources[i])
}
return nil
}

View File

@@ -0,0 +1,41 @@
// Copyright 2019 The Kubernetes Authors.
// SPDX-License-Identifier: Apache-2.0
//go:generate go run sigs.k8s.io/kustomize/plugin/pluginator
package main
import (
"sigs.k8s.io/kustomize/pkg/ifc"
"sigs.k8s.io/kustomize/pkg/resmap"
"sigs.k8s.io/kustomize/pkg/resource"
"sort"
)
// Sort the resources using an ordering defined in the Gvk class.
// This puts cluster-wide basic resources with no
// dependencies (like Namespace, StorageClass, etc.)
// first, and resources with a high number of dependencies
// (like ValidatingWebhookConfiguration) last.
type plugin struct {}
var KustomizePlugin plugin
// Nothing needed for configuration.
func (p *plugin) Config(
ldr ifc.Loader, rf *resmap.Factory, c []byte) (err error) {
return nil
}
func (p *plugin) Transform(m resmap.ResMap) error {
resources := make([]*resource.Resource, m.Size())
ids := m.AllIds()
sort.Sort(resmap.IdSlice(ids))
for i, id := range ids {
resources[i] = m.GetById(id)
}
m.Clear()
for i, id := range ids {
m.AppendWithId(id, resources[i])
}
return nil
}

View File

@@ -0,0 +1,119 @@
// Copyright 2019 The Kubernetes Authors.
// SPDX-License-Identifier: Apache-2.0
package main_test
import (
"testing"
"sigs.k8s.io/kustomize/pkg/kusttest"
"sigs.k8s.io/kustomize/plugin"
)
func TestPreferredOrderTransformer(t *testing.T) {
tc := plugin.NewEnvForTest(t).Set()
defer tc.Reset()
tc.BuildGoPlugin(
"builtin", "", "PreferredOrderTransformer")
th := kusttest_test.NewKustTestPluginHarness(t, "/app")
rm := th.LoadAndRunTransformer(`
apiVersion: builtin
kind: PreferredOrderTransformer
metadata:
name: notImportantHere
`, `
apiVersion: v1
kind: Service
metadata:
name: papaya
---
apiVersion: v1
kind: Role
metadata:
name: banana
---
apiVersion: v1
kind: ValidatingWebhookConfiguration
metadata:
name: pomegranate
---
apiVersion: v1
kind: LimitRange
metadata:
name: peach
---
apiVersion: v1
kind: Deployment
metadata:
name: pear
---
apiVersion: v1
kind: Namespace
metadata:
name: apple
---
apiVersion: v1
kind: Secret
metadata:
name: quince
---
apiVersion: v1
kind: Ingress
metadata:
name: durian
---
apiVersion: v1
kind: ConfigMap
metadata:
name: apricot
`)
th.AssertActualEqualsExpected(rm, `
apiVersion: v1
kind: Namespace
metadata:
name: apple
---
apiVersion: v1
kind: Role
metadata:
name: banana
---
apiVersion: v1
kind: ConfigMap
metadata:
name: apricot
---
apiVersion: v1
kind: Secret
metadata:
name: quince
---
apiVersion: v1
kind: Service
metadata:
name: papaya
---
apiVersion: v1
kind: LimitRange
metadata:
name: peach
---
apiVersion: v1
kind: Deployment
metadata:
name: pear
---
apiVersion: v1
kind: Ingress
metadata:
name: durian
---
apiVersion: v1
kind: ValidatingWebhookConfiguration
metadata:
name: pomegranate
`)
}

View File

@@ -8,7 +8,6 @@ import (
"testing"
"sigs.k8s.io/kustomize/pkg/kusttest"
"sigs.k8s.io/kustomize/pkg/resmap"
"sigs.k8s.io/kustomize/plugin"
)
@@ -33,7 +32,7 @@ kind: PrintWorkDir
metadata:
name: whatever
`)
a, err := m.AsYaml(resmap.Identity)
a, err := m.AsYaml()
if err != nil {
t.Error(err)
}