添加和删除props的react高阶组件

唠叨几句啦

在看程墨老师的深入浅出高阶组件，开头一点提了一个需要，创建两个高阶组件，一个能给传入的元素自定义添加 props，一个是删除特定的 props。我刚刚做了一下，发现高阶组件需要区分好传入的是 class 还是 react element, 同时也需要注意好 return 回去的是啥。顺便提一下高阶组件的概念，就说一个函数，能够接受一个组件作为参数，然后返回的时候，这个组件就带有这个高阶组件给的某些特性。我理解就跟掉泥坑了，得带点土出来一个道理。

对比一下两个组件，贴代码时刻来啦

删除属性的高阶组件
我们需要传入任何组件和参数，都把 user 参数给删除了，所以返回值是一个接收 props 属性的函数。

import React from "react"

function removeUserProp(WrappedComponent) {return function newRender(props) {const {user, ...otherProps} = props;// 删除 user 这个属性值
        return <WrappedComponent {...otherProps} />
    }
}

export default removeUserProp

使用的时候

 const RemoveComponent = removeUserProp(reactComponentClass)({user: "aa"});// 这里返回的是一个 react component
 render () {
    return <div>
                {RemoveComponent}
           </div>
 }

增加属性的高阶组件

import React from "react"

const addNewProps = function (WrappedComponent, newProps) {// 接收的是一个 class 作为参数，返回一个 class
    return class WrappingComponent extends React.Component {render () {return <WrappedComponent {...this.props} {...newProps}/>
        } 
    }
}

export default addNewProps

使用的时候, 返回值是 class，所以要用 <ReactClassName/> 转换成可以渲染的 react 组件

    const AddUserComponent = addNewProps(SampleComponent, {user: "aa"})
    render () {return <AddUserComponent />}

完整的使用的例子代码：

import React from "react"
import addNewProps from './addNewProps'
import removeUserProp from './removeUserProp'

class SampleComponent extends React.Component {constructor(props) {console.log(props)
        super(props)
    }

    render () {console.log(this.props)
        return <div>
                {this.props.user ? <p> 哈哈哈 </p> : <p> 哈哈哈 2 </p>}     
                </div>
    }
}


class Test extends React.Component {render () {var obj = {aa: "aa"}
        const AddUserComponent = addNewProps(SampleComponent, {user: "aa"})
        const RemoveUserComponent = removeUserProp(SampleComponent)({user: "aa"})
      
        return <div>
                  <AddUserComponent />
                  {RemoveUserComponent}
                </div>
       
    }
}

export default Test

一点点小收获就是明白了高阶组件要看清楚输入输出。class 跟 react element 的区别。