db import test + fix for postgres

This commit is contained in:
Jan Prochazka
2021-11-11 08:06:01 +01:00
parent d8cd3a78e4
commit 6baeb58500
5 changed files with 57 additions and 12 deletions

View File

@@ -5,7 +5,7 @@ async function fakeObjectReader({ delay = 0 } = {}) {
objectMode: true,
});
function doWrite() {
pass.write({ columns: [{ columnName: 'id' }, { columnName: 'country' }] });
pass.write({ columns: [{ columnName: 'id' }, { columnName: 'country' }], __isStreamHeader: true });
pass.write({ id: 1, country: 'Czechia' });
pass.write({ id: 2, country: 'Austria' });
pass.write({ country: 'Germany', id: 3 });

View File

@@ -1,13 +1,15 @@
const { fullNameToString } = require('dbgate-tools');
const requireEngineDriver = require('../utility/requireEngineDriver');
const { decryptConnection } = require('../utility/crypting');
const connectUtility = require('../utility/connectUtility');
async function tableWriter({ connection, schemaName, pureName, ...options }) {
async function tableWriter({ connection, schemaName, pureName, driver, systemConnection, ...options }) {
console.log(`Writing table ${fullNameToString({ schemaName, pureName })}`);
const driver = requireEngineDriver(connection);
const pool = await connectUtility(driver, connection);
if (!driver) {
driver = requireEngineDriver(connection);
}
const pool = systemConnection || (await connectUtility(driver, connection));
console.log(`Connected.`);
return await driver.writeTable(pool, { schemaName, pureName }, options);
}