2424import java .util .Optional ;
2525
2626import org .springframework .data .falkordb .core .FalkorDBTemplate ;
27+ import org .springframework .data .falkordb .core .mapping .FalkorDBMappingContext ;
2728import org .springframework .data .falkordb .core .mapping .FalkorDBPersistentEntity ;
2829import org .springframework .data .falkordb .core .mapping .FalkorDBPersistentProperty ;
2930import org .springframework .data .falkordb .repository .query .FalkorDBQueryMethod ;
30- import org .springframework .data .mapping .context .MappingContext ;
3131import org .springframework .data .projection .ProjectionFactory ;
3232import org .springframework .data .repository .core .NamedQueries ;
3333import org .springframework .data .repository .core .RepositoryInformation ;
3434import org .springframework .data .repository .core .RepositoryMetadata ;
3535import org .springframework .data .repository .core .support .RepositoryFactorySupport ;
3636import org .springframework .data .repository .query .QueryLookupStrategy ;
37- import org .springframework .data .repository .query .QueryMethodEvaluationContextProvider ;
3837import org .springframework .data .repository .query .RepositoryQuery ;
3938import org .springframework .lang .Nullable ;
4039import org .springframework .util .Assert ;
@@ -52,7 +51,7 @@ public class FalkorDBRepositoryFactory extends RepositoryFactorySupport {
5251
5352 private final FalkorDBTemplate falkorDBTemplate ;
5453
55- private final MappingContext <? extends FalkorDBPersistentEntity <?>, FalkorDBPersistentProperty > mappingContext ;
54+ private final FalkorDBMappingContext mappingContext ;
5655
5756 /**
5857 * Creates a new {@link FalkorDBRepositoryFactory} with the given
@@ -64,7 +63,7 @@ public FalkorDBRepositoryFactory(FalkorDBTemplate falkorDBTemplate) {
6463 Assert .notNull (falkorDBTemplate , "FalkorDBTemplate must not be null" );
6564
6665 this .falkorDBTemplate = falkorDBTemplate ;
67- this .mappingContext = falkorDBTemplate .getConverter (). getMappingContext ();
66+ this .mappingContext = falkorDBTemplate .getMappingContext ();
6867 }
6968
7069 @ Override
@@ -87,9 +86,7 @@ protected Class<?> getRepositoryBaseClass(RepositoryMetadata metadata) {
8786 return SimpleFalkorDBRepository .class ;
8887 }
8988
90- @ Override
91- protected Optional <QueryLookupStrategy > getQueryLookupStrategy (@ Nullable QueryLookupStrategy .Key key ,
92- QueryMethodEvaluationContextProvider evaluationContextProvider ) {
89+ protected Optional <QueryLookupStrategy > getQueryLookupStrategy (@ Nullable QueryLookupStrategy .Key key ) {
9390
9491 return Optional .of (new FalkorDBQueryLookupStrategy ());
9592 }
@@ -104,7 +101,7 @@ public RepositoryQuery resolveQuery(Method method, RepositoryMetadata metadata,
104101 NamedQueries namedQueries ) {
105102
106103 FalkorDBQueryMethod queryMethod = new FalkorDBQueryMethod (method , metadata , factory ,
107- FalkorDBRepositoryFactory . this . mappingContext );
104+ mappingContext );
108105
109106 // TODO: Implement query resolution logic
110107 // For now, return a simple implementation that throws an exception
@@ -155,6 +152,21 @@ public Class<T> getJavaType() {
155152 return entity .getType ();
156153 }
157154
155+ @ Override
156+ public String getPrimaryLabel () {
157+ return entity .getPrimaryLabel ();
158+ }
159+
160+ @ Override
161+ public String [] getLabels () {
162+ return entity .getLabels ();
163+ }
164+
165+ @ Override
166+ public boolean isNew (T t ) {
167+ return getId (t ) == null ;
168+ }
169+
158170 }
159171
160172}
0 commit comments