@eggjs/dal-runtime 3.52.0 → 4.0.0-beta.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (47) hide show
  1. package/package.json +27 -25
  2. package/{dist/src → src}/BaseSqlMap.d.ts +1 -1
  3. package/src/BaseSqlMap.js +303 -0
  4. package/{dist/src → src}/CodeGenerator.d.ts +1 -1
  5. package/src/CodeGenerator.js +138 -0
  6. package/{dist/src → src}/DaoLoader.d.ts +1 -1
  7. package/src/DaoLoader.js +13 -0
  8. package/{dist/src → src}/DataSource.d.ts +5 -5
  9. package/src/DataSource.js +79 -0
  10. package/{dist/src → src}/DatabaseForker.d.ts +1 -1
  11. package/src/DatabaseForker.js +49 -0
  12. package/{dist/src → src}/MySqlDataSource.d.ts +1 -1
  13. package/{dist/src → src}/MySqlDataSource.js +5 -12
  14. package/{dist/src → src}/NunjucksConverter.js +3 -7
  15. package/src/NunjucksUtil.js +74 -0
  16. package/{dist/src → src}/SqlGenerator.d.ts +1 -1
  17. package/src/SqlGenerator.js +391 -0
  18. package/{dist/src → src}/SqlMapLoader.d.ts +2 -2
  19. package/src/SqlMapLoader.js +19 -0
  20. package/{dist/src → src}/SqlUtil.js +2 -6
  21. package/{dist/src → src}/TableModelInstanceBuilder.d.ts +1 -1
  22. package/src/TableModelInstanceBuilder.js +22 -0
  23. package/src/TableSqlMap.js +91 -0
  24. package/{dist/src → src}/TemplateUtil.d.ts +3 -12
  25. package/src/TemplateUtil.js +86 -0
  26. package/src/index.d.ts +15 -0
  27. package/src/index.js +16 -0
  28. package/dist/index.d.ts +0 -9
  29. package/dist/index.js +0 -28
  30. package/dist/src/BaseSqlMap.js +0 -310
  31. package/dist/src/CodeGenerator.js +0 -145
  32. package/dist/src/DaoLoader.js +0 -17
  33. package/dist/src/DataSource.js +0 -83
  34. package/dist/src/DatabaseForker.js +0 -56
  35. package/dist/src/NunjucksUtil.js +0 -104
  36. package/dist/src/SqlGenerator.js +0 -395
  37. package/dist/src/SqlMapLoader.js +0 -23
  38. package/dist/src/TableModelInstanceBuilder.js +0 -26
  39. package/dist/src/TableSqlMap.js +0 -95
  40. package/dist/src/TemplateUtil.js +0 -93
  41. package/dist/src/templates/base_dao.njk +0 -164
  42. package/dist/src/templates/dao.njk +0 -17
  43. package/dist/src/templates/extension.njk +0 -17
  44. /package/{dist/src → src}/NunjucksConverter.d.ts +0 -0
  45. /package/{dist/src → src}/NunjucksUtil.d.ts +0 -0
  46. /package/{dist/src → src}/SqlUtil.d.ts +0 -0
  47. /package/{dist/src → src}/TableSqlMap.d.ts +0 -0
@@ -1,17 +0,0 @@
1
- "use strict";
2
- Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.DaoLoader = void 0;
4
- const tegg_types_1 = require("@eggjs/tegg-types");
5
- const dal_1 = require("@eggjs/tegg/dal");
6
- const helper_1 = require("@eggjs/tegg/helper");
7
- class DaoLoader {
8
- static loadDaos(moduleDir) {
9
- const loader = helper_1.LoaderFactory.createLoader(moduleDir, tegg_types_1.EggLoadUnitType.MODULE);
10
- const clazzList = loader.load();
11
- return clazzList.filter((t) => {
12
- return dal_1.DaoInfoUtil.getIsDao(t);
13
- });
14
- }
15
- }
16
- exports.DaoLoader = DaoLoader;
17
- //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiRGFvTG9hZGVyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL0Rhb0xvYWRlci50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7QUFBQSxrREFBb0Q7QUFDcEQseUNBQThDO0FBRTlDLCtDQUFtRDtBQUVuRCxNQUFhLFNBQVM7SUFDcEIsTUFBTSxDQUFDLFFBQVEsQ0FBQyxTQUFpQjtRQUMvQixNQUFNLE1BQU0sR0FBRyxzQkFBYSxDQUFDLFlBQVksQ0FBQyxTQUFTLEVBQUUsNEJBQWUsQ0FBQyxNQUFNLENBQUMsQ0FBQztRQUM3RSxNQUFNLFNBQVMsR0FBRyxNQUFNLENBQUMsSUFBSSxFQUFFLENBQUM7UUFDaEMsT0FBTyxTQUFTLENBQUMsTUFBTSxDQUFDLENBQUMsQ0FBQyxFQUFvQixFQUFFO1lBQzlDLE9BQU8saUJBQVcsQ0FBQyxRQUFRLENBQUMsQ0FBQyxDQUFDLENBQUM7UUFDakMsQ0FBQyxDQUFDLENBQUM7SUFDTCxDQUFDO0NBQ0Y7QUFSRCw4QkFRQyJ9
@@ -1,83 +0,0 @@
1
- "use strict";
2
- var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
3
- if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
4
- if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
5
- return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
6
- };
7
- var _DataSource_instances, _DataSource_paginateCount;
8
- Object.defineProperty(exports, "__esModule", { value: true });
9
- exports.DataSource = void 0;
10
- const TableModelInstanceBuilder_1 = require("./TableModelInstanceBuilder");
11
- const PAGINATE_COUNT_WRAPPER = ['SELECT COUNT(0) as count FROM (', ') AS T'];
12
- class DataSource {
13
- constructor(tableModel, mysqlDataSource, sqlMap) {
14
- _DataSource_instances.add(this);
15
- this.tableModel = tableModel;
16
- this.mysqlDataSource = mysqlDataSource;
17
- this.sqlMap = sqlMap;
18
- }
19
- /**
20
- * public for aop execute to implement sql hint append
21
- * @param sqlName
22
- * @param data
23
- */
24
- async generateSql(sqlName, data) {
25
- const sql = this.sqlMap.generate(sqlName, data, this.mysqlDataSource.timezone);
26
- const sqlType = this.sqlMap.getType(sqlName);
27
- const template = this.sqlMap.getTemplateString(sqlName);
28
- return {
29
- sql,
30
- sqlType,
31
- template,
32
- };
33
- }
34
- async count(sqlName, data) {
35
- const newData = Object.assign({ $$count: true }, data);
36
- const executeSql = await this.generateSql(sqlName, newData);
37
- return await __classPrivateFieldGet(this, _DataSource_instances, "m", _DataSource_paginateCount).call(this, executeSql.sql);
38
- }
39
- async execute(sqlName, data) {
40
- const executeSql = await this.generateSql(sqlName, data);
41
- const rows = await this.mysqlDataSource.query(executeSql.sql);
42
- return rows.map(t => {
43
- return TableModelInstanceBuilder_1.TableModelInstanceBuilder.buildInstance(this.tableModel, t);
44
- });
45
- }
46
- async executeRaw(sqlName, data) {
47
- const executeSql = await this.generateSql(sqlName, data);
48
- return await this.mysqlDataSource.query(executeSql.sql);
49
- }
50
- async executeScalar(sqlName, data) {
51
- const ret = await this.execute(sqlName, data);
52
- if (!Array.isArray(ret))
53
- return ret || null;
54
- return ret[0] || null;
55
- }
56
- async executeRawScalar(sqlName, data) {
57
- const ret = await this.executeRaw(sqlName, data);
58
- if (!Array.isArray(ret))
59
- return (ret || null);
60
- return ret[0] || null;
61
- }
62
- async paginate(sqlName, data, currentPage, perPageCount) {
63
- const limit = `LIMIT ${(currentPage - 1) * perPageCount}, ${perPageCount}`;
64
- const sql = (await this.generateSql(sqlName, data)).sql + ' ' + limit;
65
- const countSql = (await this.generateSql(sqlName, Object.assign({ $$count: true }, data))).sql;
66
- const ret = await Promise.all([
67
- this.mysqlDataSource.query(sql),
68
- __classPrivateFieldGet(this, _DataSource_instances, "m", _DataSource_paginateCount).call(this, countSql),
69
- ]);
70
- return {
71
- total: Number(ret[1]),
72
- pageNum: currentPage,
73
- rows: ret[0].map(t => TableModelInstanceBuilder_1.TableModelInstanceBuilder.buildInstance(this.tableModel, t)),
74
- };
75
- }
76
- }
77
- exports.DataSource = DataSource;
78
- _DataSource_instances = new WeakSet(), _DataSource_paginateCount = async function _DataSource_paginateCount(baseSQL) {
79
- const sql = `${PAGINATE_COUNT_WRAPPER[0]}${baseSQL}${PAGINATE_COUNT_WRAPPER[1]}`;
80
- const result = await this.mysqlDataSource.query(sql);
81
- return result[0].count;
82
- };
83
- //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiRGF0YVNvdXJjZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9EYXRhU291cmNlLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7OztBQUlBLDJFQUF3RTtBQVF4RSxNQUFNLHNCQUFzQixHQUFHLENBQUUsaUNBQWlDLEVBQUUsUUFBUSxDQUFFLENBQUM7QUFFL0UsTUFBYSxVQUFVO0lBS3JCLFlBQVksVUFBeUIsRUFBRSxlQUFnQyxFQUFFLE1BQW1COztRQUMxRixJQUFJLENBQUMsVUFBVSxHQUFHLFVBQVUsQ0FBQztRQUM3QixJQUFJLENBQUMsZUFBZSxHQUFHLGVBQWUsQ0FBQztRQUN2QyxJQUFJLENBQUMsTUFBTSxHQUFHLE1BQU0sQ0FBQztJQUN2QixDQUFDO0lBRUQ7Ozs7T0FJRztJQUNILEtBQUssQ0FBQyxXQUFXLENBQUMsT0FBZSxFQUFFLElBQVk7UUFDN0MsTUFBTSxHQUFHLEdBQUcsSUFBSSxDQUFDLE1BQU0sQ0FBQyxRQUFRLENBQUMsT0FBTyxFQUFFLElBQUksRUFBRSxJQUFJLENBQUMsZUFBZSxDQUFDLFFBQVMsQ0FBQyxDQUFDO1FBQ2hGLE1BQU0sT0FBTyxHQUFHLElBQUksQ0FBQyxNQUFNLENBQUMsT0FBTyxDQUFDLE9BQU8sQ0FBQyxDQUFDO1FBQzdDLE1BQU0sUUFBUSxHQUFHLElBQUksQ0FBQyxNQUFNLENBQUMsaUJBQWlCLENBQUMsT0FBTyxDQUFDLENBQUM7UUFDeEQsT0FBTztZQUNMLEdBQUc7WUFDSCxPQUFPO1lBQ1AsUUFBUTtTQUNULENBQUM7SUFDSixDQUFDO0lBRUQsS0FBSyxDQUFDLEtBQUssQ0FBQyxPQUFlLEVBQUUsSUFBVTtRQUNyQyxNQUFNLE9BQU8sR0FBRyxNQUFNLENBQUMsTUFBTSxDQUFDLEVBQUUsT0FBTyxFQUFFLElBQUksRUFBRSxFQUFFLElBQUksQ0FBQyxDQUFDO1FBQ3ZELE1BQU0sVUFBVSxHQUFHLE1BQU0sSUFBSSxDQUFDLFdBQVcsQ0FBQyxPQUFPLEVBQUUsT0FBTyxDQUFDLENBQUM7UUFDNUQsT0FBTyxNQUFNLHVCQUFBLElBQUksd0RBQWUsTUFBbkIsSUFBSSxFQUFnQixVQUFVLENBQUMsR0FBRyxDQUFDLENBQUM7SUFDbkQsQ0FBQztJQUVELEtBQUssQ0FBQyxPQUFPLENBQUMsT0FBZSxFQUFFLElBQVU7UUFDdkMsTUFBTSxVQUFVLEdBQUcsTUFBTSxJQUFJLENBQUMsV0FBVyxDQUFDLE9BQU8sRUFBRSxJQUFJLENBQUMsQ0FBQztRQUN6RCxNQUFNLElBQUksR0FBRyxNQUFNLElBQUksQ0FBQyxlQUFlLENBQUMsS0FBSyxDQUFDLFVBQVUsQ0FBQyxHQUFHLENBQUMsQ0FBQztRQUM5RCxPQUFPLElBQUksQ0FBQyxHQUFHLENBQUMsQ0FBQyxDQUFDLEVBQUU7WUFDbEIsT0FBTyxxREFBeUIsQ0FBQyxhQUFhLENBQUMsSUFBSSxDQUFDLFVBQVUsRUFBRSxDQUFDLENBQUMsQ0FBQztRQUNyRSxDQUFDLENBQUMsQ0FBQztJQUNMLENBQUM7SUFFRCxLQUFLLENBQUMsVUFBVSxDQUFDLE9BQWUsRUFBRSxJQUFVO1FBQzFDLE1BQU0sVUFBVSxHQUFHLE1BQU0sSUFBSSxDQUFDLFdBQVcsQ0FBQyxPQUFPLEVBQUUsSUFBSSxDQUFDLENBQUM7UUFDekQsT0FBTyxNQUFNLElBQUksQ0FBQyxlQUFlLENBQUMsS0FBSyxDQUFDLFVBQVUsQ0FBQyxHQUFHLENBQUMsQ0FBQztJQUMxRCxDQUFDO0lBRUQsS0FBSyxDQUFDLGFBQWEsQ0FBQyxPQUFlLEVBQUUsSUFBVTtRQUM3QyxNQUFNLEdBQUcsR0FBRyxNQUFNLElBQUksQ0FBQyxPQUFPLENBQUMsT0FBTyxFQUFFLElBQUksQ0FBQyxDQUFDO1FBQzlDLElBQUksQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLEdBQUcsQ0FBQztZQUFFLE9BQU8sR0FBRyxJQUFJLElBQUksQ0FBQztRQUM1QyxPQUFPLEdBQUcsQ0FBQyxDQUFDLENBQUMsSUFBSSxJQUFJLENBQUM7SUFDeEIsQ0FBQztJQUVELEtBQUssQ0FBQyxnQkFBZ0IsQ0FBQyxPQUFlLEVBQUUsSUFBVTtRQUNoRCxNQUFNLEdBQUcsR0FBRyxNQUFNLElBQUksQ0FBQyxVQUFVLENBQUMsT0FBTyxFQUFFLElBQUksQ0FBQyxDQUFDO1FBQ2pELElBQUksQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLEdBQUcsQ0FBQztZQUFFLE9BQU8sQ0FBQyxHQUFHLElBQUksSUFBSSxDQUFRLENBQUM7UUFDckQsT0FBTyxHQUFHLENBQUMsQ0FBQyxDQUFDLElBQUksSUFBSSxDQUFDO0lBQ3hCLENBQUM7SUFFRCxLQUFLLENBQUMsUUFBUSxDQUFDLE9BQWUsRUFBRSxJQUFTLEVBQUUsV0FBbUIsRUFBRSxZQUFvQjtRQUNsRixNQUFNLEtBQUssR0FBRyxTQUFTLENBQUMsV0FBVyxHQUFHLENBQUMsQ0FBQyxHQUFHLFlBQVksS0FBSyxZQUFZLEVBQUUsQ0FBQztRQUMzRSxNQUFNLEdBQUcsR0FBRyxDQUFDLE1BQU0sSUFBSSxDQUFDLFdBQVcsQ0FBQyxPQUFPLEVBQUUsSUFBSSxDQUFDLENBQUMsQ0FBQyxHQUFHLEdBQUcsR0FBRyxHQUFHLEtBQUssQ0FBQztRQUN0RSxNQUFNLFFBQVEsR0FBRyxDQUFDLE1BQU0sSUFBSSxDQUFDLFdBQVcsQ0FBQyxPQUFPLEVBQUUsTUFBTSxDQUFDLE1BQU0sQ0FBQyxFQUFFLE9BQU8sRUFBRSxJQUFJLEVBQUUsRUFBRSxJQUFJLENBQUMsQ0FBQyxDQUFDLENBQUMsR0FBRyxDQUFDO1FBRy9GLE1BQU0sR0FBRyxHQUFHLE1BQU0sT0FBTyxDQUFDLEdBQUcsQ0FBQztZQUM1QixJQUFJLENBQUMsZUFBZSxDQUFDLEtBQUssQ0FBQyxHQUFHLENBQUM7WUFDL0IsdUJBQUEsSUFBSSx3REFBZSxNQUFuQixJQUFJLEVBQWdCLFFBQVEsQ0FBQztTQUM5QixDQUFDLENBQUM7UUFFSCxPQUFPO1lBQ0wsS0FBSyxFQUFFLE1BQU0sQ0FBQyxHQUFHLENBQUMsQ0FBQyxDQUFDLENBQUM7WUFDckIsT0FBTyxFQUFFLFdBQVc7WUFDcEIsSUFBSSxFQUFFLEdBQUcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxHQUFHLENBQUMsQ0FBQyxDQUFDLEVBQUUsQ0FBQyxxREFBeUIsQ0FBQyxhQUFhLENBQUMsSUFBSSxDQUFDLFVBQVUsRUFBRSxDQUFDLENBQUMsQ0FBQztTQUNuRixDQUFDO0lBQ0osQ0FBQztDQVNGO0FBbkZELGdDQW1GQzttRUFQQyxLQUFLLG9DQUFnQixPQUFlO0lBQ2xDLE1BQU0sR0FBRyxHQUFHLEdBQUcsc0JBQXNCLENBQUMsQ0FBQyxDQUFDLEdBQUcsT0FBTyxHQUFHLHNCQUFzQixDQUFDLENBQUMsQ0FBQyxFQUFFLENBQUM7SUFFakYsTUFBTSxNQUFNLEdBQUcsTUFBTSxJQUFJLENBQUMsZUFBZSxDQUFDLEtBQUssQ0FBQyxHQUFHLENBQUMsQ0FBQztJQUVyRCxPQUFPLE1BQU0sQ0FBQyxDQUFDLENBQUMsQ0FBQyxLQUFLLENBQUM7QUFDekIsQ0FBQyJ9
@@ -1,56 +0,0 @@
1
- "use strict";
2
- var __importDefault = (this && this.__importDefault) || function (mod) {
3
- return (mod && mod.__esModule) ? mod : { "default": mod };
4
- };
5
- Object.defineProperty(exports, "__esModule", { value: true });
6
- exports.DatabaseForker = void 0;
7
- const node_assert_1 = __importDefault(require("node:assert"));
8
- const rds_1 = require("@eggjs/rds");
9
- const DaoLoader_1 = require("./DaoLoader");
10
- class DatabaseForker {
11
- constructor(env, options) {
12
- this.env = env;
13
- this.options = options;
14
- }
15
- shouldFork() {
16
- return this.env === 'unittest' && this.options.forkDb;
17
- }
18
- async forkDb(moduleDir) {
19
- (0, node_assert_1.default)(this.shouldFork(), 'fork db only run in unittest');
20
- // 尽早判断不应该 fork,避免对 rds pool 配置造成污染
21
- // eslint-disable-next-line @typescript-eslint/no-unused-vars
22
- const { name, initSql, forkDb, database, ...mysqlOptions } = this.options;
23
- const client = new rds_1.RDSClient(Object.assign(mysqlOptions));
24
- const conn = await client.getConnection();
25
- await this.doCreateUtDb(conn);
26
- await this.forkTables(conn, moduleDir);
27
- conn.release();
28
- await client.end();
29
- }
30
- async forkTables(conn, moduleDir) {
31
- const daoClazzList = DaoLoader_1.DaoLoader.loadDaos(moduleDir);
32
- for (const clazz of daoClazzList) {
33
- await this.doForkTable(conn, clazz.tableSql);
34
- }
35
- }
36
- async doForkTable(conn, sqlFile) {
37
- const sqls = sqlFile.split(';').filter(t => !!t.trim());
38
- for (const sql of sqls) {
39
- await conn.query(sql);
40
- }
41
- }
42
- async doCreateUtDb(conn) {
43
- await conn.query(`CREATE DATABASE IF NOT EXISTS ${this.options.database};`);
44
- await conn.query(`use ${this.options.database};`);
45
- }
46
- async destroy() {
47
- (0, node_assert_1.default)(this.shouldFork(), 'fork db only run in unittest');
48
- // eslint-disable-next-line @typescript-eslint/no-unused-vars
49
- const { name, initSql, forkDb, database, ...mysqlOptions } = this.options;
50
- const client = new rds_1.RDSClient(Object.assign(mysqlOptions));
51
- await client.query(`DROP DATABASE ${database}`);
52
- await client.end();
53
- }
54
- }
55
- exports.DatabaseForker = DatabaseForker;
56
- //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiRGF0YWJhc2VGb3JrZXIuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvRGF0YWJhc2VGb3JrZXIudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7Ozs7O0FBQUEsOERBQWlDO0FBQ2pDLG9DQUF1QztBQUV2QywyQ0FBd0M7QUFFeEMsTUFBYSxjQUFjO0lBSXpCLFlBQVksR0FBVyxFQUFFLE9BQTBCO1FBQ2pELElBQUksQ0FBQyxHQUFHLEdBQUcsR0FBRyxDQUFDO1FBQ2YsSUFBSSxDQUFDLE9BQU8sR0FBRyxPQUFPLENBQUM7SUFDekIsQ0FBQztJQUVELFVBQVU7UUFDUixPQUFPLElBQUksQ0FBQyxHQUFHLEtBQUssVUFBVSxJQUFJLElBQUksQ0FBQyxPQUFPLENBQUMsTUFBTSxDQUFDO0lBQ3hELENBQUM7SUFFRCxLQUFLLENBQUMsTUFBTSxDQUFDLFNBQWlCO1FBQzVCLElBQUEscUJBQU0sRUFBQyxJQUFJLENBQUMsVUFBVSxFQUFFLEVBQUUsOEJBQThCLENBQUMsQ0FBQztRQUMxRCxtQ0FBbUM7UUFDbkMsNkRBQTZEO1FBQzdELE1BQU0sRUFBRSxJQUFJLEVBQUUsT0FBTyxFQUFFLE1BQU0sRUFBRSxRQUFRLEVBQUUsR0FBRyxZQUFZLEVBQUUsR0FBRyxJQUFJLENBQUMsT0FBTyxDQUFDO1FBQzFFLE1BQU0sTUFBTSxHQUFHLElBQUksZUFBUyxDQUFDLE1BQU0sQ0FBQyxNQUFNLENBQUMsWUFBWSxDQUFDLENBQUMsQ0FBQztRQUMxRCxNQUFNLElBQUksR0FBRyxNQUFNLE1BQU0sQ0FBQyxhQUFhLEVBQUUsQ0FBQztRQUMxQyxNQUFNLElBQUksQ0FBQyxZQUFZLENBQUMsSUFBSSxDQUFDLENBQUM7UUFDOUIsTUFBTSxJQUFJLENBQUMsVUFBVSxDQUFDLElBQUksRUFBRSxTQUFTLENBQUMsQ0FBQztRQUN2QyxJQUFJLENBQUMsT0FBTyxFQUFFLENBQUM7UUFDZixNQUFNLE1BQU0sQ0FBQyxHQUFHLEVBQUUsQ0FBQztJQUNyQixDQUFDO0lBRU8sS0FBSyxDQUFDLFVBQVUsQ0FBQyxJQUFJLEVBQUUsU0FBaUI7UUFDOUMsTUFBTSxZQUFZLEdBQUcscUJBQVMsQ0FBQyxRQUFRLENBQUMsU0FBUyxDQUFDLENBQUM7UUFDbkQsS0FBSyxNQUFNLEtBQUssSUFBSSxZQUFZLEVBQUUsQ0FBQztZQUNqQyxNQUFNLElBQUksQ0FBQyxXQUFXLENBQUMsSUFBSSxFQUFFLEtBQUssQ0FBQyxRQUFRLENBQUMsQ0FBQztRQUMvQyxDQUFDO0lBQ0gsQ0FBQztJQUVPLEtBQUssQ0FBQyxXQUFXLENBQUMsSUFBSSxFQUFFLE9BQWU7UUFDN0MsTUFBTSxJQUFJLEdBQUcsT0FBTyxDQUFDLEtBQUssQ0FBQyxHQUFHLENBQUMsQ0FBQyxNQUFNLENBQUMsQ0FBQyxDQUFDLEVBQUUsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQUFDLElBQUksRUFBRSxDQUFDLENBQUM7UUFDeEQsS0FBSyxNQUFNLEdBQUcsSUFBSSxJQUFJLEVBQUUsQ0FBQztZQUN2QixNQUFNLElBQUksQ0FBQyxLQUFLLENBQUMsR0FBRyxDQUFDLENBQUM7UUFDeEIsQ0FBQztJQUNILENBQUM7SUFFTyxLQUFLLENBQUMsWUFBWSxDQUFDLElBQUk7UUFDN0IsTUFBTSxJQUFJLENBQUMsS0FBSyxDQUFDLGlDQUFpQyxJQUFJLENBQUMsT0FBTyxDQUFDLFFBQVEsR0FBRyxDQUFDLENBQUM7UUFDNUUsTUFBTSxJQUFJLENBQUMsS0FBSyxDQUFDLE9BQU8sSUFBSSxDQUFDLE9BQU8sQ0FBQyxRQUFRLEdBQUcsQ0FBQyxDQUFDO0lBQ3BELENBQUM7SUFFRCxLQUFLLENBQUMsT0FBTztRQUNYLElBQUEscUJBQU0sRUFBQyxJQUFJLENBQUMsVUFBVSxFQUFFLEVBQUUsOEJBQThCLENBQUMsQ0FBQztRQUMxRCw2REFBNkQ7UUFDN0QsTUFBTSxFQUFFLElBQUksRUFBRSxPQUFPLEVBQUUsTUFBTSxFQUFFLFFBQVEsRUFBRSxHQUFHLFlBQVksRUFBRSxHQUFHLElBQUksQ0FBQyxPQUFPLENBQUM7UUFDMUUsTUFBTSxNQUFNLEdBQUcsSUFBSSxlQUFTLENBQUMsTUFBTSxDQUFDLE1BQU0sQ0FBQyxZQUFZLENBQUMsQ0FBQyxDQUFDO1FBQzFELE1BQU0sTUFBTSxDQUFDLEtBQUssQ0FBQyxpQkFBaUIsUUFBUSxFQUFFLENBQUMsQ0FBQztRQUNoRCxNQUFNLE1BQU0sQ0FBQyxHQUFHLEVBQUUsQ0FBQztJQUNyQixDQUFDO0NBQ0Y7QUFyREQsd0NBcURDIn0=
@@ -1,104 +0,0 @@
1
- "use strict";
2
- var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
- if (k2 === undefined) k2 = k;
4
- var desc = Object.getOwnPropertyDescriptor(m, k);
5
- if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
- desc = { enumerable: true, get: function() { return m[k]; } };
7
- }
8
- Object.defineProperty(o, k2, desc);
9
- }) : (function(o, m, k, k2) {
10
- if (k2 === undefined) k2 = k;
11
- o[k2] = m[k];
12
- }));
13
- var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
14
- Object.defineProperty(o, "default", { enumerable: true, value: v });
15
- }) : function(o, v) {
16
- o["default"] = v;
17
- });
18
- var __importStar = (this && this.__importStar) || function (mod) {
19
- if (mod && mod.__esModule) return mod;
20
- var result = {};
21
- if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
22
- __setModuleDefault(result, mod);
23
- return result;
24
- };
25
- var __importDefault = (this && this.__importDefault) || function (mod) {
26
- return (mod && mod.__esModule) ? mod : { "default": mod };
27
- };
28
- Object.defineProperty(exports, "__esModule", { value: true });
29
- exports.NunjucksUtils = void 0;
30
- const nunjucks_1 = __importStar(require("nunjucks"));
31
- const sqlstring_1 = __importDefault(require("sqlstring"));
32
- const NunjucksConverter_1 = require("./NunjucksConverter");
33
- const SqlUtil_1 = require("./SqlUtil");
34
- const compiler = nunjucks_1.default.compiler;
35
- const envs = {};
36
- const ROOT_RENDER_FUNC = Symbol('rootRenderFunc');
37
- const RUNTIME = Object.assign({}, nunjucks_1.default.runtime, {
38
- escapeSQL: function escapeSQL(key, value) {
39
- // 如果是预定义 block 则不转义
40
- if (this.env.globals[key])
41
- return value;
42
- return sqlstring_1.default.escape(value, true, this.env.timezone);
43
- },
44
- });
45
- function _replaceCodeWithSQLFeature(source) {
46
- const funcs = [
47
- 'convertNormalVariableCode', // 普通变量
48
- 'convertTernaryCode', // 三目运算
49
- 'convertNestedObjectCode', // 对象中的变量,如 `user.id`
50
- 'convertValueInsideFor', // for 中的值需要转义
51
- ];
52
- return funcs.reduce((source, func) => NunjucksConverter_1.NunjucksConverter[func](source), source);
53
- }
54
- /**
55
- * compile the string into function
56
- * @see https://github.com/mozilla/nunjucks/blob/2fd547f/src/environment.js#L571-L592
57
- */
58
- function _compile() {
59
- let source = compiler.compile(this.tmplStr, this.env.asyncFilters, this.env.extensionsList, this.path, this.env.opts);
60
- /**
61
- * 将一些 Nunjucks 的 HTML 转义的代码转换成 SQL 防注入的代码
62
- */
63
- source = _replaceCodeWithSQLFeature(source);
64
- // eslint-disable-next-line
65
- const props = (new Function(source))();
66
- this.blocks = this._getBlocks(props);
67
- this[ROOT_RENDER_FUNC] = props.root;
68
- this.rootRenderFunc = function (env, context, frame, _runtime, cb) {
69
- /**
70
- * 1. 将 runtime 遗弃,用新的
71
- * 2. 移除 SQL 语句中多余空白符
72
- */
73
- return this[ROOT_RENDER_FUNC](env, context, frame, RUNTIME, function (err, ret) {
74
- // istanbul ignore if
75
- if (err)
76
- return cb(err, ret);
77
- return cb(err, SqlUtil_1.SqlUtil.minify(ret || ''));
78
- });
79
- };
80
- this.compiled = true;
81
- }
82
- class NunjucksUtils {
83
- static createEnv(modelName) {
84
- if (envs[modelName])
85
- return envs[modelName];
86
- const env = envs[modelName] = nunjucks_1.default.configure({
87
- autoescape: false,
88
- });
89
- return env;
90
- }
91
- static compile(modelName, sqlName, sql) {
92
- // istanbul ignore if
93
- if (!envs[modelName]) {
94
- throw new Error(`you should create an Environment for ${modelName} first.`);
95
- }
96
- const template = new nunjucks_1.Template(sql, envs[modelName], `egg-dal:MySQL:${modelName}:${sqlName}`, false);
97
- // 做一些 hack,使得支持 MySQL 的一些 Escape
98
- template._compile = _compile;
99
- template.compile();
100
- return template;
101
- }
102
- }
103
- exports.NunjucksUtils = NunjucksUtils;
104
- //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiTnVuanVja3NVdGlsLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL051bmp1Y2tzVXRpbC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQUFBLHFEQUFnRTtBQUNoRSwwREFBa0M7QUFDbEMsMkRBQXdEO0FBQ3hELHVDQUFvQztBQUVwQyxNQUFNLFFBQVEsR0FBSSxrQkFBZ0IsQ0FBQyxRQUFRLENBQUM7QUFDNUMsTUFBTSxJQUFJLEdBQWdDLEVBQUUsQ0FBQztBQUU3QyxNQUFNLGdCQUFnQixHQUFHLE1BQU0sQ0FBQyxnQkFBZ0IsQ0FBQyxDQUFDO0FBQ2xELE1BQU0sT0FBTyxHQUFHLE1BQU0sQ0FBQyxNQUFNLENBQUMsRUFBRSxFQUFFLGtCQUFRLENBQUMsT0FBTyxFQUFFO0lBQ2xELFNBQVMsRUFBRSxTQUFTLFNBQVMsQ0FBWSxHQUFHLEVBQUUsS0FBSztRQUNqRCxvQkFBb0I7UUFDcEIsSUFBSSxJQUFJLENBQUMsR0FBRyxDQUFDLE9BQU8sQ0FBQyxHQUFHLENBQUM7WUFBRSxPQUFPLEtBQUssQ0FBQztRQUN4QyxPQUFPLG1CQUFTLENBQUMsTUFBTSxDQUFDLEtBQUssRUFBRSxJQUFJLEVBQUUsSUFBSSxDQUFDLEdBQUcsQ0FBQyxRQUFRLENBQUMsQ0FBQztJQUMxRCxDQUFDO0NBQ0YsQ0FBQyxDQUFDO0FBRUgsU0FBUywwQkFBMEIsQ0FBQyxNQUFNO0lBQ3hDLE1BQU0sS0FBSyxHQUFHO1FBQ1osMkJBQTJCLEVBQUUsT0FBTztRQUNwQyxvQkFBb0IsRUFBRSxPQUFPO1FBQzdCLHlCQUF5QixFQUFFLHFCQUFxQjtRQUNoRCx1QkFBdUIsRUFBRSxjQUFjO0tBQ3hDLENBQUM7SUFFRixPQUFPLEtBQUssQ0FBQyxNQUFNLENBQUMsQ0FBQyxNQUFNLEVBQUUsSUFBSSxFQUFFLEVBQUUsQ0FBQyxxQ0FBaUIsQ0FBQyxJQUFJLENBQUMsQ0FBQyxNQUFNLENBQUMsRUFBRSxNQUFNLENBQUMsQ0FBQztBQUNqRixDQUFDO0FBRUQ7OztHQUdHO0FBQ0gsU0FBUyxRQUFRO0lBQ2YsSUFBSSxNQUFNLEdBQUcsUUFBUSxDQUFDLE9BQU8sQ0FDM0IsSUFBSSxDQUFDLE9BQU8sRUFDWixJQUFJLENBQUMsR0FBRyxDQUFDLFlBQVksRUFDckIsSUFBSSxDQUFDLEdBQUcsQ0FBQyxjQUFjLEVBQ3ZCLElBQUksQ0FBQyxJQUFJLEVBQ1QsSUFBSSxDQUFDLEdBQUcsQ0FBQyxJQUFJLENBQUMsQ0FBQztJQUVqQjs7T0FFRztJQUNILE1BQU0sR0FBRywwQkFBMEIsQ0FBQyxNQUFNLENBQUMsQ0FBQztJQUU1QywyQkFBMkI7SUFDM0IsTUFBTSxLQUFLLEdBQUcsQ0FBQyxJQUFJLFFBQVEsQ0FBQyxNQUFNLENBQUMsQ0FBQyxFQUFFLENBQUM7SUFFdkMsSUFBSSxDQUFDLE1BQU0sR0FBRyxJQUFJLENBQUMsVUFBVSxDQUFDLEtBQUssQ0FBQyxDQUFDO0lBQ3JDLElBQUksQ0FBQyxnQkFBZ0IsQ0FBQyxHQUFHLEtBQUssQ0FBQyxJQUFJLENBQUM7SUFDcEMsSUFBSSxDQUFDLGNBQWMsR0FBRyxVQUFTLEdBQUcsRUFBRSxPQUFPLEVBQUUsS0FBSyxFQUFFLFFBQVEsRUFBRSxFQUFFO1FBQzlEOzs7V0FHRztRQUNILE9BQU8sSUFBSSxDQUFDLGdCQUFnQixDQUFDLENBQUMsR0FBRyxFQUFFLE9BQU8sRUFBRSxLQUFLLEVBQUUsT0FBTyxFQUFFLFVBQVMsR0FBRyxFQUFFLEdBQUc7WUFDM0UscUJBQXFCO1lBQ3JCLElBQUksR0FBRztnQkFBRSxPQUFPLEVBQUUsQ0FBQyxHQUFHLEVBQUUsR0FBRyxDQUFDLENBQUM7WUFDN0IsT0FBTyxFQUFFLENBQUMsR0FBRyxFQUFFLGlCQUFPLENBQUMsTUFBTSxDQUFDLEdBQUcsSUFBSSxFQUFFLENBQUMsQ0FBQyxDQUFDO1FBQzVDLENBQUMsQ0FBQyxDQUFDO0lBQ0wsQ0FBQyxDQUFDO0lBQ0YsSUFBSSxDQUFDLFFBQVEsR0FBRyxJQUFJLENBQUM7QUFDdkIsQ0FBQztBQUVELE1BQWEsYUFBYTtJQUN4QixNQUFNLENBQUMsU0FBUyxDQUFDLFNBQWlCO1FBQ2hDLElBQUksSUFBSSxDQUFDLFNBQVMsQ0FBQztZQUFFLE9BQU8sSUFBSSxDQUFDLFNBQVMsQ0FBQyxDQUFDO1FBRTVDLE1BQU0sR0FBRyxHQUFHLElBQUksQ0FBQyxTQUFTLENBQUMsR0FBRyxrQkFBUSxDQUFDLFNBQVMsQ0FBQztZQUMvQyxVQUFVLEVBQUUsS0FBSztTQUNsQixDQUFDLENBQUM7UUFFSCxPQUFPLEdBQUcsQ0FBQztJQUNiLENBQUM7SUFFRCxNQUFNLENBQUMsT0FBTyxDQUFDLFNBQWlCLEVBQUUsT0FBZSxFQUFFLEdBQVc7UUFDNUQscUJBQXFCO1FBQ3JCLElBQUksQ0FBQyxJQUFJLENBQUMsU0FBUyxDQUFDLEVBQUUsQ0FBQztZQUNyQixNQUFNLElBQUksS0FBSyxDQUFDLHdDQUF3QyxTQUFTLFNBQVMsQ0FBQyxDQUFDO1FBQzlFLENBQUM7UUFFRCxNQUFNLFFBQVEsR0FBRyxJQUFJLG1CQUFRLENBQUMsR0FBRyxFQUFFLElBQUksQ0FBQyxTQUFTLENBQUMsRUFBRSxpQkFBaUIsU0FBUyxJQUFJLE9BQU8sRUFBRSxFQUFFLEtBQUssQ0FBQyxDQUFDO1FBRXBHLGlDQUFpQztRQUNoQyxRQUFnQixDQUFDLFFBQVEsR0FBRyxRQUFRLENBQUM7UUFDckMsUUFBZ0IsQ0FBQyxPQUFPLEVBQUUsQ0FBQztRQUU1QixPQUFPLFFBQVEsQ0FBQztJQUNsQixDQUFDO0NBQ0Y7QUF6QkQsc0NBeUJDIn0=