feat: 데이터베이스 및 Redis 연결 테스트 스크립트 추가

2025-12-28 12:10:39 +00:00
parent 2caba52b7a
commit 04c250fed5
1 changed files with 148 additions and 0 deletions
@@ -0,0 +1,148 @@
+import fs from 'node:fs/promises';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { createRequire } from 'node:module';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+const DEFAULT_ENV_FILE = path.resolve(__dirname, '..', '..', '.env.ci');
+const COMMON_PACKAGE_JSON = path.resolve(
+    __dirname,
+    '..',
+    '..',
+    'packages',
+    'common',
+    'package.json'
+);
+const commonRequire = createRequire(COMMON_PACKAGE_JSON);
+const { PrismaClient } = commonRequire('@prisma/client');
+const { createClient } = commonRequire('redis');
+
+const parseEnvFile = (rawText) => {
+    const env = {};
+    const lines = rawText.split(/\r?\n/);
+    for (const line of lines) {
+        const trimmed = line.trim();
+        if (!trimmed || trimmed.startsWith('#')) {
+            continue;
+        }
+        const index = trimmed.indexOf('=');
+        if (index < 0) {
+            continue;
+        }
+        const key = trimmed.slice(0, index).trim();
+        let value = trimmed.slice(index + 1).trim();
+        if (
+            (value.startsWith('"') && value.endsWith('"')) ||
+            (value.startsWith("'") && value.endsWith("'"))
+        ) {
+            value = value.slice(1, -1);
+        }
+        env[key] = value;
+    }
+    return env;
+};
+
+const loadEnvFile = async (envFile) => {
+    const text = await fs.readFile(envFile, 'utf8');
+    return parseEnvFile(text);
+};
+
+const maskUrlPassword = (url) => url.replace(/:(?:[^@]+)@/, ':***@');
+
+const resolveDatabaseUrl = (env) => {
+    if (process.env.DATABASE_URL) {
+        return process.env.DATABASE_URL;
+    }
+    if (env.DATABASE_URL) {
+        return env.DATABASE_URL;
+    }
+    const host = env.POSTGRES_HOST ?? '127.0.0.1';
+    const port = env.POSTGRES_PORT ?? '15432';
+    const user = env.POSTGRES_USER ?? 'sammo';
+    const password = env.POSTGRES_PASSWORD ?? '';
+    const dbName = env.POSTGRES_DB ?? 'sammo';
+    return `postgresql://${user}:${password}@${host}:${port}/${dbName}?schema=public`;
+};
+
+const resolveRedisUrl = (env) => {
+    if (process.env.REDIS_URL) {
+        return process.env.REDIS_URL;
+    }
+    if (env.REDIS_URL) {
+        return env.REDIS_URL;
+    }
+    const host = env.REDIS_HOST ?? '127.0.0.1';
+    const port = env.REDIS_PORT ?? '16379';
+    const db = env.REDIS_DB ?? '0';
+    const password = env.REDIS_PASSWORD ?? '';
+    return `redis://:${password}@${host}:${port}/${db}`;
+};
+
+const testPostgres = async (databaseUrl) => {
+    const prisma = new PrismaClient({
+        datasources: {
+            db: {
+                url: databaseUrl,
+            },
+        },
+    });
+
+    await prisma.$connect();
+    await prisma.$queryRawUnsafe('SELECT 1');
+    await prisma.$disconnect();
+};
+
+const testRedis = async (redisUrl) => {
+    const client = createClient({ url: redisUrl });
+    await client.connect();
+    await client.ping();
+    await client.quit();
+};
+
+const main = async () => {
+    const envFile = process.env.SAMMO_ENV_FILE ?? DEFAULT_ENV_FILE;
+    let env = {};
+
+    try {
+        env = await loadEnvFile(envFile);
+    } catch (error) {
+        if (error?.code !== 'ENOENT') {
+            throw error;
+        }
+    }
+
+    const databaseUrl = resolveDatabaseUrl(env);
+    const redisUrl = resolveRedisUrl(env);
+
+    console.log(`Using env file: ${envFile}`);
+    console.log(`DATABASE_URL: ${maskUrlPassword(databaseUrl)}`);
+    console.log(`REDIS_URL: ${maskUrlPassword(redisUrl)}`);
+
+    const errors = [];
+
+    try {
+        await testPostgres(databaseUrl);
+        console.log('Postgres connection: OK');
+    } catch (error) {
+        console.error('Postgres connection: FAILED');
+        console.error(error?.message ?? error);
+        errors.push('postgres');
+    }
+
+    try {
+        await testRedis(redisUrl);
+        console.log('Redis connection: OK');
+    } catch (error) {
+        console.error('Redis connection: FAILED');
+        console.error(error?.message ?? error);
+        errors.push('redis');
+    }
+
+    if (errors.length > 0) {
+        process.exitCode = 1;
+    }
+};
+
+await main();