Replace ApplicationResource with ConnectorResource (#9419)

1 files changed, 2 insertions, 13 deletions

diff --git a/server/src/com/vaadin/ui/AbstractEmbedded.java b/server/src/com/vaadin/ui/AbstractEmbedded.java
index f3256dcde6..d94f62120f 100644
--- a/server/src/com/vaadin/ui/AbstractEmbedded.java
+++ b/server/src/com/vaadin/ui/AbstractEmbedded.java
@@ -5,7 +5,6 @@
 package com.vaadin.ui;
 
 import com.vaadin.server.Resource;
-import com.vaadin.server.ResourceReference;
 import com.vaadin.shared.ui.AbstractEmbeddedState;
 
 /**
@@ -32,12 +31,7 @@ public abstract class AbstractEmbedded extends AbstractComponent {
      *            the source to set.
      */
     public void setSource(Resource source) {
-        if (source == null) {
-            getState().setSource(null);
-        } else {
-            getState().setSource(new ResourceReference(source));
-        }
-        requestRepaint();
+        setResource(AbstractEmbeddedState.SOURCE_RESOURCE, source);
     }
 
     /**
@@ -46,12 +40,7 @@ public abstract class AbstractEmbedded extends AbstractComponent {
      * @return the source
      */
     public Resource getSource() {
-        ResourceReference ref = ((ResourceReference) getState().getSource());
-        if (ref == null) {
-            return null;
-        } else {
-            return ref.getResource();
-        }
+        return getResource(AbstractEmbeddedState.SOURCE_RESOURCE);
     }
 
     /**