@@ -4,7 +4,7 @@ import useId from '@rc-component/util/lib/hooks/useId';
4
4
import type { IconType } from '@rc-component/tree/lib/interface' ;
5
5
import type { ExpandAction } from '@rc-component/tree/lib/Tree' ;
6
6
import { conductCheck } from '@rc-component/tree/lib/utils/conductUtil' ;
7
- import useMergedState from '@rc-component/util/lib/hooks/useMergedState ' ;
7
+ import useControlledState from '@rc-component/util/lib/hooks/useControlledState ' ;
8
8
import * as React from 'react' ;
9
9
import useCache from './hooks/useCache' ;
10
10
import useCheckedKeys from './hooks/useCheckedKeys' ;
@@ -224,7 +224,7 @@ const TreeSelect = React.forwardRef<BaseSelectRef, TreeSelectProps>((props, ref)
224
224
treeNodeFilterProp = 'value' ,
225
225
} = searchConfig ;
226
226
227
- const [ internalValue , setInternalValue ] = useMergedState ( defaultValue , { value } ) ;
227
+ const [ internalValue , setInternalValue ] = useControlledState ( defaultValue , value ) ;
228
228
229
229
// `multiple` && `!treeCheckable` should be show all
230
230
const mergedShowCheckedStrategy = React . useMemo ( ( ) => {
@@ -249,10 +249,8 @@ const TreeSelect = React.forwardRef<BaseSelectRef, TreeSelectProps>((props, ref)
249
249
) ;
250
250
251
251
// =========================== Search ===========================
252
- const [ mergedSearchValue , setSearchValue ] = useMergedState ( '' , {
253
- value : searchValue ,
254
- postState : search => search || '' ,
255
- } ) ;
252
+ const [ internalSearchValue , setSearchValue ] = useControlledState ( '' , searchValue ) ;
253
+ const mergedSearchValue = internalSearchValue || '' ;
256
254
257
255
const onInternalSearch = searchText => {
258
256
setSearchValue ( searchText ) ;
0 commit comments