Merge branch 'develop'

package.json

@@ -1,6 +1,6 @@
 {
   "private": true,
-  "version": "5.3.5-beta.3",
+  "version": "5.3.5-beta.6",
   "name": "dbgate-all",
   "workspaces": [
     "packages/*",

packages/datalib/src/CollectionGridDisplay.ts

@@ -66,6 +66,10 @@ function getDisplayColumn(basePath, columnName, display: CollectionGridDisplay)
     filterBehaviour: display?.driver?.getFilterBehaviour(null, standardFilterBehaviours) ?? mongoFilterBehaviour,
     pureName: display.collection?.pureName,
     schemaName: display.collection?.schemaName,
+
+    isPartitionKey: !!display?.collection?.partitionKey?.find(x => x.columnName == uniqueName),
+    isClusterKey: !!display?.collection?.clusterKey?.find(x => x.columnName == uniqueName),
+    isUniqueKey: !!display?.collection?.uniqueKey?.find(x => x.columnName == uniqueName),
   };
 }
 

packages/datalib/src/GridDisplay.ts

@@ -28,6 +28,12 @@ export interface DisplayColumn {
   notNull?: boolean;
   autoIncrement?: boolean;
   isPrimaryKey?: boolean;
+
+  // NoSQL specific
+  isPartitionKey?: boolean;
+  isClusterKey?: boolean;
+  isUniqueKey?: boolean;
+
   foreignKey?: ForeignKeyInfo;
   isForeignKeyUnique?: boolean;
   isExpandable?: boolean;

packages/web/src/datagrid/DataGridCore.svelte

@@ -559,7 +559,7 @@
         for (const column of display.columns) {
           if (column.uniquePath.length > 1) continue;
           if (column.autoIncrement) continue;
-          if (column.columnName == '_id' && isDynamicStructure) continue;
+          if (column.isClusterKey) continue;
 
           grider.setCellValue(rowIndex, column.uniqueName, grider.getRowData(index)[column.uniqueName]);
         }
@@ -959,7 +959,9 @@
 
   export async function mergeSelectionIntoMirror({ fullRows, mergeMode = 'merge' }) {
     const file = display.baseTableOrSimilar?.pureName;
-    const mergeKey = display.baseCollection ? ['_id'] : display.baseTable?.primaryKey.columns.map(x => x.columnName);
+    const mergeKey = display.baseCollection
+      ? display.baseCollection?.uniqueKey?.map(x => x.columnName)
+      : display.baseTable?.primaryKey.columns.map(x => x.columnName);
 
     const cells = cellsToRegularCells(selectedCells);
     const rowIndexes = _.sortBy(_.uniq(cells.map(x => x[0])));

plugins/dbgate-plugin-mongo/src/backend/Analyser.js

@@ -35,10 +35,14 @@ class Analyser extends DatabaseAnalyser {
           pureName: x.name,
           tableRowCount: stats[index]?.count,
           uniqueKey: [{ columnName: '_id' }],
+          partitionKey: [{ columnName: '_id' }],
+          clusterKey: [{ columnName: '_id' }],
         })),
         ...views.map((x, index) => ({
           pureName: x.name,
           uniqueKey: [{ columnName: '_id' }],
+          partitionKey: [{ columnName: '_id' }],
+          clusterKey: [{ columnName: '_id' }],
         })),
       ],
     });

plugins/dbgate-plugin-mongo/src/frontend/driver.js

@@ -112,8 +112,8 @@ const driver = {
 
   getCollectionExportQueryScript(collection, condition, sort) {
     return `db.collection('${collection}')
-      .find(${JSON.stringify(convertToMongoCondition(condition) || {})})
-      .sort(${JSON.stringify(convertToMongoSort(sort) || {})})`;
+  .find(${JSON.stringify(convertToMongoCondition(condition) || {})})
+  .sort(${JSON.stringify(convertToMongoSort(sort) || {})})`;
   },
   getCollectionExportQueryJson(collection, condition, sort) {
     return {