From 41b0e678c54c0e92270a088e3bb38e9097e41e06 Mon Sep 17 00:00:00 2001
From: simplejason <tangjie@yinspect.com>
Date: Thu, 28 Feb 2019 19:18:24 +0800
Subject: [PATCH] fix(module: tree): fix remove and clearChildren missing
 origin (#2995)

---
 components/tree/nz-tree-node.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/components/tree/nz-tree-node.ts b/components/tree/nz-tree-node.ts
index 6c37b8e17a..dcdb8467cb 100644
--- a/components/tree/nz-tree-node.ts
+++ b/components/tree/nz-tree-node.ts
@@ -194,6 +194,7 @@ export class NzTreeNode {
       this.treeService.afterRemove(n, false);
     });
     this.getChildren().splice(0, this.getChildren().length);
+    this.origin.children = [];
     // refresh checked state
     this.treeService.calcCheckedKeys(this.treeService.checkedNodeList.map(v => v.key), this.treeService.rootNodes, this.treeService.isCheckStrictly);
     if (this.component) {
@@ -205,6 +206,7 @@ export class NzTreeNode {
     if (this.getParentNode()) {
       const index = this.getParentNode().getChildren().findIndex(n => n.key === this.key);
       this.getParentNode().getChildren().splice(index, 1);
+      this.getParentNode().origin.children.splice(index, 1);
       this.treeService.afterRemove(this);
       if (this.component) {
         this.component.markForCheck();