Author: achabatar Date: 2007-12-07 09:21:12 -0500 (Fri, 07 Dec 2007) New Revision: 5223 Modified: trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/ui/veditor/editors/model/OrmDiagram.java Log: http://jira.jboss.com/jira/browse/JBIDE-1411 Modified: trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/ui/veditor/editors/model/OrmDiagram.java =================================================================== --- trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/ui/veditor/editors/model/OrmDiagram.java 2007-12-07 13:43:47 UTC (rev 5222) +++ trunk/hibernatetools/plugins/org.jboss.tools.hibernate.ui.veditor/src/org/jboss/tools/hibernate/ui/veditor/editors/model/OrmDiagram.java 2007-12-07 14:21:12 UTC (rev 5223) @@ -59,12 +59,14 @@ private Configuration configuration; private ConsoleConfiguration consoleConfiguration; private IJavaProject javaProject; + private String entityName; public static final String HIBERNATE_MAPPING_LAYOUT_FOLDER_NAME = "hibernateMapping"; public OrmDiagram(ConsoleConfiguration configuration, RootClass ioe, IJavaProject javaProject) { consoleConfiguration = configuration; this.configuration = configuration.getConfiguration(); ormElement = (RootClass)ioe; + entityName = ioe.getEntityName(); this.javaProject = javaProject; if (ormElement instanceof RootClass) { @@ -98,9 +100,16 @@ } public void refresh() { + RootClass newOrmElement = (RootClass) consoleConfiguration + .getConfiguration().getClassMapping(entityName); + if (ormElement.equals(newOrmElement)) return; + ormElement = newOrmElement; saveHelper(); getChildren().clear(); elements.clear(); + getOrCreatePersistentClass(ormElement, null); + expandModel(this); + load(); firePropertyChange(REFRESH, null, null); }