OperationsCollector.java

package com.appmetr.hercules.operations;

import com.appmetr.hercules.manager.WideEntityManager;
import com.appmetr.hercules.profile.DataOperationsProfile;
import com.google.inject.Inject;

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

public class OperationsCollector<T extends ExecutableOperation> {
    @Inject WideEntityManager entityManager;

    public enum Type {
        GET, SAVE, DELETE
    }

    private List<T> operations = new ArrayList<T>();

    public void add(T operation) {
        operations.add(operation);
    }

    public void add(OperationsCollector<T> collector) {
        operations.addAll(collector.getOperations());
    }

    public OperationsResult execute() {
        return execute(null);
    }

    public OperationsResult execute(DataOperationsProfile dataOperationsProfile) {
        return entityManager.executeOperations(this, dataOperationsProfile);
    }

    public List<T> getOperations() {
        return operations;
    }

    public Type getOperationType(ExecutableOperation operation) {
        if (operation instanceof GetExecutableOperation) {
            return Type.GET;
        } else if (operation instanceof SaveExecutableOperation) {
            return Type.SAVE;
        } else if (operation instanceof DeleteExecutableOperation) {
            return Type.DELETE;
        } else {
            throw new RuntimeException("Unknown operation type");
        }
    }
}