78 lines
2.5 KiB
JavaScript
78 lines
2.5 KiB
JavaScript
const UserError = require('vn-loopback/util/user-error');
|
|
const fs = require('fs-extra');
|
|
const fastCsv = require("fast-csv");
|
|
const axios = require('axios');
|
|
const path = require('path');
|
|
const { pipeline } = require('stream/promises');
|
|
|
|
module.exports = Self => {
|
|
Self.remoteMethodCtx('syncData', {
|
|
description: 'Sync schema data from external provider',
|
|
accessType: 'WRITE',
|
|
returns: {
|
|
type: 'object',
|
|
root: true
|
|
},
|
|
http: {
|
|
path: `/syncData`,
|
|
verb: 'POST'
|
|
}
|
|
});
|
|
|
|
Self.syncData = async ctx => {
|
|
const models = Self.app.models;
|
|
let tx;
|
|
try {
|
|
const floricodeConfig = await models.FloricodeConfig.findOne();
|
|
if (!floricodeConfig) throw new UserError(`Floricode service is not configured`);
|
|
|
|
const tables = await models.TableMultiConfig.find();
|
|
if (!tables?.length) throw new UserError(`No tables to sync`);
|
|
|
|
const token = await getToken(floricodeConfig);
|
|
for (const table of tables) {
|
|
const data = await getData(floricodeConfig.url, table.method, token);
|
|
if (!data) continue;
|
|
|
|
tx = await Self.beginTransaction({});
|
|
const options = {transaction: tx, userId: ctx.req.accessToken.userId};
|
|
|
|
await Self.rawSql(`DELETE FROM edi.??`, [table.toTable]);
|
|
const ws = fs.createWriteStream(path.join(__dirname, `/${table.toTable}.csv`));
|
|
await pipeline(fastCsv.write(data, { delimiter: ';' }), ws);
|
|
const templatePath = path.join(__dirname, `./syncSql/${table.toTable}.sql`);
|
|
const sqlTemplate = await fs.readFile(templatePath, 'utf8');
|
|
await Self.rawSql(sqlTemplate, [ws.path], options);
|
|
|
|
fs.remove(ws.path);
|
|
await table.updateAttribute('updated', Date.vnNew(), options);
|
|
await tx.commit();
|
|
}
|
|
} catch (e) {
|
|
if (tx) await tx.rollback();
|
|
throw e;
|
|
}
|
|
};
|
|
|
|
const getToken = async ({ url, user, password }) =>
|
|
(await axios.post(`${url}/oauth/token`,
|
|
{ grant_type: 'client_credentials', client_id: user, client_secret: password },
|
|
{ headers: { 'Content-Type': 'application/x-www-form-urlencoded' } }
|
|
)).data.access_token;
|
|
|
|
async function getData(url, method, token) {
|
|
let data = [];
|
|
let count = 0;
|
|
const maxCount = (await get(`${url}/v2/${method}?$count=true`, token))["@odata.count"];
|
|
while (count < maxCount) {
|
|
const response = await get(`${url}/v2/${method}?$skip=${count}`, token)
|
|
data.push(...response.value);
|
|
count += response.value.length;
|
|
}
|
|
return data;
|
|
}
|
|
|
|
const get = async (url, token) =>
|
|
(await axios.get(url, { headers: { Authorization: `Bearer ${token}` } })).data;
|
|
};
|