core: allow to customize GetOwnerID

plgd-dev · Jun 9, 2022 · c1f49e4 · c1f49e4
1 parent e287cc7
commit c1f49e4
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 7 deletions.
diff --git a/client/core/device.go b/client/core/device.go
@@ -16,12 +16,13 @@ import (
 )
 
 type DeviceConfiguration struct {
-	DialDTLS  DialDTLS
-	DialTLS   DialTLS
-	DialUDP   DialUDP
-	DialTCP   DialTCP
-	ErrFunc   ErrFunc
-	TLSConfig *TLSConfig
+	DialDTLS   DialDTLS
+	DialTLS    DialTLS
+	DialUDP    DialUDP
+	DialTCP    DialTCP
+	ErrFunc    ErrFunc
+	TLSConfig  *TLSConfig
+	GetOwnerID func() (string, error)
 }
 
 type Device struct {
@@ -53,6 +54,7 @@ func NewDevice(
 	deviceTypes []string,
 	getEndpoints func() schema.Endpoints,
 ) *Device {
+
 	return &Device{
 		cfg:          cfg,
 		deviceID:     deviceID,

diff --git a/client/core/getSdkOwnerID.go b/client/core/getSdkOwnerID.go
@@ -43,8 +43,12 @@ func (c *Client) GetSdkOwnerID() (string, error) {
 	return id, nil
 }
 
-// GetSdkOwnerID returns sdk ownerID from sdk identity certificate.
+// GetSdkOwnerID returns sdk ownerID
 func (d *Device) GetSdkOwnerID() (string, error) {
+	if d.cfg.GetOwnerID != nil {
+		return d.cfg.GetOwnerID()
+	}
+
 	id, err := getSdkOwnerID(d.cfg.TLSConfig.GetCertificate)
 	if err != nil {
 		return "", fmt.Errorf("cannot get sdk id: %w", err)