refactor: add pdkIdMap state and related mutations for improved data handling

Author: feynmx

packages/assets/icons/node/memory.svg

@@ -129,6 +129,9 @@ export function getNodeIconSrc(node) {
     }
     icon = map[node.externaltype]
   }
+  if (node.type === 'mem_cache') {
+    return getIcon('memory')
+  }
   return icon ? getIcon(icon) : null
 }
 

packages/business/src/shared/task.ts

@@ -174,6 +174,7 @@ export default {
       'setPdkDoubleActiveMap',
       'setPdkCapabilitiesMap',
       'setMaterializedViewVisible',
+      'setPdkIdMap',
     ]),
 
     ...mapActions('dataflow', ['addNodeAsync', 'updateDag', 'loadCustomNode']),
@@ -228,6 +229,15 @@ export default {
           setting: this.loadNodeHiddenSetting(node),
           totalData: this.loadNodeHiddenTotalData(node),
         }
+
+        if (
+          this.dataflow.syncType === 'shareCache' &&
+          node.databaseType &&
+          !node.attrs.pdkHash
+        ) {
+          node.attrs.pdkHash =
+            this.$store.state.dataflow.pdkIdMap[node.databaseType]
+        }
         this.addNode(node)
       })
 
@@ -2882,30 +2892,36 @@ export default {
       const doubleActiveMap = {}
       const propertiesMap = {}
       const capabilitiesMap = {}
+      const pdkIdMap = {}
 
-      databaseItems.forEach(({ properties, pdkHash, tags, capabilities }) => {
-        const nodeProperties = properties?.node
+      databaseItems.forEach(
+        ({ properties, pdkHash, pdkId, tags, capabilities }) => {
+          const nodeProperties = properties?.node
 
-        if (nodeProperties) {
-          propertiesMap[pdkHash] = nodeProperties
-        }
-        if (tags?.includes('schema-free')) {
-          tagsMap[pdkHash] = true
-        }
-        if (tags?.includes('doubleActive')) {
-          doubleActiveMap[pdkHash] = true
-        }
-        if (capabilities?.length) {
-          capabilitiesMap[pdkHash] = capabilities.reduce((map, item) => {
-            map[item.id] = item
-            return map
-          }, {})
-        }
-      })
+          pdkIdMap[pdkId] = pdkHash
+
+          if (nodeProperties) {
+            propertiesMap[pdkHash] = nodeProperties
+          }
+          if (tags?.includes('schema-free')) {
+            tagsMap[pdkHash] = true
+          }
+          if (tags?.includes('doubleActive')) {
+            doubleActiveMap[pdkHash] = true
+          }
+          if (capabilities?.length) {
+            capabilitiesMap[pdkHash] = capabilities.reduce((map, item) => {
+              map[item.id] = item
+              return map
+            }, {})
+          }
+        },
+      )
       this.setPdkPropertiesMap(propertiesMap)
       this.setPdkSchemaFreeMap(tagsMap)
       this.setPdkDoubleActiveMap(doubleActiveMap)
       this.setPdkCapabilitiesMap(capabilitiesMap)
+      this.setPdkIdMap(pdkIdMap)
     },
 
     getIsDataflow() {

packages/dag/src/mixins/editor.js

@@ -85,6 +85,7 @@ const getState = () => ({
   pdkSchemaFreeMap: {},
   pdkDoubleActiveMap: {},
   pdkCapabilitiesMap: {},
+  pdkIdMap: {},
   taskSaving: false,
   materializedViewVisible: false,
   schemaRefreshing: false,
@@ -1018,6 +1019,10 @@ const mutations = {
     state.pdkCapabilitiesMap = map
   },
 
+  setPdkIdMap(state, map) {
+    state.pdkIdMap = map
+  },
+
   toggleTaskSaving(state, flag = !state.taskSaving) {
     state.taskSaving = flag
   },