import { DbInst } from '../db'; import { commonCustomKeywords, DataType, DbDialect, DialectInfo, DuplicateStrategy, EditorCompletion, EditorCompletionItem, IndexDefinition, QuoteEscape, RowDefinition, sqlColumnType, } from './index'; import { language as sqlLanguage } from 'monaco-editor/esm/vs/basic-languages/sql/sql.js'; export { OracleDialect, ORACLE_TYPE_LIST }; // 参考官方文档:https://docs.oracle.com/cd/B19306_01/server.102/b14200/sql_elements001.htm const ORACLE_TYPE_LIST: sqlColumnType[] = [ // 字符数据类型 { udtName: 'CHAR', dataType: 'CHAR', desc: '定长字符串,自动在末尾用空格补全,非unicode', space: '', range: '1 - 2000' }, { udtName: 'NCHAR', dataType: 'NCHAR', desc: '定长字符串,自动在末尾用空格补全,unicode', space: '', range: '1 - 1000' }, { udtName: 'VARCHAR2', dataType: 'VARCHAR2', desc: '变长字符串,不自动补全空格,非unicode', space: '', range: '1 - 4000' }, { udtName: 'NVARCHAR2', dataType: 'NVARCHAR2', desc: '变长字符串,不自动补全空格,unicode', space: '', range: '1 - 2000' }, // 精确数值数据类型 NUMERIC、DECIMAL、DEC 类型、NUMBER 类型、INTEGER 类型、INT 类型、BIGINT 类型、TINYINT 类型、BYTE 类型、SMALLINT { udtName: 'NUMBER', dataType: 'NUMBER', desc: 'NUMBER(p,s)', space: '1-38', range: '' }, { udtName: 'INTEGER', dataType: 'INTEGER', desc: '同于number(38)', space: '', range: '' }, { udtName: 'INT', dataType: 'INT', desc: '同INTEGER', space: '10', range: '' }, { udtName: 'SMALLINT', dataType: 'SMALLINT', desc: '同于number(38)', space: '', range: '' }, { udtName: 'DECIMAL', dataType: 'DECIMAL', desc: 'decimal(p,s) 默认number(38)', space: '', range: '' }, { udtName: 'FLOAT', dataType: 'FLOAT', desc: 'float(b二进制进度),b的取值范围[1,126],默认126', space: '', range: '' }, { udtName: 'REAL', dataType: 'REAL', desc: '同FLOAT(63)', space: '', range: '' }, { udtName: 'BINARY_FLOAT', dataType: 'BINARY_FLOAT', desc: '32位单精度浮点数数据类型', space: '', range: '' }, { udtName: 'BINARY_DOUBLE', dataType: 'BINARY_DOUBLE', desc: '64位双精度浮点数数据类型', space: '', range: '' }, // 一般日期时间数据类型 DATE TIME TIMESTAMP 默认精度 6 // 多媒体数据类型 TEXT/LONG/LONGVARCHAR 类型:变长字符串类型 IMAGE/LONGVARBINARY 类型 BLOB CLOB BFILE 100G-1 { udtName: 'DATE', dataType: 'DATE', desc: '世纪,年,月,日,时,分,秒', space: '', range: '' }, { udtName: 'TIMESTAMP', dataType: 'TIMESTAMP', desc: '', space: '', range: '' }, // { udtName: 'timestamp(precision) with time zone', dataType: 'TIMESTAMP', desc: '在timestamp(precison)的基础上加入了时区偏移量的值', space: '', range: '' }, // { udtName: 'timestamp with local time zone', dataType: 'TIMESTAMP', desc: '存储时转化为数据库时区进行规范化存储,但不存储时区信息,客户端检索时,按客户端时区的时间数据返回给客户端', space: '', range: '' }, // { udtName: 'interval year(precision) to month', dataType: 'interval year(precision) to month', desc: '可以用来表示几年几月的时间间隔', space: '', range: '' }, // { udtName: 'nterval day(days_precision) to second(seconds_precision)', dataType: 'nterval day(days_precision) to second(seconds_precision)', desc: '可以用来存储天、小时、分和秒的时间间隔', space: '', range: '' }, { udtName: 'LONG', dataType: 'LONG', desc: '文本类型,不能作为主键或唯一约束', space: '', range: '最多达2GB' }, { udtName: 'LONG RAW', dataType: 'LONG RAW', desc: '可变长二进制数据,不用进行字符集转换的数据', space: '', range: '最多达2GB' }, { udtName: 'BLOB', dataType: 'BLOB', desc: '二进制大型对象', space: '', range: '最大长度4G' }, { udtName: 'CLOB', dataType: 'CLOB', desc: '字符大型对象', space: '', range: '最大长度4G' }, { udtName: 'NCLOB', dataType: 'NCLOB', desc: 'Unicode类型的数据', space: '', range: '最大长度4G' }, { udtName: 'BFILE', dataType: 'BFILE', desc: '二进制文件', space: '', range: '' }, ]; // 参考官方文档:https://docs.oracle.com/cd/B19306_01/server.102/b14200/functions001.htm const replaceFunctions: EditorCompletionItem[] = [ // 字符函数 { label: 'ASCII', insertText: 'ASCII(x)', description: '返回字符X的ASCII码' }, { label: 'CONCAT', insertText: 'CONCAT(x,y)', description: '连接字符串X和Y' }, { label: 'INSTR', insertText: 'INSTR(X,STR[,START][,N)', description: '从X中查找str,可以指定从start开始,也可以指定从n开始' }, { label: 'LENGTH', insertText: 'LENGTH(x)', description: '返回X的长度' }, { label: 'LOWER', insertText: 'LOWER(X)', description: 'X转换成小写' }, { label: 'UPPER', insertText: 'UPPER(X)', description: 'X转换成大写' }, { label: 'LTRIM', insertText: 'LTRIM(X[,TRIM_STR])', description: '把X的左边截去trim_str字符串,缺省截去空格' }, { label: 'RTRIM', insertText: 'RTRIM(X[,TRIM_STR])', description: '把X的右边截去trim_str字符串,缺省截去空格' }, { label: 'TRIM', insertText: 'TRIM(X[,TRIM_STR])', description: '把X的两边截去trim_str字符串,缺省截去空格' }, { label: 'REPLACE', insertText: 'REPLACE(X,old,new)', description: '在X中查找old,并替换成new' }, { label: 'SUBSTR', insertText: 'SUBSTR(X,start[,length])', description: '返回X的字串,从start处开始,截取length个字符,缺省length,默认到结尾' }, // 数值函数 { label: 'ABS', insertText: 'ABS(X)', description: 'X的绝对值' }, { label: 'ACOS', insertText: 'ACOS(X)', description: 'X的反余弦' }, { label: 'COS', insertText: 'COS(X)', description: '余弦' }, { label: 'CEIL', insertText: 'CEIL(X)', description: '大于或等于X的最小值' }, { label: 'FLOOR', insertText: 'FLOOR(X)', description: '小于或等于X的最大值' }, { label: 'LOG', insertText: 'LOG(X,Y)', description: 'X为底Y的对数' }, { label: 'MOD', insertText: 'MOD(X,Y)', description: 'X除以Y的余数' }, { label: 'POWER', insertText: 'POWER(X,Y)', description: 'X的Y次幂' }, { label: 'ROUND', insertText: 'ROUND(X [,Y]})', description: 'X在第Y位四舍五入' }, { label: 'SQRT', insertText: 'SQRT(n)', description: '求数值 n 的平方根' }, { label: 'TRUNC', insertText: 'TRUNC(n [,m])', description: "截取数值函数,str 内只能为数字和'-', '+', '.' 的组合" }, //日期时间函数 { label: 'ADD_MONTHS', insertText: 'ADD_MONTHS(date,n)', description: '在输入日期上加上指定的几个月返回一个新日期' }, { label: 'LAST_DAY', insertText: 'LAST_DAY(date)', description: '返回输入日期所在月份最后一天的日期' }, { label: 'EXTRACT', insertText: 'EXTRACT(fmt FROM d)', description: '提取日期中的特定部分' }, { label: 'CURRENT_DATE', insertText: 'CURRENT_DATE', description: '获取当前日期' }, { label: 'CURRENT_TIMESTAMP', insertText: 'TIMESTAMP', description: '获取当前时间' }, // 转换函数 { label: 'TO_CHAR', insertText: `TO_CHAR(d|n, 'yyyy-MM-dd HH24:mi:ss')`, description: '把日期和数字转换为制定格式的字符串' }, { label: 'TO_DATE', insertText: `TO_DATE(X, 'yyyy-MM-dd HH24:mi:ss')`, description: '把一个字符串以fmt格式转换成一个日期类型' }, { label: 'TO_NUMBER', insertText: `TO_NUMBER(X, 'yyyy-MM-dd HH24:mi:ss')`, description: '把一个字符串以fmt格式转换为一个数字' }, { label: 'TO_TIMESTAMP', insertText: `TO_TIMESTAMP(X, 'yyyy-MM-dd HH24:mi:ss.ff')`, description: '把一个字符串以fmt格式转换为日期类型' }, // 其他 { label: 'NVL', insertText: 'NVL(X,VALUE)', description: '如果X为空,返回value,否则返回X' }, { label: 'NVL2', insertText: 'NVL2(x,value1,value2)', description: '如果x非空,返回value1,否则返回value2' }, ]; const addCustomKeywords: EditorCompletionItem[] = [ { label: 'ROWNUM', description: 'keyword', insertText: 'ROWNUM', }, { label: 'DUAL', description: 'keyword', insertText: 'DUAL', }, // 分页代码块 { label: 'SELECT ROWNUM', description: 'code block', insertText: 'SELECT * from table_name where rownum <= 10', }, { label: 'SELECT PAGE', description: 'code block', insertText: ` SELECT * FROM ( SELECT t.*, ROWNUM AS rn FROM table_name t WHERE ROWNUM <= 25 ) WHERE rn > 0 \n`, }, ]; let oracleDialectInfo: DialectInfo; class OracleDialect implements DbDialect { getInfo(): DialectInfo { if (oracleDialectInfo) { return oracleDialectInfo; } let { keywords, operators, builtinVariables } = sqlLanguage; let functionNames = replaceFunctions.map((a) => a.label); let excludeKeywords = new Set(functionNames.concat(operators)); excludeKeywords.add('SELECT'); let editorCompletions: EditorCompletion = { keywords: keywords .filter((a: string) => !excludeKeywords.has(a)) // 移除已存在的operator、function .map((a: string): EditorCompletionItem => ({ label: a, description: 'keyword' })) .concat( // 加上自定义的关键字 commonCustomKeywords.map( (a): EditorCompletionItem => ({ label: a, description: 'keyword', }) ) ) .concat(addCustomKeywords), operators: operators.map((a: string): EditorCompletionItem => ({ label: a, description: 'operator' })), functions: replaceFunctions, variables: builtinVariables.map((a: string): EditorCompletionItem => ({ label: a, description: 'var' })), }; oracleDialectInfo = { name: 'Oracle', icon: 'iconfont icon-oracle', defaultPort: 1521, formatSqlDialect: 'plsql', columnTypes: ORACLE_TYPE_LIST.sort((a, b) => a.udtName.localeCompare(b.udtName)), editorCompletions, }; return oracleDialectInfo; } getDefaultSelectSql(db: string, table: string, condition: string, orderBy: string, pageNum: number, limit: number) { return ` SELECT * FROM ( SELECT t.*, ROWNUM AS rn FROM "${table}" t WHERE ROWNUM <=${pageNum * limit} ${condition ? ' and ' + condition : ''} ${orderBy ? orderBy : ''} ) WHERE rn > ${(pageNum - 1) * limit} `; } // eslint-disable-next-line @typescript-eslint/no-unused-vars,no-unused-vars getPageSql(pageNum: number, limit: number) { return ``; } getDefaultRows(): RowDefinition[] { return [ { name: 'ID', type: 'NUMBER', length: '', numScale: '', value: '', notNull: true, pri: true, auto_increment: true, remark: '主键ID' }, { name: 'CREATOR_ID', type: 'NUMBER', length: '', numScale: '', value: '', notNull: true, pri: false, auto_increment: false, remark: '创建人id' }, { name: 'CREATOR', type: 'VARCHAR2', length: '100', numScale: '', value: '', notNull: true, pri: false, auto_increment: false, remark: '创建人姓名', }, { name: 'CREATE_TIME', type: 'DATE', length: '', numScale: '', value: 'CURRENT_TIMESTAMP', notNull: true, pri: false, auto_increment: false, remark: '创建时间', }, { name: 'UPDATOR_ID', type: 'NUMBER', length: '', numScale: '', value: '', notNull: true, pri: false, auto_increment: false, remark: '修改人id' }, { name: 'UPDATOR', type: 'VARCHAR2', length: '100', numScale: '', value: '', notNull: true, pri: false, auto_increment: false, remark: '修改人姓名', }, { name: 'UPDATE_TIME', type: 'DATE', length: '', numScale: '', value: 'CURRENT_TIMESTAMP', notNull: true, pri: false, auto_increment: false, remark: '修改时间', }, ]; } getDefaultIndex(): IndexDefinition { return { indexName: '', columnNames: [], unique: false, indexType: 'NORMAL', indexComment: '', }; } quoteIdentifier = (name: string) => { return `"${name}"`; }; matchType(text: string, arr: string[]): boolean { if (!text || !arr || arr.length === 0) { return false; } for (let i = 0; i < arr.length; i++) { if (text.indexOf(arr[i]) > -1) { return true; } } return false; } getDefaultValueSql(cl: any): string { if (cl.value && cl.value.length > 0) { // 哪些字段默认值需要加引号 let marks = false; if (this.matchType(cl.type, ['CHAR', 'TIME', 'DATE', 'LONG', 'CLOB', 'BLOB', 'BFILE'])) { // 默认值是时间日期函数的必须要加引号 let val = cl.value.toUpperCase().replace(' ', ''); if (this.matchType(cl.type, ['DATE', 'TIMESTAMP']) && ['CURRENT_DATE', 'CURRENT_TIMESTAMP'].includes(val)) { marks = false; } else { marks = true; } } return ` DEFAULT ${marks ? "'" : ''}${cl.value}${marks ? "'" : ''}`; } return ''; } getTypeLengthSql(cl: any) { // 哪些字段可以指定长度 VARCHAR/VARCHAR2/CHAR/BIT/NUMBER/NUMERIC/TIME、TIMESTAMP(可以指定小数秒精度) if (cl.length && this.matchType(cl.type, ['CHAR', 'BIT', 'TIME', 'NUM', 'DEC'])) { // 哪些字段类型可以指定小数点 if (cl.numScale && this.matchType(cl.type, ['NUM', 'DEC'])) { return `(${cl.length}, ${cl.numScale})`; } else { return `(${cl.length})`; } } return ''; } genColumnBasicSql(cl: RowDefinition, create: boolean, data = {}): string { let length = this.getTypeLengthSql(cl); // 默认值 let defVal = this.getDefaultValueSql(cl); let incr = ''; if (cl.auto_increment && create) { cl.type = 'number'; length = ''; incr = 'generated by default as IDENTITY'; } // 如果有原名以原名为准 let name = cl.oldName && cl.name !== cl.oldName ? cl.oldName : cl.name; let baseSql = ` ${this.quoteIdentifier(name)} ${cl.type}${length} ${incr}`; return incr ? baseSql : ` ${baseSql} ${defVal} ${cl.notNull ? 'NOT NULL' : ''} `; } // eslint-disable-next-line @typescript-eslint/no-unused-vars,no-unused-vars getOtherCreateTableSql(data: any) { return ''; } getCreateTableSql(data: any): string { let schemaArr = data.db.split('/'); let schema = schemaArr.length > 1 ? schemaArr[schemaArr.length - 1] : schemaArr[0]; let dbTable = `${this.quoteIdentifier(schema)}.${this.quoteIdentifier(data.tableName)}`; let createSql = ''; let tableCommentSql = ''; let columCommentSql = ''; let pris = [] as string[]; // 创建表结构 let fields: string[] = []; data.fields.res.forEach((item: any) => { item.name && fields.push(this.genColumnBasicSql(item, true, data)); // 列注释 if (item.remark) { columCommentSql += ` COMMENT ON COLUMN ${dbTable}.${this.quoteIdentifier(item.name)} is '${QuoteEscape(item.remark)}'; `; } // 主键 if (item.pri) { pris.push(this.quoteIdentifier(item.name)); } }); // 主键语句 let prisql = ''; if (pris.length > 0) { prisql = ` PRIMARY KEY (${pris.join(',')})`; } // 建表 createSql = `CREATE TABLE ${dbTable} ( ${fields.join(',')} ${prisql ? ',' + prisql : ''} ) ;`; // 表注释 if (data.tableComment) { tableCommentSql = ` COMMENT ON TABLE ${dbTable} is '${QuoteEscape(data.tableComment)}'; `; } // 其余建表信息,如:自增字段在老版本的使用方式是创建自增序列 let other = this.getOtherCreateTableSql(data); return createSql + tableCommentSql + columCommentSql + other; } getCreateIndexSql(tableData: any): string { // CREATE UNIQUE INDEX idx_column_name ON your_table (column1, column2); // COMMENT ON INDEX idx_column_name IS 'Your index comment here'; let schemaArr = tableData.db.split('/'); let schema = schemaArr.length > 1 ? schemaArr[schemaArr.length - 1] : schemaArr[0]; let dbTable = `${this.quoteIdentifier(schema)}.${this.quoteIdentifier(tableData.tableName)}`; let sql: string[] = []; tableData.indexs.res.forEach((a: any) => { sql.push(` CREATE ${a.unique ? 'UNIQUE' : ''} INDEX ${a.indexName} ON ${dbTable} ("${a.columnNames.join('","')})"`); }); return sql.join(';'); } getModifyColumnSql(tableData: any, tableName: string, changeData: { del: RowDefinition[]; add: RowDefinition[]; upd: RowDefinition[] }): string { let schemaArr = tableData.db.split('/'); let schema = schemaArr.length > 1 ? schemaArr[schemaArr.length - 1] : schemaArr[0]; let dbTable = `${this.quoteIdentifier(schema)}.${this.quoteIdentifier(tableName)}`; let baseSql = `ALTER TABLE ${dbTable} `; let modifyArr: string[] = []; let dropArr: string[] = []; // 重命名的sql要一条条执行 let renameArr: string[] = []; let commentArr: string[] = []; // 主键字段 let priArr = new Set(); if (changeData.upd.length > 0) { changeData.upd.forEach((a) => { let commentSql = `COMMENT ON COLUMN ${dbTable}.${this.quoteIdentifier(a.name)} IS '${QuoteEscape(a.remark)}'`; if (a.remark && a.oldName === a.name) { commentArr.push(commentSql); } // 修改了字段名 if (a.oldName !== a.name) { renameArr.push(baseSql + ` RENAME COLUMN ${this.quoteIdentifier(a.oldName!)} TO ${this.quoteIdentifier(a.name)} ;`); if (a.remark) { commentArr.push(commentSql); } } modifyArr.push(` MODIFY (${this.genColumnBasicSql(a, false, tableData)})`); if (a.pri) { priArr.add(`${this.quoteIdentifier(a.name)}`); } }); } if (changeData.add.length > 0) { changeData.add.forEach((a) => { modifyArr.push(` ADD (${this.genColumnBasicSql(a, false, tableData)})`); if (a.remark) { commentArr.push(`COMMENT ON COLUMN ${dbTable}.${this.quoteIdentifier(a.name)} is '${QuoteEscape(a.remark)}'`); } if (a.pri) { priArr.add(`"${a.name}"`); } }); } if (changeData.del.length > 0) { changeData.del.forEach((a) => { dropArr.push(`${this.quoteIdentifier(a.name)}`); }); } let dropPkSql = ''; if (priArr.size > 0) { let resPri = tableData.fields.res.find((a: RowDefinition) => a.pri); if (resPri) { priArr.add(`"${resPri.name}"`); } // 如果有编辑主键字段,则删除主键,再添加主键 // 解析表字段中是否含有主键,有的话就删除主键 if (tableData.fields.oldFields.find((a: RowDefinition) => a.pri)) { dropPkSql = `ALTER TABLE ${dbTable} DROP PRIMARY KEY;`; } } let modifySql = modifyArr.length > 0 ? baseSql + modifyArr.join(' ') + ';' : ''; let dropSql = dropArr.length > 0 ? baseSql + ` DROP (${dropArr.join(',')}) ;` : ''; let renameSql = renameArr.join(''); let addPkSql = priArr.size > 0 ? `ALTER TABLE ${dbTable} ADD CONSTRAINT "PK_${tableName}" PRIMARY KEY (${Array.from(priArr).join(',')});` : ''; let commentSql = commentArr.join(';'); return dropPkSql + modifySql + dropSql + renameSql + addPkSql + commentSql; } getModifyIndexSql(tableData: any, tableName: string, changeData: { del: any[]; add: any[]; upd: any[] }): string { // 不能直接修改索引名或字段、需要先删后加 let dropIndexNames: string[] = []; let addIndexs: any[] = []; if (changeData.upd.length > 0) { changeData.upd.forEach((a) => { dropIndexNames.push(a.indexName); addIndexs.push(a); }); } if (changeData.del.length > 0) { changeData.del.forEach((a) => { dropIndexNames.push(a.indexName); }); } if (changeData.add.length > 0) { changeData.add.forEach((a) => { addIndexs.push(a); }); } if (dropIndexNames.length > 0 || addIndexs.length > 0) { let sql: string[] = []; if (dropIndexNames.length > 0) { dropIndexNames.forEach((a) => { sql.push(`DROP INDEX ${a}`); }); } if (addIndexs.length > 0) { addIndexs.forEach((a) => { sql.push(`CREATE ${a.unique ? 'UNIQUE' : ''} INDEX ${a.indexName} ON "${tableName}" (${a.columnNames.join(',')})`); }); } return sql.join(';'); } return ''; } getModifyTableInfoSql(tableData: any): string { let schemaArr = tableData.db.split('/'); let schema = schemaArr.length > 1 ? schemaArr[schemaArr.length - 1] : schemaArr[0]; let sql = ''; if (tableData.tableComment != tableData.oldTableComment) { let dbTable = `${this.quoteIdentifier(schema)}.${this.quoteIdentifier(tableData.oldTableName)}`; sql = `COMMENT ON TABLE ${dbTable} is '${QuoteEscape(tableData.tableComment)}';`; } if (tableData.tableName != tableData.oldTableName) { let dbTable = `${this.quoteIdentifier(schema)}.${this.quoteIdentifier(tableData.oldTableName)}`; sql += `ALTER TABLE ${dbTable} RENAME TO ${this.quoteIdentifier(tableData.tableName)}`; } return sql; } getDataType(columnType: string): DataType { if (DbInst.isNumber(columnType)) { return DataType.Number; } // 日期时间类型 oracle只有date和timestamp类型 if (/timestamp|date/gi.test(columnType)) { return DataType.DateTime; } return DataType.String; } wrapValue(columnType: string, value: any): any { if (value == null) { return 'NULL'; } if (DbInst.isNumber(columnType)) { return value; } if (value && this.getDataType(columnType) === DataType.DateTime) { return `to_timestamp('${value}', 'yyyy-mm-dd hh24:mi:ss')`; } return `'${value}'`; } getBatchInsertPreviewSql(tableName: string, fieldArr: string[], duplicateStrategy: DuplicateStrategy): string { if (duplicateStrategy == DuplicateStrategy.REPLACE) { // 字段数组生成占位符sql let phs = []; let values = []; let insertFields = []; for (let i = 0; i < fieldArr.length; i++) { phs.push(`:${i + 1} ${fieldArr[i]}`); values.push(`T2.${fieldArr[i]}`); insertFields.push(`T1.${fieldArr[i]}`); } let placeholder = phs.join(','); let sql = `MERGE INTO ${this.quoteIdentifier(tableName)} T1 USING ( SELECT ${placeholder} FROM dual ) T2 ON (T1.id = T2.id) WHEN NOT MATCHED THEN INSERT (${insertFields.join(',')}) VALUES (${values.join(',')}) WHEN MATCHED THEN UPDATE SET ${fieldArr.map((a) => `T1.${a} = T2.${a}`).join(',')}`; return sql; } else { // 字段数组生成占位符sql let phs = []; for (let i = 0; i < fieldArr.length; i++) { phs.push(`:${i + 1} ${fieldArr[i]}`); } let ignore = ''; if (duplicateStrategy == DuplicateStrategy.IGNORE) { ignore = `/*+ IGNORE_ROW_ON_DUPKEY_INDEX(${tableName}(id)) */`; } let placeholder = phs.join(','); return `INSERT ${ignore} INTO ${tableName} (${fieldArr.join(',')}) VALUES (${placeholder});`; } } }