
etcd_tools.go is not dependent on the specific implementation (which is provided by pkg/api). All EtcdHelpers are created with an encoding object which handles Encode/Decode/DecodeInto. Additional tests added to verify simple atomic flows. Begins to break up api singleton pattern.
129 lines
3.2 KiB
Go
129 lines
3.2 KiB
Go
/*
|
|
Copyright 2014 Google Inc. All rights reserved.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
)
|
|
|
|
// versionedJSONBase allows access to the version state of a JSONBase object
|
|
type JSONBaseVersioning struct{}
|
|
|
|
func (v JSONBaseVersioning) ResourceVersion(obj interface{}) (uint64, error) {
|
|
json, err := FindJSONBaseRO(obj)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return json.ResourceVersion, nil
|
|
}
|
|
|
|
func (v JSONBaseVersioning) SetResourceVersion(obj interface{}, version uint64) error {
|
|
json, err := FindJSONBase(obj)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
json.SetResourceVersion(version)
|
|
return nil
|
|
}
|
|
|
|
// JSONBase lets you work with a JSONBase from any of the versioned or
|
|
// internal APIObjects.
|
|
type JSONBaseInterface interface {
|
|
APIVersion() string
|
|
SetAPIVersion(version string)
|
|
Kind() string
|
|
SetKind(kind string)
|
|
ResourceVersion() uint64
|
|
SetResourceVersion(version uint64)
|
|
}
|
|
|
|
type genericJSONBase struct {
|
|
apiVersion *string
|
|
kind *string
|
|
resourceVersion *uint64
|
|
}
|
|
|
|
func (g genericJSONBase) APIVersion() string {
|
|
return *g.apiVersion
|
|
}
|
|
|
|
func (g genericJSONBase) SetAPIVersion(version string) {
|
|
*g.apiVersion = version
|
|
}
|
|
|
|
func (g genericJSONBase) Kind() string {
|
|
return *g.kind
|
|
}
|
|
|
|
func (g genericJSONBase) SetKind(kind string) {
|
|
*g.kind = kind
|
|
}
|
|
|
|
func (g genericJSONBase) ResourceVersion() uint64 {
|
|
return *g.resourceVersion
|
|
}
|
|
|
|
func (g genericJSONBase) SetResourceVersion(version uint64) {
|
|
*g.resourceVersion = version
|
|
}
|
|
|
|
// fieldPtr puts the address address of fieldName, which must be a member of v,
|
|
// into dest, which must be an address of a variable to which this field's address
|
|
// can be assigned.
|
|
func fieldPtr(v reflect.Value, fieldName string, dest interface{}) error {
|
|
field := v.FieldByName(fieldName)
|
|
if !field.IsValid() {
|
|
return fmt.Errorf("Couldn't find %v field in %#v", fieldName, v.Interface())
|
|
}
|
|
v = reflect.ValueOf(dest)
|
|
if v.Kind() != reflect.Ptr {
|
|
return fmt.Errorf("dest should be ptr")
|
|
}
|
|
v = v.Elem()
|
|
field = field.Addr()
|
|
if field.Type().AssignableTo(v.Type()) {
|
|
v.Set(field)
|
|
return nil
|
|
}
|
|
if field.Type().ConvertibleTo(v.Type()) {
|
|
v.Set(field.Convert(v.Type()))
|
|
return nil
|
|
}
|
|
return fmt.Errorf("Couldn't assign/convert %v to %v", field.Type(), v.Type())
|
|
}
|
|
|
|
// newGenericJSONBase makes a new generic JSONBase from v, which must be an
|
|
// addressable/setable reflect.Value having the same fields as api.JSONBase.
|
|
// Returns an error if this isn't the case.
|
|
func newGenericJSONBase(v reflect.Value) (genericJSONBase, error) {
|
|
g := genericJSONBase{}
|
|
err := fieldPtr(v, "APIVersion", &g.apiVersion)
|
|
if err != nil {
|
|
return g, err
|
|
}
|
|
err = fieldPtr(v, "Kind", &g.kind)
|
|
if err != nil {
|
|
return g, err
|
|
}
|
|
err = fieldPtr(v, "ResourceVersion", &g.resourceVersion)
|
|
if err != nil {
|
|
return g, err
|
|
}
|
|
return g, nil
|
|
}
|