53 lines
1.6 KiB
JavaScript
53 lines
1.6 KiB
JavaScript
const query = require("../db.js");
|
|
const fs = require('fs');
|
|
const zlib = require('zlib');
|
|
const { toSnakeCase } = require("./utils.js");
|
|
|
|
const getGzFile = async (req, res) => {
|
|
try {
|
|
const { client, tableName } = req.params;
|
|
const dBtableName = `${client}_${toSnakeCase(tableName)}`;
|
|
|
|
let data;
|
|
|
|
// Check if table exists
|
|
const tableCheck = await query(`
|
|
SELECT EXISTS (SELECT 1 FROM pg_tables WHERE tablename = $1)
|
|
AS table_exists`, [dBtableName]);
|
|
if (!tableCheck.rows[0]["table_exists"]) {
|
|
return res.status(404).json({ error: `Data collection ${tableName} not found.` });
|
|
}
|
|
|
|
// Get data and create the json file
|
|
const result = await query(`SELECT data FROM ${dBtableName}`);
|
|
if (result.rows.length > 0) {
|
|
data = result.rows.map(row => row.data);
|
|
} else {
|
|
res.status(404).json({ message: "Not found." });
|
|
}
|
|
|
|
// Make json file
|
|
const jsonFile = `./${tableName}.json.gz`;
|
|
const compressedData = zlib.gzipSync(JSON.stringify(data));
|
|
fs.writeFileSync(jsonFile, compressedData);
|
|
|
|
// Deliver the json file
|
|
if (fs.existsSync(jsonFile)) {
|
|
const compressedData = fs.readFileSync(jsonFile);
|
|
res.setHeader('Content-Type', 'application/gzip');
|
|
res.setHeader('Content-Disposition', `attachment; filename=${tableName}.json.gz`);
|
|
res.send(compressedData);
|
|
fs.unlinkSync(jsonFile);
|
|
} else {
|
|
res.sendStatus(404);
|
|
}
|
|
} catch (err) {
|
|
console.error("Error handling the request:", err);
|
|
res
|
|
.status(500)
|
|
.json({ error: "An error occurred while processing the request" });
|
|
}
|
|
}
|
|
|
|
module.exports = getGzFile;
|