Change missing keys error handling

handler/manage_device_key.go

@@ -100,7 +100,9 @@ func (s *KeyStorageServerImpl) updateAll(ctx context.Context, uuid string, updat
 	for _, kType := range []api.KeyType{api.KeyType_PRIVATE, api.KeyType_PUBLIC, api.KeyType_CERTIFICATE} {
 		err := updateKey(client, uuid, kType, updateFunc)
 		if err != nil {
-			return err
+			if !errors.Wraps(err, keyutils.ErrNoData) {
+				return err
+			}
 		}
 	}
 

utils/utils.go

@@ -37,6 +37,8 @@ import (
 	"code.vereign.com/code/viam-apis/key-storage-agent/api"
 )
 
+var ErrNoData = errors.New("No data")
+
 func GenerateUnusedUUID(client *client.DataStorageClientImpl) (string, error) {
 	count := 0
 	for {
@@ -145,8 +147,7 @@ func GetKey(client *client.DataStorageClientImpl, uuid string, keyType api.KeyTy
 	}
 
 	if !hasData {
-		err = errors.New("No data")
-		return nil, errors.WrapUserFormat(err, "Could not get key %s", uuid+"/"+api.KeyType.String(keyType))
+		return nil, errors.WrapUserFormat(ErrNoData, "Could not get key %s", uuid+"/"+api.KeyType.String(keyType))
 	}
 
 	return key, nil