react高阶组件添加和删除props

唠叨几句啦

在看程墨老师的深入浅出高阶组件，开头一点提了一个需要，创建两个高阶组件，一个能给传入的元素自定义添加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的区别。