fix #1246, for case 3

AllenFang · May 8, 2017 · 7104c6b · 7104c6b
1 parent b6321fa
commit 7104c6b
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/TableEditColumn.js b/src/TableEditColumn.js
@@ -10,6 +10,7 @@ class TableEditColumn extends Component {
     super(props);
     this.timeouteClear = 0;
     const { fieldValue, row, className } = this.props;
+    this.focusInEditor = this.focusInEditor.bind(this);
     this.state = {
       shakeEditor: false,
       className: typeof className === 'function' ? className(fieldValue, row) : className
@@ -74,7 +75,6 @@ class TableEditColumn extends Component {
     const ts = this;
     let valid = true;
     if (ts.props.editable.validator) {
-      const input = ts.refs.inputRef;
       const checkVal = ts.props.editable.validator(value, this.props.row);
       const responseType = typeof checkVal;
       if (responseType !== 'object' && checkVal !== true) {
@@ -97,7 +97,7 @@ class TableEditColumn extends Component {
         ts.timeouteClear = setTimeout(() => {
           ts.setState({ shakeEditor: false });
         }, 300);
-        input.focus();
+        this.focusInEditor();
         return valid;
       }
     }
@@ -124,7 +124,7 @@ class TableEditColumn extends Component {
   }
 
   componentDidMount() {
-    this.refs.inputRef.focus();
+    this.focusInEditor();
     const dom = ReactDOM.findDOMNode(this);
     if (this.props.isFocus) {
       dom.focus();
@@ -146,6 +146,12 @@ class TableEditColumn extends Component {
     this.clearTimeout();
   }
 
+  focusInEditor() {
+    if (typeof this.refs.inputRef.focus === 'function') {
+      this.refs.inputRef.focus();
+    }
+  }
+
   handleClick = e => {
     if (e.target.tagName !== 'TD') {
       e.stopPropagation();