11e9ecac1a
* wip * wip * wip * wip * wip * add buffer * wip * wip * fixes * fix * wip * group validation * fix group issues * docs: add groups
67 lines
1.6 KiB
TypeScript
67 lines
1.6 KiB
TypeScript
import fs from 'node:fs';
|
|
import path from 'node:path';
|
|
import { TABLE_NAMES } from '../src/clickhouse/client';
|
|
import {
|
|
addColumns,
|
|
createTable,
|
|
runClickhouseMigrationCommands,
|
|
} from '../src/clickhouse/migration';
|
|
import { getIsCluster } from './helpers';
|
|
|
|
export async function up() {
|
|
const isClustered = getIsCluster();
|
|
|
|
const sqls: string[] = [
|
|
...addColumns(
|
|
'events',
|
|
['`groups` Array(String) DEFAULT [] CODEC(ZSTD(3)) AFTER session_id'],
|
|
isClustered
|
|
),
|
|
...addColumns(
|
|
'sessions',
|
|
['`groups` Array(String) DEFAULT [] CODEC(ZSTD(3)) AFTER device_id'],
|
|
isClustered
|
|
),
|
|
...addColumns(
|
|
'profiles',
|
|
['`groups` Array(String) DEFAULT [] CODEC(ZSTD(3)) AFTER project_id'],
|
|
isClustered
|
|
),
|
|
...createTable({
|
|
name: TABLE_NAMES.groups,
|
|
columns: [
|
|
'`id` String',
|
|
'`project_id` String',
|
|
'`type` String',
|
|
'`name` String',
|
|
'`properties` Map(String, String)',
|
|
'`created_at` DateTime',
|
|
'`version` UInt64',
|
|
'`deleted` UInt8 DEFAULT 0',
|
|
],
|
|
engine: 'ReplacingMergeTree(version, deleted)',
|
|
orderBy: ['project_id', 'id'],
|
|
distributionHash: 'cityHash64(project_id, id)',
|
|
replicatedVersion: '1',
|
|
isClustered,
|
|
}),
|
|
];
|
|
|
|
fs.writeFileSync(
|
|
path.join(import.meta.filename.replace('.ts', '.sql')),
|
|
sqls
|
|
.map((sql) =>
|
|
sql
|
|
.trim()
|
|
.replace(/;$/, '')
|
|
.replace(/\n{2,}/g, '\n')
|
|
.concat(';')
|
|
)
|
|
.join('\n\n---\n\n')
|
|
);
|
|
|
|
if (!process.argv.includes('--dry')) {
|
|
await runClickhouseMigrationCommands(sqls);
|
|
}
|
|
}
|