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,7 @@ 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 ? findQuery . toImplementation ( this . model ) : findQuery ) ;
5758 }
5859
5960 /**
@@ -67,9 +68,9 @@ class Repository {
6768 findQuery = dataSource . createQueryBuilder ( ) ;
6869 }
6970
70- if ( findQuery instanceof QueryBuilder ) {
71+ if ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder ) {
7172 findQuery . set ( 'distinct' , true ) ;
72- findQuery = findQuery . toImplementation ( ) ;
73+ findQuery = findQuery . toImplementation ( this . model ) ;
7374 }
7475
7576 return this . model . findAndCountAll ( findQuery ) ;
@@ -82,8 +83,8 @@ class Repository {
8283 * @returns {Promise<Object|null> } Promise object representing the full mock data
8384 */
8485 async findOne ( findQuery = { } ) {
85- if ( findQuery instanceof QueryBuilder ) {
86- findQuery = findQuery . toImplementation ( ) ;
86+ if ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder ) {
87+ findQuery = findQuery . toImplementation ( this . model ) ;
8788 }
8889 findQuery . limit = 1 ;
8990 return this . model . findOne ( findQuery ) ;
@@ -132,8 +133,8 @@ class Repository {
132133 * @returns {Promise<*[]> } promise with list of removed entities
133134 */
134135 async removeAll ( findQuery ) {
135- if ( findQuery instanceof QueryBuilder ) {
136- findQuery = findQuery . toImplementation ( ) ;
136+ if ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder ) {
137+ findQuery = findQuery . toImplementation ( this . model ) ;
137138 }
138139
139140 const entities = await this . findAll ( findQuery ) ;
@@ -151,8 +152,8 @@ class Repository {
151152 * @returns {Promise|Null } Promise object representing the full mock data
152153 */
153154 async removeOne ( findQuery ) {
154- if ( findQuery instanceof QueryBuilder ) {
155- findQuery = findQuery . toImplementation ( ) ;
155+ if ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder ) {
156+ findQuery = findQuery . toImplementation ( this . model ) ;
156157 }
157158 findQuery . limit = 1 ;
158159
@@ -183,8 +184,8 @@ class Repository {
183184 * @return {Promise<void> } promise that resolves when the patch has been applied
184185 */
185186 async updateAll ( patch , findQuery ) {
186- if ( findQuery instanceof QueryBuilder ) {
187- findQuery = findQuery . toImplementation ( ) ;
187+ if ( findQuery instanceof QueryBuilder || findQuery instanceof OldQueryBuilder ) {
188+ findQuery = findQuery . toImplementation ( this . model ) ;
188189 }
189190
190191 return this . model . update ( patch , findQuery ) ;
0 commit comments