Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
apoc.create.Create Maven / Gradle / Ivy
package apoc.create;
import org.neo4j.procedure.*;
import apoc.get.Get;
import apoc.result.*;
import apoc.util.Util;
import org.neo4j.graphdb.*;
import org.neo4j.kernel.internal.GraphDatabaseAPI;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import java.util.stream.LongStream;
import java.util.stream.Stream;
public class Create {
@Context
public GraphDatabaseService db;
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.node(['Label'], {key:value,...}) - create node with dynamic labels")
public Stream node(@Name("label") List labelNames, @Name("props") Map props) {
return Stream.of(new NodeResult(setProperties(db.createNode(Util.labels(labelNames)),props)));
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.addLabels( [node,id,ids,nodes], ['Label',...]) - adds the given labels to the node or nodes")
public Stream addLabels(@Name("nodes") Object nodes, @Name("label") List labelNames) {
Label[] labels = Util.labels(labelNames);
return new Get((GraphDatabaseAPI) db).nodes(nodes).map((r) -> {
Node node = r.node;
for (Label label : labels) {
node.addLabel(label);
}
return r;
});
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.setProperty( [node,id,ids,nodes], key, value) - sets the given property on the node(s)")
public Stream setProperty(@Name("nodes") Object nodes, @Name("key") String key, @Name("value") Object value) {
return new Get((GraphDatabaseAPI) db).nodes(nodes).map((r) -> {
r.node.setProperty(key,value);
return r;
});
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.setRelProperty( [rel,id,ids,rels], key, value) - sets the given property on the relationship(s)")
public Stream setRelProperty(@Name("relationships") Object rels, @Name("key") String key, @Name("value") Object value) {
return new Get((GraphDatabaseAPI) db).rels(rels).map((r) -> {
r.rel.setProperty(key,value);
return r;
});
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.setProperties( [node,id,ids,nodes], [keys], [values]) - sets the given property on the nodes(s)")
public Stream setProperties(@Name("nodes") Object nodes, @Name("keys") List keys, @Name("values") List values) {
return new Get((GraphDatabaseAPI) db).nodes(nodes).map((r) -> {
setProperties(r.node, Util.mapFromLists(keys,values));
return r;
});
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.setRelProperties( [rel,id,ids,rels], [keys], [values]) - sets the given property on the relationship(s)")
public Stream setRelProperties(@Name("rels") Object rels, @Name("keys") List keys, @Name("values") List values) {
return new Get((GraphDatabaseAPI) db).rels(rels).map((r) -> {
setProperties(r.rel, Util.mapFromLists(keys,values));
return r;
});
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.setLabels( [node,id,ids,nodes], ['Label',...]) - sets the given labels, non matching labels are removed on the node or nodes")
public Stream setLabels(@Name("nodes") Object nodes, @Name("label") List labelNames) {
Label[] labels = Util.labels(labelNames);
return new Get((GraphDatabaseAPI) db).nodes(nodes).map((r) -> {
Node node = r.node;
for (Label label : node.getLabels()) {
if (labelNames.contains(label.name())) continue;
node.removeLabel(label);
}
for (Label label : labels) {
if (node.hasLabel(label)) continue;
node.addLabel(label);
}
return r;
});
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.removeLabels( [node,id,ids,nodes], ['Label',...]) - removes the given labels from the node or nodes")
public Stream removeLabels(@Name("nodes") Object nodes, @Name("label") List labelNames) {
Label[] labels = Util.labels(labelNames);
return new Get((GraphDatabaseAPI) db).nodes(nodes).map((r) -> {
Node node = r.node;
for (Label label : labels) {
node.removeLabel(label);
}
return r;
});
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.nodes(['Label'], [{key:value,...}]) create multiple nodes with dynamic labels")
public Stream nodes(@Name("label") List labelNames, @Name("props") List> props) {
Label[] labels = Util.labels(labelNames);
return props.stream().map(p -> new NodeResult(setProperties(db.createNode(labels), p)));
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.relationship(person1,'KNOWS',{key:value,...}, person2) create relationship with dynamic rel-type")
public Stream relationship(@Name("from") Node from,
@Name("relType") String relType, @Name("props") Map props,
@Name("to") Node to) {
return Stream.of(new RelationshipResult(setProperties(from.createRelationshipTo(to,RelationshipType.withName(relType)),props)));
}
@Procedure
@Description("apoc.create.vNode(['Label'], {key:value,...}) returns a virtual node")
public Stream vNode(@Name("label") List labelNames, @Name("props") Map props) {
Label[] labels = Util.labels(labelNames);
return Stream.of(new NodeResult(new VirtualNode(labels, props, db)));
}
@Procedure
@Description("apoc.create.vNodes(['Label'], [{key:value,...}]) returns virtual nodes")
public Stream vNodes(@Name("label") List labelNames, @Name("props") List> props) {
Label[] labels = Util.labels(labelNames);
return props.stream().map(p -> new NodeResult(new VirtualNode(labels, p, db)));
}
@Procedure
@Description("apoc.create.vRelationship(nodeFrom,'KNOWS',{key:value,...}, nodeTo) returns a virtual relationship")
public Stream vRelationship(@Name("from") Node from, @Name("relType") String relType, @Name("props") Map props, @Name("to") Node to) {
RelationshipType type = RelationshipType.withName(relType);
return Stream.of(new RelationshipResult(new VirtualRelationship(from,to,type).withProperties(props)));
}
@Procedure
@Description("apoc.create.vPattern({_labels:['LabelA'],key:value},'KNOWS',{key:value,...}, {_labels:['LabelB'],key:value}) returns a virtual pattern")
public Stream vPattern(@Name("from") Map n,
@Name("relType") String relType, @Name("props") Map props,
@Name("to") Map m) {
n = new LinkedHashMap<>(n); m=new LinkedHashMap<>(m);
RelationshipType type = RelationshipType.withName(relType);
VirtualNode from = new VirtualNode(Util.labels(n.remove("_labels")), n, db);
VirtualNode to = new VirtualNode(Util.labels(m.remove("_labels")), m, db);
Relationship rel = new VirtualRelationship(from, to, RelationshipType.withName(relType)).withProperties(props);
return Stream.of(new VirtualPathResult(from, rel, to));
}
@Procedure
@Description("apoc.create.vPatternFull(['LabelA'],{key:value},'KNOWS',{key:value,...},['LabelB'],{key:value}) returns a virtual pattern")
public Stream vPatternFull(@Name("labelsN") List labelsN, @Name("n") Map n,
@Name("relType") String relType, @Name("props") Map props,
@Name("labelsM") List labelsM, @Name("m") Map m) {
RelationshipType type = RelationshipType.withName(relType);
VirtualNode from = new VirtualNode(Util.labels(labelsN), n, db);
VirtualNode to = new VirtualNode(Util.labels(labelsM), m, db);
Relationship rel = new VirtualRelationship(from, to, type).withProperties(props);
return Stream.of(new VirtualPathResult(from,rel,to));
}
private T setProperties(T pc, Map p) {
if (p == null) return pc;
for (Map.Entry entry : p.entrySet()) pc.setProperty(entry.getKey(), entry.getValue());
return pc;
}
@UserFunction
@Description("apoc.create.uuid() - creates an UUID")
public String uuid() {
return UUID.randomUUID().toString();
}
@Procedure
@Description("apoc.create.uuids(count) yield uuid - creates 'count' UUIDs ")
public Stream uuids(@Name("count") long count) {
return LongStream.range(0,count).mapToObj(UUIDResult::new);
}
public static class UUIDResult {
public final long row;
public final String uuid;
public UUIDResult(long row) {
this.row = row;
this.uuid = UUID.randomUUID().toString();
// TODO Long.toHexString(uuid.getMostSignificantBits())+Long.toHexString(uuid.getLeastSignificantBits());
}
}
}
