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 apoc.get.Get;
import apoc.result.*;
import apoc.util.Util;
import org.neo4j.graphdb.*;
import org.neo4j.helpers.collection.Iterables;
import org.neo4j.procedure.*;
import java.util.*;
import java.util.stream.LongStream;
import java.util.stream.Stream;
import static org.neo4j.graphdb.RelationshipType.withName;
public class Create {
public static final String[] EMPTY_ARRAY = new String[0];
@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(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(db).nodes(nodes).map((r) -> {
setProperty(r.node, key, toPropertyValue(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(db).rels(rels).map((r) -> {
setProperty(r.rel, key, toPropertyValue(value));
return r;
});
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.setProperties( [node,id,ids,nodes], [keys], [values]) - sets the given properties on the nodes(s)")
public Stream setProperties(@Name("nodes") Object nodes, @Name("keys") List keys, @Name("values") List values) {
return new Get(db).nodes(nodes).map((r) -> {
setProperties(r.node, Util.mapFromLists(keys, values));
return r;
});
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.removeProperties( [node,id,ids,nodes], [keys]) - removes the given properties from the nodes(s)")
public Stream removeProperties(@Name("nodes") Object nodes, @Name("keys") List keys) {
return new Get(db).nodes(nodes).map((r) -> {
keys.forEach(r.node::removeProperty);
return r;
});
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.setRelProperties( [rel,id,ids,rels], [keys], [values]) - sets the given properties on the relationship(s)")
public Stream setRelProperties(@Name("rels") Object rels, @Name("keys") List keys, @Name("values") List values) {
return new Get(db).rels(rels).map((r) -> {
setProperties(r.rel, Util.mapFromLists(keys, values));
return r;
});
}
@Procedure(mode = Mode.WRITE)
@Description("apoc.create.removeRelProperties( [rel,id,ids,rels], [keys]) - removes the given properties from the relationship(s)")
public Stream removeRelProperties(@Name("rels") Object rels, @Name("keys") List keys) {
return new Get(db).rels(rels).map((r) -> {
keys.forEach(r.rel::removeProperty);
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(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(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, 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) {
return Stream.of(new NodeResult(vNodeFunction(labelNames, props)));
}
@UserFunction("apoc.create.vNode")
@Description("apoc.create.vNode(['Label'], {key:value,...}) returns a virtual node")
public Node vNodeFunction(@Name("label") List labelNames, @Name(value = "props", defaultValue = "{}") Map props) {
return new VirtualNode(Util.labels(labelNames), props, db);
}
@UserFunction("apoc.create.virtual.fromNode")
@Description("apoc.create.virtual.fromNode(node, [propertyNames]) returns a virtual node built from an existing node with only the requested properties")
public Node virtualFromNodeFunction(@Name("node") Node node, @Name("propertyNames") List propertyNames) {
return new VirtualNode(node, propertyNames);
}
@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) {
return Stream.of(new RelationshipResult(vRelationshipFunction(from, relType, props, to)));
}
@UserFunction("apoc.create.vRelationship")
@Description("apoc.create.vRelationship(nodeFrom,'KNOWS',{key:value,...}, nodeTo) returns a virtual relationship")
public Relationship vRelationshipFunction(@Name("from") Node from, @Name("relType") String relType, @Name("props") Map props, @Name("to") Node to) {
return new VirtualRelationship(from, to, withName(relType)).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 = 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, 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 = 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()) {
setProperty(pc, entry.getKey(), entry.getValue());
}
return pc;
}
private void setProperty(T pc, String key, Object value) {
if (value == null) pc.removeProperty(key);
else pc.setProperty(key, toPropertyValue(value));
}
@UserFunction
@Description("apoc.create.uuid() - creates an UUID")
public String uuid() {
return UUID.randomUUID().toString();
}
private Object toPropertyValue(Object value) {
if (value instanceof Iterable) {
Iterable it = (Iterable) value;
Object first = Iterables.firstOrNull(it);
if (first == null) return EMPTY_ARRAY;
return Iterables.asArray(first.getClass(), it);
}
return value;
}
@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());
}
}
}
