Merged in bugfix/remove-modal (pull request #310)

SHIBUI Fixed issue with removing drafts
TIER · Feb 28, 2019 · 44490e8 · 44490e8
2 parents 6c05a36 + 93b5a81
commit 44490e8
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/ui/src/app/metadata/resolver/effect/draft-collection.effects.ts b/ui/src/app/metadata/resolver/effect/draft-collection.effects.ts
@@ -104,11 +104,15 @@ export class DraftCollectionEffects {
     removeDraft$ = this.actions$.pipe(
         ofType<actions.RemoveDraftRequest>(DraftActionTypes.REMOVE_DRAFT),
         map(getPayload),
-            switchMap(provider => this.draftService.find(provider.id, 'id').pipe(
+        switchMap(provider => {
+            let hasEntityId = !!provider.entityId;
+            let prop = hasEntityId ? 'entityId' : 'id';
+            let val = hasEntityId ? provider.entityId : provider.id;
+            return this.draftService.find(val, prop).pipe(
                 switchMap(selected => this.draftService.remove(selected)),
                 map(p => new actions.RemoveDraftSuccess(p))
-            )
-        )
+            );
+        })
     );
     @Effect()
     removeDraftSuccessReload$ = this.actions$.pipe(