feat: add auto-increment for raw table type (#929)

Author: walborn

src/containers/Tenant/Schema/SchemaViewer/SchemaViewer.tsx

@@ -54,6 +54,10 @@ export const SchemaViewer = ({type, path, tenantName, extended = false}: SchemaV
         return prepareSchemaData(type, currentObjectData);
     }, [currentObjectData, type, viewColumnsData]);
 
+    const hasAutoIncrement = React.useMemo(() => {
+        return tableData.some((i) => i.autoIncrement);
+    }, [tableData]);
+
     const columns = React.useMemo(() => {
         if (isViewType(type)) {
             return getViewColumns();
@@ -65,11 +69,11 @@ export const SchemaViewer = ({type, path, tenantName, extended = false}: SchemaV
             return getColumnTableColumns();
         }
         if (isRowTableType(type)) {
-            return getRowTableColumns(extended);
+            return getRowTableColumns(extended, hasAutoIncrement);
         }
 
         return [];
-    }, [type, extended]);
+    }, [type, extended, hasAutoIncrement]);
 
     const renderContent = () => {
         if (loading || isViewSchemaLoading) {

src/containers/Tenant/Schema/SchemaViewer/columns.tsx

@@ -15,6 +15,7 @@ export const SCHEMA_TABLE_COLUMS_IDS = {
     isKeyColumn: 'isKeyColumn',
     type: 'type',
     notNull: 'notNull',
+    autoIncrement: 'autoIncrement',
     familyName: 'familyName',
     prefferedPoolKind: 'prefferedPoolKind',
     columnCodec: 'columnCodec',
@@ -78,6 +79,22 @@ const notNullColumn: SchemaColumn = {
         return undefined;
     },
 };
+const autoIncrementColumn: SchemaColumn = {
+    name: SCHEMA_TABLE_COLUMS_IDS.autoIncrement,
+    get header() {
+        return i18n('column-title.autoIncrement');
+    },
+    width: 100,
+    // Table should start with notNull columns on sort click
+    defaultOrder: DataTable.DESCENDING,
+    render: ({row}) => {
+        if (row.autoIncrement) {
+            return '\u2713';
+        }
+
+        return undefined;
+    },
+};
 const familyColumn: SchemaColumn = {
     name: SCHEMA_TABLE_COLUMS_IDS.familyName,
     get header() {
@@ -112,11 +129,14 @@ export function getExternalTableColumns(): SchemaColumn[] {
 export function getColumnTableColumns(): SchemaColumn[] {
     return [idColumn, keyColumn, nameColumn, typeColumn, notNullColumn];
 }
-export function getRowTableColumns(extended: boolean): SchemaColumn[] {
+export function getRowTableColumns(extended: boolean, hasAutoIncrement: boolean): SchemaColumn[] {
     const rowTableColumns = [idColumn, keyColumn, nameColumn, typeColumn, notNullColumn];
 
     if (extended) {
-        return rowTableColumns.concat(familyColumn, mediaColumn, compressionColumn);
+        rowTableColumns.push(familyColumn, mediaColumn, compressionColumn);
+    }
+    if (hasAutoIncrement) {
+        rowTableColumns.push(autoIncrementColumn);
     }
 
     return rowTableColumns;

src/containers/Tenant/Schema/SchemaViewer/i18n/en.json

@@ -4,6 +4,7 @@
   "column-title.key": "Key",
   "column-title.type": "Type",
   "column-title.notNull": "NotNull",
+  "column-title.autoIncrement": "AutoIncrement",
   "column-title.family": "Family",
   "column-title.media": "Media",
   "column-title.compression": "Compression"

src/containers/Tenant/Schema/SchemaViewer/prepareData.ts

@@ -61,7 +61,7 @@ function prepareRowTableSchema(data: TTableDescription = {}): SchemaData[] {
     const keyAccessorsMap = getKeyColumnsSortAccessorMap(KeyColumnIds);
 
     const preparedColumns = Columns?.map((column) => {
-        const {Id, Name, NotNull, Type, Family} = column;
+        const {Id, Name, NotNull, Type, Family, DefaultFromSequence} = column;
 
         const isKeyColumn = Boolean(KeyColumnIds?.find((keyColumnId) => keyColumnId === Id));
         // Values in keyAccessorsMap are always negative, so it will be 1 for not key columns
@@ -80,6 +80,7 @@ function prepareRowTableSchema(data: TTableDescription = {}): SchemaData[] {
             keyAccessor,
             type: Type,
             notNull: NotNull,
+            autoIncrement: Boolean(DefaultFromSequence),
             familyName,
             prefferedPoolKind,
             columnCodec,

src/containers/Tenant/Schema/SchemaViewer/types.ts

@@ -8,6 +8,7 @@ export interface SchemaData {
     keyAccessor?: number;
     type?: string;
     notNull?: boolean;
+    autoIncrement?: boolean;
     familyName?: string;
     prefferedPoolKind?: string;
     columnCodec?: string;