mirror of
https://github.com/DeNNiiInc/dbgate.git
synced 2026-04-19 20:06:00 +00:00
load recursive model
This commit is contained in:
@@ -4,13 +4,14 @@ const yaml = require('js-yaml');
|
||||
const { databaseInfoFromYamlModel, DatabaseAnalyser } = require('dbgate-tools');
|
||||
const { startsWith } = require('lodash');
|
||||
const { archivedir, resolveArchiveFolder } = require('./directories');
|
||||
const loadFilesRecursive = require('./loadFilesRecursive');
|
||||
|
||||
async function importDbModel(inputDir) {
|
||||
const files = [];
|
||||
|
||||
const dir = inputDir.startsWith('archive:') ? resolveArchiveFolder(inputDir.substring('archive:'.length)) : inputDir;
|
||||
|
||||
for (const name of await fs.readdir(dir)) {
|
||||
for (const name of await loadFilesRecursive(dir)) {
|
||||
if (name.endsWith('.table.yaml') || name.endsWith('.sql')) {
|
||||
const text = await fs.readFile(path.join(dir, name), { encoding: 'utf-8' });
|
||||
|
||||
|
||||
20
packages/api/src/utility/loadFilesRecursive.js
Normal file
20
packages/api/src/utility/loadFilesRecursive.js
Normal file
@@ -0,0 +1,20 @@
|
||||
const fs = require('fs-extra');
|
||||
const stream = require('stream');
|
||||
const readline = require('readline');
|
||||
const path = require('path');
|
||||
|
||||
async function loadFilesRecursive(dir, root = null) {
|
||||
if (!root) root = dir;
|
||||
if (!root.endsWith(path.sep)) root += path.sep;
|
||||
const dirents = await fs.readdir(dir, { withFileTypes: true });
|
||||
const files = await Promise.all(
|
||||
dirents.map(dirent => {
|
||||
const res = path.join(dir, dirent.name);
|
||||
return dirent.isDirectory() ? loadFilesRecursive(res, root) : res;
|
||||
})
|
||||
);
|
||||
const flatten = Array.prototype.concat(...files);
|
||||
return flatten.map(file => (file.startsWith(root) ? file.substr(root.length) : file));
|
||||
}
|
||||
|
||||
module.exports = loadFilesRecursive;
|
||||
Reference in New Issue
Block a user