implement SQL export in prisma api

This commit is contained in:
Dobromir Popov
2024-04-30 14:58:07 +03:00
parent 60a3e770f9
commit e6883428c2

View File

@ -17,6 +17,37 @@ const parseQueryParams = (query: any) => {
}; };
}; };
// Helper function to escape and quote strings, handle nulls and other types
const serializeValue = (value) => {
if (value === null || value === undefined) {
return 'NULL';
}
if (typeof value === 'string') {
return `'${value.replace(/'/g, "''")}'`; // escape single quotes in SQL string
}
if (typeof value === 'boolean') {
return value ? 'true' : 'false';
}
if (typeof value === 'number') {
return value;
}
if (Array.isArray(value) || typeof value === 'object') {
// Convert arrays and objects to JSON strings and escape them
return `'${JSON.stringify(value).replace(/'/g, "''")}'`;
}
return value;
};
// Function to generate SQL INSERT statements from data
const generateSQL = (data, tableName) => {
return data.map(item => {
const columns = Object.keys(item).join(", ");
const values = Object.values(item).map(serializeValue).join(", ");
return `INSERT INTO ${tableName} (${columns}) VALUES (${values});`;
}).join("\n");
};
export default async function handler(req: NextApiRequest, res: NextApiResponse) { export default async function handler(req: NextApiRequest, res: NextApiResponse) {
const prisma: PrismaClient = common.getPrismaClient(); const prisma: PrismaClient = common.getPrismaClient();
const modelArray = (req.query.model || (req.body && req.body.model)) as string[]; const modelArray = (req.query.model || (req.body && req.body.model)) as string[];
@ -39,7 +70,15 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
throw new Error(`Model ${modelName} not found in Prisma client.`); throw new Error(`Model ${modelName} not found in Prisma client.`);
} }
const result = await prisma[modelName].findMany(queryOptions); const result = await prisma[modelName].findMany(queryOptions);
res.status(200).json(result); if (req.query.format === 'sql') {
// Generate SQL if requested via query parameter
const sql = generateSQL(result, modelName);
res.setHeader('Content-Type', 'application/sql');
res.send(sql);
} else {
// Normal JSON response
res.status(200).json(result);
}
} catch (error) { } catch (error) {
console.error(error); console.error(error);
res.status(500).json({ error: error.message }); res.status(500).json({ error: error.message });