diff --git a/gui/app/(dashboard)/database/hooks.ts b/gui/app/(dashboard)/database/hooks.ts index 27c36b1323..b60c54f72f 100644 --- a/gui/app/(dashboard)/database/hooks.ts +++ b/gui/app/(dashboard)/database/hooks.ts @@ -1,6 +1,7 @@ import { ITableColumns, ITableIndex } from '@/lib/databse-interface'; import { useRouter } from 'next/navigation'; import { useCallback, useEffect, useRef, useState } from 'react'; +import { INode } from 'react-accessible-treeview'; import { listDatabase, listTable, @@ -9,7 +10,7 @@ import { showTableSegments } from '../actions'; import { initialData } from './constants'; -import { DatabaseRouteParams, TreeNode, TreeParentId } from './interface'; +import { DatabaseRouteParams, TreeParentId } from './interface'; import { buildLeafData, getParentIdById, updateTreeData } from './utils'; export const useHandleClickTreeName = () => { @@ -25,7 +26,7 @@ export const useHandleClickTreeName = () => { level: number; name: string; parent: TreeParentId; - data: TreeNode[]; + data: INode[]; }) => () => { if (level === 3) { @@ -43,7 +44,7 @@ export const useHandleClickTreeName = () => { export const useBuildTreeData = () => { const loadedAlertElement = useRef(null); - const [data, setData] = useState(initialData); + const [data, setData] = useState(initialData); const [nodesAlreadyLoaded, setNodesAlreadyLoaded] = useState([]); const [loading, setLoading] = useState(true); @@ -76,7 +77,7 @@ export const useBuildTreeData = () => { const ret = await listTable(databaseName); if (ret?.tables?.length > 0) { setData((value) => { - const tablePropertyList: TreeNode[] = []; + const tablePropertyList: INode[] = []; const tableList = ret.tables.map((x: string) => { const leafs = buildLeafData(x); tablePropertyList.push(...leafs); @@ -95,6 +96,16 @@ export const useBuildTreeData = () => { ...tablePropertyList ]; }); + } else { + setData((value) => + value.map((x) => { + let metadata = x.metadata ?? {}; + if (x.id === databaseName) { + metadata['isEmpty'] = true; + } + return { ...x, metadata }; + }) + ); } }, []); @@ -102,7 +113,7 @@ export const useBuildTreeData = () => { fetchDatabases(); }, [fetchDatabases]); - const onLoadData = async ({ element }: { element: TreeNode }) => { + const onLoadData = async ({ element }: { element: INode }) => { if (element.children.length > 0) { return; } diff --git a/gui/app/(dashboard)/database/interface.ts b/gui/app/(dashboard)/database/interface.ts index b5ed168aca..06a01b4bc4 100644 --- a/gui/app/(dashboard)/database/interface.ts +++ b/gui/app/(dashboard)/database/interface.ts @@ -2,14 +2,6 @@ import { Leaf } from './constants'; export type TreeParentId = string | number | null; -export interface TreeNode { - name: string; - id: string | number; - children: Array; - parent: TreeParentId; - isBranch?: boolean; -} - export interface DatabaseRouteParams { params: { databaseId: string; tableId: string }; searchParams: { tab: Leaf }; diff --git a/gui/app/(dashboard)/database/tree.tsx b/gui/app/(dashboard)/database/tree.tsx index f7c61ae2f4..05eed9cd5e 100644 --- a/gui/app/(dashboard)/database/tree.tsx +++ b/gui/app/(dashboard)/database/tree.tsx @@ -1,7 +1,7 @@ 'use client'; import cx from 'classnames'; -import TreeView from 'react-accessible-treeview'; +import TreeView, { INode } from 'react-accessible-treeview'; import { AiOutlineLoading } from 'react-icons/ai'; import { IoMdArrowDropright } from 'react-icons/io'; import { Leaf, LeafIconMap } from './constants'; @@ -56,8 +56,10 @@ function AsyncTree() { handleSelect, handleExpand }) => { - const branchNode = (isExpanded: any, element: any) => { - return isExpanded && element.children.length === 0 ? ( + const branchNode = (isExpanded: boolean, element: INode) => { + return isExpanded && + !element.metadata?.isEmpty && + element.children.length === 0 ? ( <> + children: Array ) => { const data = list.map((node) => { if (node.id === id) { @@ -40,6 +41,6 @@ export const buildLeafData = (parent: string) => { ]; }; -export const getParentIdById = (data: TreeNode[], id: TreeParentId) => { +export const getParentIdById = (data: INode[], id: TreeParentId) => { return data.find((x) => x.id === id)?.parent; };