Триггеры в Neo4j

Одной из часто упускаемых функций в Neo4j являются возможности «TransactionEventHandler»… более известные в мире баз данных как « Триггеры» . Когда происходит транзакция, мы можем проанализировать это событие и принять решение о некоторых действиях. Для этого мы будем напишите «расширение ядра» (немного отличающееся от неуправляемых расширений, которые мы видели в этом блоге), чтобы связать наш триггер.

Представьте, что вы — специальный агент Эйвери Райан, работающий в команде ФБР по расследованию киберпреступлений, занимающейся расследованием убийств в Интернете, кражи, взлома, сексуальных преступлений, шантажа и любых других преступлений, которые считаются кибер-преступлениями в пределах юрисдикции ФБР.

Ваши бывшие коллеги-хакеры Равен Рамирез и Броди Нельсон создали социальную сеть известных подозреваемых и их знакомых. Но бывший старший специальный агент морской пехоты США Элайджа Мундо не может потрудиться выучить Сайфера , все, что он хочет — это электронное письмо, когда будет выявлен новый подозреваемый или выяснится новая связь с известным подозреваемым. Итак, вы назначаете хакерского ученого Дэниела Крумица, чтобы добавить эту функцию в вашу систему.

Даниэль просматривает документацию TransactionEventHandler и узнает, что есть 3 хука: beforeCommit, afterCommit и afterRollback, которые можно использовать:

beforeCommit(TransactionData data)
Invoked when a transaction is about to be committed.

afterCommit(TransactionData data, T state)
Invoked after the transaction has been committed successfully.

afterRollback(TransactionData data, T state)
Invoked after the transaction has been rolled back if committing the transaction failed for some reason.

public class MyTransactionEventHandler implements TransactionEventHandler {

    public static GraphDatabaseService db;
    private static ExecutorService ex;

    public MyTransactionEventHandler(GraphDatabaseService graphDatabaseService, ExecutorService executor) {
        db = graphDatabaseService;
        ex = executor;
    }

    @Override
    public Object beforeCommit(TransactionData transactionData) throws Exception {
        return null;
    }

    @Override
    public void afterCommit(TransactionData transactionData, Object o) {
        ex.submit(new SuspectRunnable(transactionData, db));
    }

    @Override
    public void afterRollback(TransactionData transactionData, Object o) {

    }
}

@Override
public void run() {
    try (Transaction tx = db.beginTx()) {
        Set<Node> suspects = new HashSet<>();
        for (Node node : td.createdNodes()) {
            if (node.hasLabel(Labels.Suspect)) {
                suspects.add(node);
                System.out.println("A new Suspect has been created!");
            }
        }

for (LabelEntry labelEntry : td.assignedLabels()) {
    if (labelEntry.label().equals(Labels.Suspect) && !suspects.contains(labelEntry.node())) {
        System.out.println("A new Suspect has been identified!");
        suspects.add(labelEntry.node());
    }
}

for (Relationship relationship : td.createdRelationships()) {
    if (relationship.isType(RelationshipTypes.KNOWS)) {
        for (Node user : relationship.getNodes()) {
            if (user.hasLabel(Labels.Suspect)) {
                System.out.println("A new direct relationship to a Suspect has been created!");
            }

            for (Relationship knows : user.getRelationships(Direction.BOTH, 
                                                            RelationshipTypes.KNOWS)) {
                Node otherUser = knows.getOtherNode(user);
                if (otherUser.hasLabel(Labels.Suspect) &&
                   !otherUser.equals(relationship.getOtherNode(user))) {
                    System.out.println("A new indirect relationship to a Suspect has been created!");
                }
            }
        }
    }
}

@Override
public Lifecycle newKernelExtension(final Dependencies dependencies) throws Throwable {
    return new LifecycleAdapter() {

        private MyTransactionEventHandler handler;
        private ExecutorService executor;

        @Override
        public void start() throws Throwable {
            executor = Executors.newFixedThreadPool(2);
            handler = new MyTransactionEventHandler(dependencies.getGraphDatabaseService(), executor);
            dependencies.getGraphDatabaseService().registerTransactionEventHandler(handler);
        }

        @Override
        public void shutdown() throws Throwable {
            executor.shutdown();
            dependencies.getGraphDatabaseService().unregisterTransactionEventHandler(handler);
        }
    };
}

mvn clean package

CREATE (max:User {name:"Max"}) RETURN max;

CREATE (al:Suspect {name:"Al Capone"}) RETURN al;

MATCH (max:User),(al:Suspect)
WHERE max.name = "Max" AND al.name = "Al Capone"
CREATE (max)-[r:KNOWS]->(al)
RETURN r;

CREATE (monica:User {name:"Monica"}) RETURN monica;

MATCH (max:User),(monica:User)
WHERE max.name = "Max" AND monica.name = "Monica"
CREATE (max)-[r:KNOWS]->(monica)
RETURN r;

MATCH (monica:User)
WHERE monica.name = "Monica"
SET monica :Suspect
RETURN monica;