SHIBUI Fixed issue with removing drafts

TIER · Feb 28, 2019 · 93b5a81 · 93b5a81
1 parent 0b8b92b
commit 93b5a81
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(