semi-design/packages/semi-ui/treeSelect/_story/CustomTrigger/index.jsx

import React from 'react';
import { TreeSelect, Button } from '../../../index';

export default function Demo() {
    const treeData = [
        {
            label: '亚洲',
            value: 'Asia',
            key: '0',
            children: [
                {
                    label: '中国',
                    value: 'China',
                    key: '0-0',
                    children: [
                        {
                            label: '北京',
                            value: 'Beijing',
                            key: '0-0-0',
                        },
                        {
                            label: '上海',
                            value: 'Shanghai',
                            key: '0-0-1',
                        },
                    ],
                },
            ],
        },
        {
            label: '北美洲',
            value: 'North America',
            key: '1',
        },
    ];
    return (
        <TreeSelect
            style={{ width: 300 }}
            dropdownStyle={{ maxHeight: 400, overflow: 'auto' }}
            treeData={treeData}
            placeholder="请选择"
            triggerRender={({ value, placeholder }) => (
                <Button block>{(value && value[0] && value[0].label) || placeholder}</Button>
            )}
        />
    );
}