fix #845

AllenFang · Nov 21, 2016 · 07b8560 · 07b8560
1 parent 8aa21f5
commit 07b8560
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/TableHeaderColumn.js b/src/TableHeaderColumn.js
@@ -60,8 +60,10 @@ class TableHeaderColumn extends Component {
       );
     }
     case Const.FILTER_TYPE.CUSTOM: {
-      return this.props.filter.getElement(this.handleFilter,
+      const elm = this.props.filter.getElement(this.handleFilter,
           this.props.filter.customFilterParameters);
+
+      return React.cloneElement(elm, { ref: 'customFilter' });
     }
     }
   }
@@ -150,6 +152,10 @@ class TableHeaderColumn extends Component {
       this.refs.dateFilter.cleanFiltered();
       break;
     }
+    case Const.FILTER_TYPE.CUSTOM: {
+      this.refs.customFilter.cleanFiltered();
+      break;
+    }
     }
   }