TEIIDDES-3044 (cntd)

* added pk/fk decorator on column image when applicable
Teiid-Designer · Jun 8, 2017 · 60e21e2 · 60e21e2
1 parent 35ec616
commit 60e21e2
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/...teiid.designer.metamodels.relational/icons/full/obj16/column-is-foreign-key.gif b/...teiid.designer.metamodels.relational/icons/full/obj16/column-is-foreign-key.gif
diff --git a/...ins/org.teiid.designer.metamodels.relational/icons/full/obj16/column-is-key.gif b/...ins/org.teiid.designer.metamodels.relational/icons/full/obj16/column-is-key.gif
diff --git a/....relational/src/org/teiid/designer/metamodels/relational/provider/ColumnItemProvider.java b/....relational/src/org/teiid/designer/metamodels/relational/provider/ColumnItemProvider.java
@@ -1111,6 +1111,15 @@ protected void addAccessPatternsPropertyDescriptorGen( Object object ) { // NO_U
      */
     @Override
     public Object getImage( Object object ) {
+    	// check the column's reference to a primary key value?
+    	final Column column = (Column)object;
+
+    	if( column.getUniqueKeys().size() > 0 ) {
+    		return getResourceLocator().getImage("full/obj16/column-is-key"); //$NON-NLS-1$
+    	} else if( column.getForeignKeys().size() > 0 ) {
+    		return getResourceLocator().getImage("full/obj16/column-is-foreign-key"); //$NON-NLS-1$
+    	}
+
         return getResourceLocator().getImage("full/obj16/Column"); //$NON-NLS-1$
     }