添加和删除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的区别。