TEIIDDES-2980 now clearing mat view ext. props.

Teiid-Designer · Jan 5, 2017 · 88aff02 · 88aff02
1 parent 16dac2b
commit 88aff02
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/...ation.ui/src/org/teiid/designer/transformation/ui/actions/ClearMaterializationAction.java b/...ation.ui/src/org/teiid/designer/transformation/ui/actions/ClearMaterializationAction.java
@@ -12,7 +12,12 @@
 import org.teiid.designer.core.ModelerCore;
 import org.teiid.designer.core.workspace.ModelResource;
 import org.teiid.designer.core.workspace.ModelWorkspaceException;
+import org.teiid.designer.extension.ExtensionPlugin;
+import org.teiid.designer.extension.definition.ModelObjectExtensionAssistant;
+import org.teiid.designer.extension.registry.ModelExtensionRegistry;
 import org.teiid.designer.metamodels.relational.Table;
+import org.teiid.designer.metamodels.relational.extension.RelationalModelExtensionConstants;
+import org.teiid.designer.transformation.reverseeng.ReverseEngConstants;
 import org.teiid.designer.transformation.ui.UiPlugin;
 import org.teiid.designer.ui.UiConstants;
 import org.teiid.designer.ui.actions.SortableSelectionAction;
@@ -74,6 +79,13 @@ public void run() {
 				Table table = (Table)(SelectionUtilities.getSelectedEObject(getSelection()));
 				table.setMaterialized(false);
 				table.setMaterializedTable(null);
+
+				// Clear extension properties
+	            final ModelExtensionRegistry registry = ExtensionPlugin.getInstance().getRegistry();
+	            ModelObjectExtensionAssistant assistant =  (ModelObjectExtensionAssistant)registry.getModelExtensionAssistant(ReverseEngConstants.RELATIONAL_EXT_ASSISTANT_NS);
+	            assistant.removeProperty(table, RelationalModelExtensionConstants.PropertyIds.MATVIEW_AFTER_LOAD_SCRIPT);
+	            assistant.removeProperty(table, RelationalModelExtensionConstants.PropertyIds.MATVIEW_BEFORE_LOAD_SCRIPT);
+	            assistant.removeProperty(table, RelationalModelExtensionConstants.PropertyIds.MATERIALIZED_STAGE_TABLE);
 				succeeded = true;
 			}
 		} catch (Exception ex) {