AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-05-13 14:58:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: status query not stop when leaving document embedding detail page (#4754)

Browse Source
This commit is contained in:
doufa 2024-05-31 08:07:48 +00:00 committed by GitHub
parent 7aaa1ff270
commit b1c9671a60
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
web/app/components/datasets/documents/detail/embedding/index.tsx
View File

@ -126,17 +126,13 @@ const EmbeddingDetail: FC<Props> = ({ detail, stopPosition = 'top', datasetId: d
return status return status
} }
const [isStopQuery, setIsStopQuery] = useState(false) const isStopQuery = useRef(false)
const isStopQueryRef = useRef(isStopQuery) const stopQueryStatus = useCallback(() => {
useEffect(() => { isStopQuery.current = true
isStopQueryRef.current = isStopQuery }, [])
}, [isStopQuery])
const stopQueryStatus = () => {
setIsStopQuery(true)
}
const startQueryStatus = async () => { const startQueryStatus = useCallback(async () => {
if (isStopQueryRef.current) if (isStopQuery.current)
return return
try { try {
@ -146,6 +142,7 @@ const EmbeddingDetail: FC<Props> = ({ detail, stopPosition = 'top', datasetId: d
detailUpdate() detailUpdate()
return return
} }
await sleep(2500) await sleep(2500)
await startQueryStatus() await startQueryStatus()
} }
@ -153,16 +150,15 @@ const EmbeddingDetail: FC<Props> = ({ detail, stopPosition = 'top', datasetId: d
await sleep(2500) await sleep(2500)
await startQueryStatus() await startQueryStatus()
} }
} }, [stopQueryStatus])
useEffect(() => { useEffect(() => {
setIsStopQuery(false) isStopQuery.current = false
startQueryStatus() startQueryStatus()
return () => { return () => {
stopQueryStatus() stopQueryStatus()
} }
// eslint-disable-next-line react-hooks/exhaustive-deps }, [startQueryStatus, stopQueryStatus])
}, [])
const { data: indexingEstimateDetail, error: indexingEstimateErr } = useSWR({ const { data: indexingEstimateDetail, error: indexingEstimateErr } = useSWR({
action: 'fetchIndexingEstimate', action: 'fetchIndexingEstimate',