github/dbx
4
0 Fork 0
mirror of https://github.com/t8y2/dbx.git synced 2026-11-15 14:52:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(vector): add qdrant and milvus support

Browse Source
This commit is contained in:
t8y2 committed 2026-06-19 10:07:56 +08:00
1 parent fffdcb9db1
commit e27d34a34a
45 files changed
+1116 -29
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

apps/desktop/src/stores/connectionStore.ts
+40 -1
View File
@@ -293,6 +293,8 @@ export const useConnectionStore = defineStore("connection", () => {
mongodb: "MongoDB",
oracle: "Oracle",
elasticsearch: "Elasticsearch",
qdrant: "Qdrant",
milvus: "Milvus",
doris: "Doris",
starrocks: "StarRocks",
manticoresearch: "Manticore Search",
@@ -788,6 +790,8 @@ export const useConnectionStore = defineStore("connection", () => {
await loadMongoDatabases(connectionId);
} else if (config.db_type === "elasticsearch") {
await loadElasticsearchIndices(connectionId);
} else if (config.db_type === "qdrant" || config.db_type === "milvus") {
await loadVectorCollections(connectionId);
} else if (config.db_type === "mq") {
await loadMqTenants(connectionId, { force: true });
} else {
@@ -1198,7 +1202,7 @@ export const useConnectionStore = defineStore("connection", () => {
withSavedSqlRoot(
connectionId,
sortSidebarNames(indices).map((index) => ({
id: `${connectionId}:__es_index:${index}`,
id: `${connectionId}:__collection:${index}`,
label: index,
type: "elasticsearch-index" as const,
connectionId,
@@ -1217,6 +1221,38 @@ export const useConnectionStore = defineStore("connection", () => {
}
}
async function loadVectorCollections(connectionId: string) {
const node = findNode(treeNodes.value, connectionId);
if (!node) return;
node.isLoading = true;
try {
await ensureConnected(connectionId);
const collections = await api.vectorListCollections(connectionId);
setChildren(
node,
withSavedSqlRoot(
connectionId,
sortSidebarNames(collections).map((collection) => ({
id: `${connectionId}:__vector_collection:${collection}`,
label: collection,
type: "vector-collection" as const,
connectionId,
database: "default",
isExpanded: false,
})),
node,
),
);
node.isExpanded = true;
} catch (e) {
recordMetadataLoadError(connectionId, e);
throw e;
} finally {
node.isLoading = false;
}
}
async function loadMongoCollections(connectionId: string, database: string) {
const nodeId = `${connectionId}:${database}`;
const node = findNode(treeNodes.value, nodeId);
@@ -1822,6 +1858,8 @@ export const useConnectionStore = defineStore("connection", () => {
await loadMongoDatabases(node.connectionId);
} else if (config?.db_type === "elasticsearch") {
await loadElasticsearchIndices(node.connectionId);
} else if (config?.db_type === "qdrant" || config?.db_type === "milvus") {
await loadVectorCollections(node.connectionId);
} else if (config?.db_type === "mq") {
await loadMqTenants(node.connectionId, options);
} else {
@@ -2967,6 +3005,7 @@ export const useConnectionStore = defineStore("connection", () => {
updateRedisDbKeyStats,
loadMongoDatabases,
loadElasticsearchIndices,
loadVectorCollections,
loadMongoCollections,
loadSchemas,
loadSqlServerDatabaseObjects,