ru.taskurotta.service.hz.serialization.bson.DecisionRowBSerializer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of services-hz Show documentation
The newest version!
package ru.taskurotta.service.hz.serialization.bson;

import ru.taskurotta.mongodb.driver.BDataInput;
import ru.taskurotta.mongodb.driver.BDataOutput;
import ru.taskurotta.mongodb.driver.CString;
import ru.taskurotta.mongodb.driver.StreamBSerializer;
import ru.taskurotta.service.dependency.links.Modification;
import ru.taskurotta.service.hz.dependency.HzGraphDao;

import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.UUID;

public class DecisionRowBSerializer implements StreamBSerializer {

    public static final CString TASK_ID = new CString("t");
    public static final CString PROCESS_ID = new CString("p");
    public static final CString WAIT_FOR_AFTER_RELEASE = new CString("wait");
    public static final CString NEW_ITEMS = new CString("new");
    public static final CString READY_ITEMS = new CString("ready");

    @Override
    public Class getObjectClass() {
        return HzGraphDao.DecisionRow.class;
    }

    @Override
    public void write(BDataOutput out, HzGraphDao.DecisionRow decisionRow) {
        int writeIdLabel = out.writeObject(_ID);
        out.writeUUID(TASK_ID, decisionRow.getTaskId());
        out.writeUUID(PROCESS_ID, decisionRow.getProcessId());
        out.writeObjectStop(writeIdLabel);

        Modification modification = decisionRow.getModification();
        out.writeUUID(WAIT_FOR_AFTER_RELEASE, modification.getWaitForAfterRelease());

        Set newItems = modification.getNewItems();
        if (newItems != null) {
            int newItemsLabel = out.writeArray(NEW_ITEMS);
            int i = 0;
            for (UUID item : newItems) {
                out.writeUUID(i, item);
                i++;
            }
            out.writeArrayStop(newItemsLabel);
        }

        Map> links = modification.getLinks();
        GraphBSerializer.serializeLinks(out, links);

        UUID[] readyItems = decisionRow.getReadyItems();
        if (readyItems != null) {
            int readyItemsLabel = out.writeArray(READY_ITEMS);
            for (int i = 0; i < readyItems.length; i++) {
                out.writeUUID(i, readyItems[i]);
            }
            out.writeArrayStop(readyItemsLabel);
        }
    }

    @Override
    public HzGraphDao.DecisionRow read(BDataInput in) {
        int readIdLabel = in.readObject(_ID);
        UUID taskId = in.readUUID(TASK_ID);
        UUID processId = in.readUUID(PROCESS_ID);
        in.readObjectStop(readIdLabel);

        UUID waitForAfterRelease = in.readUUID(WAIT_FOR_AFTER_RELEASE);
        Set newItems = null;

        int newItemsLabel = in.readArray(NEW_ITEMS);
        if (newItemsLabel != -1) {
            int newItemsSize = in.readArraySize();
            newItems = new HashSet<>(newItemsSize);
            for (int i = 0; i < newItemsSize; i++) {
                newItems.add(in.readUUID(i));
            }
            in.readArrayStop(newItemsLabel);
        }

        Map> links = GraphBSerializer.deserializeLinks(in);

        int readyItemsLabel = in.readArray(READY_ITEMS);
        UUID[] readyItems = null;
        if (readyItemsLabel != -1) {
            int readyItemsSize = in.readArraySize();
            readyItems = new UUID[readyItemsSize];
            for (int i = 0; i < readyItemsSize; i++) {
                readyItems[i] = in.readUUID(i);
            }
        }

        Modification modification = new Modification(taskId, waitForAfterRelease, links, newItems);

        return new HzGraphDao.DecisionRow(taskId, processId, modification, readyItems);
    }
}