From 1351dfda27934ea20496ff95902bbaf30301bb08 Mon Sep 17 00:00:00 2001
From: Bruce An <bruce.an@marklogic.com>
Date: Fri, 1 Feb 2019 11:29:16 -0800
Subject: [PATCH] fixed primary key displaying properly when mapping

---
 .../app/shared/components/mappings/map-ui.component.html | 2 +-
 .../app/shared/components/mappings/map-ui.component.ts   | 9 +++++++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/quick-start/src/main/ui/app/shared/components/mappings/map-ui.component.html b/quick-start/src/main/ui/app/shared/components/mappings/map-ui.component.html
index 5c99f7f39c..eb63d82cf4 100644
--- a/quick-start/src/main/ui/app/shared/components/mappings/map-ui.component.html
+++ b/quick-start/src/main/ui/app/shared/components/mappings/map-ui.component.html
@@ -121,7 +121,7 @@ <h1>
           <span class="prop-name">{{prop.name}}</span>
           <span class="prop-type">{{prop.datatype}}</span>
           <span class="entity-icons">
-            <span class="entity-icon" tooltip="Primary Key" container="body"><i class="fa fa-key fa-fw" [ngClass]="prop.name === entityPrimaryKey ? 'entity-icon-selected' : 'icon-hide'"></i></span>
+            <span class="entity-icon" tooltip="Primary Key" container="body"><i class="fa fa-key fa-fw" [ngClass]="isPrimaryKey(prop.name) ? 'entity-icon-selected' : 'icon-hide'"></i></span>
             <span class="entity-icon" tooltip="Element Range Index" container="body"><i class="fa fa-bolt fa-fw" [ngClass]="hasElementRangeIndex(prop.name) ? 'entity-icon-selected' : 'icon-hide'"></i></span>
             <span class="entity-icon" tooltip="Path Range Index" container="body"><i class="fa fa-code fa-fw" [ngClass]="hasRangeIndex(prop.name) ? 'entity-icon-selected' : 'icon-hide'"></i></span>
             <span class="entity-icon" tooltip="Word Lexicon" container="body"><i class="fa fa-krw fa-fw" [ngClass]="hasWordLexicon(prop.name) ? 'entity-icon-selected' : 'icon-hide'"></i></span>
diff --git a/quick-start/src/main/ui/app/shared/components/mappings/map-ui.component.ts b/quick-start/src/main/ui/app/shared/components/mappings/map-ui.component.ts
index 9dee1623c3..cf0fedf786 100644
--- a/quick-start/src/main/ui/app/shared/components/mappings/map-ui.component.ts
+++ b/quick-start/src/main/ui/app/shared/components/mappings/map-ui.component.ts
@@ -357,4 +357,13 @@ export class MapUiComponent implements OnChanges {
     return _.includes(this.chosenEntity.definition.pii, name);
   }
 
+      /**
+   * Is an entity property the primary key?
+   * @param name Name of property
+   * @returns {boolean}
+   */
+  isPrimaryKey(name) {
+    return _.includes(this.chosenEntity.definition.primaryKey, name);
+  }
+
 }