ru.taskurotta.service.hz.serialization.TaskOptionsContainerSerializer 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;

import com.hazelcast.nio.ObjectDataInput;
import com.hazelcast.nio.ObjectDataOutput;
import com.hazelcast.nio.serialization.StreamSerializer;
import ru.taskurotta.transport.model.TaskConfigContainer;
import ru.taskurotta.transport.model.ArgContainer;
import ru.taskurotta.internal.core.ArgType;
import ru.taskurotta.transport.model.TaskOptionsContainer;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import static ru.taskurotta.service.hz.serialization.SerializationTools.readArgsContainerArray;
import static ru.taskurotta.service.hz.serialization.SerializationTools.writeArgsContainerArray;

/**
 * User: greg
 */
public class TaskOptionsContainerSerializer implements StreamSerializer {

    private TaskConfigContainerStreamSerializer taskConfigContainerStreamSerializer = new TaskConfigContainerStreamSerializer();


    @Override
    public void write(ObjectDataOutput out, TaskOptionsContainer object) throws IOException {
        if (object.getTaskConfigContainer() != null) {
            out.writeBoolean(true);
            taskConfigContainerStreamSerializer.write(out, object.getTaskConfigContainer());
        } else {
            out.writeBoolean(false);
        }
        int argTypesCount = (object.getArgTypes() != null) ? object.getArgTypes().length : -1;
        if (argTypesCount > 0) {
            out.writeInt(argTypesCount);
            for (ArgType i : object.getArgTypes()) {
                if (i == null) {
                    out.writeInt(-1);
                } else {
                    out.writeInt(i.getValue());
                }
            }
        } else {
            out.writeInt(-1);
        }

        writeArgsContainerArray(out, object.getPromisesWaitFor());
    }

    @Override
    public TaskOptionsContainer read(ObjectDataInput in) throws IOException {
        TaskConfigContainer taskConfigContainer = null;
        if (in.readBoolean()) {
            taskConfigContainer = taskConfigContainerStreamSerializer.read(in);
        }
        int argTypesCount = in.readInt();
        List argTypeList = new ArrayList<>();
        ArgType[] argTypeArray = null;
        if (argTypesCount != -1) {
            for (int i = 0; i < argTypesCount; i++) {
                int rd = in.readInt();
                if (rd == -1) {
                    argTypeList.add(null);
                } else {
                    argTypeList.add(ArgType.fromInt(rd));
                }
            }
            argTypeArray = new ArgType[argTypeList.size()];
            argTypeList.toArray(argTypeArray);
        }

        ArgContainer[] argContainersArray = readArgsContainerArray(in);

        return new TaskOptionsContainer(argTypeArray, taskConfigContainer, argContainersArray);
    }

    @Override
    public int getTypeId() {
        return ObjectTypes.TASK_OPTIONS_CONTAINER;
    }

    @Override
    public void destroy() {

    }
}