提交 449ef2db authored 作者: 王鹏飞's avatar 王鹏飞

chore: update

上级 d2e4e5c7
import { useEffect, useState } from 'react'
import { useQuery } from '@tanstack/react-query'
import { useQuery, useQueryClient } from '@tanstack/react-query'
import { getUser, getMapList, getMyList, getMyField, getProcessProgress } from '@/api/base'
import { useUserStore } from '@/stores/user'
import { useMapStore } from '@/stores/map'
......@@ -111,7 +111,9 @@ export function useDataFieldQuery() {
// 进度查询
export function useProcessProgressQuery(params: { function_name: string }) {
const queryClient = useQueryClient()
const [enabled, setEnabled] = useState(false)
const query = useQuery({
queryKey: ['processProgress', params],
queryFn: () => {
......@@ -119,11 +121,18 @@ export function useProcessProgressQuery(params: { function_name: string }) {
},
select: (res) => res.data,
enabled,
refetchInterval: enabled ? 10000 : false,
refetchInterval: enabled ? 1000 : false,
staleTime: 0,
gcTime: 0,
placeholderData: (): any => {
return { data: { progress: -1, message: {} } }
},
})
const remove = () => {
queryClient.removeQueries({ queryKey: ['processProgress', params] })
}
// 开始轮询的方法
const start = () => {
setEnabled(true)
......@@ -133,7 +142,6 @@ export function useProcessProgressQuery(params: { function_name: string }) {
const stop = () => {
setEnabled(false)
}
console.log(query.data)
// 组件卸载时清理
useEffect(() => {
if (query.data?.result === 'SUCCESS' || query.data?.result === 'FAIL' || query.data?.result === null) {
......@@ -142,7 +150,7 @@ export function useProcessProgressQuery(params: { function_name: string }) {
return () => {
stop()
}
}, [query.data])
}, [query.data, enabled])
return { ...query, start, stop }
return { ...query, start, stop, remove }
}
......@@ -65,14 +65,12 @@ export default function ButtonModal() {
const rule = Form.useWatch('rule', form)
const rules = Form.useWatch('rules', form) || {}
const { mutate, isPending, progress, message } = useProcessData()
const { mutate, isPending, progress, message, remove } = useProcessData()
// 开始处理
const handleStart = () => {
form.validateFields().then((values) => {
const params = { ...values, fields: values.fields.join(','), rules: JSON.stringify(values.rules) }
mutate(params, {
onSuccess: handleClose,
})
mutate(params)
})
}
......@@ -80,7 +78,7 @@ export default function ButtonModal() {
const handleClose = () => {
setOpen(false)
setCurrent(0)
form.resetFields()
remove()
}
const steps = [
......
......@@ -7,7 +7,7 @@ import { useProcessProgressQuery } from '@/hooks/useQuery'
// 处理数据
export function useProcessData() {
const queryClient = useQueryClient()
const { data, start } = useProcessProgressQuery({ function_name: 'missing' })
const { data, start, remove } = useProcessProgressQuery({ function_name: 'missing' })
const query = useMutation({
mutationFn: (data: ProcessDataParams) => {
......@@ -19,5 +19,6 @@ export function useProcessData() {
queryClient.invalidateQueries({ queryKey: ['data'] })
},
})
return { ...query, progress: data?.progress ?? -1, message: data?.message ?? {} }
return { ...query, progress: data.progress, message: data.message, remove }
}
......@@ -220,7 +220,6 @@ export default function ButtonModal() {
setCurrent(0)
setStep(-1)
setConcatenatedFields([])
form.resetFields()
}
return (
......
......@@ -31,7 +31,7 @@ export default function ButtonModal() {
setCurrent(current + 1)
}
const { mutate, isPending, progress, message } = useProcessData()
const { mutate, isPending, progress, message, remove } = useProcessData()
// 开始处理
const handleStart = () => {
......@@ -46,8 +46,8 @@ export default function ButtonModal() {
const handleClose = () => {
setOpen(false)
setCurrent(0)
remove()
}
const [groupName, setGroupName] = useState('')
// 步骤定义
......
......@@ -7,7 +7,7 @@ import { useProcessProgressQuery } from '@/hooks/useQuery'
// 处理数据
export function useProcessData() {
const queryClient = useQueryClient()
const { data, start } = useProcessProgressQuery({ function_name: 'mapping' })
const { data, start, remove } = useProcessProgressQuery({ function_name: 'mapping' })
const query = useMutation({
mutationFn: (data: ProcessDataParams) => {
......@@ -19,5 +19,5 @@ export function useProcessData() {
queryClient.invalidateQueries({ queryKey: ['data'] })
},
})
return { ...query, progress: data?.progress ?? -1, message: data?.message ?? {} }
return { ...query, progress: data.progress, message: data.message, remove }
}
......@@ -103,7 +103,7 @@ export default function ButtonModal() {
setCurrent(current + 1)
}
const { mutate, isPending, progress, message } = useProcessData()
const { mutate, isPending, progress, message, remove } = useProcessData()
// 开始处理
const handleStart = () => {
......@@ -126,8 +126,8 @@ export default function ButtonModal() {
const handleClose = () => {
setOpen(false)
setCurrent(0)
remove()
}
const handleFunctionChange = (value: string) => {
form.setFieldsValue({ content: `${value}()` })
}
......
......@@ -7,7 +7,7 @@ import { useProcessProgressQuery } from '@/hooks/useQuery'
// 处理数据
export function useProcessData() {
const queryClient = useQueryClient()
const { data, start } = useProcessProgressQuery({ function_name: 'date' })
const { data, start, remove } = useProcessProgressQuery({ function_name: 'date' })
const query = useMutation({
mutationFn: (data: ProcessDataParams) => {
......@@ -19,5 +19,5 @@ export function useProcessData() {
queryClient.invalidateQueries({ queryKey: ['data'] })
},
})
return { ...query, progress: data?.progress ?? -1, message: data?.message ?? {} }
return { ...query, progress: data.progress, message: data.message, remove }
}
......@@ -27,7 +27,7 @@ export default function ButtonModal() {
setCurrent(current + 1)
}
const { mutate, isPending, progress, message } = useProcessData()
const { mutate, isPending, progress, message, remove } = useProcessData()
// 开始处理
const handleStart = () => {
......@@ -42,8 +42,8 @@ export default function ButtonModal() {
const handleClose = () => {
setOpen(false)
setCurrent(0)
remove()
}
// 步骤定义
const steps = [
{
......
......@@ -7,7 +7,7 @@ import { useProcessProgressQuery } from '@/hooks/useQuery'
// 处理数据
export function useProcessData() {
const queryClient = useQueryClient()
const { data, start } = useProcessProgressQuery({ function_name: 'anonymization' })
const { data, start, remove } = useProcessProgressQuery({ function_name: 'anonymization' })
const query = useMutation({
mutationFn: (data: ProcessDataParams) => {
......@@ -19,5 +19,5 @@ export function useProcessData() {
queryClient.invalidateQueries({ queryKey: ['data'] })
},
})
return { ...query, progress: data?.progress ?? -1, message: data?.message ?? {} }
return { ...query, progress: data.progress, message: data.message, remove }
}
......@@ -44,7 +44,7 @@ export default function ButtonModal() {
setCurrent(current + 1)
}
const { mutate, isPending, progress, message } = useProcessData()
const { mutate, isPending, progress, message, remove } = useProcessData()
// 开始处理
const handleStart = () => {
......@@ -69,8 +69,8 @@ export default function ButtonModal() {
const handleClose = () => {
setOpen(false)
setCurrent(0)
remove()
}
const [groupName, setGroupName] = useState('')
// 步骤定义
const steps = [
......
......@@ -7,7 +7,7 @@ import { useProcessProgressQuery } from '@/hooks/useQuery'
// 处理数据
export function useProcessData() {
const queryClient = useQueryClient()
const { data, start } = useProcessProgressQuery({ function_name: 'grouping' })
const { data, start, remove } = useProcessProgressQuery({ function_name: 'grouping' })
const query = useMutation({
mutationFn: (data: ProcessDataParams) => {
......@@ -19,5 +19,5 @@ export function useProcessData() {
queryClient.invalidateQueries({ queryKey: ['data'] })
},
})
return { ...query, progress: data?.progress ?? -1, message: data?.message ?? {} }
return { ...query, progress: data.progress, message: data.message, remove }
}
......@@ -43,7 +43,7 @@ export default function ButtonModal() {
setCurrent(current + 1)
}
const { mutate, isPending, progress, message } = useProcessData()
const { mutate, isPending, progress, message, remove } = useProcessData()
// 开始处理
const handleStart = () => {
......@@ -63,8 +63,8 @@ export default function ButtonModal() {
const handleClose = () => {
setOpen(false)
setCurrent(0)
remove()
}
const handleFunctionChange = (value: string) => {
form.setFieldsValue({ content: `${value}()` })
}
......
......@@ -7,7 +7,7 @@ import { useProcessProgressQuery } from '@/hooks/useQuery'
// 处理数据
export function useProcessData() {
const queryClient = useQueryClient()
const { data, start } = useProcessProgressQuery({ function_name: 'logic' })
const { data, start, remove } = useProcessProgressQuery({ function_name: 'logic' })
const query = useMutation({
mutationFn: (data: ProcessDataParams) => {
......@@ -19,5 +19,5 @@ export function useProcessData() {
queryClient.invalidateQueries({ queryKey: ['data'] })
},
})
return { ...query, progress: data?.progress ?? -1, message: data?.message ?? {} }
return { ...query, progress: data.progress, message: data.message, remove }
}
......@@ -45,7 +45,7 @@ export default function ButtonModal() {
setCurrent(current + 1)
}
const { mutate, isPending, progress, message } = useProcessData()
const { mutate, isPending, progress, message, remove } = useProcessData()
// 开始处理
const handleStart = () => {
......@@ -63,8 +63,8 @@ export default function ButtonModal() {
const handleClose = () => {
setOpen(false)
setCurrent(0)
remove()
}
// 步骤定义
const steps = [
{
......
......@@ -7,7 +7,7 @@ import { useProcessProgressQuery } from '@/hooks/useQuery'
// 处理数据
export function useProcessData() {
const queryClient = useQueryClient()
const { data, start } = useProcessProgressQuery({ function_name: 'mapping' })
const { data, start, remove } = useProcessProgressQuery({ function_name: 'mapping' })
const query = useMutation({
mutationFn: (data: ProcessDataParams) => {
......@@ -19,5 +19,5 @@ export function useProcessData() {
queryClient.invalidateQueries({ queryKey: ['data'] })
},
})
return { ...query, progress: data?.progress ?? -1, message: data?.message ?? {} }
return { ...query, progress: data.progress, message: data.message, remove }
}
......@@ -43,7 +43,7 @@ export default function ButtonModal() {
setCurrent(current + 1)
}
const { mutate, isPending, progress, message } = useProcessData()
const { mutate, isPending, progress, message, remove } = useProcessData()
// 开始处理
const handleStart = () => {
......@@ -63,8 +63,8 @@ export default function ButtonModal() {
const handleClose = () => {
setOpen(false)
setCurrent(0)
remove()
}
const handleFunctionChange = (value: string) => {
form.setFieldsValue({ content: `${value}()` })
}
......
......@@ -7,7 +7,7 @@ import { useProcessProgressQuery } from '@/hooks/useQuery'
// 处理数据
export function useProcessData() {
const queryClient = useQueryClient()
const { data, start } = useProcessProgressQuery({ function_name: 'numeric' })
const { data, start, remove } = useProcessProgressQuery({ function_name: 'numeric' })
const query = useMutation({
mutationFn: (data: ProcessDataParams) => {
......@@ -19,5 +19,5 @@ export function useProcessData() {
queryClient.invalidateQueries({ queryKey: ['data'] })
},
})
return { ...query, progress: data?.progress ?? -1, message: data?.message ?? {} }
return { ...query, progress: data.progress, message: data.message, remove }
}
......@@ -44,7 +44,7 @@ export default function ButtonModal() {
setCurrent(current + 1)
}
const { mutate, isPending, progress, message } = useProcessData()
const { mutate, isPending, progress, message, remove } = useProcessData()
// 开始处理
const handleStart = () => {
......@@ -64,6 +64,7 @@ export default function ButtonModal() {
const handleClose = () => {
setOpen(false)
setCurrent(0)
remove()
}
const handleFunctionChange = (value: string) => {
......
......@@ -7,7 +7,7 @@ import { useProcessProgressQuery } from '@/hooks/useQuery'
// 处理数据
export function useProcessData() {
const queryClient = useQueryClient()
const { data, start } = useProcessProgressQuery({ function_name: 'text' })
const { data, start, remove } = useProcessProgressQuery({ function_name: 'text' })
const query = useMutation({
mutationFn: (data: ProcessDataParams) => {
......@@ -19,5 +19,5 @@ export function useProcessData() {
queryClient.invalidateQueries({ queryKey: ['data'] })
},
})
return { ...query, progress: data?.progress ?? -1, message: data?.message ?? {} }
return { ...query, progress: data.progress, message: data.message, remove }
}
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论