Merge pull request taozhi8833998#1317 from taozhi8833998/feat-create-table-mysql

feat: create table index with length in mysql

Author: taozhi8833998

pegjs/mariadb.pegjs

@@ -798,12 +798,38 @@ ALTER_CHANGE_COLUMN
         },
       }
   }
+column_idx_ref
+  = col:column_without_kw __ LPAREN __ l:[0-9]+ __ RPAREN __ ob:(KW_ASC / KW_DESC)? {
+      return {
+        type: 'column_ref',
+        column: col,
+        suffix: `(${parseInt(l.join(''), 10)})`,
+        order_by: ob
+      };
+    }
+  / col:column_without_kw __ ob:(KW_ASC / KW_DESC)? {
+      return {
+        type: 'column_ref',
+        column: col,
+        order_by: ob
+      };
+    }
+
+column_ref_idx_list
+  = head:column_idx_ref tail:(__ COMMA __ column_idx_ref)* {
+      return createList(head, tail);
+    }
+
+cte_idx_column_definition
+  = LPAREN __ l:column_ref_idx_list __ RPAREN {
+      return l
+    }
 
 create_index_definition
   = kc:(KW_INDEX / KW_KEY) __
     c:column? __
     t:index_type? __
-    de:cte_column_definition __
+    de:cte_idx_column_definition __
     id:index_options? __
      {
       return {

pegjs/mysql.pegjs

@@ -1518,7 +1518,15 @@ cte_column_definition
     }
 
 column_idx_ref
-  = col:column_without_kw __ ob:(KW_ASC / KW_DESC)? {
+  = col:column_without_kw __ LPAREN __ l:[0-9]+ __ RPAREN __ ob:(KW_ASC / KW_DESC)? {
+      return {
+        type: 'column_ref',
+        column: col,
+        suffix: `(${parseInt(l.join(''), 10)})`,
+        order_by: ob
+      };
+    }
+  / col:column_without_kw __ ob:(KW_ASC / KW_DESC)? {
       return {
         type: 'column_ref',
         column: col,

src/index-definition.js

@@ -73,10 +73,7 @@ function indexTypeAndOptionToSQL(indexDefinition) {
 
 function indexDefinitionToSQL(indexDefinition) {
   const indexSQL = []
-  const {
-    keyword,
-    index,
-  } = indexDefinition
+  const { keyword, index } = indexDefinition
   indexSQL.push(toUpper(keyword))
   indexSQL.push(index)
   indexSQL.push(...indexTypeAndOptionToSQL(indexDefinition))

test/mysql-mariadb.spec.js

@@ -505,6 +505,13 @@ describe('mysql', () => {
           'SELECT * FROM (SELECT 1) MINUS SELECT * FROM (SELECT 2)'
         ]
       },
+      {
+        title: 'index column length',
+        sql: [
+          'CREATE TABLE `Translation` (`id` char(36) CHARACTER SET ascii COLLATE ascii_bin NOT NULL,`en-GB` text,PRIMARY KEY (`id`),KEY `Translation_en-GB_btree_idx` (`en-GB`(768))) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci',
+          'CREATE TABLE `Translation` (`id` CHAR(36) NOT NULL CHARACTER SET ASCII COLLATE ASCII_BIN, `en-GB` TEXT, PRIMARY KEY (`id`), KEY Translation_en-GB_btree_idx (`en-GB` (768))) ENGINE = INNODB DEFAULT CHARSET = utf8mb4 COLLATE = utf8mb4_0900_ai_ci',
+        ]
+      },
     ]
     SQL_LIST.forEach(sqlInfo => {
       const { title, sql } = sqlInfo