mirror of
https://github.com/Syngnat/GoNavi.git
synced 2026-06-16 03:29:55 +08:00
569edbb11a
- 后端接入：新增独立 starrocks 可选驱动，复用 MySQL wire 协议并支持默认 9030 端口 - 驱动管理：补齐 manifest、build tag、revision、driver-agent provider 和构建脚本 - 前端接入：新增 StarRocks 连接类型、图标、能力矩阵、URI 解析、保存回显和 SQL 自动 LIMIT - 方言增强：新增 StarRocks 类型、关键字、函数补全和专属建表 SQL 生成 - 高级对象：支持物化视图对象浏览、Rollup 模板、外部 Catalog 模板和高级表设计器参数 - CI 发布：将 StarRocks driver-agent 纳入 dev/release 构建与 release 资产校验
334 lines
13 KiB
TypeScript
334 lines
13 KiB
TypeScript
import { describe, expect, it } from 'vitest';
|
|
|
|
import {
|
|
buildCreateTablePreviewSql,
|
|
buildAlterTablePreviewSql,
|
|
buildStarRocksMaterializedViewPreviewSql,
|
|
hasAlterTableDraftChanges,
|
|
type BuildAlterTablePreviewInput,
|
|
type EditableColumnSnapshot,
|
|
} from './tableDesignerSchemaSql';
|
|
|
|
const baseColumn = (overrides: Partial<EditableColumnSnapshot>): EditableColumnSnapshot => ({
|
|
_key: overrides._key || 'col',
|
|
name: overrides.name || 'id',
|
|
type: overrides.type || 'int',
|
|
nullable: overrides.nullable || 'NO',
|
|
default: overrides.default || '',
|
|
extra: overrides.extra || '',
|
|
comment: overrides.comment || '',
|
|
key: overrides.key || '',
|
|
isAutoIncrement: overrides.isAutoIncrement || false,
|
|
});
|
|
|
|
const buildInput = (overrides: Partial<BuildAlterTablePreviewInput>): BuildAlterTablePreviewInput => ({
|
|
dbType: overrides.dbType || 'mysql',
|
|
tableName: overrides.tableName || 'users',
|
|
originalColumns: overrides.originalColumns || [baseColumn({ _key: 'id', name: 'id', key: 'PRI', nullable: 'NO' })],
|
|
columns: overrides.columns || [
|
|
baseColumn({ _key: 'id', name: 'id', key: 'PRI', nullable: 'NO' }),
|
|
baseColumn({ _key: 'age', name: 'age', nullable: 'YES', comment: '年龄' }),
|
|
],
|
|
});
|
|
|
|
describe('tableDesignerSchemaSql', () => {
|
|
it('detects when alter table drafts contain unsaved column changes', () => {
|
|
expect(hasAlterTableDraftChanges(buildInput({ dbType: 'mysql' }))).toBe(true);
|
|
expect(
|
|
hasAlterTableDraftChanges(
|
|
buildInput({
|
|
dbType: 'mysql',
|
|
columns: [baseColumn({ _key: 'id', name: 'id', key: 'PRI', nullable: 'NO' })],
|
|
}),
|
|
),
|
|
).toBe(false);
|
|
});
|
|
|
|
it('keeps mysql alter preview syntax with column position clauses', () => {
|
|
const sql = buildAlterTablePreviewSql(buildInput({ dbType: 'mysql' }));
|
|
|
|
expect(sql).toContain('ALTER TABLE `users`');
|
|
expect(sql).toContain('ADD COLUMN `age` int NULL');
|
|
expect(sql).toContain("COMMENT '年龄'");
|
|
expect(sql).toContain('AFTER `id`');
|
|
});
|
|
|
|
it('builds kingbase alter preview without mysql-only syntax', () => {
|
|
const sql = buildAlterTablePreviewSql(buildInput({
|
|
dbType: 'kingbase',
|
|
tableName: 'public.users',
|
|
}));
|
|
|
|
expect(sql).toContain('ALTER TABLE public.users');
|
|
expect(sql).toContain('ADD COLUMN age int');
|
|
expect(sql).toContain("COMMENT ON COLUMN public.users.age IS '年龄';");
|
|
expect(sql).not.toContain('`');
|
|
expect(sql).not.toContain('AFTER');
|
|
expect(sql).not.toContain(' FIRST');
|
|
});
|
|
|
|
it('uses mysql change column syntax when renaming a column', () => {
|
|
const sql = buildAlterTablePreviewSql(buildInput({
|
|
dbType: 'mysql',
|
|
originalColumns: [baseColumn({ _key: 'name', name: 'name', type: 'varchar(64)', nullable: 'YES' })],
|
|
columns: [baseColumn({ _key: 'name', name: 'display_name', type: 'varchar(64)', nullable: 'YES' })],
|
|
}));
|
|
|
|
expect(sql).toContain('CHANGE COLUMN `name` `display_name` varchar(64) NULL');
|
|
expect(sql).toContain('FIRST');
|
|
expect(sql).not.toContain('MODIFY COLUMN `display_name`');
|
|
});
|
|
|
|
it('builds oracle alter preview with oracle rename and modify syntax', () => {
|
|
const sql = buildAlterTablePreviewSql(buildInput({
|
|
dbType: 'oracle',
|
|
tableName: 'HR.EMPLOYEES',
|
|
originalColumns: [
|
|
baseColumn({ _key: 'name', name: 'NAME', type: 'VARCHAR2(64)', nullable: 'YES', comment: '旧名称' }),
|
|
],
|
|
columns: [
|
|
baseColumn({
|
|
_key: 'name',
|
|
name: 'DISPLAY_NAME',
|
|
type: 'VARCHAR2(128)',
|
|
nullable: 'NO',
|
|
default: 'guest',
|
|
comment: '显示名',
|
|
}),
|
|
],
|
|
}));
|
|
|
|
expect(sql).toContain('ALTER TABLE "HR"."EMPLOYEES"\nRENAME COLUMN "NAME" TO "DISPLAY_NAME";');
|
|
expect(sql).toContain(`ALTER TABLE "HR"."EMPLOYEES"\nMODIFY ("DISPLAY_NAME" VARCHAR2(128) DEFAULT 'guest' NOT NULL);`);
|
|
expect(sql).toContain(`COMMENT ON COLUMN "HR"."EMPLOYEES"."DISPLAY_NAME" IS '显示名';`);
|
|
expect(sql).not.toContain('`');
|
|
expect(sql).not.toContain('CHANGE COLUMN');
|
|
expect(sql).not.toContain('AUTO_INCREMENT');
|
|
});
|
|
|
|
it('builds sqlserver alter preview with sp_rename and alter column syntax', () => {
|
|
const sql = buildAlterTablePreviewSql(buildInput({
|
|
dbType: 'sqlserver',
|
|
tableName: 'dbo.Users',
|
|
originalColumns: [
|
|
baseColumn({ _key: 'name', name: 'name', type: 'nvarchar(64)', nullable: 'YES' }),
|
|
],
|
|
columns: [
|
|
baseColumn({ _key: 'name', name: 'display_name', type: 'nvarchar(128)', nullable: 'NO' }),
|
|
],
|
|
}));
|
|
|
|
expect(sql).toContain(`EXEC sp_rename 'dbo.Users.name', 'display_name', 'COLUMN';`);
|
|
expect(sql).toContain('ALTER TABLE [dbo].[Users]\nALTER COLUMN [display_name] nvarchar(128) NOT NULL;');
|
|
expect(sql).not.toContain('CHANGE COLUMN');
|
|
expect(sql).not.toContain('MODIFY COLUMN');
|
|
expect(sql).not.toContain('`');
|
|
});
|
|
|
|
it('keeps sqlite alter preview limited to sqlite-supported operations', () => {
|
|
const sql = buildAlterTablePreviewSql(buildInput({
|
|
dbType: 'sqlite',
|
|
tableName: 'users',
|
|
originalColumns: [
|
|
baseColumn({ _key: 'name', name: 'name', type: 'TEXT', nullable: 'YES' }),
|
|
],
|
|
columns: [
|
|
baseColumn({ _key: 'name', name: 'display_name', type: 'INTEGER', nullable: 'NO' }),
|
|
],
|
|
}));
|
|
|
|
expect(sql).toContain('ALTER TABLE "users"\nRENAME COLUMN "name" TO "display_name";');
|
|
expect(sql).toContain('-- SQLite 不支持直接修改字段属性');
|
|
expect(sql).not.toContain('CHANGE COLUMN');
|
|
expect(sql).not.toContain('MODIFY COLUMN');
|
|
expect(sql).not.toContain('AFTER');
|
|
});
|
|
|
|
it('builds duckdb alter preview without mysql-only syntax', () => {
|
|
const sql = buildAlterTablePreviewSql(buildInput({
|
|
dbType: 'duckdb',
|
|
tableName: 'main.users',
|
|
originalColumns: [
|
|
baseColumn({ _key: 'score', name: 'score', type: 'INTEGER', nullable: 'YES', default: '0' }),
|
|
],
|
|
columns: [
|
|
baseColumn({ _key: 'score', name: 'score', type: 'BIGINT', nullable: 'NO', default: '1' }),
|
|
],
|
|
}));
|
|
|
|
expect(sql).toContain('ALTER TABLE "main"."users"\nALTER COLUMN "score" SET DATA TYPE BIGINT;');
|
|
expect(sql).toContain('ALTER TABLE "main"."users"\nALTER COLUMN "score" SET DEFAULT 1;');
|
|
expect(sql).toContain('ALTER TABLE "main"."users"\nALTER COLUMN "score" SET NOT NULL;');
|
|
expect(sql).not.toContain('CHANGE COLUMN');
|
|
expect(sql).not.toContain('MODIFY COLUMN');
|
|
});
|
|
|
|
it('builds doris alter preview without mysql-only syntax or metadata extra', () => {
|
|
const sql = buildAlterTablePreviewSql(buildInput({
|
|
dbType: 'doris',
|
|
tableName: 'sales.orders',
|
|
originalColumns: [
|
|
baseColumn({
|
|
_key: 'carrier',
|
|
name: 'carrier_id',
|
|
type: 'bigint',
|
|
nullable: 'YES',
|
|
extra: 'NONE',
|
|
comment: '承运商id',
|
|
}),
|
|
],
|
|
columns: [
|
|
baseColumn({
|
|
_key: 'carrier',
|
|
name: 'carrier_code',
|
|
type: 'bigint',
|
|
nullable: 'YES',
|
|
extra: 'NONE',
|
|
comment: '承运商id1',
|
|
}),
|
|
],
|
|
}));
|
|
|
|
expect(sql).toContain('ALTER TABLE `sales`.`orders`\nRENAME COLUMN `carrier_id` `carrier_code`;');
|
|
expect(sql).toContain("ALTER TABLE `sales`.`orders`\nMODIFY COLUMN `carrier_code` bigint NULL COMMENT '承运商id1';");
|
|
expect(sql).not.toContain('CHANGE COLUMN');
|
|
expect(sql).not.toContain('AFTER');
|
|
expect(sql).not.toContain(' FIRST');
|
|
expect(sql).not.toContain('NONE');
|
|
});
|
|
|
|
it('uses native limited alter syntax for clickhouse and tdengine instead of mysql syntax', () => {
|
|
const clickhouseSql = buildAlterTablePreviewSql(buildInput({
|
|
dbType: 'clickhouse',
|
|
tableName: 'events',
|
|
originalColumns: [baseColumn({ _key: 'name', name: 'name', type: 'String', nullable: 'YES' })],
|
|
columns: [baseColumn({ _key: 'name', name: 'display_name', type: 'String', nullable: 'YES' })],
|
|
}));
|
|
const tdengineSql = buildAlterTablePreviewSql(buildInput({
|
|
dbType: 'tdengine',
|
|
tableName: 'meters',
|
|
originalColumns: [baseColumn({ _key: 'value', name: 'value', type: 'FLOAT', nullable: 'YES' })],
|
|
columns: [baseColumn({ _key: 'value', name: 'value', type: 'DOUBLE', nullable: 'YES' })],
|
|
}));
|
|
|
|
expect(clickhouseSql).toContain('ALTER TABLE `events`\nRENAME COLUMN `name` TO `display_name`;');
|
|
expect(tdengineSql).toContain('ALTER TABLE `meters`\nMODIFY COLUMN `value` DOUBLE;');
|
|
expect(clickhouseSql).not.toContain('CHANGE COLUMN');
|
|
expect(tdengineSql).not.toContain('CHANGE COLUMN');
|
|
expect(clickhouseSql).not.toContain('AFTER');
|
|
expect(tdengineSql).not.toContain('AFTER');
|
|
});
|
|
|
|
it('builds StarRocks create table preview with OLAP engine and conservative distribution', () => {
|
|
const sql = buildCreateTablePreviewSql({
|
|
tableName: 'sales.orders',
|
|
dbType: 'starrocks',
|
|
columns: [
|
|
baseColumn({ _key: 'id', name: 'id', type: 'BIGINT', nullable: 'NO', key: 'PRI' }),
|
|
baseColumn({ _key: 'amount', name: 'amount', type: 'DECIMAL(10,2)', nullable: 'YES' }),
|
|
],
|
|
});
|
|
|
|
expect(sql).toContain('CREATE TABLE `sales`.`orders`');
|
|
expect(sql).toContain('ENGINE=OLAP');
|
|
expect(sql).toContain('DUPLICATE KEY (`id`)');
|
|
expect(sql).toContain('DISTRIBUTED BY HASH(`id`) BUCKETS AUTO');
|
|
expect(sql).not.toContain('ENGINE=InnoDB');
|
|
});
|
|
|
|
it('builds StarRocks advanced OLAP table preview with key model, partition, buckets, properties and rollup', () => {
|
|
const sql = buildCreateTablePreviewSql({
|
|
tableName: 'sales.events',
|
|
dbType: 'starrocks',
|
|
columns: [
|
|
baseColumn({ _key: 'dt', name: 'dt', type: 'DATE', nullable: 'NO' }),
|
|
baseColumn({ _key: 'user_id', name: 'user_id', type: 'BIGINT', nullable: 'NO' }),
|
|
baseColumn({ _key: 'amount', name: 'amount', type: 'DECIMAL(10,2)', nullable: 'YES', extra: 'SUM' }),
|
|
],
|
|
starRocksOptions: {
|
|
keyModel: 'AGGREGATE',
|
|
keyColumnNames: ['dt', 'user_id'],
|
|
partitionClause: 'PARTITION BY date_trunc(\'day\', `dt`)',
|
|
distributionColumnNames: ['user_id'],
|
|
bucketMode: 'NUMBER',
|
|
bucketCount: 12,
|
|
properties: '"replication_num" = "1"',
|
|
rollups: [{ name: 'rollup_dt', columnNames: ['dt', 'amount'] }],
|
|
},
|
|
});
|
|
|
|
expect(sql).toContain('AGGREGATE KEY (`dt`, `user_id`)');
|
|
expect(sql).toContain("PARTITION BY date_trunc('day', `dt`)");
|
|
expect(sql).toContain('DISTRIBUTED BY HASH(`user_id`) BUCKETS 12');
|
|
expect(sql).toContain('PROPERTIES (');
|
|
expect(sql).toContain('ALTER TABLE `sales`.`events`\nADD ROLLUP `rollup_dt` (`dt`, `amount`);');
|
|
});
|
|
|
|
it('builds StarRocks external table preview with external engine and properties', () => {
|
|
const sql = buildCreateTablePreviewSql({
|
|
tableName: 'ext.raw_orders',
|
|
dbType: 'starrocks',
|
|
columns: [
|
|
baseColumn({ _key: 'id', name: 'id', type: 'BIGINT', nullable: 'NO' }),
|
|
baseColumn({ _key: 'payload', name: 'payload', type: 'STRING', nullable: 'YES' }),
|
|
],
|
|
starRocksOptions: {
|
|
tableKind: 'external',
|
|
externalEngine: 'hive',
|
|
externalProperties: '"resource" = "hive0"\n"database" = "ods"\n"table" = "orders"',
|
|
},
|
|
});
|
|
|
|
expect(sql).toContain('CREATE EXTERNAL TABLE `ext`.`raw_orders`');
|
|
expect(sql).toContain('ENGINE=HIVE');
|
|
expect(sql).toContain('"resource" = "hive0"');
|
|
expect(sql).not.toContain('ENGINE=OLAP');
|
|
});
|
|
|
|
it('builds StarRocks materialized view preview with refresh and distribution clauses', () => {
|
|
const sql = buildStarRocksMaterializedViewPreviewSql({
|
|
name: 'sales.mv_user_amount',
|
|
query: 'SELECT user_id, SUM(amount) AS total_amount FROM sales.events GROUP BY user_id',
|
|
distributionColumnNames: ['user_id'],
|
|
bucketCount: 8,
|
|
refreshClause: 'REFRESH SCHEDULE EVERY(INTERVAL 10 MINUTE)',
|
|
properties: '"replication_num" = "1"',
|
|
});
|
|
|
|
expect(sql).toContain('CREATE MATERIALIZED VIEW `sales`.`mv_user_amount`');
|
|
expect(sql).toContain('REFRESH SCHEDULE EVERY(INTERVAL 10 MINUTE)');
|
|
expect(sql).toContain('DISTRIBUTED BY HASH(`user_id`) BUCKETS 8');
|
|
expect(sql).toContain('AS\nSELECT user_id, SUM(amount) AS total_amount FROM sales.events GROUP BY user_id;');
|
|
});
|
|
|
|
it('treats mariadb and sphinx as mysql-family only where mysql syntax is intended', () => {
|
|
for (const dbType of ['mariadb', 'sphinx']) {
|
|
const sql = buildAlterTablePreviewSql(buildInput({ dbType }));
|
|
expect(sql).toContain('ALTER TABLE `users`');
|
|
expect(sql).toContain('ADD COLUMN `age` int NULL');
|
|
}
|
|
});
|
|
|
|
it('builds oracle create table preview without mysql table options', () => {
|
|
const sql = buildCreateTablePreviewSql({
|
|
dbType: 'oracle',
|
|
tableName: 'HR.EMPLOYEES',
|
|
charset: 'utf8mb4',
|
|
collation: 'utf8mb4_unicode_ci',
|
|
columns: [
|
|
baseColumn({ _key: 'id', name: 'ID', type: 'NUMBER(10)', nullable: 'NO', key: 'PRI', isAutoIncrement: true }),
|
|
baseColumn({ _key: 'name', name: 'NAME', type: 'VARCHAR2(255)', nullable: 'YES', comment: '姓名' }),
|
|
],
|
|
});
|
|
|
|
expect(sql).toContain('CREATE TABLE "HR"."EMPLOYEES"');
|
|
expect(sql).toContain('"ID" NUMBER(10) GENERATED BY DEFAULT AS IDENTITY NOT NULL');
|
|
expect(sql).toContain('PRIMARY KEY ("ID")');
|
|
expect(sql).toContain(`COMMENT ON COLUMN "HR"."EMPLOYEES"."NAME" IS '姓名';`);
|
|
expect(sql).not.toContain('ENGINE=InnoDB');
|
|
expect(sql).not.toContain('DEFAULT CHARSET');
|
|
expect(sql).not.toContain('AUTO_INCREMENT');
|
|
expect(sql).not.toContain('`');
|
|
});
|
|
});
|