diff --git a/packages/g6/src/behaviors/brush-select.ts b/packages/g6/src/behaviors/brush-select.ts
index 96ce5ecb094..775dd438819 100644
--- a/packages/g6/src/behaviors/brush-select.ts
+++ b/packages/g6/src/behaviors/brush-select.ts
@@ -90,9 +90,8 @@ export interface BrushSelectOptions extends BaseBehaviorOptions {
*
* Callback when brush select elements.
* @param states - 选中的元素状态
- * @returns 选中的元素状态
*/
- onSelect?: (states: Record) => Record;
+ onSelect?: (states: Record) => void;
}
/**
* 框选一组元素
@@ -246,7 +245,7 @@ export class BrushSelect extends BaseBehavior {
const selectedIds = this.selector(graph, points, enableElements);
- let states: Record = {};
+ const states: Record = {};
switch (mode) {
case 'union':
@@ -274,7 +273,7 @@ export class BrushSelect extends BaseBehavior {
break;
}
- if (isFunction(onSelect)) states = onSelect(states);
+ if (isFunction(onSelect)) onSelect(states);
graph.setElementState(states, this.options.animation);
}