Skip to content

Commit 873f604

Browse files
committed
feat: update custom test
Signed-off-by: Otavio Santana <[email protected]>
1 parent a46b76c commit 873f604

File tree

3 files changed

+6
-22
lines changed

3 files changed

+6
-22
lines changed

jnosql-tinkerpop/src/main/java/org/eclipse/jnosql/databases/tinkerpop/mapping/query/TinkerpopRepositoryBean.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ public class TinkerpopRepositoryBean<T, K> extends AbstractBean<TinkerPopReposit
5656
private final Set<Annotation> qualifiers = Collections.singleton(new AnnotationLiteral<Default>() {
5757
});
5858

59-
TinkerpopRepositoryBean(Class<T> type) {
59+
public TinkerpopRepositoryBean(Class<T> type) {
6060
this.type = type;
6161
this.types = Collections.singleton(type);
6262
}

jnosql-tinkerpop/src/main/java/org/eclipse/jnosql/databases/tinkerpop/mapping/spi/TinkerpopExtension.java

Lines changed: 3 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
import jakarta.enterprise.inject.spi.Extension;
2020
import jakarta.enterprise.inject.spi.ProcessProducer;
2121
import org.apache.tinkerpop.gremlin.structure.Graph;
22+
import org.eclipse.jnosql.databases.tinkerpop.mapping.TinkerPopRepository;
2223
import org.eclipse.jnosql.databases.tinkerpop.mapping.query.TinkerpopRepositoryBean;
2324
import org.eclipse.jnosql.mapping.DatabaseMetadata;
2425
import org.eclipse.jnosql.mapping.Databases;
@@ -47,13 +48,9 @@ <T, X extends Graph> void observes(@Observes final ProcessProducer<T, X> pp) {
4748
void onAfterBeanDiscovery(@Observes final AfterBeanDiscovery afterBeanDiscovery) {
4849

4950
ClassScanner scanner = ClassScanner.load();
50-
Set<Class<?>> crudTypes = scanner.repositoriesStandard();
51+
Set<Class<?>> crudTypes = scanner.repositories(TinkerPopRepository.class);
5152

5253

53-
Set<Class<?>> customRepositories = scanner.customRepositories();
54-
55-
LOGGER.info(String.format("Processing graph extension: %d databases crud %d found, custom repositories: %d",
56-
databases.size(), crudTypes.size(), customRepositories.size()));
5754

5855
LOGGER.info("Processing repositories as a Graph implementation: " + crudTypes);
5956
databases.forEach(type -> {
@@ -63,21 +60,7 @@ void onAfterBeanDiscovery(@Observes final AfterBeanDiscovery afterBeanDiscovery)
6360
}
6461
});
6562

63+
crudTypes.forEach(type -> afterBeanDiscovery.addBean(new TinkerpopRepositoryBean<>(type)));
6664

67-
crudTypes.forEach(type -> {
68-
if (!databases.contains(DatabaseMetadata.DEFAULT_GRAPH)) {
69-
afterBeanDiscovery.addBean(new TinkerpopRepositoryBean<>(type, ""));
70-
}
71-
databases.forEach(database -> afterBeanDiscovery
72-
.addBean(new TinkerpopRepositoryBean<>(type, database.getProvider())));
73-
});
74-
75-
customRepositories.forEach(type -> {
76-
if (!databases.contains(DatabaseMetadata.DEFAULT_DOCUMENT)) {
77-
afterBeanDiscovery.addBean(new CustomRepositoryGraphBean<>(type, ""));
78-
}
79-
databases.forEach(database ->
80-
afterBeanDiscovery.addBean(new CustomRepositoryGraphBean<>(type, database.getProvider())));
81-
});
8265
}
8366
}

jnosql-tinkerpop/src/test/java/org/eclipse/jnosql/databases/tinkerpop/mapping/spi/GraphCustomExtensionTest.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
import org.eclipse.jnosql.mapping.Database;
2424
import org.eclipse.jnosql.mapping.DatabaseType;
2525
import org.eclipse.jnosql.mapping.core.Converters;
26+
import org.eclipse.jnosql.mapping.graph.spi.GraphExtension;
2627
import org.eclipse.jnosql.mapping.reflection.Reflections;
2728
import org.eclipse.jnosql.mapping.reflection.spi.ReflectionEntityMetadataExtension;
2829
import org.eclipse.jnosql.mapping.semistructured.EntityConverter;
@@ -38,7 +39,7 @@
3839
@AddPackages(value = {Converters.class, EntityConverter.class, TinkerpopTemplate.class})
3940
@AddPackages(GraphProducer.class)
4041
@AddPackages(Reflections.class)
41-
@AddExtensions({ReflectionEntityMetadataExtension.class, TinkerpopExtension.class})
42+
@AddExtensions({ReflectionEntityMetadataExtension.class, TinkerpopExtension.class, GraphExtension.class})
4243
class GraphCustomExtensionTest {
4344

4445
@Inject

0 commit comments

Comments
 (0)