Skip to content

Commit 4f8866a

Browse files
soerenreichardtDarthMax
authored andcommitted
Apply usage of RelationshipType to new features after rebase
1 parent c101dd9 commit 4f8866a

File tree

4 files changed

+14
-14
lines changed

4 files changed

+14
-14
lines changed

core/src/main/java/org/neo4j/graphalgo/config/DeleteRelationshipsConfig.java

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
*/
2020
package org.neo4j.graphalgo.config;
2121

22+
import org.neo4j.graphalgo.RelationshipType;
2223
import org.neo4j.graphalgo.annotation.Configuration;
2324
import org.neo4j.graphalgo.annotation.ValueClass;
2425
import org.neo4j.graphalgo.core.loading.GraphStore;
@@ -49,7 +50,7 @@ static DeleteRelationshipsConfig of(
4950
@Configuration.Ignore
5051
default void validate(GraphStore graphStore) {
5152

52-
Set<String> relationshipTypes = graphStore.relationshipTypes();
53+
Set<RelationshipType> relationshipTypes = graphStore.relationshipTypes();
5354

5455
if (relationshipTypes.size() == 1) {
5556
throw new IllegalArgumentException(String.format(
@@ -60,7 +61,7 @@ default void validate(GraphStore graphStore) {
6061
));
6162
}
6263

63-
if (!relationshipTypes.contains(relationshipType())) {
64+
if (!relationshipTypes.contains(RelationshipType.of(relationshipType()))) {
6465
throw new IllegalArgumentException(String.format(
6566
"No relationship type '%s' found in graph '%s'.",
6667
relationshipType(),

core/src/main/java/org/neo4j/graphalgo/core/loading/GraphStore.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@
2323
import org.immutables.builder.Builder.AccessibleFields;
2424
import org.neo4j.graphalgo.ElementIdentifier;
2525
import org.neo4j.graphalgo.NodeLabel;
26-
import org.neo4j.graphalgo.annotation.ValueClass;
2726
import org.neo4j.graphalgo.RelationshipType;
27+
import org.neo4j.graphalgo.annotation.ValueClass;
2828
import org.neo4j.graphalgo.api.Graph;
2929
import org.neo4j.graphalgo.api.IdMapGraph;
3030
import org.neo4j.graphalgo.api.NodeProperties;
@@ -341,7 +341,7 @@ private void addRelationshipProperty(
341341
});
342342
}
343343

344-
public DeletionResult deleteRelationships(String relationshipType) {
344+
public DeletionResult deleteRelationships(RelationshipType relationshipType) {
345345
return DeletionResult.of(builder ->
346346
updateGraphStore(graphStore -> {
347347
builder.deletedRelationships(graphStore.relationships.get(relationshipType).elementCount());

core/src/test/java/org/neo4j/graphalgo/core/loading/GraphStoreTest.java

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
package org.neo4j.graphalgo.core.loading;
2121

2222
import org.jetbrains.annotations.NotNull;
23-
import org.hamcrest.Matcher;
2423
import org.junit.jupiter.api.AfterEach;
2524
import org.junit.jupiter.api.BeforeEach;
2625
import org.junit.jupiter.api.Test;
@@ -47,27 +46,25 @@
4746
import java.time.LocalDateTime;
4847
import java.util.Arrays;
4948
import java.util.Collections;
50-
import java.util.List;
5149
import java.util.HashSet;
52-
import java.util.Map;
50+
import java.util.List;
5351
import java.util.Optional;
5452
import java.util.stream.Stream;
5553

5654
import static java.util.Collections.singletonList;
5755
import static org.hamcrest.MatcherAssert.assertThat;
5856
import static org.hamcrest.Matchers.equalTo;
59-
import static org.hamcrest.Matchers.greaterThan;
6057
import static org.junit.jupiter.api.Assertions.assertEquals;
6158
import static org.junit.jupiter.api.Assertions.assertFalse;
6259
import static org.junit.jupiter.api.Assertions.assertTrue;
6360
import static org.neo4j.graphalgo.NodeLabel.ALL_NODES;
6461
import static org.neo4j.graphalgo.QueryRunner.runQuery;
6562
import static org.neo4j.graphalgo.TestGraph.Builder.fromGdl;
6663
import static org.neo4j.graphalgo.TestSupport.assertGraphEquals;
67-
import static org.neo4j.graphalgo.config.AlgoBaseConfig.ALL_RELATIONSHIP_TYPE_IDENTIFIERS;
68-
import static org.neo4j.values.storable.NumberType.FLOATING_POINT;
6964
import static org.neo4j.graphalgo.TestSupport.mapEquals;
7065
import static org.neo4j.graphalgo.compat.MapUtil.map;
66+
import static org.neo4j.graphalgo.config.AlgoBaseConfig.ALL_RELATIONSHIP_TYPE_IDENTIFIERS;
67+
import static org.neo4j.values.storable.NumberType.FLOATING_POINT;
7168

7269
class GraphStoreTest {
7370

@@ -246,10 +243,10 @@ void deleteRelationshipsAndProperties() throws InterruptedException {
246243
// relationshipTypes containing properties from other relationshipTypes
247244
assertThat(graphStore.relationshipPropertyCount(), equalTo(8L));
248245

249-
DeletionResult deletionResult = graphStore.deleteRelationships("LER");
246+
DeletionResult deletionResult = graphStore.deleteRelationships(RelationshipType.of("LER"));
250247

251-
assertEquals(new HashSet<>(singletonList("REL")), graphStore.relationshipTypes());
252-
assertFalse(graphStore.hasRelationshipType("LER"));
248+
assertEquals(new HashSet<>(singletonList(RelationshipType.of("REL"))), graphStore.relationshipTypes());
249+
assertFalse(graphStore.hasRelationshipType(RelationshipType.of("LER")));
253250
assertEquals(1, graphStore.relationshipCount());
254251
// should expect 1 instead of two, but currently properties are global across relationship types
255252
assertEquals(2, graphStore.relationshipPropertyCount());

proc/catalog/src/main/java/org/neo4j/graphalgo/catalog/GraphDeleteRelationshipProc.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,9 @@ public Stream<Result> delete(
5353

5454
DeleteRelationshipsConfig.of(graphName, relationshipType).validate(graphStoreWithConfig.graphStore());
5555

56-
DeletionResult deletionResult = graphStoreWithConfig.graphStore().deleteRelationships(relationshipType);
56+
DeletionResult deletionResult = graphStoreWithConfig
57+
.graphStore()
58+
.deleteRelationships(RelationshipType.of(relationshipType));
5759

5860
return Stream.of(new Result(
5961
graphName,

0 commit comments

Comments
 (0)