WIP

nucleus/device.go

@@ -6,6 +6,18 @@ import (
 	"github.com/openconfig/ygot/ygot"
 )
 
+type DeviceIdentifier interface {
+	IsDeviceIdentifier()
+}
+
+type StringIdentifier string
+
+func (si StringIdentifier) IsDeviceIdentifier() {}
+
+type UUIDIdentifier uuid.UUID
+
+func (ui UUIDIdentifier) IsDeviceIdentifier() {}
+
 // Device represents an Orchestrated Network Device (OND) which is managed by
 // nucleus
 type Device struct {

nucleus/principalNetworkDomain.go

@@ -223,6 +223,20 @@ func (pnd *pndImplementation) addDevice(device *Device) error {
 	return nil
 }
 
+func (pnd *pndImplementation) getDevice(identifier DeviceIdentifier) (*Device, error) {
+	stringIdentifier, ok := identifier.(StringIdentifier)
+	if ok {
+		return pnd.getDeviceByName(string(stringIdentifier))
+	}
+
+	uuidIdentifier, ok := identifier.(UUIDIdentifier)
+	if ok {
+		return pnd.getDeviceByUUID(uuid.UUID(uuidIdentifier))
+	}
+
+	return nil, &ErrNotFound{id: identifier}
+}
+
 func (pnd *pndImplementation) getDeviceByUUID(id uuid.UUID) (*Device, error) {
 	return pnd.devices.get(id)
 }