1313
1414const { dataSource } = require ( '../DataSource.js' ) ;
1515const { QueryBuilder } = require ( '../utilities/QueryBuilder.js' ) ;
16+ const { OldQueryBuilder } = require ( '../utilities/OldQueryBuilder.js' ) ;
1617
1718/**
1819 * Sequelize implementation of the Repository.
@@ -39,8 +40,8 @@ class Repository {
3940 findQuery . selectOnly ( ) ;
4041 }
4142
42- if ( findQuery instanceof QueryBuilder ) {
43- findQuery = findQuery . toImplementation ( ) ;
43+ if ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder ) {
44+ findQuery = findQuery . toImplementation ( this . model ) ;
4445 }
4546
4647 return this . model . count ( findQuery ) ;
@@ -53,7 +54,9 @@ class Repository {
5354 * @returns {Promise<array> } Promise object representing the full mock data
5455 */
5556 async findAll ( findQuery = { } ) {
56- return this . model . findAll ( findQuery instanceof QueryBuilder ? findQuery . toImplementation ( ) : findQuery ) ;
57+ return this . model . findAll ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder
58+ ? findQuery . toImplementation ( this . model )
59+ : findQuery ) ;
5760 }
5861
5962 /**
@@ -67,9 +70,9 @@ class Repository {
6770 findQuery = dataSource . createQueryBuilder ( ) ;
6871 }
6972
70- if ( findQuery instanceof QueryBuilder ) {
73+ if ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder ) {
7174 findQuery . set ( 'distinct' , true ) ;
72- findQuery = findQuery . toImplementation ( ) ;
75+ findQuery = findQuery . toImplementation ( this . model ) ;
7376 }
7477
7578 return this . model . findAndCountAll ( findQuery ) ;
@@ -82,8 +85,8 @@ class Repository {
8285 * @returns {Promise<Object|null> } Promise object representing the full mock data
8386 */
8487 async findOne ( findQuery = { } ) {
85- if ( findQuery instanceof QueryBuilder ) {
86- findQuery = findQuery . toImplementation ( ) ;
88+ if ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder ) {
89+ findQuery = findQuery . toImplementation ( this . model ) ;
8790 }
8891 findQuery . limit = 1 ;
8992 return this . model . findOne ( findQuery ) ;
@@ -132,8 +135,8 @@ class Repository {
132135 * @returns {Promise<*[]> } promise with list of removed entities
133136 */
134137 async removeAll ( findQuery ) {
135- if ( findQuery instanceof QueryBuilder ) {
136- findQuery = findQuery . toImplementation ( ) ;
138+ if ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder ) {
139+ findQuery = findQuery . toImplementation ( this . model ) ;
137140 }
138141
139142 const entities = await this . findAll ( findQuery ) ;
@@ -151,8 +154,8 @@ class Repository {
151154 * @returns {Promise|Null } Promise object representing the full mock data
152155 */
153156 async removeOne ( findQuery ) {
154- if ( findQuery instanceof QueryBuilder ) {
155- findQuery = findQuery . toImplementation ( ) ;
157+ if ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder ) {
158+ findQuery = findQuery . toImplementation ( this . model ) ;
156159 }
157160 findQuery . limit = 1 ;
158161
@@ -183,8 +186,8 @@ class Repository {
183186 * @return {Promise<void> } promise that resolves when the patch has been applied
184187 */
185188 async updateAll ( patch , findQuery ) {
186- if ( findQuery instanceof QueryBuilder ) {
187- findQuery = findQuery . toImplementation ( ) ;
189+ if ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder ) {
190+ findQuery = findQuery . toImplementation ( this . model ) ;
188191 }
189192
190193 return this . model . update ( patch , findQuery ) ;
0 commit comments