【New Features】Edit table structure supports setting primary keys in columns

shanhexi · shanhexi · commit 96a635792622 · 2023-11-02T12:38:40.000+08:00
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -15,6 +15,7 @@
     "chmod",
     "CLOB",
     "datasource",
+    "Datetime",
     "DBAI",
     "DBHUB",
     "Dchat",
@@ -46,6 +47,7 @@
     "pgsql",
     "pnpm",
     "Popconfirm",
+    "Prec",
     "quickinput",
     "redownload",
     "remaininguses",
@@ -62,6 +64,7 @@
     "umirc",
     "USERANDPASSWORD",
     "uuidv",
+    "VARCHAR",
     "VIEWCOLUMN",
     "VIEWCOLUMNS",
     "wechat",
diff --git a/chat2db-client/src/blocks/DatabaseTableEditor/ColumnList/index.less b/chat2db-client/src/blocks/DatabaseTableEditor/ColumnList/index.less
@@ -85,6 +85,18 @@
 //   }
 // }
 
+.keyBox{
+  width: 26px;
+  height: 26px;
+  display: flex;
+  justify-content: center;
+  align-items: center;
+  cursor: pointer;
+  i{
+    color: var(--color-warning);
+  }
+}
+
 .operationBar {
   display: flex;
   justify-content: end;
diff --git a/chat2db-client/src/blocks/DatabaseTableEditor/ColumnList/index.tsx b/chat2db-client/src/blocks/DatabaseTableEditor/ColumnList/index.tsx
@@ -226,6 +226,23 @@ const ColumnList = forwardRef((props: IProps, ref: ForwardedRef<IColumnListRef>)
         );
       },
     },
+    {
+      title: i18n('editTable.label.primaryKey'),
+      dataIndex: 'primaryKey',
+      width: '50px',
+      render: (primaryKey: boolean, record: IColumnItemNew) => {
+        return (
+          <div>
+            <div className={styles.keyBox} onClick={()=>{handelPrimaryKey(record)} }>
+              {
+                primaryKey &&
+                <Iconfont code="&#xe775;" />
+              }
+            </div>
+          </div>
+        );
+      },
+    },
     {
       title: i18n('editTable.label.comment'),
       dataIndex: 'comment',
@@ -263,6 +280,26 @@ const ColumnList = forwardRef((props: IProps, ref: ForwardedRef<IColumnListRef>)
     },
   ];
 
+  const handelPrimaryKey = (_data: IColumnItemNew)=>{
+    const newData = dataSource.map((item) => {
+      if (item.key === _data?.key) {
+        // 判断当前数据是新增的数据还是编辑后的数据
+        let editStatus = item.editStatus;
+        if (editStatus !== EditColumnOperationType.Add) {
+          editStatus = EditColumnOperationType.Modify;
+        }
+        const editingDataItem = {
+          ...item,
+          primaryKey: !item.primaryKey,
+          editStatus,
+        };
+        return editingDataItem;
+      }
+      return item;
+    });
+    setDataSource(newData);
+  }
+
   const onDragEnd = ({ active, over }: DragEndEvent) => {
     if (active.id !== over?.id) {
       setDataSource((previous) => {
diff --git a/chat2db-client/src/i18n/en-us/editTable.ts b/chat2db-client/src/i18n/en-us/editTable.ts
@@ -31,6 +31,7 @@ export default {
   'editTable.label.engine': 'Engine',
   'editTable.label.incrementValue': 'Increment value',
   'editTable.label.order': 'Order',
+  'editTable.label.primaryKey': 'Key',
   'editTable.title.sqlPreview': 'SQL preview',
   'editTable.button.addColumn': 'Add column',
   'editTable.button.addIndex': 'Add Index',
diff --git a/chat2db-client/src/i18n/zh-cn/editTable.ts b/chat2db-client/src/i18n/zh-cn/editTable.ts
@@ -31,6 +31,7 @@ export default {
   'editTable.label.engine': '引擎',
   'editTable.label.incrementValue': '自增值',
   'editTable.label.order': '排序',
+  'editTable.label.primaryKey': '键',
   'editTable.title.sqlPreview': 'sql预览',
   'editTable.button.addColumn': '添加列',
   'editTable.button.addIndex': '添加索引',
diff --git a/chat2db-client/src/typings/editTable.ts b/chat2db-client/src/typings/editTable.ts
@@ -25,7 +25,7 @@ export interface IColumnItemNew {
   defaultValue: string | null; // 默认值
   autoIncrement: string | null; // 是否自增
   comment: string | null; // 注释
-  primaryKey: string | null; // 是否主键
+  primaryKey: boolean | null; // 是否主键
   typeName: string | null; // 类型名
   columnSize: number | null; // 列的长度
   bufferLength: number | null; // 缓冲区长度
