actor4j.core.persistence.connectors.MongoDBPersistenceAdapter.java Source code

Introduction

Here is the source code for actor4j.core.persistence.connectors.MongoDBPersistenceAdapter.java

Source

/*
 * Copyright (c) 2015-2018, David A. Bauer. All rights reserved.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package actor4j.core.persistence.connectors;

import static actor4j.core.protocols.ActorProtocolTag.INTERNAL_PERSISTENCE_FAILURE;
import static actor4j.core.protocols.ActorProtocolTag.INTERNAL_PERSISTENCE_RECOVER;
import static actor4j.core.protocols.ActorProtocolTag.INTERNAL_PERSISTENCE_SUCCESS;

import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

import org.bson.Document;
import org.json.JSONArray;
import org.json.JSONObject;

import com.mongodb.client.FindIterable;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoCursor;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.InsertOneModel;
import com.mongodb.client.model.WriteModel;

import actor4j.core.ActorSystem;
import actor4j.core.messages.ActorMessage;
import static actor4j.core.persistence.actor.PersistenceServiceActor.*;

public class MongoDBPersistenceAdapter extends PersistenceAdapter {
    protected MongoDatabase database;
    protected MongoCollection<Document> events;
    protected MongoCollection<Document> states;

    public MongoDBPersistenceAdapter(ActorSystem parent, PersistenceConnector connector) {
        super(parent, connector);
    }

    @Override
    public void preStart(UUID id) {
        super.preStart(id);

        database = ((MongoDBPersistenceConnector) connector).client.getDatabase(connector.databaseName);
        events = database.getCollection("persistence.events");
        states = database.getCollection("persistence.states");
    }

    @Override
    public void receive(ActorMessage<?> message) {
        if (message.tag == PERSIST_EVENTS) {
            try {
                JSONArray array = new JSONArray(message.valueAsString());
                if (array.length() == 1) {
                    Document document = Document.parse(array.get(0).toString());
                    events.insertOne(document);
                } else {
                    List<WriteModel<Document>> requests = new ArrayList<WriteModel<Document>>();
                    for (Object obj : array) {
                        Document document = Document.parse(obj.toString());
                        requests.add(new InsertOneModel<Document>(document));
                    }
                    events.bulkWrite(requests);
                }
                parent.send(new ActorMessage<Object>(null, INTERNAL_PERSISTENCE_SUCCESS, self(), message.source));
            } catch (Exception e) {
                e.printStackTrace();
                parent.send(new ActorMessage<Exception>(e, INTERNAL_PERSISTENCE_FAILURE, self(), message.source));
            }
        } else if (message.tag == PERSIST_STATE) {
            try {
                Document document = Document.parse(message.valueAsString());
                states.insertOne(document);
                parent.send(new ActorMessage<Object>(null, INTERNAL_PERSISTENCE_SUCCESS, self(), message.source));
            } catch (Exception e) {
                e.printStackTrace();
                parent.send(new ActorMessage<Exception>(e, INTERNAL_PERSISTENCE_FAILURE, self(), message.source));
            }
        } else if (message.tag == RECOVER) {
            try {
                JSONObject obj = new JSONObject();
                Document document = null;

                FindIterable<Document> statesIterable = states
                        .find(new Document("persistenceId", message.valueAsString()))
                        .sort(new Document("timeStamp", -1)).limit(1);
                document = statesIterable.first();
                if (document != null) {
                    JSONObject stateValue = new JSONObject(document.toJson());
                    stateValue.remove("_id");
                    long timeStamp = stateValue.getJSONObject("timeStamp").getLong("$numberLong");
                    stateValue.put("timeStamp", timeStamp);
                    obj.put("state", stateValue);

                    FindIterable<Document> eventsIterable = events
                            .find(new Document("persistenceId", message.valueAsString()).append("timeStamp",
                                    new Document("$gte", timeStamp)))
                            .sort(new Document("timeStamp", -1));
                    JSONArray array = new JSONArray();
                    MongoCursor<Document> cursor = eventsIterable.iterator();
                    while (cursor.hasNext()) {
                        document = cursor.next();
                        JSONObject eventValue = new JSONObject(document.toJson());
                        eventValue.remove("_id");
                        timeStamp = eventValue.getJSONObject("timeStamp").getLong("$numberLong");
                        eventValue.put("timeStamp", timeStamp);
                        array.put(eventValue);
                    }
                    cursor.close();
                    obj.put("events", array);
                } else
                    obj.put("state", new JSONObject());

                parent.send(new ActorMessage<String>(obj.toString(), INTERNAL_PERSISTENCE_RECOVER, self(),
                        message.source));
            } catch (Exception e) {
                e.printStackTrace();
                JSONObject obj = new JSONObject();
                obj.put("error", e.getMessage());
                parent.send(new ActorMessage<String>(obj.toString(), INTERNAL_PERSISTENCE_RECOVER, self(),
                        message.source));
            }
        }
    }
}