Implement RPC-related methods

Author: dalbani

registry/interface.go

@@ -39,8 +39,8 @@ type Registry interface {
 	SetMachineState(ms machine.MachineState, ttl time.Duration) (uint64, error)
 	MachineState(machID string) (machine.MachineState, error)
 	UnscheduleUnit(name, machID string) error
-	SetMachineMetadata(machID, key, value string) error
-	DeleteMachineMetadata(machID, key string) error
+	SetMachineMetadata(machID string, key string, value string) error
+	DeleteMachineMetadata(machID string, key string) error
 
 	IsRegistryReady() bool
 	UseEtcdRegistry() bool

registry/rpc/registrymux.go

@@ -290,6 +290,10 @@ func (r *RegistryMux) SetUnitTargetState(name string, state job.JobState) error
 	return r.getRegistry().SetUnitTargetState(name, state)
 }
 
+func (r *RegistryMux) MachineState(machID string) (machine.MachineState, error) {
+	return r.getRegistry().MachineState(machID)
+}
+
 func (r *RegistryMux) SetMachineState(ms machine.MachineState, ttl time.Duration) (uint64, error) {
 	return r.etcdRegistry.SetMachineState(ms, ttl)
 }
@@ -333,3 +337,11 @@ func (r *RegistryMux) EngineVersion() (int, error) {
 func (r *RegistryMux) UpdateEngineVersion(from int, to int) error {
 	return r.etcdRegistry.UpdateEngineVersion(from, to)
 }
+
+func (r *RegistryMux) SetMachineMetadata(machID string, key string, value string) error {
+	return r.etcdRegistry.SetMachineMetadata(machID, key, value)
+}
+
+func (r *RegistryMux) DeleteMachineMetadata(machID string, key string) error {
+	return r.etcdRegistry.DeleteMachineMetadata(machID, key)
+}

registry/rpc/rpcregistry.go

@@ -244,6 +244,14 @@ func (r *RPCRegistry) UnscheduleUnit(unitName, machID string) error {
 	return err
 }
 
+func (r *RPCRegistry) SetMachineMetadata(machID string, key string, value string) error {
+	panic("Set machine metadata function not implemented")
+}
+
+func (r *RPCRegistry) DeleteMachineMetadata(machID string, key string) error {
+	panic("Delete machine metadata function not implemented")
+}
+
 func (r *RPCRegistry) Machines() ([]machine.MachineState, error) {
 	panic("Machines function not implemented")
 }
@@ -252,6 +260,10 @@ func (r *RPCRegistry) SetMachineState(ms machine.MachineState, ttl time.Duration
 	panic("Set machine state function not implemented")
 }
 
+func (r *RPCRegistry) MachineState(machID string) (machine.MachineState, error) {
+	panic("Machine state function not implemented")
+}
+
 func (r *RPCRegistry) CreateMachineState(ms machine.MachineState, ttl time.Duration) (uint64, error) {
 	panic("Set machine state function not implemented")
 }