@eggjs/dal-runtime 4.0.0-beta.35 → 4.0.0-beta.36

package/dist/CodeGenerator.js

@@ -1,150 +1,119 @@
-import fs from 'node:fs/promises';
-import path from 'node:path';
-import { PrototypeUtil } from '@eggjs/core-decorator';
-import { ColumnModel, TableModel } from '@eggjs/dal-decorator';
-import { Templates } from '@eggjs/tegg-types';
-import js_beautify from 'js-beautify';
-import _ from 'lodash';
-import nunjucks, {} from 'nunjucks';
-import { SqlGenerator } from "./SqlGenerator.js";
 import { TemplateUtil } from "./TemplateUtil.js";
-export class CodeGenerator {
-    moduleDir;
-    moduleName;
-    teggPkg;
-    dalPkg;
-    constructor(options) {
-        this.moduleDir = options.moduleDir;
-        this.moduleName = options.moduleName;
-        this.teggPkg = options.teggPkg ?? '@eggjs/tegg';
-        this.dalPkg = options.dalPkg ?? '@eggjs/tegg/dal';
-        this.createNunjucksEnv();
-    }
-    njkEnv;
-    createNunjucksEnv() {
-        this.njkEnv = nunjucks.configure(path.join(__dirname, './templates'), {
-            autoescape: false,
-        });
-        this.njkEnv.addFilter('pascalCase', (name) => _.upperFirst(_.camelCase(name)));
-        this.njkEnv.addFilter('camelCase', (name) => _.camelCase(name));
-        this.njkEnv.addFilter('dbTypeToTSType', TemplateUtil.dbTypeToTsType);
-    }
-    genCode(tplName, filePath, tableModel) {
-        let tableModelAbsolutePath = PrototypeUtil.getFilePath(tableModel.clazz);
-        tableModelAbsolutePath = tableModelAbsolutePath.substring(0, tableModelAbsolutePath.length - 3);
-        const data = {
-            table: tableModel,
-            file: filePath,
-            fileName: path.basename(filePath),
-            clazzName: tableModel.clazz.name,
-            moduleName: this.moduleName,
-            teggPkg: this.teggPkg,
-            dalPkg: this.dalPkg,
-            id: tableModel.columns.find((t) => t.propertyName === 'id'),
-            primaryIndex: tableModel.getPrimary(),
-            tableModelPath: TemplateUtil.importPath(tableModelAbsolutePath, path.dirname(filePath)) + '.ts',
-            extensionPath: `../../extension/${tableModel.clazz.name}Extension.ts`,
-            structurePath: `../../structure/${tableModel.clazz.name}.json`,
-            // FIXME: not support with { type: 'json' } in nunjucks
-            // importStructurePathWithJSON: `import Structure from '../../structure/${tableModel.clazz.name}.json' with { type: 'json' }`,
-            sqlPath: `../../structure/${tableModel.clazz.name}.sql`,
-            columnMap: tableModel.columns.reduce((p, c) => {
-                p[c.propertyName] = c;
-                return p;
-            }, {}),
-        };
-        return this.njkEnv.render(`${tplName}.njk`, data);
-    }
-    async generate(tableModel) {
-        let dalDir;
-        try {
-            await fs.access(path.join(this.moduleDir, 'src'));
-            dalDir = path.join(this.moduleDir, 'src/dal');
-        }
-        catch {
-            dalDir = path.join(this.moduleDir, 'dal');
-        }
-        // const tableName = tableModel.name;
-        // const clazzName = tableModel.clazz.name;
-        const clazzFileName = path.basename(PrototypeUtil.getFilePath(tableModel.clazz));
-        const baseFileName = path.basename(clazzFileName, '.ts');
-        // 要动的一些文件
-        const paths = {
-            // e.g. app/dal/dao/base/example.ts
-            baseBizDAO: path.join(dalDir, `dao/base/Base${baseFileName}DAO.ts`),
-            // e.g. app/dal/dao/example.ts
-            bizDAO: path.join(dalDir, `dao/${baseFileName}DAO.ts`),
-            // e.g. app/dal/extension/example.ts
-            extension: path.join(dalDir, `extension/${baseFileName}Extension.ts`),
-            // e.g. app/dal/structure/example.json
-            structure: path.join(dalDir, `structure/${baseFileName}.json`),
-            // e.g. app/dal/structure/example.sql
-            structureSql: path.join(dalDir, `structure/${baseFileName}.sql`),
-        };
-        // 建立 structure 文件
-        await fs.mkdir(path.dirname(paths.structure), {
-            recursive: true,
-        });
-        await fs.writeFile(paths.structure, JSON.stringify(tableModel, null, 2), 'utf8');
-        const sqlGenerator = new SqlGenerator();
-        const structureSql = sqlGenerator.generate(tableModel);
-        await fs.writeFile(paths.structureSql, structureSql, 'utf8');
-        const codes = [
-            {
-                templates: Templates.BASE_DAO,
-                filePath: paths.baseBizDAO,
-                beautify: true,
-                overwrite: true,
-            },
-            {
-                templates: Templates.DAO,
-                filePath: paths.bizDAO,
-                beautify: true,
-                overwrite: false,
-            },
-            {
-                templates: Templates.EXTENSION,
-                filePath: paths.extension,
-                beautify: false,
-                overwrite: false,
-            },
-        ];
-        for (const { templates, filePath, beautify, overwrite } of codes) {
-            await fs.mkdir(path.dirname(filePath), {
-                recursive: true,
-            });
-            const code = this.genCode(templates, filePath, tableModel);
-            let beautified;
-            if (beautify) {
-                beautified = js_beautify(code, {
-                    brace_style: 'preserve-inline',
-                    indent_size: 2,
-                    jslint_happy: true,
-                    preserve_newlines: false,
-                });
-            }
-            else {
-                beautified = code;
-            }
-            beautified = beautified
-                .replace(/( )*\/\/ empty-line( )*/g, '')
-                .replace(/Promise( )*<( )*(.+?)( )*>/g, 'Promise<$3>')
-                .replace(/Optional( )*<( )*(.+?)( )*>/g, 'Optional<$3>')
-                .replace(/Record( )*<( )*(.+?)( )*>/g, 'Record<$3>')
-                .replace(/Partial( )*<( )*(.+?)( )*>/g, 'Partial<$3>')
-                .replace(/DataSource( )*<( )*(.+?)( )*>/g, 'DataSource<$3>')
-                .replace(/ \? :/g, '?:');
-            if (overwrite !== true) {
-                try {
-                    await fs.access(filePath);
-                    continue;
-                }
-                catch {
-                    // file not exists
-                }
-            }
-            await fs.writeFile(filePath, beautified, 'utf8');
-        }
-    }
-}
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiQ29kZUdlbmVyYXRvci5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uL3NyYy9Db2RlR2VuZXJhdG9yLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxNQUFNLGtCQUFrQixDQUFDO0FBQ2xDLE9BQU8sSUFBSSxNQUFNLFdBQVcsQ0FBQztBQUU3QixPQUFPLEVBQUUsYUFBYSxFQUFFLE1BQU0sdUJBQXVCLENBQUM7QUFDdEQsT0FBTyxFQUFFLFdBQVcsRUFBRSxVQUFVLEVBQUUsTUFBTSxzQkFBc0IsQ0FBQztBQUMvRCxPQUFPLEVBQUUsU0FBUyxFQUFFLE1BQU0sbUJBQW1CLENBQUM7QUFFOUMsT0FBTyxXQUFXLE1BQU0sYUFBYSxDQUFDO0FBQ3RDLE9BQU8sQ0FBQyxNQUFNLFFBQVEsQ0FBQztBQUN2QixPQUFPLFFBQVEsRUFBRSxFQUFvQixNQUFNLFVBQVUsQ0FBQztBQUV0RCxPQUFPLEVBQUUsWUFBWSxFQUFFLE1BQU0sbUJBQW1CLENBQUM7QUFDakQsT0FBTyxFQUFFLFlBQVksRUFBRSxNQUFNLG1CQUFtQixDQUFDO0FBRWpELE1BQU0sT0FBTyxhQUFhO0lBQ1AsU0FBUyxDQUFTO0lBQ2xCLFVBQVUsQ0FBUztJQUNuQixPQUFPLENBQVM7SUFDaEIsTUFBTSxDQUFTO0lBRWhDLFlBQVksT0FBNkI7UUFDdkMsSUFBSSxDQUFDLFNBQVMsR0FBRyxPQUFPLENBQUMsU0FBUyxDQUFDO1FBQ25DLElBQUksQ0FBQyxVQUFVLEdBQUcsT0FBTyxDQUFDLFVBQVUsQ0FBQztRQUNyQyxJQUFJLENBQUMsT0FBTyxHQUFHLE9BQU8sQ0FBQyxPQUFPLElBQUksYUFBYSxDQUFDO1FBQ2hELElBQUksQ0FBQyxNQUFNLEdBQUcsT0FBTyxDQUFDLE1BQU0sSUFBSSxpQkFBaUIsQ0FBQztRQUNsRCxJQUFJLENBQUMsaUJBQWlCLEVBQUUsQ0FBQztJQUMzQixDQUFDO0lBRU8sTUFBTSxDQUFjO0lBRTVCLGlCQUFpQjtRQUNmLElBQUksQ0FBQyxNQUFNLEdBQUcsUUFBUSxDQUFDLFNBQVMsQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLFNBQVMsRUFBRSxhQUFhLENBQUMsRUFBRTtZQUNwRSxVQUFVLEVBQUUsS0FBSztTQUNsQixDQUFDLENBQUM7UUFDSCxJQUFJLENBQUMsTUFBTSxDQUFDLFNBQVMsQ0FBQyxZQUFZLEVBQUUsQ0FBQyxJQUFJLEVBQUUsRUFBRSxDQUFDLENBQUMsQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDLFNBQVMsQ0FBQyxJQUFJLENBQUMsQ0FBQyxDQUFDLENBQUM7UUFDL0UsSUFBSSxDQUFDLE1BQU0sQ0FBQyxTQUFTLENBQUMsV0FBVyxFQUFFLENBQUMsSUFBSSxFQUFFLEVBQUUsQ0FBQyxDQUFDLENBQUMsU0FBUyxDQUFDLElBQUksQ0FBQyxDQUFDLENBQUM7UUFDaEUsSUFBSSxDQUFDLE1BQU0sQ0FBQyxTQUFTLENBQUMsZ0JBQWdCLEVBQUUsWUFBWSxDQUFDLGNBQWMsQ0FBQyxDQUFDO0lBQ3ZFLENBQUM7SUFFRCxPQUFPLENBQUMsT0FBa0IsRUFBRSxRQUFnQixFQUFFLFVBQXNCO1FBQ2xFLElBQUksc0JBQXNCLEdBQUcsYUFBYSxDQUFDLFdBQVcsQ0FBQyxVQUFVLENBQUMsS0FBSyxDQUFFLENBQUM7UUFDMUUsc0JBQXNCLEdBQUcsc0JBQXNCLENBQUMsU0FBUyxDQUFDLENBQUMsRUFBRSxzQkFBc0IsQ0FBQyxNQUFNLEdBQUcsQ0FBQyxDQUFDLENBQUM7UUFDaEcsTUFBTSxJQUFJLEdBQUc7WUFDWCxLQUFLLEVBQUUsVUFBVTtZQUNqQixJQUFJLEVBQUUsUUFBUTtZQUNkLFFBQVEsRUFBRSxJQUFJLENBQUMsUUFBUSxDQUFDLFFBQVEsQ0FBQztZQUNqQyxTQUFTLEVBQUUsVUFBVSxDQUFDLEtBQUssQ0FBQyxJQUFJO1lBQ2hDLFVBQVUsRUFBRSxJQUFJLENBQUMsVUFBVTtZQUMzQixPQUFPLEVBQUUsSUFBSSxDQUFDLE9BQU87WUFDckIsTUFBTSxFQUFFLElBQUksQ0FBQyxNQUFNO1lBQ25CLEVBQUUsRUFBRSxVQUFVLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxDQUFDLENBQUMsRUFBRSxFQUFFLENBQUMsQ0FBQyxDQUFDLFlBQVksS0FBSyxJQUFJLENBQUM7WUFDM0QsWUFBWSxFQUFFLFVBQVUsQ0FBQyxVQUFVLEVBQUU7WUFDckMsY0FBYyxFQUFFLFlBQVksQ0FBQyxVQUFVLENBQUMsc0JBQXNCLEVBQUUsSUFBSSxDQUFDLE9BQU8sQ0FBQyxRQUFRLENBQUMsQ0FBQyxHQUFHLEtBQUs7WUFDL0YsYUFBYSxFQUFFLG1CQUFtQixVQUFVLENBQUMsS0FBSyxDQUFDLElBQUksY0FBYztZQUNyRSxhQUFhLEVBQUUsbUJBQW1CLFVBQVUsQ0FBQyxLQUFLLENBQUMsSUFBSSxPQUFPO1lBQzlELHVEQUF1RDtZQUN2RCw4SEFBOEg7WUFDOUgsT0FBTyxFQUFFLG1CQUFtQixVQUFVLENBQUMsS0FBSyxDQUFDLElBQUksTUFBTTtZQUN2RCxTQUFTLEVBQUUsVUFBVSxDQUFDLE9BQU8sQ0FBQyxNQUFNLENBQThCLENBQUMsQ0FBQyxFQUFFLENBQUMsRUFBRSxFQUFFO2dCQUN6RSxDQUFDLENBQUMsQ0FBQyxDQUFDLFlBQVksQ0FBQyxHQUFHLENBQUMsQ0FBQztnQkFDdEIsT0FBTyxDQUFDLENBQUM7WUFDWCxDQUFDLEVBQUUsRUFBRSxDQUFDO1NBQ1AsQ0FBQztRQUNGLE9BQU8sSUFBSSxDQUFDLE1BQU0sQ0FBQyxNQUFNLENBQUMsR0FBRyxPQUFPLE1BQU0sRUFBRSxJQUFJLENBQUMsQ0FBQztJQUNwRCxDQUFDO0lBRUQsS0FBSyxDQUFDLFFBQVEsQ0FBQyxVQUFzQjtRQUNuQyxJQUFJLE1BQWMsQ0FBQztRQUNuQixJQUFJLENBQUM7WUFDSCxNQUFNLEVBQUUsQ0FBQyxNQUFNLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxJQUFJLENBQUMsU0FBUyxFQUFFLEtBQUssQ0FBQyxDQUFDLENBQUM7WUFDbEQsTUFBTSxHQUFHLElBQUksQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLFNBQVMsRUFBRSxTQUFTLENBQUMsQ0FBQztRQUNoRCxDQUFDO1FBQUMsTUFBTSxDQUFDO1lBQ1AsTUFBTSxHQUFHLElBQUksQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLFNBQVMsRUFBRSxLQUFLLENBQUMsQ0FBQztRQUM1QyxDQUFDO1FBRUQscUNBQXFDO1FBQ3JDLDJDQUEyQztRQUMzQyxNQUFNLGFBQWEsR0FBRyxJQUFJLENBQUMsUUFBUSxDQUFDLGFBQWEsQ0FBQyxXQUFXLENBQUMsVUFBVSxDQUFDLEtBQUssQ0FBRSxDQUFDLENBQUM7UUFDbEYsTUFBTSxZQUFZLEdBQUcsSUFBSSxDQUFDLFFBQVEsQ0FBQyxhQUFhLEVBQUUsS0FBSyxDQUFDLENBQUM7UUFFekQsVUFBVTtRQUNWLE1BQU0sS0FBSyxHQUFHO1lBQ1osbUNBQW1DO1lBQ25DLFVBQVUsRUFBRSxJQUFJLENBQUMsSUFBSSxDQUFDLE1BQU0sRUFBRSxnQkFBZ0IsWUFBWSxRQUFRLENBQUM7WUFDbkUsOEJBQThCO1lBQzlCLE1BQU0sRUFBRSxJQUFJLENBQUMsSUFBSSxDQUFDLE1BQU0sRUFBRSxPQUFPLFlBQVksUUFBUSxDQUFDO1lBQ3RELG9DQUFvQztZQUNwQyxTQUFTLEVBQUUsSUFBSSxDQUFDLElBQUksQ0FBQyxNQUFNLEVBQUUsYUFBYSxZQUFZLGNBQWMsQ0FBQztZQUNyRSxzQ0FBc0M7WUFDdEMsU0FBUyxFQUFFLElBQUksQ0FBQyxJQUFJLENBQUMsTUFBTSxFQUFFLGFBQWEsWUFBWSxPQUFPLENBQUM7WUFDOUQscUNBQXFDO1lBQ3JDLFlBQVksRUFBRSxJQUFJLENBQUMsSUFBSSxDQUFDLE1BQU0sRUFBRSxhQUFhLFlBQVksTUFBTSxDQUFDO1NBQ2pFLENBQUM7UUFFRixrQkFBa0I7UUFDbEIsTUFBTSxFQUFFLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsS0FBSyxDQUFDLFNBQVMsQ0FBQyxFQUFFO1lBQzVDLFNBQVMsRUFBRSxJQUFJO1NBQ2hCLENBQUMsQ0FBQztRQUNILE1BQU0sRUFBRSxDQUFDLFNBQVMsQ0FBQyxLQUFLLENBQUMsU0FBUyxFQUFFLElBQUksQ0FBQyxTQUFTLENBQUMsVUFBVSxFQUFFLElBQUksRUFBRSxDQUFDLENBQUMsRUFBRSxNQUFNLENBQUMsQ0FBQztRQUVqRixNQUFNLFlBQVksR0FBRyxJQUFJLFlBQVksRUFBRSxDQUFDO1FBQ3hDLE1BQU0sWUFBWSxHQUFHLFlBQVksQ0FBQyxRQUFRLENBQUMsVUFBVSxDQUFDLENBQUM7UUFDdkQsTUFBTSxFQUFFLENBQUMsU0FBUyxDQUFDLEtBQUssQ0FBQyxZQUFZLEVBQUUsWUFBWSxFQUFFLE1BQU0sQ0FBQyxDQUFDO1FBRTdELE1BQU0sS0FBSyxHQUFHO1lBQ1o7Z0JBQ0UsU0FBUyxFQUFFLFNBQVMsQ0FBQyxRQUFRO2dCQUM3QixRQUFRLEVBQUUsS0FBSyxDQUFDLFVBQVU7Z0JBQzFCLFFBQVEsRUFBRSxJQUFJO2dCQUNkLFNBQVMsRUFBRSxJQUFJO2FBQ2hCO1lBQ0Q7Z0JBQ0UsU0FBUyxFQUFFLFNBQVMsQ0FBQyxHQUFHO2dCQUN4QixRQUFRLEVBQUUsS0FBSyxDQUFDLE1BQU07Z0JBQ3RCLFFBQVEsRUFBRSxJQUFJO2dCQUNkLFNBQVMsRUFBRSxLQUFLO2FBQ2pCO1lBQ0Q7Z0JBQ0UsU0FBUyxFQUFFLFNBQVMsQ0FBQyxTQUFTO2dCQUM5QixRQUFRLEVBQUUsS0FBSyxDQUFDLFNBQVM7Z0JBQ3pCLFFBQVEsRUFBRSxLQUFLO2dCQUNmLFNBQVMsRUFBRSxLQUFLO2FBQ2pCO1NBQ0YsQ0FBQztRQUNGLEtBQUssTUFBTSxFQUFFLFNBQVMsRUFBRSxRQUFRLEVBQUUsUUFBUSxFQUFFLFNBQVMsRUFBRSxJQUFJLEtBQUssRUFBRSxDQUFDO1lBQ2pFLE1BQU0sRUFBRSxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQUMsT0FBTyxDQUFDLFFBQVEsQ0FBQyxFQUFFO2dCQUNyQyxTQUFTLEVBQUUsSUFBSTthQUNoQixDQUFDLENBQUM7WUFDSCxNQUFNLElBQUksR0FBRyxJQUFJLENBQUMsT0FBTyxDQUFDLFNBQVMsRUFBRSxRQUFRLEVBQUUsVUFBVSxDQUFDLENBQUM7WUFDM0QsSUFBSSxVQUFrQixDQUFDO1lBQ3ZCLElBQUksUUFBUSxFQUFFLENBQUM7Z0JBQ2IsVUFBVSxHQUFHLFdBQVcsQ0FBQyxJQUFJLEVBQUU7b0JBQzdCLFdBQVcsRUFBRSxpQkFBaUI7b0JBQzlCLFdBQVcsRUFBRSxDQUFDO29CQUNkLFlBQVksRUFBRSxJQUFJO29CQUNsQixpQkFBaUIsRUFBRSxLQUFLO2lCQUN6QixDQUFDLENBQUM7WUFDTCxDQUFDO2lCQUFNLENBQUM7Z0JBQ04sVUFBVSxHQUFHLElBQUksQ0FBQztZQUNwQixDQUFDO1lBQ0QsVUFBVSxHQUFHLFVBQVU7aUJBQ3BCLE9BQU8sQ0FBQywwQkFBMEIsRUFBRSxFQUFFLENBQUM7aUJBQ3ZDLE9BQU8sQ0FBQyw2QkFBNkIsRUFBRSxhQUFhLENBQUM7aUJBQ3JELE9BQU8sQ0FBQyw4QkFBOEIsRUFBRSxjQUFjLENBQUM7aUJBQ3ZELE9BQU8sQ0FBQyw0QkFBNEIsRUFBRSxZQUFZLENBQUM7aUJBQ25ELE9BQU8sQ0FBQyw2QkFBNkIsRUFBRSxhQUFhLENBQUM7aUJBQ3JELE9BQU8sQ0FBQyxnQ0FBZ0MsRUFBRSxnQkFBZ0IsQ0FBQztpQkFDM0QsT0FBTyxDQUFDLFFBQVEsRUFBRSxJQUFJLENBQUMsQ0FBQztZQUMzQixJQUFJLFNBQVMsS0FBSyxJQUFJLEVBQUUsQ0FBQztnQkFDdkIsSUFBSSxDQUFDO29CQUNILE1BQU0sRUFBRSxDQUFDLE1BQU0sQ0FBQyxRQUFRLENBQUMsQ0FBQztvQkFDMUIsU0FBUztnQkFDWCxDQUFDO2dCQUFDLE1BQU0sQ0FBQztvQkFDUCxrQkFBa0I7Z0JBQ3BCLENBQUM7WUFDSCxDQUFDO1lBQ0QsTUFBTSxFQUFFLENBQUMsU0FBUyxDQUFDLFFBQVEsRUFBRSxVQUFVLEVBQUUsTUFBTSxDQUFDLENBQUM7UUFDbkQsQ0FBQztJQUNILENBQUM7Q0FDRiJ9
+import { SqlGenerator } from "./SqlGenerator.js";
+import { ColumnModel, TableModel } from "@eggjs/dal-decorator";
+import { Templates } from "@eggjs/tegg-types";
+import _ from "lodash";
+import path from "node:path";
+import fs from "node:fs/promises";
+import { PrototypeUtil } from "@eggjs/core-decorator";
+import js_beautify from "js-beautify";
+import nunjucks from "nunjucks";
+
+//#region src/CodeGenerator.ts
+var CodeGenerator = class {
+	moduleDir;
+	moduleName;
+	teggPkg;
+	dalPkg;
+	constructor(options) {
+		this.moduleDir = options.moduleDir;
+		this.moduleName = options.moduleName;
+		this.teggPkg = options.teggPkg ?? "@eggjs/tegg";
+		this.dalPkg = options.dalPkg ?? "@eggjs/tegg/dal";
+		this.createNunjucksEnv();
+	}
+	njkEnv;
+	createNunjucksEnv() {
+		this.njkEnv = nunjucks.configure(path.join(__dirname, "./templates"), { autoescape: false });
+		this.njkEnv.addFilter("pascalCase", (name) => _.upperFirst(_.camelCase(name)));
+		this.njkEnv.addFilter("camelCase", (name) => _.camelCase(name));
+		this.njkEnv.addFilter("dbTypeToTSType", TemplateUtil.dbTypeToTsType);
+	}
+	genCode(tplName, filePath, tableModel) {
+		let tableModelAbsolutePath = PrototypeUtil.getFilePath(tableModel.clazz);
+		tableModelAbsolutePath = tableModelAbsolutePath.substring(0, tableModelAbsolutePath.length - 3);
+		const data = {
+			table: tableModel,
+			file: filePath,
+			fileName: path.basename(filePath),
+			clazzName: tableModel.clazz.name,
+			moduleName: this.moduleName,
+			teggPkg: this.teggPkg,
+			dalPkg: this.dalPkg,
+			id: tableModel.columns.find((t) => t.propertyName === "id"),
+			primaryIndex: tableModel.getPrimary(),
+			tableModelPath: TemplateUtil.importPath(tableModelAbsolutePath, path.dirname(filePath)) + ".ts",
+			extensionPath: `../../extension/${tableModel.clazz.name}Extension.ts`,
+			structurePath: `../../structure/${tableModel.clazz.name}.json`,
+			sqlPath: `../../structure/${tableModel.clazz.name}.sql`,
+			columnMap: tableModel.columns.reduce((p, c) => {
+				p[c.propertyName] = c;
+				return p;
+			}, {})
+		};
+		return this.njkEnv.render(`${tplName}.njk`, data);
+	}
+	async generate(tableModel) {
+		let dalDir;
+		try {
+			await fs.access(path.join(this.moduleDir, "src"));
+			dalDir = path.join(this.moduleDir, "src/dal");
+		} catch {
+			dalDir = path.join(this.moduleDir, "dal");
+		}
+		const clazzFileName = path.basename(PrototypeUtil.getFilePath(tableModel.clazz));
+		const baseFileName = path.basename(clazzFileName, ".ts");
+		const paths = {
+			baseBizDAO: path.join(dalDir, `dao/base/Base${baseFileName}DAO.ts`),
+			bizDAO: path.join(dalDir, `dao/${baseFileName}DAO.ts`),
+			extension: path.join(dalDir, `extension/${baseFileName}Extension.ts`),
+			structure: path.join(dalDir, `structure/${baseFileName}.json`),
+			structureSql: path.join(dalDir, `structure/${baseFileName}.sql`)
+		};
+		await fs.mkdir(path.dirname(paths.structure), { recursive: true });
+		await fs.writeFile(paths.structure, JSON.stringify(tableModel, null, 2), "utf8");
+		const structureSql = new SqlGenerator().generate(tableModel);
+		await fs.writeFile(paths.structureSql, structureSql, "utf8");
+		const codes = [
+			{
+				templates: Templates.BASE_DAO,
+				filePath: paths.baseBizDAO,
+				beautify: true,
+				overwrite: true
+			},
+			{
+				templates: Templates.DAO,
+				filePath: paths.bizDAO,
+				beautify: true,
+				overwrite: false
+			},
+			{
+				templates: Templates.EXTENSION,
+				filePath: paths.extension,
+				beautify: false,
+				overwrite: false
+			}
+		];
+		for (const { templates, filePath, beautify, overwrite } of codes) {
+			await fs.mkdir(path.dirname(filePath), { recursive: true });
+			const code = this.genCode(templates, filePath, tableModel);
+			let beautified;
+			if (beautify) beautified = js_beautify(code, {
+				brace_style: "preserve-inline",
+				indent_size: 2,
+				jslint_happy: true,
+				preserve_newlines: false
+			});
+			else beautified = code;
+			beautified = beautified.replace(/( )*\/\/ empty-line( )*/g, "").replace(/Promise( )*<( )*(.+?)( )*>/g, "Promise<$3>").replace(/Optional( )*<( )*(.+?)( )*>/g, "Optional<$3>").replace(/Record( )*<( )*(.+?)( )*>/g, "Record<$3>").replace(/Partial( )*<( )*(.+?)( )*>/g, "Partial<$3>").replace(/DataSource( )*<( )*(.+?)( )*>/g, "DataSource<$3>").replace(/ \? :/g, "?:");
+			if (overwrite !== true) try {
+				await fs.access(filePath);
+				continue;
+			} catch {}
+			await fs.writeFile(filePath, beautified, "utf8");
+		}
+	}
+};
+
+//#endregion
+export { CodeGenerator };

package/dist/DaoLoader.d.ts

@@ -1,4 +1,8 @@
-import { type BaseDaoType } from '@eggjs/tegg-types/dal';
-export declare class DaoLoader {
-    static loadDaos(moduleDir: string): Promise<Array<BaseDaoType>>;
+import { BaseDaoType } from "@eggjs/tegg-types/dal";
+
+//#region src/DaoLoader.d.ts
+declare class DaoLoader {
+  static loadDaos(moduleDir: string): Promise<Array<BaseDaoType>>;
 }
+//#endregion
+export { DaoLoader };

package/dist/DaoLoader.js

@@ -1,14 +1,16 @@
-import { DaoInfoUtil } from '@eggjs/dal-decorator';
-import { LoaderFactory } from '@eggjs/tegg-loader';
-import { EggLoadUnitType } from '@eggjs/tegg-types';
-import {} from '@eggjs/tegg-types/dal';
-export class DaoLoader {
-    static async loadDaos(moduleDir) {
-        const loader = LoaderFactory.createLoader(moduleDir, EggLoadUnitType.MODULE);
-        const clazzList = await loader.load();
-        return clazzList.filter((t) => {
-            return DaoInfoUtil.getIsDao(t);
-        });
-    }
-}
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiRGFvTG9hZGVyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vc3JjL0Rhb0xvYWRlci50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSxPQUFPLEVBQUUsV0FBVyxFQUFFLE1BQU0sc0JBQXNCLENBQUM7QUFDbkQsT0FBTyxFQUFFLGFBQWEsRUFBRSxNQUFNLG9CQUFvQixDQUFDO0FBQ25ELE9BQU8sRUFBRSxlQUFlLEVBQUUsTUFBTSxtQkFBbUIsQ0FBQztBQUNwRCxPQUFPLEVBQW9CLE1BQU0sdUJBQXVCLENBQUM7QUFFekQsTUFBTSxPQUFPLFNBQVM7SUFDcEIsTUFBTSxDQUFDLEtBQUssQ0FBQyxRQUFRLENBQUMsU0FBaUI7UUFDckMsTUFBTSxNQUFNLEdBQUcsYUFBYSxDQUFDLFlBQVksQ0FBQyxTQUFTLEVBQUUsZUFBZSxDQUFDLE1BQU0sQ0FBQyxDQUFDO1FBQzdFLE1BQU0sU0FBUyxHQUFHLE1BQU0sTUFBTSxDQUFDLElBQUksRUFBRSxDQUFDO1FBQ3RDLE9BQU8sU0FBUyxDQUFDLE1BQU0sQ0FBQyxDQUFDLENBQUMsRUFBb0IsRUFBRTtZQUM5QyxPQUFPLFdBQVcsQ0FBQyxRQUFRLENBQUMsQ0FBQyxDQUFDLENBQUM7UUFDakMsQ0FBQyxDQUFDLENBQUM7SUFDTCxDQUFDO0NBQ0YifQ==
+import { DaoInfoUtil } from "@eggjs/dal-decorator";
+import { EggLoadUnitType } from "@eggjs/tegg-types";
+import { LoaderFactory } from "@eggjs/tegg-loader";
+import "@eggjs/tegg-types/dal";
+
+//#region src/DaoLoader.ts
+var DaoLoader = class {
+	static async loadDaos(moduleDir) {
+		return (await LoaderFactory.createLoader(moduleDir, EggLoadUnitType.MODULE).load()).filter((t) => {
+			return DaoInfoUtil.getIsDao(t);
+		});
+	}
+};
+
+//#endregion
+export { DaoLoader };

package/dist/DataSource.d.ts

@@ -1,28 +1,32 @@
-import { TableModel } from '@eggjs/dal-decorator';
-import type { DataSource as IDataSource, PaginateData, SqlType } from '@eggjs/tegg-types';
-import { MysqlDataSource } from './MySqlDataSource.ts';
-import { TableSqlMap } from './TableSqlMap.ts';
-export interface ExecuteSql {
-    sql: string;
-    template: string;
-    sqlType: SqlType;
+import { MysqlDataSource } from "./MySqlDataSource.js";
+import { TableSqlMap } from "./TableSqlMap.js";
+import { TableModel } from "@eggjs/dal-decorator";
+import { DataSource, PaginateData, SqlType } from "@eggjs/tegg-types";
+
+//#region src/DataSource.d.ts
+interface ExecuteSql {
+  sql: string;
+  template: string;
+  sqlType: SqlType;
 }
-export declare class DataSource<T> implements IDataSource<T> {
-    #private;
-    private readonly tableModel;
-    private readonly mysqlDataSource;
-    private readonly sqlMap;
-    constructor(tableModel: TableModel<T>, mysqlDataSource: MysqlDataSource, sqlMap: TableSqlMap);
-    /**
-     * public for aop execute to implement sql hint append
-     * @param sqlName - sql name
-     * @param data - sql data
-     */
-    generateSql(sqlName: string, data: object): Promise<ExecuteSql>;
-    count(sqlName: string, data?: any): Promise<number>;
-    execute(sqlName: string, data?: any): Promise<Array<T>>;
-    executeRaw(sqlName: string, data?: any): Promise<Array<any>>;
-    executeScalar(sqlName: string, data?: any): Promise<T | null>;
-    executeRawScalar(sqlName: string, data?: any): Promise<any | null>;
-    paginate(sqlName: string, data: any, currentPage: number, perPageCount: number): Promise<PaginateData<T>>;
+declare class DataSource$1<T> implements DataSource<T> {
+  #private;
+  private readonly tableModel;
+  private readonly mysqlDataSource;
+  private readonly sqlMap;
+  constructor(tableModel: TableModel<T>, mysqlDataSource: MysqlDataSource, sqlMap: TableSqlMap);
+  /**
+  * public for aop execute to implement sql hint append
+  * @param sqlName - sql name
+  * @param data - sql data
+  */
+  generateSql(sqlName: string, data: object): Promise<ExecuteSql>;
+  count(sqlName: string, data?: any): Promise<number>;
+  execute(sqlName: string, data?: any): Promise<Array<T>>;
+  executeRaw(sqlName: string, data?: any): Promise<Array<any>>;
+  executeScalar(sqlName: string, data?: any): Promise<T | null>;
+  executeRawScalar(sqlName: string, data?: any): Promise<any | null>;
+  paginate(sqlName: string, data: any, currentPage: number, perPageCount: number): Promise<PaginateData<T>>;
 }
+//#endregion
+export { DataSource$1 as DataSource, ExecuteSql };

package/dist/DataSource.js

@@ -1,75 +1,72 @@
-import { TableModel } from '@eggjs/dal-decorator';
-import { MysqlDataSource } from "./MySqlDataSource.js";
+import "./MySqlDataSource.js";
 import { TableModelInstanceBuilder } from "./TableModelInstanceBuilder.js";
-import { TableSqlMap } from "./TableSqlMap.js";
-const PAGINATE_COUNT_WRAPPER = ['SELECT COUNT(0) as count FROM (', ') AS T'];
-export class DataSource {
-    tableModel;
-    mysqlDataSource;
-    sqlMap;
-    constructor(tableModel, mysqlDataSource, sqlMap) {
-        this.tableModel = tableModel;
-        this.mysqlDataSource = mysqlDataSource;
-        this.sqlMap = sqlMap;
-    }
-    /**
-     * public for aop execute to implement sql hint append
-     * @param sqlName - sql name
-     * @param data - sql data
-     */
-    async generateSql(sqlName, data) {
-        const sql = this.sqlMap.generate(sqlName, data, this.mysqlDataSource.timezone);
-        const sqlType = this.sqlMap.getType(sqlName);
-        const template = this.sqlMap.getTemplateString(sqlName);
-        return {
-            sql,
-            sqlType,
-            template,
-        };
-    }
-    async count(sqlName, data) {
-        const newData = Object.assign({ $$count: true }, data);
-        const executeSql = await this.generateSql(sqlName, newData);
-        return await this.#paginateCount(executeSql.sql);
-    }
-    async execute(sqlName, data) {
-        const executeSql = await this.generateSql(sqlName, data);
-        const rows = await this.mysqlDataSource.query(executeSql.sql);
-        return rows.map((t) => {
-            return TableModelInstanceBuilder.buildInstance(this.tableModel, t);
-        });
-    }
-    async executeRaw(sqlName, data) {
-        const executeSql = await this.generateSql(sqlName, data);
-        return await this.mysqlDataSource.query(executeSql.sql);
-    }
-    async executeScalar(sqlName, data) {
-        const ret = await this.execute(sqlName, data);
-        if (!Array.isArray(ret))
-            return ret || null;
-        return ret[0] || null;
-    }
-    async executeRawScalar(sqlName, data) {
-        const ret = await this.executeRaw(sqlName, data);
-        if (!Array.isArray(ret))
-            return (ret || null);
-        return ret[0] || null;
-    }
-    async paginate(sqlName, data, currentPage, perPageCount) {
-        const limit = `LIMIT ${(currentPage - 1) * perPageCount}, ${perPageCount}`;
-        const sql = (await this.generateSql(sqlName, data)).sql + ' ' + limit;
-        const countSql = (await this.generateSql(sqlName, Object.assign({ $$count: true }, data))).sql;
-        const ret = await Promise.all([this.mysqlDataSource.query(sql), this.#paginateCount(countSql)]);
-        return {
-            total: Number(ret[1]),
-            pageNum: currentPage,
-            rows: ret[0].map((t) => TableModelInstanceBuilder.buildInstance(this.tableModel, t)),
-        };
-    }
-    async #paginateCount(baseSQL) {
-        const sql = `${PAGINATE_COUNT_WRAPPER[0]}${baseSQL}${PAGINATE_COUNT_WRAPPER[1]}`;
-        const result = await this.mysqlDataSource.query(sql);
-        return result[0].count;
-    }
-}
-//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiRGF0YVNvdXJjZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uL3NyYy9EYXRhU291cmNlLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxVQUFVLEVBQUUsTUFBTSxzQkFBc0IsQ0FBQztBQUdsRCxPQUFPLEVBQUUsZUFBZSxFQUFFLE1BQU0sc0JBQXNCLENBQUM7QUFDdkQsT0FBTyxFQUFFLHlCQUF5QixFQUFFLE1BQU0sZ0NBQWdDLENBQUM7QUFDM0UsT0FBTyxFQUFFLFdBQVcsRUFBRSxNQUFNLGtCQUFrQixDQUFDO0FBUS9DLE1BQU0sc0JBQXNCLEdBQUcsQ0FBQyxpQ0FBaUMsRUFBRSxRQUFRLENBQUMsQ0FBQztBQUU3RSxNQUFNLE9BQU8sVUFBVTtJQUNKLFVBQVUsQ0FBZ0I7SUFDMUIsZUFBZSxDQUFrQjtJQUNqQyxNQUFNLENBQWM7SUFFckMsWUFBWSxVQUF5QixFQUFFLGVBQWdDLEVBQUUsTUFBbUI7UUFDMUYsSUFBSSxDQUFDLFVBQVUsR0FBRyxVQUFVLENBQUM7UUFDN0IsSUFBSSxDQUFDLGVBQWUsR0FBRyxlQUFlLENBQUM7UUFDdkMsSUFBSSxDQUFDLE1BQU0sR0FBRyxNQUFNLENBQUM7SUFDdkIsQ0FBQztJQUVEOzs7O09BSUc7SUFDSCxLQUFLLENBQUMsV0FBVyxDQUFDLE9BQWUsRUFBRSxJQUFZO1FBQzdDLE1BQU0sR0FBRyxHQUFHLElBQUksQ0FBQyxNQUFNLENBQUMsUUFBUSxDQUFDLE9BQU8sRUFBRSxJQUFJLEVBQUUsSUFBSSxDQUFDLGVBQWUsQ0FBQyxRQUFTLENBQUMsQ0FBQztRQUNoRixNQUFNLE9BQU8sR0FBRyxJQUFJLENBQUMsTUFBTSxDQUFDLE9BQU8sQ0FBQyxPQUFPLENBQUMsQ0FBQztRQUM3QyxNQUFNLFFBQVEsR0FBRyxJQUFJLENBQUMsTUFBTSxDQUFDLGlCQUFpQixDQUFDLE9BQU8sQ0FBQyxDQUFDO1FBQ3hELE9BQU87WUFDTCxHQUFHO1lBQ0gsT0FBTztZQUNQLFFBQVE7U0FDVCxDQUFDO0lBQ0osQ0FBQztJQUVELEtBQUssQ0FBQyxLQUFLLENBQUMsT0FBZSxFQUFFLElBQVU7UUFDckMsTUFBTSxPQUFPLEdBQUcsTUFBTSxDQUFDLE1BQU0sQ0FBQyxFQUFFLE9BQU8sRUFBRSxJQUFJLEVBQUUsRUFBRSxJQUFJLENBQUMsQ0FBQztRQUN2RCxNQUFNLFVBQVUsR0FBRyxNQUFNLElBQUksQ0FBQyxXQUFXLENBQUMsT0FBTyxFQUFFLE9BQU8sQ0FBQyxDQUFDO1FBQzVELE9BQU8sTUFBTSxJQUFJLENBQUMsY0FBYyxDQUFDLFVBQVUsQ0FBQyxHQUFHLENBQUMsQ0FBQztJQUNuRCxDQUFDO0lBRUQsS0FBSyxDQUFDLE9BQU8sQ0FBQyxPQUFlLEVBQUUsSUFBVTtRQUN2QyxNQUFNLFVBQVUsR0FBRyxNQUFNLElBQUksQ0FBQyxXQUFXLENBQUMsT0FBTyxFQUFFLElBQUksQ0FBQyxDQUFDO1FBQ3pELE1BQU0sSUFBSSxHQUFHLE1BQU0sSUFBSSxDQUFDLGVBQWUsQ0FBQyxLQUFLLENBQUMsVUFBVSxDQUFDLEdBQUcsQ0FBQyxDQUFDO1FBQzlELE9BQU8sSUFBSSxDQUFDLEdBQUcsQ0FBQyxDQUFDLENBQU0sRUFBRSxFQUFFO1lBQ3pCLE9BQU8seUJBQXlCLENBQUMsYUFBYSxDQUFDLElBQUksQ0FBQyxVQUFVLEVBQUUsQ0FBQyxDQUFDLENBQUM7UUFDckUsQ0FBQyxDQUFDLENBQUM7SUFDTCxDQUFDO0lBRUQsS0FBSyxDQUFDLFVBQVUsQ0FBQyxPQUFlLEVBQUUsSUFBVTtRQUMxQyxNQUFNLFVBQVUsR0FBRyxNQUFNLElBQUksQ0FBQyxXQUFXLENBQUMsT0FBTyxFQUFFLElBQUksQ0FBQyxDQUFDO1FBQ3pELE9BQU8sTUFBTSxJQUFJLENBQUMsZUFBZSxDQUFDLEtBQUssQ0FBQyxVQUFVLENBQUMsR0FBRyxDQUFDLENBQUM7SUFDMUQsQ0FBQztJQUVELEtBQUssQ0FBQyxhQUFhLENBQUMsT0FBZSxFQUFFLElBQVU7UUFDN0MsTUFBTSxHQUFHLEdBQUcsTUFBTSxJQUFJLENBQUMsT0FBTyxDQUFDLE9BQU8sRUFBRSxJQUFJLENBQUMsQ0FBQztRQUM5QyxJQUFJLENBQUMsS0FBSyxDQUFDLE9BQU8sQ0FBQyxHQUFHLENBQUM7WUFBRSxPQUFPLEdBQUcsSUFBSSxJQUFJLENBQUM7UUFDNUMsT0FBTyxHQUFHLENBQUMsQ0FBQyxDQUFDLElBQUksSUFBSSxDQUFDO0lBQ3hCLENBQUM7SUFFRCxLQUFLLENBQUMsZ0JBQWdCLENBQUMsT0FBZSxFQUFFLElBQVU7UUFDaEQsTUFBTSxHQUFHLEdBQUcsTUFBTSxJQUFJLENBQUMsVUFBVSxDQUFDLE9BQU8sRUFBRSxJQUFJLENBQUMsQ0FBQztRQUNqRCxJQUFJLENBQUMsS0FBSyxDQUFDLE9BQU8sQ0FBQyxHQUFHLENBQUM7WUFBRSxPQUFPLENBQUMsR0FBRyxJQUFJLElBQUksQ0FBUSxDQUFDO1FBQ3JELE9BQU8sR0FBRyxDQUFDLENBQUMsQ0FBQyxJQUFJLElBQUksQ0FBQztJQUN4QixDQUFDO0lBRUQsS0FBSyxDQUFDLFFBQVEsQ0FBQyxPQUFlLEVBQUUsSUFBUyxFQUFFLFdBQW1CLEVBQUUsWUFBb0I7UUFDbEYsTUFBTSxLQUFLLEdBQUcsU0FBUyxDQUFDLFdBQVcsR0FBRyxDQUFDLENBQUMsR0FBRyxZQUFZLEtBQUssWUFBWSxFQUFFLENBQUM7UUFDM0UsTUFBTSxHQUFHLEdBQUcsQ0FBQyxNQUFNLElBQUksQ0FBQyxXQUFXLENBQUMsT0FBTyxFQUFFLElBQUksQ0FBQyxDQUFDLENBQUMsR0FBRyxHQUFHLEdBQUcsR0FBRyxLQUFLLENBQUM7UUFDdEUsTUFBTSxRQUFRLEdBQUcsQ0FBQyxNQUFNLElBQUksQ0FBQyxXQUFXLENBQUMsT0FBTyxFQUFFLE1BQU0sQ0FBQyxNQUFNLENBQUMsRUFBRSxPQUFPLEVBQUUsSUFBSSxFQUFFLEVBQUUsSUFBSSxDQUFDLENBQUMsQ0FBQyxDQUFDLEdBQUcsQ0FBQztRQUUvRixNQUFNLEdBQUcsR0FBRyxNQUFNLE9BQU8sQ0FBQyxHQUFHLENBQUMsQ0FBQyxJQUFJLENBQUMsZUFBZSxDQUFDLEtBQUssQ0FBQyxHQUFHLENBQUMsRUFBRSxJQUFJLENBQUMsY0FBYyxDQUFDLFFBQVEsQ0FBQyxDQUFDLENBQUMsQ0FBQztRQUVoRyxPQUFPO1lBQ0wsS0FBSyxFQUFFLE1BQU0sQ0FBQyxHQUFHLENBQUMsQ0FBQyxDQUFDLENBQUM7WUFDckIsT0FBTyxFQUFFLFdBQVc7WUFDcEIsSUFBSSxFQUFFLEdBQUcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxHQUFHLENBQUMsQ0FBQyxDQUFNLEVBQUUsRUFBRSxDQUFDLHlCQUF5QixDQUFDLGFBQWEsQ0FBQyxJQUFJLENBQUMsVUFBVSxFQUFFLENBQUMsQ0FBQyxDQUFDO1NBQzFGLENBQUM7SUFDSixDQUFDO0lBRUQsS0FBSyxDQUFDLGNBQWMsQ0FBQyxPQUFlO1FBQ2xDLE1BQU0sR0FBRyxHQUFHLEdBQUcsc0JBQXNCLENBQUMsQ0FBQyxDQUFDLEdBQUcsT0FBTyxHQUFHLHNCQUFzQixDQUFDLENBQUMsQ0FBQyxFQUFFLENBQUM7UUFFakYsTUFBTSxNQUFNLEdBQUcsTUFBTSxJQUFJLENBQUMsZUFBZSxDQUFDLEtBQUssQ0FBQyxHQUFHLENBQUMsQ0FBQztRQUVyRCxPQUFPLE1BQU0sQ0FBQyxDQUFDLENBQUMsQ0FBQyxLQUFLLENBQUM7SUFDekIsQ0FBQztDQUNGIn0=
+import "./TableSqlMap.js";
+import { TableModel } from "@eggjs/dal-decorator";
+
+//#region src/DataSource.ts
+const PAGINATE_COUNT_WRAPPER = ["SELECT COUNT(0) as count FROM (", ") AS T"];
+var DataSource = class {
+	tableModel;
+	mysqlDataSource;
+	sqlMap;
+	constructor(tableModel, mysqlDataSource, sqlMap) {
+		this.tableModel = tableModel;
+		this.mysqlDataSource = mysqlDataSource;
+		this.sqlMap = sqlMap;
+	}
+	/**
+	* public for aop execute to implement sql hint append
+	* @param sqlName - sql name
+	* @param data - sql data
+	*/
+	async generateSql(sqlName, data) {
+		return {
+			sql: this.sqlMap.generate(sqlName, data, this.mysqlDataSource.timezone),
+			sqlType: this.sqlMap.getType(sqlName),
+			template: this.sqlMap.getTemplateString(sqlName)
+		};
+	}
+	async count(sqlName, data) {
+		const newData = Object.assign({ $$count: true }, data);
+		const executeSql = await this.generateSql(sqlName, newData);
+		return await this.#paginateCount(executeSql.sql);
+	}
+	async execute(sqlName, data) {
+		const executeSql = await this.generateSql(sqlName, data);
+		return (await this.mysqlDataSource.query(executeSql.sql)).map((t) => {
+			return TableModelInstanceBuilder.buildInstance(this.tableModel, t);
+		});
+	}
+	async executeRaw(sqlName, data) {
+		const executeSql = await this.generateSql(sqlName, data);
+		return await this.mysqlDataSource.query(executeSql.sql);
+	}
+	async executeScalar(sqlName, data) {
+		const ret = await this.execute(sqlName, data);
+		if (!Array.isArray(ret)) return ret || null;
+		return ret[0] || null;
+	}
+	async executeRawScalar(sqlName, data) {
+		const ret = await this.executeRaw(sqlName, data);
+		if (!Array.isArray(ret)) return ret || null;
+		return ret[0] || null;
+	}
+	async paginate(sqlName, data, currentPage, perPageCount) {
+		const limit = `LIMIT ${(currentPage - 1) * perPageCount}, ${perPageCount}`;
+		const sql = (await this.generateSql(sqlName, data)).sql + " " + limit;
+		const countSql = (await this.generateSql(sqlName, Object.assign({ $$count: true }, data))).sql;
+		const ret = await Promise.all([this.mysqlDataSource.query(sql), this.#paginateCount(countSql)]);
+		return {
+			total: Number(ret[1]),
+			pageNum: currentPage,
+			rows: ret[0].map((t) => TableModelInstanceBuilder.buildInstance(this.tableModel, t))
+		};
+	}
+	async #paginateCount(baseSQL) {
+		const sql = `${PAGINATE_COUNT_WRAPPER[0]}${baseSQL}${PAGINATE_COUNT_WRAPPER[1]}`;
+		return (await this.mysqlDataSource.query(sql))[0].count;
+	}
+};
+
+//#endregion
+export { DataSource };

package/dist/DatabaseForker.d.ts

@@ -1,12 +1,16 @@
-import { type DataSourceOptions } from './MySqlDataSource.ts';
-export declare class DatabaseForker {
-    private readonly env;
-    private readonly options;
-    constructor(env: string, options: DataSourceOptions);
-    shouldFork(): boolean;
-    forkDb(moduleDir: string): Promise<void>;
-    private forkTables;
-    private doForkTable;
-    private doCreateUtDb;
-    destroy(): Promise<void>;
+import { DataSourceOptions } from "./MySqlDataSource.js";
+
+//#region src/DatabaseForker.d.ts
+declare class DatabaseForker {
+  private readonly env;
+  private readonly options;
+  constructor(env: string, options: DataSourceOptions);
+  shouldFork(): boolean;
+  forkDb(moduleDir: string): Promise<void>;
+  private forkTables;
+  private doForkTable;
+  private doCreateUtDb;
+  destroy(): Promise<void>;
 }
+//#endregion
+export { DatabaseForker };