watson/umami
1
0
Fork 0
mirror of https://github.com/umami-software/umami.git synced 2026-02-21 04:55:36 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

feat: support mongodb connection

Browse source
This commit is contained in:
Joseph Lee 2023-05-15 15:49:21 +09:00
parent 586529a5ca
commit bb30b43a8e
9 changed files with 370 additions and 30 deletions
Download patch file Download diff file Expand all files Collapse all files

27
scripts/check-db.js
View file

@ -15,6 +15,8 @@ function getDatabaseType(url = process.env.DATABASE_URL) {
if (type === 'postgres') {
return 'postgresql';
} else if (type === 'mongodb+srv') {
return 'mongodb';
}
return type;
@ -50,10 +52,19 @@ async function checkConnection() {
}
async function checkDatabaseVersion(databaseType) {
const query = await prisma.$queryRaw`select version() as version`;
const version = semver.valid(semver.coerce(query[0].version));
let version;
if (databaseType === 'mongodb') {
const query = await prisma.$runCommandRaw({
serverStatus: 1,
});
version = semver.valid(query.version);
} else {
const query = await prisma.$queryRaw`select version() as version`;
version = semver.valid(semver.coerce(query[0].version));
}
const minVersion = databaseType === 'postgresql' ? '9.4.0' : '5.7.0';
const minVersion =
databaseType === 'postgresql' ? '9.4.0' : databaseType === 'mongodb' ? '3.0.0' : '5.7.0';
if (semver.lt(version, minVersion)) {
throw new Error(
@ -65,6 +76,10 @@ async function checkDatabaseVersion(databaseType) {
}
async function checkV1Tables() {
if (databaseType === 'mongodb') {
// Ignore
}
try {
await prisma.$queryRaw`select * from account limit 1`;
@ -78,7 +93,11 @@ async function checkV1Tables() {
}
async function applyMigration() {
console.log(execSync('prisma migrate deploy').toString());
if (databaseType === 'mongodb') {
console.log(execSync('prisma db push').toString());
} else {
console.log(execSync('prisma migrate deploy').toString());
}
success('Database is up to date.');
}

4
scripts/copy-db-files.js
View file

@ -9,6 +9,8 @@ function getDatabaseType(url = process.env.DATABASE_URL) {
if (type === 'postgres') {
return 'postgresql';
} else if (type === 'mongodb+srv') {
return 'mongodb';
}
return type;
@ -16,7 +18,7 @@ function getDatabaseType(url = process.env.DATABASE_URL) {
const databaseType = getDatabaseType();
if (!databaseType || !['mysql', 'postgresql'].includes(databaseType)) {
if (!databaseType || !['mysql', 'postgresql', 'mongodb'].includes(databaseType)) {
throw new Error('Missing or invalid database');
}