fix: #78

hemengke1997 · Aug 29, 2024 · a2659f6 · a2659f6
1 parent b947649
commit a2659f6
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 4 deletions.
diff --git a/docs/3.md b/docs/3.md
@@ -13,6 +13,7 @@
 
 - 新增 `refresh` 方法，用于手动刷新组件
 - 新增 `debounceWaitTime`，用于设置 resize窗口时 debounce 的等待时间
+- `resetColumns` 新增入参 `resetStorage` 选项，可以重置storage中的列宽度，默认为 `true`
 - 可给单独列设置以下选项
 ```tsx
 type ResizableConfig = {

diff --git a/playground/app/routes/_layout/basic/index.tsx b/playground/app/routes/_layout/basic/index.tsx
@@ -103,6 +103,10 @@ export function Component() {
 
   const { components, resizableColumns, tableWidth, resetColumns } = useAntdResizableHeader({
     columns: useMemo(() => columns, [times]),
+    columnsState: {
+      persistenceKey: 'basic',
+      persistenceType: 'sessionStorage',
+    },
   })
 
   return (

diff --git a/src/hooks/useLocalColumns.ts b/src/hooks/useLocalColumns.ts
@@ -102,8 +102,8 @@ function useLocalColumns<T extends ResizableColumnType>({
   }, [resizableColumns])
 
   // reset
-  const resetLocalColumns = useMemoizedFn(() => {
-    setLocalColumns([...(columnsProp || [])])
+  const resetLocalColumns = useMemoizedFn((resetStorage: boolean = true) => {
+    setLocalColumns([...((resetStorage ? columns : columnsProp) || [])])
   })
 
   return {

diff --git a/src/useAntdResizableHeader.tsx b/src/useAntdResizableHeader.tsx
@@ -46,9 +46,9 @@ function useAntdResizableHeader<ColumnType extends ResizableColumnType = Resizab
 
   const [shouldRender, forceRender] = useReducer((s) => s + 1, 0)
 
-  const resetColumns = useMemoizedFn(() => {
+  const resetColumns = useMemoizedFn((resetStorage: boolean = true) => {
     widthCache.current = new Map()
-    resetLocalColumns()
+    resetLocalColumns(resetStorage)
   })
 
   const _onResizeEnd = useMemoizedFn((id?: string | number) => (width: number) => {