add PatchJson6902Factory to make transformer

This commit is contained in:
Jingfang Liu
2018-08-31 10:21:58 -07:00
parent 7f0e9e3a6a
commit cedf215445
9 changed files with 320 additions and 153 deletions

View File

@@ -17,15 +17,9 @@ limitations under the License.
package transformer
import (
"fmt"
"log"
"github.com/ghodss/yaml"
yamlpatch "github.com/krishicks/yaml-patch"
"k8s.io/apimachinery/pkg/runtime/schema"
"github.com/kubernetes-sigs/kustomize/pkg/patch"
"github.com/kubernetes-sigs/kustomize/pkg/resmap"
"github.com/kubernetes-sigs/kustomize/pkg/resource"
"github.com/kubernetes-sigs/kustomize/pkg/transformers"
@@ -33,49 +27,26 @@ import (
// patchJson6902Transformer applies patches.
type patchJson6902YAMLTransformer struct {
target *patch.Target
target resource.ResId
patch yamlpatch.Patch
}
var _ transformers.Transformer = &patchJson6902YAMLTransformer{}
// NewPatchJson6902YAMLTransformer constructs a PatchJson6902 transformer.
func NewPatchJson6902YAMLTransformer(t *patch.Target, p yamlpatch.Patch) (transformers.Transformer, error) {
// newPatchJson6902YAMLTransformer constructs a PatchJson6902 transformer.
func newPatchJson6902YAMLTransformer(t resource.ResId, p yamlpatch.Patch) (transformers.Transformer, error) {
if len(p) == 0 {
return transformers.NewNoOpTransformer(), nil
}
return &patchJson6902YAMLTransformer{target: t, patch: p}, nil
}
// Transform apply the json patches on top of the base resources.
func (t *patchJson6902YAMLTransformer) Transform(baseResourceMap resmap.ResMap) error {
targetId := resource.NewResIdWithPrefixNamespace(
schema.GroupVersionKind{
Group: t.target.Group,
Version: t.target.Version,
Kind: t.target.Kind,
},
t.target.Name,
"",
t.target.Namespace,
)
matchedIds := baseResourceMap.FindByGVKN(targetId)
if targetId.Namespace() != "" {
ids := []resource.ResId{}
for _, id := range matchedIds {
if id.Namespace() == targetId.Namespace() {
ids = append(ids, id)
}
}
matchedIds = ids
obj, err := findTargetObj(baseResourceMap, t.target)
if obj == nil {
return err
}
if len(matchedIds) == 0 {
log.Printf("Couldn't find any object to apply the json patch %v, skipping it.", targetId)
return nil
}
if len(matchedIds) > 1 {
return fmt.Errorf("found multiple objects that the patch can apply %v", matchedIds)
}
obj := baseResourceMap[matchedIds[0]]
rawObj, err := yaml.Marshal(obj.Unstructured.Object)
if err != nil {
return err