Merge pull request #12414 from krousey/versioned_codec

Adding a way to decode to a specified version.
This commit is contained in:
Marek Grabowski 2015-08-10 16:32:36 +02:00
commit d239d51f3a
7 changed files with 49 additions and 7 deletions

View File

@ -33,6 +33,10 @@ func (fakeCodec) Decode([]byte) (runtime.Object, error) {
return nil, nil return nil, nil
} }
func (fakeCodec) DecodeToVersion([]byte, string) (runtime.Object, error) {
return nil, nil
}
func (fakeCodec) DecodeInto([]byte, runtime.Object) error { func (fakeCodec) DecodeInto([]byte, runtime.Object) error {
return nil return nil
} }

View File

@ -50,7 +50,16 @@ func (s *Scheme) DecodeToVersionedObject(data []byte) (obj interface{}, version,
// s.InternalVersion type before being returned. Decode will not decode // s.InternalVersion type before being returned. Decode will not decode
// objects without version set unless InternalVersion is also "". // objects without version set unless InternalVersion is also "".
func (s *Scheme) Decode(data []byte) (interface{}, error) { func (s *Scheme) Decode(data []byte) (interface{}, error) {
obj, version, kind, err := s.DecodeToVersionedObject(data) return s.DecodeToVersion(data, s.InternalVersion)
}
// DecodeToVersion converts a JSON string back into a pointer to an api object.
// Deduces the type based upon the fields added by the MetaInsertionFactory
// technique. The object will be converted, if necessary, into the versioned
// type before being returned. Decode will not decode objects without version
// set unless version is also "".
func (s *Scheme) DecodeToVersion(data []byte, version string) (interface{}, error) {
obj, sourceVersion, kind, err := s.DecodeToVersionedObject(data)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -60,12 +69,12 @@ func (s *Scheme) Decode(data []byte) (interface{}, error) {
} }
// Convert if needed. // Convert if needed.
if s.InternalVersion != version { if version != sourceVersion {
objOut, err := s.NewObject(s.InternalVersion, kind) objOut, err := s.NewObject(version, kind)
if err != nil { if err != nil {
return nil, err return nil, err
} }
flags, meta := s.generateConvertMeta(version, s.InternalVersion, obj) flags, meta := s.generateConvertMeta(sourceVersion, version, obj)
if err := s.converter.Convert(obj, objOut, flags, meta); err != nil { if err := s.converter.Convert(obj, objOut, flags, meta); err != nil {
return nil, err return nil, err
} }

View File

@ -77,3 +77,10 @@ type codecWrapper struct {
func (c *codecWrapper) Encode(obj Object) ([]byte, error) { func (c *codecWrapper) Encode(obj Object) ([]byte, error) {
return c.EncodeToVersion(obj, c.version) return c.EncodeToVersion(obj, c.version)
} }
// TODO: Make this behaviour default when we move everyone away from
// the unversioned types.
//
// func (c *codecWrapper) Decode(data []byte) (Object, error) {
// return c.DecodeToVersion(data, c.version)
// }

View File

@ -37,6 +37,7 @@ type ObjectCodec interface {
// Decoder defines methods for deserializing API objects into a given type // Decoder defines methods for deserializing API objects into a given type
type Decoder interface { type Decoder interface {
Decode(data []byte) (Object, error) Decode(data []byte) (Object, error)
DecodeToVersion(data []byte, version string) (Object, error)
DecodeInto(data []byte, obj Object) error DecodeInto(data []byte, obj Object) error
DecodeIntoWithSpecifiedVersionKind(data []byte, obj Object, kind, version string) error DecodeIntoWithSpecifiedVersionKind(data []byte, obj Object, kind, version string) error
} }

View File

@ -446,6 +446,19 @@ func (s *Scheme) Decode(data []byte) (Object, error) {
return obj.(Object), nil return obj.(Object), nil
} }
// DecodeToVersion converts a YAML or JSON string back into a pointer to an api
// object. Deduces the type based upon the APIVersion and Kind fields, which
// are set by Encode. Only versioned objects (APIVersion != "") are
// accepted. The object will be converted into the in-memory versioned type
// requested before being returned.
func (s *Scheme) DecodeToVersion(data []byte, version string) (Object, error) {
obj, err := s.raw.DecodeToVersion(data, version)
if err != nil {
return nil, err
}
return obj.(Object), nil
}
// DecodeInto parses a YAML or JSON string and stores it in obj. Returns an error // DecodeInto parses a YAML or JSON string and stores it in obj. Returns an error
// if data.Kind is set and doesn't match the type of obj. Obj should be a // if data.Kind is set and doesn't match the type of obj. Obj should be a
// pointer to an api type. // pointer to an api type.

View File

@ -87,14 +87,15 @@ func TestScheme(t *testing.T) {
TestString: "foo", TestString: "foo",
} }
// Test Encode, Decode, and DecodeInto // Test Encode, Decode, DecodeInto, and DecodeToVersion
obj := runtime.Object(simple) obj := runtime.Object(simple)
data, err := scheme.EncodeToVersion(obj, "externalVersion") data, err := scheme.EncodeToVersion(obj, "externalVersion")
obj2, err2 := scheme.Decode(data) obj2, err2 := scheme.Decode(data)
obj3 := &InternalSimple{} obj3 := &InternalSimple{}
err3 := scheme.DecodeInto(data, obj3) err3 := scheme.DecodeInto(data, obj3)
if err != nil || err2 != nil { obj4, err4 := scheme.DecodeToVersion(data, "externalVersion")
t.Fatalf("Failure: '%v' '%v' '%v'", err, err2, err3) if err != nil || err2 != nil || err3 != nil || err4 != nil {
t.Fatalf("Failure: '%v' '%v' '%v' '%v'", err, err2, err3, err4)
} }
if _, ok := obj2.(*InternalSimple); !ok { if _, ok := obj2.(*InternalSimple); !ok {
t.Fatalf("Got wrong type") t.Fatalf("Got wrong type")
@ -105,6 +106,9 @@ func TestScheme(t *testing.T) {
if e, a := simple, obj3; !reflect.DeepEqual(e, a) { if e, a := simple, obj3; !reflect.DeepEqual(e, a) {
t.Errorf("Expected:\n %#v,\n Got:\n %#v", e, a) t.Errorf("Expected:\n %#v,\n Got:\n %#v", e, a)
} }
if _, ok := obj4.(*ExternalSimple); !ok {
t.Fatalf("Got wrong type")
}
// Test Convert // Test Convert
external := &ExternalSimple{} external := &ExternalSimple{}

View File

@ -78,6 +78,10 @@ func (unstructuredJSONScheme) DecodeIntoWithSpecifiedVersionKind(data []byte, ob
return nil return nil
} }
func (unstructuredJSONScheme) DecodeToVersion(data []byte, version string) (Object, error) {
return nil, nil
}
func (unstructuredJSONScheme) DataVersionAndKind(data []byte) (version, kind string, err error) { func (unstructuredJSONScheme) DataVersionAndKind(data []byte) (version, kind string, err error) {
obj := TypeMeta{} obj := TypeMeta{}
if err := json.Unmarshal(data, &obj); err != nil { if err := json.Unmarshal(data, &obj); err != nil {