@mikro-orm/knex 6.0.0-dev.24 → 6.0.0-dev.241

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 (44) hide show
  1. package/AbstractSqlConnection.d.ts +22 -4
  2. package/AbstractSqlConnection.js +48 -18
  3. package/AbstractSqlDriver.d.ts +33 -25
  4. package/AbstractSqlDriver.js +565 -274
  5. package/AbstractSqlPlatform.d.ts +5 -4
  6. package/AbstractSqlPlatform.js +34 -5
  7. package/MonkeyPatchable.d.ts +1 -0
  8. package/MonkeyPatchable.js +3 -0
  9. package/PivotCollectionPersister.d.ts +17 -0
  10. package/PivotCollectionPersister.js +128 -0
  11. package/README.md +64 -107
  12. package/SqlEntityManager.d.ts +2 -6
  13. package/SqlEntityManager.js +4 -8
  14. package/SqlEntityRepository.d.ts +7 -6
  15. package/SqlEntityRepository.js +11 -8
  16. package/index.mjs +23 -5
  17. package/package.json +8 -8
  18. package/query/ArrayCriteriaNode.d.ts +5 -4
  19. package/query/ArrayCriteriaNode.js +7 -2
  20. package/query/CriteriaNode.d.ts +12 -10
  21. package/query/CriteriaNode.js +30 -17
  22. package/query/CriteriaNodeFactory.d.ts +6 -6
  23. package/query/CriteriaNodeFactory.js +33 -17
  24. package/query/ObjectCriteriaNode.d.ts +5 -4
  25. package/query/ObjectCriteriaNode.js +67 -27
  26. package/query/QueryBuilder.d.ts +58 -26
  27. package/query/QueryBuilder.js +444 -148
  28. package/query/QueryBuilderHelper.d.ts +13 -18
  29. package/query/QueryBuilderHelper.js +194 -167
  30. package/query/ScalarCriteriaNode.d.ts +4 -3
  31. package/query/ScalarCriteriaNode.js +18 -8
  32. package/query/enums.d.ts +7 -0
  33. package/query/enums.js +10 -2
  34. package/schema/DatabaseSchema.d.ts +5 -2
  35. package/schema/DatabaseSchema.js +28 -10
  36. package/schema/DatabaseTable.d.ts +13 -5
  37. package/schema/DatabaseTable.js +385 -60
  38. package/schema/SchemaComparator.d.ts +7 -5
  39. package/schema/SchemaComparator.js +68 -35
  40. package/schema/SchemaHelper.d.ts +7 -5
  41. package/schema/SchemaHelper.js +25 -11
  42. package/schema/SqlSchemaGenerator.d.ts +3 -6
  43. package/schema/SqlSchemaGenerator.js +71 -36
  44. package/typings.d.ts +30 -11
@@ -4,10 +4,14 @@ exports.AbstractSqlDriver = void 0;
4
4
  const core_1 = require("@mikro-orm/core");
5
5
  const query_1 = require("./query");
6
6
  const SqlEntityManager_1 = require("./SqlEntityManager");
7
+ const PivotCollectionPersister_1 = require("./PivotCollectionPersister");
7
8
  class AbstractSqlDriver extends core_1.DatabaseDriver {
9
+ [core_1.EntityManagerType];
10
+ connection;
11
+ replicas = [];
12
+ platform;
8
13
  constructor(config, platform, connection, connector) {
9
14
  super(config, connector);
10
- this.replicas = [];
11
15
  this.connection = new connection(this.config);
12
16
  this.replicas = this.createReplicas(conf => new connection(this.config, conf, 'read'));
13
17
  this.platform = platform;
@@ -25,21 +29,26 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
25
29
  return this.findVirtual(entityName, where, options);
26
30
  }
27
31
  const populate = this.autoJoinOneToOneOwner(meta, options.populate, options.fields);
28
- const joinedProps = this.joinedProps(meta, populate);
29
- const qb = this.createQueryBuilder(entityName, options.ctx, options.connectionType, false);
30
- const fields = this.buildFields(meta, populate, joinedProps, qb, options.fields);
31
- const joinedPropsOrderBy = this.buildJoinedPropsOrderBy(entityName, qb, meta, joinedProps);
32
- const orderBy = [...core_1.Utils.asArray(options.orderBy), ...joinedPropsOrderBy];
32
+ const joinedProps = this.joinedProps(meta, populate, options);
33
+ const qb = this.createQueryBuilder(entityName, options.ctx, options.connectionType, false, options.logging);
34
+ const fields = this.buildFields(meta, populate, joinedProps, qb, qb.alias, options.fields, options);
35
+ const orderBy = this.buildOrderBy(qb, meta, populate, options);
36
+ core_1.Utils.asArray(options.flags).forEach(flag => qb.setFlag(flag));
33
37
  if (core_1.Utils.isPrimaryKey(where, meta.compositePK)) {
34
38
  where = { [core_1.Utils.getPrimaryKeyHash(meta.primaryKeys)]: where };
35
39
  }
36
40
  const { first, last, before, after } = options;
37
41
  const isCursorPagination = [first, last, before, after].some(v => v != null);
42
+ qb.__populateWhere = options._populateWhere;
38
43
  qb.select(fields)
44
+ // only add populateWhere if we are populate-joining, as this will be used to add `on` conditions
39
45
  .populate(populate, joinedProps.length > 0 ? options.populateWhere : undefined)
40
46
  .where(where)
41
47
  .groupBy(options.groupBy)
42
48
  .having(options.having)
49
+ .indexHint(options.indexHint)
50
+ .comment(options.comments)
51
+ .hintComment(options.hintComments)
43
52
  .withSchema(this.getSchemaName(meta, options));
44
53
  if (isCursorPagination) {
45
54
  const { orderBy: newOrderBy, where } = this.processCursorOptions(meta, options, orderBy);
@@ -48,83 +57,92 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
48
57
  else {
49
58
  qb.orderBy(orderBy);
50
59
  }
51
- if (options.limit !== undefined) {
60
+ if (options.limit != null || options.offset != null) {
52
61
  qb.limit(options.limit, options.offset);
53
62
  }
54
63
  if (options.lockMode) {
55
64
  qb.setLockMode(options.lockMode, options.lockTableAliases);
56
65
  }
57
- core_1.Utils.asArray(options.flags).forEach(flag => qb.setFlag(flag));
58
66
  const result = await this.rethrow(qb.execute('all'));
59
67
  if (isCursorPagination && !first && !!last) {
60
68
  result.reverse();
61
69
  }
62
- if (joinedProps.length > 0) {
63
- return this.mergeJoinedResult(result, meta);
64
- }
65
70
  return result;
66
71
  }
67
72
  async findOne(entityName, where, options) {
68
73
  const opts = { populate: [], ...(options || {}) };
69
74
  const meta = this.metadata.find(entityName);
70
75
  const populate = this.autoJoinOneToOneOwner(meta, opts.populate, opts.fields);
71
- const joinedProps = this.joinedProps(meta, populate);
72
- if (joinedProps.length === 0) {
76
+ const joinedProps = this.joinedProps(meta, populate, options);
77
+ const hasToManyJoins = joinedProps.some(hint => this.hasToManyJoins(hint, meta));
78
+ if (joinedProps.length === 0 || !hasToManyJoins) {
73
79
  opts.limit = 1;
74
80
  }
75
81
  if (opts.limit > 0 && !opts.flags?.includes(core_1.QueryFlag.DISABLE_PAGINATE)) {
76
- opts.flags ?? (opts.flags = []);
82
+ opts.flags ??= [];
77
83
  opts.flags.push(core_1.QueryFlag.DISABLE_PAGINATE);
78
84
  }
79
85
  const res = await this.find(entityName, where, opts);
80
86
  return res[0] || null;
81
87
  }
82
- async findVirtual(entityName, where, options) {
83
- const meta = this.metadata.get(entityName);
84
- /* istanbul ignore next */
85
- if (!meta.expression) {
86
- return [];
87
- }
88
- if (typeof meta.expression === 'string') {
89
- return this.wrapVirtualExpressionInSubquery(meta, meta.expression, where, options);
88
+ hasToManyJoins(hint, meta) {
89
+ const [propName] = hint.field.split(':', 2);
90
+ const prop = meta.properties[propName];
91
+ if (prop && [core_1.ReferenceKind.ONE_TO_MANY, core_1.ReferenceKind.MANY_TO_MANY].includes(prop.kind)) {
92
+ return true;
90
93
  }
91
- const em = this.createEntityManager(false);
92
- em.setTransactionContext(options.ctx);
93
- const res = meta.expression(em, where, options);
94
- if (res instanceof (query_1.QueryBuilder)) {
95
- return this.wrapVirtualExpressionInSubquery(meta, res.getFormattedQuery(), where, options);
94
+ if (hint.children && prop.targetMeta) {
95
+ return hint.children.some(hint => this.hasToManyJoins(hint, prop.targetMeta));
96
96
  }
97
- /* istanbul ignore next */
98
- return res;
97
+ return false;
98
+ }
99
+ async findVirtual(entityName, where, options) {
100
+ return this.findFromVirtual(entityName, where, options, query_1.QueryType.SELECT);
99
101
  }
100
102
  async countVirtual(entityName, where, options) {
103
+ return this.findFromVirtual(entityName, where, options, query_1.QueryType.COUNT);
104
+ }
105
+ async findFromVirtual(entityName, where, options, type) {
101
106
  const meta = this.metadata.get(entityName);
102
107
  /* istanbul ignore next */
103
108
  if (!meta.expression) {
104
- return 0;
109
+ return type === query_1.QueryType.SELECT ? [] : 0;
105
110
  }
106
111
  if (typeof meta.expression === 'string') {
107
- return this.wrapVirtualExpressionInSubquery(meta, meta.expression, where, options, query_1.QueryType.COUNT);
112
+ return this.wrapVirtualExpressionInSubquery(meta, meta.expression, where, options, type);
108
113
  }
109
- const em = this.createEntityManager(false);
114
+ const em = this.createEntityManager();
110
115
  em.setTransactionContext(options.ctx);
111
116
  const res = meta.expression(em, where, options);
112
- if (res instanceof (query_1.QueryBuilder)) {
113
- return this.wrapVirtualExpressionInSubquery(meta, res.getFormattedQuery(), where, options, query_1.QueryType.COUNT);
117
+ if (typeof res === 'string') {
118
+ return this.wrapVirtualExpressionInSubquery(meta, res, where, options, type);
119
+ }
120
+ if (res instanceof query_1.QueryBuilder) {
121
+ return this.wrapVirtualExpressionInSubquery(meta, res.getFormattedQuery(), where, options, type);
122
+ }
123
+ if (core_1.Utils.isObject(res)) {
124
+ const { sql, bindings } = res.toSQL();
125
+ const query = this.platform.formatQuery(sql, bindings);
126
+ return this.wrapVirtualExpressionInSubquery(meta, query, where, options, type);
114
127
  }
115
128
  /* istanbul ignore next */
116
129
  return res;
117
130
  }
118
- async wrapVirtualExpressionInSubquery(meta, expression, where, options, type = query_1.QueryType.SELECT) {
131
+ async wrapVirtualExpressionInSubquery(meta, expression, where, options, type) {
119
132
  const qb = this.createQueryBuilder(meta.className, options?.ctx, options.connectionType, options.convertCustomTypes)
120
- .limit(options?.limit, options?.offset);
121
- if (options.orderBy) {
122
- qb.orderBy(options.orderBy);
133
+ .indexHint(options.indexHint)
134
+ .comment(options.comments)
135
+ .hintComment(options.hintComments);
136
+ if (type !== query_1.QueryType.COUNT) {
137
+ qb.limit(options?.limit, options?.offset);
138
+ if (options.orderBy) {
139
+ qb.orderBy(options.orderBy);
140
+ }
123
141
  }
124
142
  qb.where(where);
125
143
  const kqb = qb.getKnexQuery().clear('select');
126
144
  if (type === query_1.QueryType.COUNT) {
127
- kqb.select(qb.raw('count(*) as count'));
145
+ kqb.select(this.connection.getKnex().raw('count(*) as count'));
128
146
  }
129
147
  else { // select
130
148
  kqb.select('*');
@@ -143,40 +161,88 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
143
161
  return null;
144
162
  }
145
163
  if (qb) {
164
+ // here we map the aliased results (cartesian product) to an object graph
146
165
  this.mapJoinedProps(ret, meta, populate, qb, ret, map);
147
166
  }
148
167
  return ret;
149
168
  }
150
169
  mapJoinedProps(result, meta, populate, qb, root, map, parentJoinPath) {
151
170
  const joinedProps = this.joinedProps(meta, populate);
152
- joinedProps.forEach(p => {
153
- const relation = meta.properties[p.field];
171
+ joinedProps.forEach(hint => {
172
+ const [propName, ref] = hint.field.split(':', 2);
173
+ const prop = meta.properties[propName];
154
174
  /* istanbul ignore next */
155
- if (!relation) {
175
+ if (!prop) {
176
+ return;
177
+ }
178
+ const pivotRefJoin = prop.kind === core_1.ReferenceKind.MANY_TO_MANY && ref;
179
+ const meta2 = this.metadata.find(prop.type);
180
+ let path = parentJoinPath ? `${parentJoinPath}.${prop.name}` : `${meta.name}.${prop.name}`;
181
+ if (!parentJoinPath) {
182
+ path = '[populate]' + path;
183
+ }
184
+ if (pivotRefJoin) {
185
+ path += '[pivot]';
186
+ }
187
+ const relationAlias = qb.getAliasForJoinPath(path, { matchPopulateJoins: true });
188
+ // pivot ref joins via joined strategy need to be handled separately here, as they dont join the target entity
189
+ if (pivotRefJoin) {
190
+ let item;
191
+ if (prop.inverseJoinColumns.length > 1) { // composite keys
192
+ item = prop.inverseJoinColumns.map(name => root[`${relationAlias}__${name}`]);
193
+ }
194
+ else {
195
+ const alias = `${relationAlias}__${prop.inverseJoinColumns[0]}`;
196
+ item = root[alias];
197
+ }
198
+ prop.joinColumns.forEach(name => delete root[`${relationAlias}__${name}`]);
199
+ prop.inverseJoinColumns.forEach(name => delete root[`${relationAlias}__${name}`]);
200
+ result[prop.name] ??= [];
201
+ if (item) {
202
+ result[prop.name].push(item);
203
+ }
156
204
  return;
157
205
  }
158
- const meta2 = this.metadata.find(relation.type);
159
- const path = parentJoinPath ? `${parentJoinPath}.${relation.name}` : `${meta.name}.${relation.name}`;
160
- const relationAlias = qb.getAliasForJoinPath(path);
161
- const relationPojo = {};
162
206
  // If the primary key value for the relation is null, we know we haven't joined to anything
163
207
  // and therefore we don't return any record (since all values would be null)
164
208
  const hasPK = meta2.primaryKeys.every(pk => meta2.properties[pk].fieldNames.every(name => {
165
209
  return root[`${relationAlias}__${name}`] != null;
166
210
  }));
167
211
  if (!hasPK) {
168
- // initialize empty collections
169
- if ([core_1.ReferenceType.MANY_TO_MANY, core_1.ReferenceType.ONE_TO_MANY].includes(relation.reference)) {
170
- result[relation.name] = result[relation.name] || [];
212
+ if ([core_1.ReferenceKind.MANY_TO_MANY, core_1.ReferenceKind.ONE_TO_MANY].includes(prop.kind)) {
213
+ result[prop.name] ??= [];
214
+ }
215
+ if ([core_1.ReferenceKind.MANY_TO_ONE, core_1.ReferenceKind.ONE_TO_ONE].includes(prop.kind)) {
216
+ result[prop.name] ??= null;
171
217
  }
172
218
  return;
173
219
  }
220
+ const relationPojo = {};
174
221
  meta2.props
175
- .filter(prop => this.platform.shouldHaveColumn(prop, p.children || []))
222
+ .filter(prop => prop.persist === false && prop.fieldNames)
223
+ .filter(prop => !prop.lazy || populate.some(p => p.field === prop.name || p.all))
176
224
  .forEach(prop => {
225
+ /* istanbul ignore if */
177
226
  if (prop.fieldNames.length > 1) { // composite keys
178
227
  relationPojo[prop.name] = prop.fieldNames.map(name => root[`${relationAlias}__${name}`]);
228
+ }
229
+ else {
230
+ const alias = `${relationAlias}__${prop.fieldNames[0]}`;
231
+ relationPojo[prop.name] = root[alias];
232
+ }
233
+ });
234
+ meta2.props
235
+ .filter(prop => this.platform.shouldHaveColumn(prop, hint.children || []))
236
+ .forEach(prop => {
237
+ if (prop.fieldNames.length > 1) { // composite keys
238
+ const fk = prop.fieldNames.map(name => root[`${relationAlias}__${name}`]);
179
239
  prop.fieldNames.map(name => delete root[`${relationAlias}__${name}`]);
240
+ relationPojo[prop.name] = core_1.Utils.mapFlatCompositePrimaryKey(fk, prop);
241
+ }
242
+ else if (prop.runtimeType === 'Date') {
243
+ const alias = `${relationAlias}__${prop.fieldNames[0]}`;
244
+ relationPojo[prop.name] = (typeof root[alias] === 'string' ? new Date(root[alias]) : root[alias]);
245
+ delete root[alias];
180
246
  }
181
247
  else {
182
248
  const alias = `${relationAlias}__${prop.fieldNames[0]}`;
@@ -184,41 +250,26 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
184
250
  delete root[alias];
185
251
  }
186
252
  });
187
- const key = `${meta.name}-${(core_1.Utils.getCompositeKeyHash(result, meta))}`;
188
- if (map[key]) {
189
- result[relation.name] = map[key][relation.name];
190
- }
191
- else {
192
- map[key] = result;
193
- }
194
- if ([core_1.ReferenceType.MANY_TO_MANY, core_1.ReferenceType.ONE_TO_MANY].includes(relation.reference)) {
195
- result[relation.name] = result[relation.name] || [];
196
- this.appendToCollection(meta2, result[relation.name], relationPojo);
253
+ if ([core_1.ReferenceKind.MANY_TO_MANY, core_1.ReferenceKind.ONE_TO_MANY].includes(prop.kind)) {
254
+ result[prop.name] ??= [];
255
+ result[prop.name].push(relationPojo);
197
256
  }
198
257
  else {
199
- result[relation.name] = relationPojo;
258
+ result[prop.name] = relationPojo;
200
259
  }
201
- const populateChildren = p.children || [];
260
+ const populateChildren = hint.children || [];
202
261
  this.mapJoinedProps(relationPojo, meta2, populateChildren, qb, root, map, path);
203
262
  });
204
263
  }
205
- appendToCollection(meta, collection, relationPojo) {
206
- if (collection.length === 0) {
207
- return void collection.push(relationPojo);
208
- }
209
- const last = collection[collection.length - 1];
210
- const pk1 = core_1.Utils.getCompositeKeyHash(last, meta);
211
- const pk2 = core_1.Utils.getCompositeKeyHash(relationPojo, meta);
212
- if (pk1 !== pk2) {
213
- collection.push(relationPojo);
214
- }
215
- }
216
264
  async count(entityName, where, options = {}) {
217
265
  const meta = this.metadata.find(entityName);
218
266
  if (meta?.virtual) {
219
267
  return this.countVirtual(entityName, where, options);
220
268
  }
221
269
  const qb = this.createQueryBuilder(entityName, options.ctx, options.connectionType, false)
270
+ .indexHint(options.indexHint)
271
+ .comment(options.comments)
272
+ .hintComment(options.hintComments)
222
273
  .groupBy(options.groupBy)
223
274
  .having(options.having)
224
275
  .populate(options.populate ?? [])
@@ -227,7 +278,7 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
227
278
  return this.rethrow(qb.getCount());
228
279
  }
229
280
  async nativeInsert(entityName, data, options = {}) {
230
- options.convertCustomTypes ?? (options.convertCustomTypes = true);
281
+ options.convertCustomTypes ??= true;
231
282
  const meta = this.metadata.find(entityName);
232
283
  const collections = this.extractManyToMany(entityName, data);
233
284
  const pks = meta?.primaryKeys ?? [this.config.getNamingStrategy().referenceColumnName()];
@@ -247,16 +298,20 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
247
298
  return res;
248
299
  }
249
300
  async nativeInsertMany(entityName, data, options = {}) {
250
- options.processCollections ?? (options.processCollections = true);
251
- options.convertCustomTypes ?? (options.convertCustomTypes = true);
252
- const meta = this.metadata.find(entityName);
301
+ options.processCollections ??= true;
302
+ options.convertCustomTypes ??= true;
303
+ const meta = this.metadata.find(entityName)?.root;
253
304
  const collections = options.processCollections ? data.map(d => this.extractManyToMany(entityName, d)) : [];
254
305
  const pks = this.getPrimaryKeyFields(entityName);
255
306
  const set = new Set();
256
- data.forEach(row => Object.keys(row).forEach(k => set.add(k)));
307
+ data.forEach(row => core_1.Utils.keys(row).forEach(k => set.add(k)));
257
308
  const props = [...set].map(name => meta?.properties[name] ?? { name, fieldNames: [name] });
258
- const fields = core_1.Utils.flatten(props.map(prop => prop.fieldNames));
309
+ let fields = core_1.Utils.flatten(props.map(prop => prop.fieldNames));
310
+ const duplicates = core_1.Utils.findDuplicates(fields);
259
311
  const params = [];
312
+ if (duplicates.length) {
313
+ fields = core_1.Utils.unique(fields);
314
+ }
260
315
  /* istanbul ignore next */
261
316
  const tableName = meta ? this.getTableName(meta, options) : this.platform.quoteIdentifier(entityName);
262
317
  let sql = `insert into ${tableName} `;
@@ -267,29 +322,62 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
267
322
  else {
268
323
  sql += ' ' + data.map(() => `select null as ${this.platform.quoteIdentifier(pks[0])}`).join(' union all ');
269
324
  }
325
+ const addParams = (prop, row) => {
326
+ let value = row[prop.name];
327
+ if (prop.kind === core_1.ReferenceKind.EMBEDDED && prop.object) {
328
+ if (prop.array) {
329
+ for (let i = 0; i < value.length; i++) {
330
+ const item = value[i];
331
+ value[i] = this.mapDataToFieldNames(item, false, prop.embeddedProps, options.convertCustomTypes);
332
+ }
333
+ }
334
+ else {
335
+ value = this.mapDataToFieldNames(value, false, prop.embeddedProps, options.convertCustomTypes);
336
+ }
337
+ }
338
+ if (options.convertCustomTypes && prop.customType) {
339
+ params.push(prop.customType.convertToDatabaseValue(value, this.platform, { key: prop.name, mode: 'query-data' }));
340
+ return;
341
+ }
342
+ params.push(value);
343
+ };
270
344
  if (fields.length > 0 || this.platform.usesDefaultKeyword()) {
271
345
  sql += data.map(row => {
272
346
  const keys = [];
347
+ const usedDups = [];
273
348
  props.forEach(prop => {
274
349
  if (prop.fieldNames.length > 1) {
275
- params.push(...(row[prop.name] ?? prop.fieldNames.map(() => null)));
276
- keys.push(...prop.fieldNames.map(_ => '?'));
277
- }
278
- else if (prop.customType && 'convertToDatabaseValueSQL' in prop.customType && !this.platform.isRaw(row[prop.name])) {
279
- keys.push(prop.customType.convertToDatabaseValueSQL('?', this.platform));
280
- params.push(row[prop.name]);
350
+ const param = core_1.Utils.flatten([...row[prop.name] ?? prop.fieldNames.map(() => null)]);
351
+ const key = param.map(() => '?');
352
+ prop.fieldNames.forEach((field, idx) => {
353
+ if (!duplicates.includes(field) || !usedDups.includes(field)) {
354
+ params.push(param[idx]);
355
+ keys.push(key[idx]);
356
+ usedDups.push(field);
357
+ }
358
+ });
281
359
  }
282
360
  else {
283
- params.push(row[prop.name]);
284
- keys.push('?');
361
+ const field = prop.fieldNames[0];
362
+ if (!duplicates.includes(field) || !usedDups.includes(field)) {
363
+ if (prop.customType && 'convertToDatabaseValueSQL' in prop.customType && !this.platform.isRaw(row[prop.name])) {
364
+ keys.push(prop.customType.convertToDatabaseValueSQL('?', this.platform));
365
+ }
366
+ else {
367
+ keys.push('?');
368
+ }
369
+ addParams(prop, row);
370
+ usedDups.push(field);
371
+ }
285
372
  }
286
373
  });
287
374
  return '(' + (keys.join(', ') || 'default') + ')';
288
375
  }).join(', ');
289
376
  }
290
- if (this.platform.usesReturningStatement()) {
291
- /* istanbul ignore next */
292
- const returningProps = meta.hydrateProps.filter(prop => prop.persist !== false && (prop.primary || prop.defaultRaw));
377
+ if (meta && this.platform.usesReturningStatement()) {
378
+ const returningProps = meta.props
379
+ .filter(prop => prop.persist !== false && prop.defaultRaw || prop.autoincrement || prop.generated)
380
+ .filter(prop => !(prop.name in data[0]) || core_1.Utils.isRawSql(data[0][prop.name]));
293
381
  const returningFields = core_1.Utils.flatten(returningProps.map(prop => prop.fieldNames));
294
382
  /* istanbul ignore next */
295
383
  sql += returningFields.length > 0 ? ` returning ${returningFields.map(field => this.platform.quoteIdentifier(field)).join(', ')}` : '';
@@ -301,8 +389,8 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
301
389
  pk = data.map(d => core_1.Utils.getPrimaryKeyCond(d, pks));
302
390
  }
303
391
  else {
304
- res.row ?? (res.row = {});
305
- res.rows ?? (res.rows = []);
392
+ res.row ??= {};
393
+ res.rows ??= [];
306
394
  pk = data.map((d, i) => d[pks[0]] ?? res.rows[i]?.[pks[0]]).map(d => [d]);
307
395
  res.insertId = res.insertId || res.row[pks[0]];
308
396
  }
@@ -312,7 +400,7 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
312
400
  return res;
313
401
  }
314
402
  async nativeUpdate(entityName, where, data, options = {}) {
315
- options.convertCustomTypes ?? (options.convertCustomTypes = true);
403
+ options.convertCustomTypes ??= true;
316
404
  const meta = this.metadata.find(entityName);
317
405
  const pks = this.getPrimaryKeyFields(entityName);
318
406
  const collections = this.extractManyToMany(entityName, data);
@@ -326,14 +414,27 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
326
414
  .withSchema(this.getSchemaName(meta, options));
327
415
  if (options.upsert) {
328
416
  /* istanbul ignore next */
329
- const uniqueFields = core_1.Utils.isPlainObject(where) ? Object.keys(where) : meta.primaryKeys;
330
- /* istanbul ignore next */
417
+ const uniqueFields = options.onConflictFields ?? (core_1.Utils.isPlainObject(where) ? core_1.Utils.keys(where) : meta.primaryKeys);
418
+ const returning = (0, core_1.getOnConflictReturningFields)(meta, data, uniqueFields, options);
331
419
  qb.insert(data)
332
- .onConflict(uniqueFields.map(p => meta?.properties[p]?.fieldNames[0] ?? p))
333
- .merge(Object.keys(data).filter(f => !uniqueFields.includes(f)));
420
+ .onConflict(uniqueFields)
421
+ .returning(returning);
422
+ if (!options.onConflictAction || options.onConflictAction === 'merge') {
423
+ const fields = (0, core_1.getOnConflictFields)(data, uniqueFields, options);
424
+ qb.merge(fields);
425
+ }
426
+ if (options.onConflictAction === 'ignore') {
427
+ qb.ignore();
428
+ }
334
429
  }
335
430
  else {
336
431
  qb.update(data).where(where);
432
+ // reload generated columns and version fields
433
+ const returning = [];
434
+ meta?.props
435
+ .filter(prop => (prop.generated && !prop.primary) || prop.version)
436
+ .forEach(prop => returning.push(prop.name));
437
+ qb.returning(returning);
337
438
  }
338
439
  res = await this.rethrow(qb.execute('run', false));
339
440
  }
@@ -343,23 +444,57 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
343
444
  return res;
344
445
  }
345
446
  async nativeUpdateMany(entityName, where, data, options = {}) {
346
- options.processCollections ?? (options.processCollections = true);
347
- options.convertCustomTypes ?? (options.convertCustomTypes = true);
447
+ options.processCollections ??= true;
448
+ options.convertCustomTypes ??= true;
348
449
  const meta = this.metadata.get(entityName);
349
450
  if (options.upsert) {
350
- const uniqueFields = core_1.Utils.isPlainObject(where[0]) ? Object.keys(where[0]) : meta.primaryKeys;
451
+ const uniqueFields = options.onConflictFields ?? (core_1.Utils.isPlainObject(where[0]) ? Object.keys(where[0]).flatMap(key => core_1.Utils.splitPrimaryKeys(key)) : meta.primaryKeys);
351
452
  const qb = this.createQueryBuilder(entityName, options.ctx, 'write', options.convertCustomTypes).withSchema(this.getSchemaName(meta, options));
453
+ const returning = (0, core_1.getOnConflictReturningFields)(meta, data[0], uniqueFields, options);
352
454
  qb.insert(data)
353
- .onConflict(uniqueFields.map(p => meta?.properties[p]?.fieldNames[0] ?? p))
354
- .merge(Object.keys(data[0]).filter(f => !uniqueFields.includes(f)));
355
- return qb.execute('run', false);
455
+ .onConflict(uniqueFields)
456
+ .returning(returning);
457
+ if (!options.onConflictAction || options.onConflictAction === 'merge') {
458
+ const fields = (0, core_1.getOnConflictFields)(data[0], uniqueFields, options);
459
+ qb.merge(fields);
460
+ }
461
+ if (options.onConflictAction === 'ignore') {
462
+ qb.ignore();
463
+ }
464
+ return this.rethrow(qb.execute('run', false));
356
465
  }
357
466
  const collections = options.processCollections ? data.map(d => this.extractManyToMany(entityName, d)) : [];
358
467
  const keys = new Set();
359
- data.forEach(row => Object.keys(row).forEach(k => keys.add(k)));
468
+ const returning = new Set();
469
+ data.forEach(row => {
470
+ core_1.Utils.keys(row).forEach(k => {
471
+ keys.add(k);
472
+ if (core_1.Utils.isRawSql(row[k])) {
473
+ returning.add(k);
474
+ }
475
+ });
476
+ });
477
+ // reload generated columns and version fields
478
+ meta?.props
479
+ .filter(prop => (prop.generated && !prop.primary) || prop.version)
480
+ .forEach(prop => returning.add(prop.name));
360
481
  const pkCond = core_1.Utils.flatten(meta.primaryKeys.map(pk => meta.properties[pk].fieldNames)).map(pk => `${this.platform.quoteIdentifier(pk)} = ?`).join(' and ');
361
482
  const params = [];
362
483
  let sql = `update ${this.getTableName(meta, options)} set `;
484
+ const addParams = (prop, value) => {
485
+ if (prop.kind === core_1.ReferenceKind.EMBEDDED && prop.object) {
486
+ if (prop.array) {
487
+ for (let i = 0; i < value.length; i++) {
488
+ const item = value[i];
489
+ value[i] = this.mapDataToFieldNames(item, false, prop.embeddedProps, options.convertCustomTypes);
490
+ }
491
+ }
492
+ else {
493
+ value = this.mapDataToFieldNames(value, false, prop.embeddedProps, options.convertCustomTypes);
494
+ }
495
+ }
496
+ params.push(value);
497
+ };
363
498
  keys.forEach(key => {
364
499
  const prop = meta.properties[key];
365
500
  prop.fieldNames.forEach((fieldName, fieldNameIdx) => {
@@ -374,7 +509,8 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
374
509
  else {
375
510
  sql += '?';
376
511
  }
377
- params.push(...pks, prop.fieldNames.length > 1 ? data[idx][key]?.[fieldNameIdx] : data[idx][key]);
512
+ params.push(...pks);
513
+ addParams(prop, prop.fieldNames.length > 1 ? data[idx][key]?.[fieldNameIdx] : data[idx][key]);
378
514
  }
379
515
  });
380
516
  sql += ` else ${this.platform.quoteIdentifier(fieldName)} end, `;
@@ -385,7 +521,7 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
385
521
  const versionProperty = meta.properties[meta.versionProperty];
386
522
  const quotedFieldName = this.platform.quoteIdentifier(versionProperty.fieldNames[0]);
387
523
  sql += `${quotedFieldName} = `;
388
- if (versionProperty.type.toLowerCase() === 'date') {
524
+ if (versionProperty.runtimeType === 'Date') {
389
525
  sql += this.platform.getCurrentTimestampSQL(versionProperty.length);
390
526
  }
391
527
  else {
@@ -416,6 +552,11 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
416
552
  return '?';
417
553
  });
418
554
  sql += ` in (${conds.join(', ')})`;
555
+ if (this.platform.usesReturningStatement() && returning.size > 0) {
556
+ const returningFields = core_1.Utils.flatten([...returning].map(prop => meta.properties[prop].fieldNames));
557
+ /* istanbul ignore next */
558
+ sql += returningFields.length > 0 ? ` returning ${returningFields.map(field => this.platform.quoteIdentifier(field)).join(', ')}` : '';
559
+ }
419
560
  const res = await this.rethrow(this.execute(sql, params, 'run', options.ctx));
420
561
  for (let i = 0; i < collections.length; i++) {
421
562
  await this.processManyToMany(meta, where[i], collections[i], false, options);
@@ -431,49 +572,81 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
431
572
  const qb = this.createQueryBuilder(entityName, options.ctx, 'write', false).delete(where).withSchema(this.getSchemaName(meta, options));
432
573
  return this.rethrow(qb.execute('run', false));
433
574
  }
434
- async syncCollection(coll, options) {
435
- const wrapped = (0, core_1.helper)(coll.owner);
436
- const meta = wrapped.__meta;
437
- const pks = wrapped.getPrimaryKeys(true);
438
- const snap = coll.getSnapshot();
439
- const includes = (arr, item) => !!arr.find(i => core_1.Utils.equals(i, item));
440
- const snapshot = snap ? snap.map(item => (0, core_1.helper)(item).getPrimaryKeys(true)) : [];
441
- const current = coll.getItems(false).map(item => (0, core_1.helper)(item).getPrimaryKeys(true));
442
- const deleteDiff = snap ? snapshot.filter(item => !includes(current, item)) : true;
443
- const insertDiff = current.filter(item => !includes(snapshot, item));
444
- const target = snapshot.filter(item => includes(current, item)).concat(...insertDiff);
445
- const equals = core_1.Utils.equals(current, target);
446
- const ctx = options?.ctx;
447
- // wrong order if we just delete and insert to the end (only owning sides can have fixed order)
448
- if (coll.property.owner && coll.property.fixedOrder && !equals && Array.isArray(deleteDiff)) {
449
- deleteDiff.length = insertDiff.length = 0;
450
- deleteDiff.push(...snapshot);
451
- insertDiff.push(...current);
452
- }
453
- if (coll.property.reference === core_1.ReferenceType.ONE_TO_MANY) {
454
- const cols = coll.property.referencedColumnNames;
455
- const qb = this.createQueryBuilder(coll.property.type, ctx, 'write')
456
- .withSchema(this.getSchemaName(meta, options))
457
- .update({ [coll.property.mappedBy]: pks })
458
- .getKnexQuery()
459
- .whereIn(cols, insertDiff);
460
- return this.rethrow(this.execute(qb));
461
- }
462
- /* istanbul ignore next */
463
- const ownerSchema = wrapped.getSchema() === '*' ? this.config.get('schema') : wrapped.getSchema();
464
- const pivotMeta = this.metadata.find(coll.property.pivotEntity);
465
- if (pivotMeta.schema === '*') {
575
+ async syncCollections(collections, options) {
576
+ const groups = {};
577
+ for (const coll of collections) {
578
+ const wrapped = (0, core_1.helper)(coll.owner);
579
+ const meta = wrapped.__meta;
580
+ const pks = wrapped.getPrimaryKeys(true);
581
+ const snap = coll.getSnapshot();
582
+ const includes = (arr, item) => !!arr.find(i => core_1.Utils.equals(i, item));
583
+ const snapshot = snap ? snap.map(item => (0, core_1.helper)(item).getPrimaryKeys(true)) : [];
584
+ const current = coll.getItems(false).map(item => (0, core_1.helper)(item).getPrimaryKeys(true));
585
+ const deleteDiff = snap ? snapshot.filter(item => !includes(current, item)) : true;
586
+ const insertDiff = current.filter(item => !includes(snapshot, item));
587
+ const target = snapshot.filter(item => includes(current, item)).concat(...insertDiff);
588
+ const equals = core_1.Utils.equals(current, target);
589
+ // wrong order if we just delete and insert to the end (only owning sides can have fixed order)
590
+ if (coll.property.owner && coll.property.fixedOrder && !equals && Array.isArray(deleteDiff)) {
591
+ deleteDiff.length = insertDiff.length = 0;
592
+ deleteDiff.push(...snapshot);
593
+ insertDiff.push(...current);
594
+ }
595
+ if (coll.property.kind === core_1.ReferenceKind.ONE_TO_MANY) {
596
+ const cols = coll.property.referencedColumnNames;
597
+ const qb = this.createQueryBuilder(coll.property.type, options?.ctx, 'write')
598
+ .withSchema(this.getSchemaName(meta, options));
599
+ if (coll.getSnapshot() === undefined) {
600
+ if (coll.property.orphanRemoval) {
601
+ const kqb = qb.delete({ [coll.property.mappedBy]: pks })
602
+ .getKnexQuery()
603
+ .whereNotIn(cols, insertDiff);
604
+ return this.rethrow(this.execute(kqb));
605
+ }
606
+ const kqb = qb.update({ [coll.property.mappedBy]: null })
607
+ .getKnexQuery()
608
+ .whereNotIn(cols, insertDiff);
609
+ return this.rethrow(this.execute(kqb));
610
+ }
611
+ const kqb = qb.update({ [coll.property.mappedBy]: pks })
612
+ .getKnexQuery()
613
+ .whereIn(cols, insertDiff);
614
+ return this.rethrow(this.execute(kqb));
615
+ }
466
616
  /* istanbul ignore next */
467
- options ?? (options = {});
468
- options.schema = ownerSchema;
617
+ const pivotMeta = this.metadata.find(coll.property.pivotEntity);
618
+ let schema = pivotMeta.schema;
619
+ if (schema === '*') {
620
+ const ownerSchema = wrapped.getSchema() === '*' ? this.config.get('schema') : wrapped.getSchema();
621
+ schema = coll.property.owner ? ownerSchema : this.config.get('schema');
622
+ }
623
+ else if (schema == null) {
624
+ schema = this.config.get('schema');
625
+ }
626
+ const tableName = `${schema ?? '_'}.${pivotMeta.tableName}`;
627
+ const persister = groups[tableName] ??= new PivotCollectionPersister_1.PivotCollectionPersister(pivotMeta, this, options?.ctx, schema);
628
+ persister.enqueueUpdate(coll.property, insertDiff, deleteDiff, pks);
629
+ }
630
+ for (const persister of core_1.Utils.values(groups)) {
631
+ await this.rethrow(persister.execute());
469
632
  }
470
- return this.rethrow(this.updateCollectionDiff(meta, coll.property, pks, deleteDiff, insertDiff, options));
471
633
  }
472
- async loadFromPivotTable(prop, owners, where = {}, orderBy, ctx, options) {
473
- const pivotProp2 = this.getPivotInverseProperty(prop);
474
- const ownerMeta = this.metadata.find(pivotProp2.type);
634
+ async loadFromPivotTable(prop, owners, where = {}, orderBy, ctx, options, pivotJoin) {
475
635
  const pivotMeta = this.metadata.find(prop.pivotEntity);
476
- const cond = { [`${prop.pivotEntity}.${pivotProp2.name}`]: { $in: ownerMeta.compositePK ? owners : owners.map(o => o[0]) } };
636
+ const pivotProp1 = pivotMeta.relations[prop.owner ? 1 : 0];
637
+ const pivotProp2 = pivotMeta.relations[prop.owner ? 0 : 1];
638
+ const ownerMeta = this.metadata.find(pivotProp2.type);
639
+ options = { ...options };
640
+ const qb = this.createQueryBuilder(prop.pivotEntity, ctx, options.connectionType)
641
+ .withSchema(this.getSchemaName(pivotMeta, options))
642
+ .indexHint(options.indexHint)
643
+ .comment(options.comments)
644
+ .hintComment(options.hintComments);
645
+ const pivotAlias = qb.alias;
646
+ const pivotKey = pivotProp2.joinColumns.map(column => `${pivotAlias}.${column}`).join(core_1.Utils.PK_SEPARATOR);
647
+ const cond = {
648
+ [pivotKey]: { $in: ownerMeta.compositePK ? owners : owners.map(o => o[0]) },
649
+ };
477
650
  /* istanbul ignore if */
478
651
  if (!core_1.Utils.isEmpty(where) && Object.keys(where).every(k => core_1.Utils.isOperator(k, false))) {
479
652
  where = cond;
@@ -481,29 +654,51 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
481
654
  else {
482
655
  where = { ...where, ...cond };
483
656
  }
484
- orderBy = this.getPivotOrderBy(prop, orderBy);
485
- const qb = this.createQueryBuilder(prop.type, ctx, options?.connectionType)
486
- .unsetFlag(core_1.QueryFlag.CONVERT_CUSTOM_TYPES)
487
- .withSchema(this.getSchemaName(prop.targetMeta, options));
488
- const populate = this.autoJoinOneToOneOwner(prop.targetMeta, [{ field: prop.pivotEntity }]);
489
- const fields = this.buildFields(prop.targetMeta, (options?.populate ?? []), [], qb, options?.fields);
490
- qb.select(fields).populate(populate).where(where).orderBy(orderBy).setLockMode(options?.lockMode, options?.lockTableAliases);
491
- if (owners.length === 1 && (options?.offset != null || options?.limit != null)) {
492
- qb.limit(options.limit, options.offset);
493
- }
494
- if (prop.targetMeta.schema !== '*' && pivotMeta.schema === '*' && options?.schema) {
495
- // eslint-disable-next-line dot-notation
496
- qb['finalize']();
497
- // eslint-disable-next-line dot-notation
498
- Object.values(qb['_joins']).forEach(join => {
499
- join.schema = options.schema;
657
+ orderBy = this.getPivotOrderBy(prop, pivotProp1, pivotAlias, orderBy);
658
+ const populate = this.autoJoinOneToOneOwner(prop.targetMeta, []);
659
+ const fields = [];
660
+ const k1 = !prop.owner ? 'joinColumns' : 'inverseJoinColumns';
661
+ const k2 = prop.owner ? 'joinColumns' : 'inverseJoinColumns';
662
+ const cols = [
663
+ ...prop[k1].map(col => `${pivotAlias}.${col} as fk__${col}`),
664
+ ...prop[k2].map(col => `${pivotAlias}.${col} as fk__${col}`),
665
+ ];
666
+ fields.push(...cols);
667
+ if (!pivotJoin) {
668
+ const targetAlias = qb.getNextAlias(prop.targetMeta.tableName);
669
+ const targetSchema = this.getSchemaName(prop.targetMeta, options) ?? this.platform.getDefaultSchemaName();
670
+ qb.innerJoin(pivotProp1.name, targetAlias, {}, targetSchema);
671
+ const targetFields = this.buildFields(prop.targetMeta, (options.populate ?? []), [], qb, targetAlias, options.fields, options);
672
+ for (const field of targetFields) {
673
+ const f = field.toString();
674
+ fields.unshift(f.includes('.') ? field : `${targetAlias}.${f}`);
675
+ }
676
+ // we need to handle 1:1 owner auto-joins explicitly, as the QB type is the pivot table, not the target
677
+ populate.forEach(hint => {
678
+ const alias = qb.getNextAlias(prop.targetMeta.tableName);
679
+ qb.leftJoin(`${targetAlias}.${hint.field}`, alias);
680
+ // eslint-disable-next-line dot-notation
681
+ Object.values(qb['_joins']).forEach(join => {
682
+ const [propName] = hint.field.split(':', 2);
683
+ if (join.alias === alias && join.prop.name === propName) {
684
+ fields.push(...qb.helper.mapJoinColumns(qb.type, join));
685
+ }
686
+ });
500
687
  });
501
688
  }
502
- const items = owners.length ? await this.rethrow(qb.execute('all')) : [];
689
+ qb.select(fields)
690
+ .where({ [pivotProp1.name]: where })
691
+ .orderBy(orderBy)
692
+ .setLockMode(options.lockMode, options.lockTableAliases);
693
+ if (owners.length === 1 && (options.offset != null || options.limit != null)) {
694
+ qb.limit(options.limit, options.offset);
695
+ }
696
+ const res = owners.length ? await this.rethrow(qb.execute('all', { mergeResults: false, mapResults: false })) : [];
697
+ const items = res.map((row) => super.mapResult(row, prop.targetMeta));
503
698
  const map = {};
504
699
  const pkProps = ownerMeta.getPrimaryProps();
505
700
  owners.forEach(owner => {
506
- const key = core_1.Utils.getPrimaryKeyHash(prop.joinColumns.map((col, idx) => {
701
+ const key = core_1.Utils.getPrimaryKeyHash(prop.joinColumns.map((_col, idx) => {
507
702
  const pkProp = pkProps[idx];
508
703
  return pkProp.customType ? pkProp.customType.convertToJSValue(owner[idx], this.platform) : owner[idx];
509
704
  }));
@@ -516,10 +711,25 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
516
711
  }));
517
712
  map[key].push(item);
518
713
  prop.joinColumns.forEach(col => delete item[`fk__${col}`]);
519
- prop.inverseJoinColumns.forEach(col => delete item[`fk__${col}`]);
714
+ prop.inverseJoinColumns.forEach((col, idx) => {
715
+ core_1.Utils.renameKey(item, `fk__${col}`, prop.targetMeta.primaryKeys[idx]);
716
+ });
520
717
  });
521
718
  return map;
522
719
  }
720
+ getPivotOrderBy(prop, pivotProp, pivotAlias, orderBy) {
721
+ // FIXME this is ignoring the rest of the array items
722
+ if (!core_1.Utils.isEmpty(orderBy)) {
723
+ return [{ [pivotProp.name]: core_1.Utils.asArray(orderBy)[0] }];
724
+ }
725
+ if (!core_1.Utils.isEmpty(prop.orderBy)) {
726
+ return [{ [pivotProp.name]: core_1.Utils.asArray(prop.orderBy)[0] }];
727
+ }
728
+ if (prop.fixedOrder) {
729
+ return [{ [`${pivotAlias}.${prop.fixedOrderColumn}`]: core_1.QueryOrder.ASC }];
730
+ }
731
+ return [];
732
+ }
523
733
  async execute(queryOrKnex, params = [], method = 'all', ctx) {
524
734
  return this.rethrow(this.connection.execute(queryOrKnex, params, method, ctx));
525
735
  }
@@ -532,39 +742,66 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
532
742
  }
533
743
  const relationsToPopulate = populate.map(({ field }) => field);
534
744
  const toPopulate = meta.relations
535
- .filter(prop => prop.reference === core_1.ReferenceType.ONE_TO_ONE && !prop.owner && !relationsToPopulate.includes(prop.name))
745
+ .filter(prop => prop.kind === core_1.ReferenceKind.ONE_TO_ONE && !prop.owner && !relationsToPopulate.includes(prop.name))
536
746
  .filter(prop => fields.length === 0 || fields.some(f => prop.name === f || prop.name.startsWith(`${String(f)}.`)))
537
- .map(prop => ({ field: prop.name, strategy: prop.strategy }));
747
+ .map(prop => ({ field: `${prop.name}:ref`, strategy: prop.strategy }));
538
748
  return [...populate, ...toPopulate];
539
749
  }
540
- joinedProps(meta, populate) {
541
- return populate.filter(p => {
542
- const prop = meta.properties[p.field] || {};
543
- return (p.strategy || prop.strategy || this.config.get('loadStrategy')) === core_1.LoadStrategy.JOINED && prop.reference !== core_1.ReferenceType.SCALAR;
750
+ /**
751
+ * @internal
752
+ */
753
+ joinedProps(meta, populate, options) {
754
+ return populate.filter(hint => {
755
+ const [propName, ref] = hint.field.split(':', 2);
756
+ const prop = meta.properties[propName] || {};
757
+ if (ref) {
758
+ // keep only pivot ref joins here, as that only triggers actual join
759
+ return prop.kind === core_1.ReferenceKind.MANY_TO_MANY;
760
+ }
761
+ return (prop.strategy || options?.strategy || hint.strategy || this.config.get('loadStrategy')) === core_1.LoadStrategy.JOINED && ![core_1.ReferenceKind.SCALAR, core_1.ReferenceKind.EMBEDDED].includes(prop.kind);
544
762
  });
545
763
  }
546
764
  /**
547
765
  * @internal
548
766
  */
549
- mergeJoinedResult(rawResults, meta) {
550
- // group by the root entity primary key first
551
- const map = {};
767
+ mergeJoinedResult(rawResults, meta, joinedProps) {
552
768
  const res = [];
553
- rawResults.forEach(item => {
769
+ const map = {};
770
+ for (const item of rawResults) {
554
771
  const pk = core_1.Utils.getCompositeKeyHash(item, meta);
555
772
  if (map[pk]) {
556
- map[pk].push(item);
773
+ for (const hint of joinedProps) {
774
+ const [propName, ref] = hint.field.split(':', 2);
775
+ const prop = meta.properties[propName];
776
+ if (!item[propName]) {
777
+ continue;
778
+ }
779
+ if (prop.kind === core_1.ReferenceKind.MANY_TO_MANY && ref) {
780
+ map[pk][propName] = [...map[pk][propName], ...item[propName]];
781
+ continue;
782
+ }
783
+ switch (prop.kind) {
784
+ case core_1.ReferenceKind.ONE_TO_MANY:
785
+ case core_1.ReferenceKind.MANY_TO_MANY:
786
+ map[pk][propName] = this.mergeJoinedResult([...map[pk][propName], ...item[propName]], prop.targetMeta, hint.children ?? []);
787
+ break;
788
+ case core_1.ReferenceKind.MANY_TO_ONE:
789
+ case core_1.ReferenceKind.ONE_TO_ONE:
790
+ map[pk][propName] = this.mergeJoinedResult([map[pk][propName], item[propName]], prop.targetMeta, hint.children ?? [])[0];
791
+ break;
792
+ }
793
+ }
557
794
  }
558
795
  else {
559
- map[pk] = [item];
796
+ map[pk] = item;
560
797
  res.push(item);
561
798
  }
562
- });
799
+ }
563
800
  return res;
564
801
  }
565
- getFieldsForJoinedLoad(qb, meta, explicitFields, populate = [], parentTableAlias, parentJoinPath) {
802
+ getFieldsForJoinedLoad(qb, meta, explicitFields, populate = [], options, parentTableAlias, parentJoinPath) {
566
803
  const fields = [];
567
- const joinedProps = this.joinedProps(meta, populate);
804
+ const joinedProps = this.joinedProps(meta, populate, options);
568
805
  if (explicitFields?.includes('*')) {
569
806
  fields.push('*');
570
807
  }
@@ -575,26 +812,41 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
575
812
  if (!fields || prop.primary) {
576
813
  return !fields?.includes('*');
577
814
  }
578
- return fields.includes(prop.name);
815
+ return fields.some(f => f === prop.name || f.toString().startsWith(prop.name + '.'));
579
816
  };
580
- // alias all fields in the primary table
581
- meta.props
582
- .filter(prop => shouldHaveColumn(prop, populate, explicitFields))
583
- .forEach(prop => fields.push(...this.mapPropToFieldNames(qb, prop, parentTableAlias)));
584
- joinedProps.forEach(relation => {
585
- const prop = meta.properties[relation.field];
817
+ const populateWhereAll = options?._populateWhere === 'all' || core_1.Utils.isEmpty(options?._populateWhere);
818
+ // root entity is already handled, skip that
819
+ if (parentJoinPath) {
820
+ // alias all fields in the primary table
821
+ meta.props
822
+ .filter(prop => shouldHaveColumn(prop, populate, explicitFields))
823
+ .forEach(prop => fields.push(...this.mapPropToFieldNames(qb, prop, parentTableAlias)));
824
+ }
825
+ joinedProps.forEach(hint => {
826
+ const [propName, ref] = hint.field.split(':', 2);
827
+ const prop = meta.properties[propName];
586
828
  const meta2 = this.metadata.find(prop.type);
829
+ const pivotRefJoin = prop.kind === core_1.ReferenceKind.MANY_TO_MANY && ref;
587
830
  const tableAlias = qb.getNextAlias(prop.name);
588
831
  const field = parentTableAlias ? `${parentTableAlias}.${prop.name}` : prop.name;
589
- const path = parentJoinPath ? `${parentJoinPath}.${prop.name}` : `${meta.name}.${prop.name}`;
590
- qb.join(field, tableAlias, {}, 'leftJoin', path);
832
+ let path = parentJoinPath ? `${parentJoinPath}.${prop.name}` : `${meta.name}.${prop.name}`;
833
+ if (!parentJoinPath && populateWhereAll && !path.startsWith('[populate]')) {
834
+ path = '[populate]' + path;
835
+ }
836
+ const joinType = pivotRefJoin ? query_1.JoinType.pivotJoin : query_1.JoinType.leftJoin;
837
+ qb.join(field, tableAlias, {}, joinType, path);
838
+ if (pivotRefJoin) {
839
+ fields.push(...prop.joinColumns.map(col => qb.helper.mapper(`${tableAlias}.${col}`, qb.type, undefined, `${tableAlias}__${col}`)), ...prop.inverseJoinColumns.map(col => qb.helper.mapper(`${tableAlias}.${col}`, qb.type, undefined, `${tableAlias}__${col}`)));
840
+ }
591
841
  const childExplicitFields = explicitFields?.filter(f => core_1.Utils.isPlainObject(f)).map(o => o[prop.name])[0] || [];
592
842
  explicitFields?.forEach(f => {
593
843
  if (typeof f === 'string' && f.startsWith(`${prop.name}.`)) {
594
844
  childExplicitFields.push(f.substring(prop.name.length + 1));
595
845
  }
596
846
  });
597
- fields.push(...this.getFieldsForJoinedLoad(qb, meta2, childExplicitFields.length === 0 ? undefined : childExplicitFields, relation.children, tableAlias, path));
847
+ if (!ref) {
848
+ fields.push(...this.getFieldsForJoinedLoad(qb, meta2, childExplicitFields.length === 0 ? undefined : childExplicitFields, hint.children, options, tableAlias, path));
849
+ }
598
850
  });
599
851
  return fields;
600
852
  }
@@ -602,24 +854,35 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
602
854
  * @internal
603
855
  */
604
856
  mapPropToFieldNames(qb, prop, tableAlias) {
605
- const aliased = qb.ref(tableAlias ? `${tableAlias}__${prop.fieldNames[0]}` : prop.fieldNames[0]).toString();
606
- if (prop.customType?.convertToJSValueSQL && tableAlias) {
607
- const prefixed = qb.ref(prop.fieldNames[0]).withSchema(tableAlias).toString();
608
- return [qb.raw(prop.customType.convertToJSValueSQL(prefixed, this.platform) + ' as ' + aliased).toString()];
857
+ const knex = this.connection.getKnex();
858
+ const aliased = knex.ref(tableAlias ? `${tableAlias}__${prop.fieldNames[0]}` : prop.fieldNames[0]).toString();
859
+ if (tableAlias && prop.customTypes?.some(type => type.convertToJSValueSQL)) {
860
+ return prop.fieldNames.map((col, idx) => {
861
+ if (!prop.customTypes[idx].convertToJSValueSQL) {
862
+ return col;
863
+ }
864
+ const prefixed = knex.ref(col).withSchema(tableAlias).toString();
865
+ const aliased = knex.ref(`${tableAlias}__${col}`).toString();
866
+ return (0, core_1.raw)(`${prop.customTypes[idx].convertToJSValueSQL(prefixed, this.platform)} as ${aliased}`);
867
+ });
868
+ }
869
+ if (tableAlias && prop.customType?.convertToJSValueSQL) {
870
+ const prefixed = knex.ref(prop.fieldNames[0]).withSchema(tableAlias).toString();
871
+ return [(0, core_1.raw)(`${prop.customType.convertToJSValueSQL(prefixed, this.platform)} as ${aliased}`)];
609
872
  }
610
873
  if (prop.formula) {
611
- const alias = qb.ref(tableAlias ?? qb.alias).toString();
612
- return [`${prop.formula(alias)} as ${aliased}`];
874
+ const alias = knex.ref(tableAlias ?? qb.alias).toString();
875
+ return [(0, core_1.raw)(`${prop.formula(alias)} as ${aliased}`)];
613
876
  }
614
877
  if (tableAlias) {
615
- return prop.fieldNames.map(fieldName => qb.ref(fieldName).withSchema(tableAlias).as(`${tableAlias}__${fieldName}`));
878
+ return prop.fieldNames.map(fieldName => knex.ref(fieldName).withSchema(tableAlias).as(`${tableAlias}__${fieldName}`));
616
879
  }
617
880
  return prop.fieldNames;
618
881
  }
619
882
  /** @internal */
620
- createQueryBuilder(entityName, ctx, preferredConnectionType, convertCustomTypes) {
883
+ createQueryBuilder(entityName, ctx, preferredConnectionType, convertCustomTypes, logging) {
621
884
  const connectionType = this.resolveConnectionType({ ctx, connectionType: preferredConnectionType });
622
- const qb = new query_1.QueryBuilder(entityName, this.metadata, this, ctx, undefined, connectionType);
885
+ const qb = new query_1.QueryBuilder(entityName, this.metadata, this, ctx, undefined, connectionType, undefined, logging);
623
886
  if (!convertCustomTypes) {
624
887
  qb.unsetFlag(core_1.QueryFlag.CONVERT_CUSTOM_TYPES);
625
888
  }
@@ -643,7 +906,7 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
643
906
  }
644
907
  const ret = {};
645
908
  this.metadata.find(entityName).relations.forEach(prop => {
646
- if (prop.reference === core_1.ReferenceType.MANY_TO_MANY && data[prop.name]) {
909
+ if (prop.kind === core_1.ReferenceKind.MANY_TO_MANY && data[prop.name]) {
647
910
  ret[prop.name] = data[prop.name].map((item) => core_1.Utils.asArray(item));
648
911
  delete data[prop.name];
649
912
  }
@@ -656,77 +919,97 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
656
919
  }
657
920
  for (const prop of meta.relations) {
658
921
  if (collections[prop.name]) {
659
- await this.rethrow(this.updateCollectionDiff(meta, prop, pks, clear, collections[prop.name], options));
660
- }
661
- }
662
- }
663
- async updateCollectionDiff(meta, prop, pks, deleteDiff, insertDiff, options) {
664
- if (!deleteDiff) {
665
- deleteDiff = [];
666
- }
667
- if (deleteDiff === true || deleteDiff.length > 0) {
668
- const qb1 = this.createQueryBuilder(prop.pivotEntity, options?.ctx, 'write')
669
- .withSchema(this.getSchemaName(meta, options))
670
- .unsetFlag(core_1.QueryFlag.CONVERT_CUSTOM_TYPES);
671
- const knex = qb1.getKnex();
672
- if (Array.isArray(deleteDiff)) {
673
- knex.whereIn(prop.inverseJoinColumns, deleteDiff);
922
+ const pivotMeta = this.metadata.find(prop.pivotEntity);
923
+ const persister = new PivotCollectionPersister_1.PivotCollectionPersister(pivotMeta, this, options?.ctx, options?.schema);
924
+ persister.enqueueUpdate(prop, collections[prop.name], clear, pks);
925
+ await this.rethrow(persister.execute());
674
926
  }
675
- prop.joinColumns.forEach((joinColumn, idx) => knex.andWhere(joinColumn, pks[idx]));
676
- await this.execute(knex.delete());
677
- }
678
- if (insertDiff.length === 0) {
679
- return;
680
- }
681
- const items = insertDiff.map(item => {
682
- const cond = {};
683
- prop.joinColumns.forEach((joinColumn, idx) => cond[joinColumn] = pks[idx]);
684
- prop.inverseJoinColumns.forEach((inverseJoinColumn, idx) => cond[inverseJoinColumn] = item[idx]);
685
- return cond;
686
- });
687
- /* istanbul ignore else */
688
- if (this.platform.allowsMultiInsert()) {
689
- await this.nativeInsertMany(prop.pivotEntity, items, {
690
- ...options,
691
- convertCustomTypes: false,
692
- processCollections: false,
693
- });
694
- }
695
- else {
696
- await core_1.Utils.runSerial(items, item => {
697
- return this.createQueryBuilder(prop.pivotEntity, options?.ctx, 'write')
698
- .unsetFlag(core_1.QueryFlag.CONVERT_CUSTOM_TYPES)
699
- .withSchema(this.getSchemaName(meta, options))
700
- .insert(item)
701
- .execute('run', false);
702
- });
703
927
  }
704
928
  }
705
929
  async lockPessimistic(entity, options) {
706
930
  const meta = (0, core_1.helper)(entity).__meta;
707
- const qb = this.createQueryBuilder(entity.constructor.name, options.ctx).unsetFlag(core_1.QueryFlag.CONVERT_CUSTOM_TYPES).withSchema(options.schema ?? meta.schema);
931
+ const qb = this.createQueryBuilder(entity.constructor.name, options.ctx).withSchema(options.schema ?? meta.schema);
708
932
  const cond = core_1.Utils.getPrimaryKeyCond(entity, meta.primaryKeys);
709
- qb.select('1').where(cond).setLockMode(options.lockMode, options.lockTableAliases);
933
+ qb.select((0, core_1.raw)('1')).where(cond).setLockMode(options.lockMode, options.lockTableAliases);
710
934
  await this.rethrow(qb.execute());
711
935
  }
712
- buildJoinedPropsOrderBy(entityName, qb, meta, populate, parentPath) {
936
+ buildOrderBy(qb, meta, populate, options) {
937
+ const joinedProps = this.joinedProps(meta, populate, options);
938
+ // `options._populateWhere` is a copy of the value provided by user with a fallback to the global config option
939
+ // as `options.populateWhere` will be always recomputed to respect filters
940
+ const populateWhereAll = options._populateWhere !== 'infer' && !core_1.Utils.isEmpty(options._populateWhere);
941
+ const path = (populateWhereAll ? '[populate]' : '') + meta.className;
942
+ const populateOrderBy = this.buildPopulateOrderBy(qb, meta, core_1.Utils.asArray(options.populateOrderBy ?? options.orderBy), path);
943
+ const joinedPropsOrderBy = this.buildJoinedPropsOrderBy(qb, meta, joinedProps, options, path);
944
+ return [...core_1.Utils.asArray(options.orderBy), ...populateOrderBy, ...joinedPropsOrderBy];
945
+ }
946
+ buildPopulateOrderBy(qb, meta, populateOrderBy, parentPath, parentAlias = qb.alias) {
713
947
  const orderBy = [];
714
- const joinedProps = this.joinedProps(meta, populate);
715
- joinedProps.forEach(relation => {
716
- const prop = meta.properties[relation.field];
948
+ for (let i = 0; i < populateOrderBy.length; i++) {
949
+ const orderHint = populateOrderBy[i];
950
+ for (const propName of core_1.Utils.keys(orderHint)) {
951
+ const prop = meta.properties[propName];
952
+ const meta2 = this.metadata.find(prop.type);
953
+ const childOrder = orderHint[prop.name];
954
+ let path = `${parentPath}.${propName}`;
955
+ if (prop.kind === core_1.ReferenceKind.MANY_TO_MANY && typeof childOrder !== 'object') {
956
+ path += '[pivot]';
957
+ }
958
+ const join = qb.getJoinForPath(path, { matchPopulateJoins: true });
959
+ const propAlias = qb.getAliasForJoinPath(join ?? path, { matchPopulateJoins: true }) ?? parentAlias;
960
+ if (!join && parentAlias === qb.alias) {
961
+ continue;
962
+ }
963
+ if (![core_1.ReferenceKind.SCALAR, core_1.ReferenceKind.EMBEDDED].includes(prop.kind) && typeof childOrder === 'object') {
964
+ const children = this.buildPopulateOrderBy(qb, meta2, core_1.Utils.asArray(childOrder), path, propAlias);
965
+ orderBy.push(...children);
966
+ continue;
967
+ }
968
+ if (prop.kind === core_1.ReferenceKind.MANY_TO_MANY && join) {
969
+ if (prop.fixedOrderColumn) {
970
+ orderBy.push({ [`${join.alias}.${prop.fixedOrderColumn}`]: childOrder });
971
+ }
972
+ else {
973
+ for (const col of prop.inverseJoinColumns) {
974
+ orderBy.push({ [`${join.ownerAlias}.${col}`]: childOrder });
975
+ }
976
+ }
977
+ continue;
978
+ }
979
+ const order = typeof childOrder === 'object' ? childOrder[propName] : childOrder;
980
+ orderBy.push({ [`${propAlias}.${propName}`]: order });
981
+ }
982
+ }
983
+ return orderBy;
984
+ }
985
+ buildJoinedPropsOrderBy(qb, meta, populate, options, parentPath) {
986
+ const orderBy = [];
987
+ const joinedProps = this.joinedProps(meta, populate, options);
988
+ for (const hint of joinedProps) {
989
+ const [propName, ref] = hint.field.split(':', 2);
990
+ const prop = meta.properties[propName];
717
991
  const propOrderBy = prop.orderBy;
718
- const path = `${parentPath ? parentPath : entityName}.${relation.field}`;
719
- const propAlias = qb.getAliasForJoinPath(path);
992
+ let path = `${parentPath}.${propName}`;
993
+ if (prop.kind === core_1.ReferenceKind.MANY_TO_MANY && ref) {
994
+ path += '[pivot]';
995
+ }
996
+ const join = qb.getJoinForPath(path, { matchPopulateJoins: true });
997
+ const propAlias = qb.getAliasForJoinPath(join ?? path, { matchPopulateJoins: true });
998
+ const meta2 = this.metadata.find(prop.type);
999
+ if (prop.kind === core_1.ReferenceKind.MANY_TO_MANY && prop.fixedOrder && join) {
1000
+ const alias = ref ? propAlias : join.ownerAlias;
1001
+ orderBy.push({ [`${alias}.${prop.fixedOrderColumn}`]: core_1.QueryOrder.ASC });
1002
+ }
720
1003
  if (propOrderBy) {
721
- Object.keys(propOrderBy).forEach(field => {
1004
+ core_1.Utils.keys(propOrderBy).forEach(field => {
722
1005
  orderBy.push({ [`${propAlias}.${field}`]: propOrderBy[field] });
723
1006
  });
724
1007
  }
725
- if (relation.children) {
726
- const meta2 = this.metadata.find(prop.type);
727
- orderBy.push(...this.buildJoinedPropsOrderBy(prop.name, qb, meta2, relation.children, path));
1008
+ if (hint.children) {
1009
+ const buildJoinedPropsOrderBy = this.buildJoinedPropsOrderBy(qb, meta2, hint.children, options, path);
1010
+ orderBy.push(...buildJoinedPropsOrderBy);
728
1011
  }
729
- });
1012
+ }
730
1013
  return orderBy;
731
1014
  }
732
1015
  normalizeFields(fields, prefix = '') {
@@ -745,12 +1028,12 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
745
1028
  return ret;
746
1029
  }
747
1030
  processField(meta, prop, field, ret, populate, joinedProps, qb) {
748
- if (!prop || (prop.reference === core_1.ReferenceType.ONE_TO_ONE && !prop.owner)) {
1031
+ if (!prop || (prop.kind === core_1.ReferenceKind.ONE_TO_ONE && !prop.owner)) {
749
1032
  return;
750
1033
  }
751
- if (prop.reference === core_1.ReferenceType.EMBEDDED) {
1034
+ if (prop.kind === core_1.ReferenceKind.EMBEDDED) {
752
1035
  if (prop.object) {
753
- ret.push(prop.fieldNames[0]);
1036
+ ret.push(prop.name);
754
1037
  return;
755
1038
  }
756
1039
  const parts = field.split('.');
@@ -762,20 +1045,17 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
762
1045
  }
763
1046
  return;
764
1047
  }
765
- if (prop.fieldNames) {
766
- ret.push(...prop.fieldNames);
767
- }
1048
+ ret.push(prop.name);
768
1049
  }
769
- buildFields(meta, populate, joinedProps, qb, fields) {
1050
+ buildFields(meta, populate, joinedProps, qb, alias, fields, options) {
770
1051
  const lazyProps = meta.props.filter(prop => prop.lazy && !populate.some(p => p.field === prop.name || p.all));
771
1052
  const hasLazyFormulas = meta.props.some(p => p.lazy && p.formula);
772
- const requiresSQLConversion = meta.props.some(p => p.customType?.convertToJSValueSQL);
1053
+ const requiresSQLConversion = meta.props.some(p => p.customType?.convertToJSValueSQL && p.persist !== false);
773
1054
  const hasExplicitFields = !!fields;
774
1055
  const ret = [];
775
- if (joinedProps.length > 0) {
776
- ret.push(...this.getFieldsForJoinedLoad(qb, meta, fields, populate));
777
- }
778
- else if (fields) {
1056
+ let addFormulas = false;
1057
+ // handle root entity properties first, this is used for both strategies in the same way
1058
+ if (fields) {
779
1059
  for (const field of this.normalizeFields(fields)) {
780
1060
  if (field === '*') {
781
1061
  ret.push('*');
@@ -792,28 +1072,39 @@ class AbstractSqlDriver extends core_1.DatabaseDriver {
792
1072
  });
793
1073
  this.processField(meta, prop, parts.join('.'), ret, populate, joinedProps, qb);
794
1074
  }
795
- ret.unshift(...meta.primaryKeys.filter(pk => !fields.includes(pk)));
1075
+ if (!fields.includes('*') && !fields.includes(`${qb.alias}.*`)) {
1076
+ ret.unshift(...meta.primaryKeys.filter(pk => !fields.includes(pk)));
1077
+ }
796
1078
  }
797
1079
  else if (lazyProps.filter(p => !p.formula).length > 0) {
798
1080
  const props = meta.props.filter(prop => this.platform.shouldHaveColumn(prop, populate, false));
799
1081
  ret.push(...core_1.Utils.flatten(props.filter(p => !lazyProps.includes(p)).map(p => p.fieldNames)));
1082
+ addFormulas = true;
800
1083
  }
801
1084
  else if (hasLazyFormulas || requiresSQLConversion) {
802
1085
  ret.push('*');
1086
+ addFormulas = true;
803
1087
  }
804
- if (ret.length > 0 && !hasExplicitFields) {
1088
+ else {
1089
+ ret.push('*');
1090
+ }
1091
+ if (ret.length > 0 && !hasExplicitFields && addFormulas) {
805
1092
  meta.props
806
1093
  .filter(prop => prop.formula && !lazyProps.includes(prop))
807
1094
  .forEach(prop => {
808
- const alias = qb.ref(qb.alias).toString();
809
- const aliased = qb.ref(prop.fieldNames[0]).toString();
810
- ret.push(`${prop.formula(alias)} as ${aliased}`);
1095
+ const a = this.connection.getKnex().ref(alias).toString();
1096
+ const aliased = this.connection.getKnex().ref(prop.fieldNames[0]).toString();
1097
+ ret.push((0, core_1.raw)(`${prop.formula(a)} as ${aliased}`));
811
1098
  });
812
1099
  meta.props
813
- .filter(prop => prop.customType?.convertToDatabaseValueSQL || prop.customType?.convertToJSValueSQL)
1100
+ .filter(prop => prop.hasConvertToDatabaseValueSQL || prop.hasConvertToJSValueSQL)
814
1101
  .forEach(prop => ret.push(prop.name));
815
1102
  }
816
- return ret.length > 0 ? core_1.Utils.unique(ret) : ['*'];
1103
+ // add joined relations after the root entity fields
1104
+ if (joinedProps.length > 0) {
1105
+ ret.push(...this.getFieldsForJoinedLoad(qb, meta, fields, populate, options, alias));
1106
+ }
1107
+ return core_1.Utils.unique(ret);
817
1108
  }
818
1109
  }
819
1110
  exports.AbstractSqlDriver = AbstractSqlDriver;