This repository has been archived by the owner on Dec 7, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 228
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
WIP: Changes for making the gitops toolkit work
- Loading branch information
Showing
9 changed files
with
286 additions
and
57 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,179 @@ | ||
| package serializer | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "io/ioutil" | ||
|
|
||
| "k8s.io/apimachinery/pkg/runtime" | ||
| "k8s.io/apimachinery/pkg/runtime/schema" | ||
| k8sserializer "k8s.io/apimachinery/pkg/runtime/serializer" | ||
| "k8s.io/apimachinery/pkg/runtime/serializer/json" | ||
| ) | ||
|
|
||
| // Serializer is an interface providing high-level decoding/encoding functionality | ||
| // for types registered in a *runtime.Scheme | ||
| type Serializer interface { | ||
| // DecodeInto takes byte content and a target object to serialize the data into | ||
| DecodeInto(content []byte, obj runtime.Object) error | ||
| // DecodeFileInto takes a file path and a target object to serialize the data into | ||
| DecodeFileInto(filePath string, obj runtime.Object) error | ||
|
|
||
| // Decode takes byte content and returns the target object | ||
| Decode(content []byte, internal bool) (runtime.Object, error) | ||
| // DecodeFile takes a file path and returns the target object | ||
| DecodeFile(filePath string, internal bool) (runtime.Object, error) | ||
|
|
||
| // EncodeYAML encodes the specified object for a specific version to YAML bytes | ||
| EncodeYAML(obj runtime.Object) ([]byte, error) | ||
| // EncodeJSON encodes the specified object for a specific version to pretty JSON bytes | ||
| EncodeJSON(obj runtime.Object) ([]byte, error) | ||
|
|
||
| // DefaultInternal populates the given internal object with the preferred external version's defaults | ||
| DefaultInternal(cfg runtime.Object) error | ||
|
|
||
| // Scheme provides access to the underlying runtime.Scheme | ||
| Scheme() *runtime.Scheme | ||
| } | ||
|
|
||
| // NewSerializer constructs a new serializer based on a scheme, and optionally a codecfactory | ||
| func NewSerializer(scheme *runtime.Scheme, codecs *k8sserializer.CodecFactory) Serializer { | ||
| if scheme == nil { | ||
| panic("scheme must not be nil") | ||
| } | ||
|
|
||
| if codecs == nil { | ||
| codecs = &k8sserializer.CodecFactory{} | ||
| *codecs = k8sserializer.NewCodecFactory(scheme) | ||
| } | ||
|
|
||
| // Allow both YAML and JSON inputs (JSON is a subset of YAML), and deserialize in strict mode | ||
| strictSerializer := json.NewSerializerWithOptions(json.DefaultMetaFactory, scheme, scheme, json.SerializerOptions{ | ||
| Yaml: true, | ||
| Strict: true, | ||
| }) | ||
|
|
||
| return &serializer{ | ||
| scheme: scheme, | ||
| codecs: codecs, | ||
| // Construct a codec that uses the strict serializer, but also performs defaulting & conversion | ||
| decoder: codecs.CodecForVersions(nil, strictSerializer, nil, runtime.InternalGroupVersioner), | ||
| } | ||
| } | ||
|
|
||
| // serializer implements the Serializer interface | ||
| type serializer struct { | ||
| scheme *runtime.Scheme | ||
| codecs *k8sserializer.CodecFactory | ||
| decoder runtime.Decoder | ||
| } | ||
|
|
||
| // Scheme provides access to the underlying runtime.Scheme | ||
| func (s *serializer) Scheme() *runtime.Scheme { | ||
| return s.scheme | ||
| } | ||
|
|
||
| // DecodeFileInto takes a file path and a target object to serialize the data into | ||
| func (s *serializer) DecodeFileInto(filePath string, obj runtime.Object) error { | ||
| content, err := ioutil.ReadFile(filePath) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return s.DecodeInto(content, obj) | ||
| } | ||
|
|
||
| // DecodeInto takes byte content and a target object to serialize the data into | ||
| func (s *serializer) DecodeInto(content []byte, obj runtime.Object) error { | ||
| return runtime.DecodeInto(s.decoder, content, obj) | ||
| } | ||
|
|
||
| // DecodeFile takes a file path and returns the target object | ||
| func (s *serializer) DecodeFile(filePath string, internal bool) (runtime.Object, error) { | ||
| content, err := ioutil.ReadFile(filePath) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return s.Decode(content, internal) | ||
| } | ||
|
|
||
| // Decode takes byte content and returns the target object | ||
| func (s *serializer) Decode(content []byte, internal bool) (runtime.Object, error) { | ||
| obj, err := runtime.Decode(s.decoder, content) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| // Default the object | ||
| s.scheme.Default(obj) | ||
|
|
||
| // If we did not request an internal conversion, return quickly | ||
| if !internal { | ||
| return obj, nil | ||
| } | ||
| // Return the internal version of the object | ||
| return s.scheme.ConvertToVersion(obj, runtime.InternalGroupVersioner) | ||
| } | ||
|
|
||
| // EncodeYAML encodes the specified object for a specific version to YAML bytes | ||
| func (s *serializer) EncodeYAML(obj runtime.Object) ([]byte, error) { | ||
| return s.encode(obj, runtime.ContentTypeYAML, false) | ||
| } | ||
|
|
||
| // EncodeJSON encodes the specified object for a specific version to pretty JSON bytes | ||
| func (s *serializer) EncodeJSON(obj runtime.Object) ([]byte, error) { | ||
| return s.encode(obj, runtime.ContentTypeJSON, true) | ||
| } | ||
|
|
||
| func (s *serializer) encode(obj runtime.Object, mediaType string, pretty bool) ([]byte, error) { | ||
| info, ok := runtime.SerializerInfoForMediaType(s.codecs.SupportedMediaTypes(), mediaType) | ||
| if !ok { | ||
| return nil, fmt.Errorf("unable to locate encoder -- %q is not a supported media type", mediaType) | ||
| } | ||
|
|
||
| serializer := info.Serializer | ||
| if pretty { | ||
| serializer = info.PrettySerializer | ||
| } | ||
|
|
||
| gvk, err := s.externalGVKForObject(obj) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| encoder := s.codecs.EncoderForVersion(serializer, gvk.GroupVersion()) | ||
| return runtime.Encode(encoder, obj) | ||
| } | ||
|
|
||
| // DefaultInternal populates the given internal object with the preferred external version's defaults | ||
| func (s *serializer) DefaultInternal(cfg runtime.Object) error { | ||
| gvk, err := s.externalGVKForObject(cfg) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| external, err := s.scheme.New(*gvk) | ||
| if err != nil { | ||
| return nil | ||
| } | ||
| if err := s.scheme.Convert(cfg, external, nil); err != nil { | ||
| return err | ||
| } | ||
| s.scheme.Default(external) | ||
| return s.scheme.Convert(external, cfg, nil) | ||
| } | ||
|
|
||
| func (s *serializer) externalGVKForObject(cfg runtime.Object) (*schema.GroupVersionKind, error) { | ||
| gvks, unversioned, err := s.scheme.ObjectKinds(cfg) | ||
| if unversioned || err != nil || len(gvks) != 1 { | ||
| return nil, fmt.Errorf("unversioned %t or err %v or invalid gvks %v", unversioned, err, gvks) | ||
| } | ||
|
|
||
| gvk := gvks[0] | ||
| gvs := s.scheme.PrioritizedVersionsForGroup(gvk.Group) | ||
| if len(gvs) < 1 { | ||
| return nil, fmt.Errorf("expected some version to be registered for group %s", gvk.Group) | ||
| } | ||
|
|
||
| // Use the preferred (external) version | ||
| gvk.Version = gvs[0].Version | ||
| return &gvk, nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,50 @@ | ||
| package serializer_test | ||
|
|
||
| import ( | ||
| "testing" | ||
|
|
||
| api "github.com/weaveworks/ignite/pkg/apis/ignite" | ||
| "github.com/weaveworks/ignite/pkg/apis/ignite/scheme" | ||
| meta "github.com/weaveworks/ignite/pkg/apis/meta/v1alpha1" | ||
| "github.com/weaveworks/ignite/pkg/serializer" | ||
| ) | ||
|
|
||
| var s = serializer.NewSerializer(scheme.Scheme, nil) | ||
| var sampleobj = &api.VM{ | ||
| ObjectMeta: meta.ObjectMeta{ | ||
| Name: "foo", | ||
| UID: meta.UID("1234"), | ||
| }, | ||
| Spec: api.VMSpec{ | ||
| CPUs: 1, | ||
| }, | ||
| } | ||
| var samplejson = []byte(`{"kind":"VM","apiVersion":"ignite.weave.works/v1alpha1","metadata":{"name":"foo","uid":"1234"},"spec":{"cpus":1}}`) | ||
| var nonstrictjson = []byte(`{"kind":"VM","apiVersion":"ignite.weave.works/v1alpha1","metadata":{"name":"foo","uid":"1234"},"spec":{"cpus":1, "foo": "bar"}}`) | ||
|
|
||
| func TestEncodeJSON(t *testing.T) { | ||
| b, err := s.EncodeJSON(sampleobj) | ||
| t.Fatal(string(b), err) | ||
| } | ||
|
|
||
| func TestEncodeYAML(t *testing.T) { | ||
| b, err := s.EncodeYAML(sampleobj) | ||
| t.Fatal(string(b), err) | ||
| } | ||
|
|
||
| func TestDecode(t *testing.T) { | ||
| obj, err := s.Decode(samplejson) | ||
| t.Fatal(obj, err) | ||
| } | ||
|
|
||
| func TestDecodeInto(t *testing.T) { | ||
| vm := &api.VM{} | ||
| err := s.DecodeInto(samplejson, vm) | ||
| t.Fatal(*vm, err) | ||
| } | ||
|
|
||
| func TestDecodeStrict(t *testing.T) { | ||
| vm := &api.VM{} | ||
| err := s.DecodeInto(nonstrictjson, vm) | ||
| t.Fatal(vm, err) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.