chore: workflow sync with hash (#4250)

This commit is contained in:
zxhlyh
2024-05-10 14:48:20 +08:00
committed by GitHub
parent f49c99937c
commit a1ab87107b
8 changed files with 74 additions and 22 deletions

View File

@@ -7,6 +7,7 @@ import {
useWorkflowStore,
} from '../store'
import { BlockEnum } from '../types'
import { useWorkflowUpdate } from '../hooks'
import { useNodesReadOnly } from './use-workflow'
import { syncWorkflowDraft } from '@/service/workflow'
import { useFeaturesStore } from '@/app/components/base/features/hooks'
@@ -17,19 +18,23 @@ export const useNodesSyncDraft = () => {
const workflowStore = useWorkflowStore()
const featuresStore = useFeaturesStore()
const { getNodesReadOnly } = useNodesReadOnly()
const { handleRefreshWorkflowDraft } = useWorkflowUpdate()
const debouncedSyncWorkflowDraft = useStore(s => s.debouncedSyncWorkflowDraft)
const params = useParams()
const getPostParams = useCallback((appIdParams?: string) => {
const getPostParams = useCallback(() => {
const {
getNodes,
edges,
transform,
} = store.getState()
const [x, y, zoom] = transform
const appId = workflowStore.getState().appId
const {
appId,
syncWorkflowDraftHash,
} = workflowStore.getState()
if (appId || appIdParams) {
if (appId) {
const nodes = getNodes()
const hasStartNode = nodes.find(node => node.data.type === BlockEnum.Start)
@@ -54,7 +59,7 @@ export const useNodesSyncDraft = () => {
})
})
return {
url: `/apps/${appId || appIdParams}/workflows/draft`,
url: `/apps/${appId}/workflows/draft`,
params: {
graph: {
nodes: producedNodes,
@@ -75,6 +80,7 @@ export const useNodesSyncDraft = () => {
sensitive_word_avoidance: features.moderation,
file_upload: features.file,
},
hash: syncWorkflowDraftHash,
},
}
}
@@ -93,23 +99,38 @@ export const useNodesSyncDraft = () => {
}
}, [getPostParams, params.appId, getNodesReadOnly])
const doSyncWorkflowDraft = useCallback(async (appId?: string) => {
const doSyncWorkflowDraft = useCallback(async (notRefreshWhenSyncError?: boolean) => {
if (getNodesReadOnly())
return
const postParams = getPostParams(appId)
const postParams = getPostParams()
if (postParams) {
const res = await syncWorkflowDraft(postParams)
workflowStore.getState().setDraftUpdatedAt(res.updated_at)
const {
setSyncWorkflowDraftHash,
setDraftUpdatedAt,
} = workflowStore.getState()
try {
const res = await syncWorkflowDraft(postParams)
setSyncWorkflowDraftHash(res.hash)
setDraftUpdatedAt(res.updated_at)
}
catch (error: any) {
if (error && error.json && !error.bodyUsed) {
error.json().then((err: any) => {
if (err.code === 'draft_workflow_not_sync' && !notRefreshWhenSyncError)
handleRefreshWorkflowDraft()
})
}
}
}
}, [workflowStore, getPostParams, getNodesReadOnly])
}, [workflowStore, getPostParams, getNodesReadOnly, handleRefreshWorkflowDraft])
const handleSyncWorkflowDraft = useCallback((sync?: boolean, appId?: string) => {
const handleSyncWorkflowDraft = useCallback((sync?: boolean, notRefreshWhenSyncError?: boolean) => {
if (getNodesReadOnly())
return
if (sync)
doSyncWorkflowDraft(appId)
doSyncWorkflowDraft(notRefreshWhenSyncError)
else
debouncedSyncWorkflowDraft(doSyncWorkflowDraft)
}, [debouncedSyncWorkflowDraft, doSyncWorkflowDraft, getNodesReadOnly])