1919import java .lang .reflect .Method ;
2020
2121import org .socialsignin .spring .data .dynamodb .core .DynamoDBOperations ;
22+ import org .springframework .data .projection .ProjectionFactory ;
2223import org .springframework .data .repository .core .NamedQueries ;
2324import org .springframework .data .repository .core .RepositoryMetadata ;
2425import org .springframework .data .repository .query .QueryLookupStrategy ;
@@ -62,13 +63,13 @@ public AbstractQueryLookupStrategy(DynamoDBOperations dynamoDBOperations) {
6263 * org.springframework.data.repository.core.NamedQueries)
6364 */
6465 @ Override
65- public final RepositoryQuery resolveQuery (Method method , RepositoryMetadata metadata , NamedQueries namedQueries ) {
66+ public final RepositoryQuery resolveQuery (Method method , RepositoryMetadata metadata , ProjectionFactory factory , NamedQueries namedQueries ) {
6667
67- return createDynamoDBQuery (method , metadata , metadata .getDomainType (), metadata .getIdType (), namedQueries );
68+ return createDynamoDBQuery (method , metadata , factory , metadata .getDomainType (), metadata .getIdType (), namedQueries );
6869 }
6970
7071 protected abstract <T , ID extends Serializable > RepositoryQuery createDynamoDBQuery (Method method ,
71- RepositoryMetadata metadata , Class <T > entityClass , Class <ID > idClass , NamedQueries namedQueries );
72+ RepositoryMetadata metadata , ProjectionFactory factory , Class <T > entityClass , Class <ID > idClass , NamedQueries namedQueries );
7273 }
7374
7475 /**
@@ -84,10 +85,10 @@ public CreateQueryLookupStrategy(DynamoDBOperations dynamoDBOperations) {
8485 }
8586
8687 @ Override
87- protected <T , ID extends Serializable > RepositoryQuery createDynamoDBQuery (Method method , RepositoryMetadata metadata ,
88+ protected <T , ID extends Serializable > RepositoryQuery createDynamoDBQuery (Method method , RepositoryMetadata metadata , ProjectionFactory factory ,
8889 Class <T > entityClass , Class <ID > idClass , NamedQueries namedQueries ) {
8990 try {
90- return new PartTreeDynamoDBQuery <T , ID >(dynamoDBOperations , new DynamoDBQueryMethod <T , ID >(method , metadata ));
91+ return new PartTreeDynamoDBQuery <T , ID >(dynamoDBOperations , new DynamoDBQueryMethod <T , ID >(method , metadata , factory ));
9192 } catch (IllegalArgumentException e ) {
9293 throw new IllegalArgumentException (String .format ("Could not create query metamodel for method %s!" ,
9394 method .toString ()), e );
@@ -110,7 +111,7 @@ public DeclaredQueryLookupStrategy(DynamoDBOperations dynamoDBOperations) {
110111 }
111112
112113 @ Override
113- protected <T , ID extends Serializable > RepositoryQuery createDynamoDBQuery (Method method , RepositoryMetadata metadata ,
114+ protected <T , ID extends Serializable > RepositoryQuery createDynamoDBQuery (Method method , RepositoryMetadata metadata , ProjectionFactory factory ,
114115 Class <T > entityClass , Class <ID > idClass , NamedQueries namedQueries ) {
115116 throw new UnsupportedOperationException ("Declared Queries not supported at this time" );
116117 }
@@ -137,14 +138,14 @@ public CreateIfNotFoundQueryLookupStrategy(DynamoDBOperations dynamoDBOperations
137138 }
138139
139140 @ Override
140- protected <T , ID extends Serializable > RepositoryQuery createDynamoDBQuery (Method method , RepositoryMetadata metadata ,
141+ protected <T , ID extends Serializable > RepositoryQuery createDynamoDBQuery (Method method , RepositoryMetadata metadata , ProjectionFactory factory ,
141142 Class <T > entityClass , Class <ID > idClass , NamedQueries namedQueries ) {
142143 try {
143- return strategy .createDynamoDBQuery (method , metadata , entityClass , idClass , namedQueries );
144+ return strategy .createDynamoDBQuery (method , metadata , factory , entityClass , idClass , namedQueries );
144145 } catch (IllegalStateException e ) {
145- return createStrategy .createDynamoDBQuery (method , metadata , entityClass , idClass , namedQueries );
146+ return createStrategy .createDynamoDBQuery (method , metadata , factory , entityClass , idClass , namedQueries );
146147 } catch (UnsupportedOperationException e ) {
147- return createStrategy .createDynamoDBQuery (method , metadata , entityClass , idClass , namedQueries );
148+ return createStrategy .createDynamoDBQuery (method , metadata , factory , entityClass , idClass , namedQueries );
148149 }
149150
150151 }
0 commit comments