OccurrenceConcrete.java

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package br.ufrgs.inf.prosoft.memoizeit;

import br.ufrgs.inf.prosoft.memoizeit.graph.Node;
import br.ufrgs.inf.prosoft.memoizeit.utils.Action;
import br.ufrgs.inf.prosoft.memoizeit.utils.ObjectUtils;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

/**
 *
 * @author romulo
 */
public class OccurrenceConcrete extends Occurrence {

    private final Object returnValue;
    private final List<Parameter> parameters;
    private boolean truncated;

    public OccurrenceConcrete(String instance, Object returnValue, List<Parameter> parameters, long startTime, long endTime) {
        super(instance, startTime, endTime);
        this.returnValue = returnValue;
        this.parameters = parameters;
        this.truncated = true;
    }

    @Override
    public Object getReturnValue() {
        return this.returnValue;
    }

    @Override
    public List<Parameter> getParameters() {
        return this.parameters;
    }

    private void removeUnusedFields(Map<String, Object> map, String parameterType, Node<String> methodNode) {
        Iterator<Map.Entry<String, Object>> iterator = map.entrySet().iterator();
        while (iterator.hasNext()) {
            Map.Entry<String, Object> entry = iterator.next();
            String getter = parameterType + ".get" + entry.getKey().substring(0, 1).toUpperCase() + entry.getKey().substring(1);
            if (!methodNode.getLinks().stream().anyMatch(node -> node.getContent().equals(getter))) {
                iterator.remove();
            }
        }

    }

    @Override
    protected Occurrence removeUnusedFields(Node<String> methodNode) {
        for (Parameter parameter : getParameters()) {
            String parameterType = parameter.getType();
            if (parameter.getData() instanceof Collection) {
                Collection collection = (Collection) parameter.getData();
                if (collection.stream().findAny().get() instanceof Map) {
                    Collection<Map<String, Object>> cast = collection;
                    cast.forEach(map -> removeUnusedFields(map, parameterType, methodNode));
                }
            } else if (parameter.getData() instanceof Map) {
                Map<String, Object> map = (Map<String, Object>) parameter.getData();
                removeUnusedFields(map, parameterType, methodNode);
            }
        }
        return this;
    }

    @Override
    protected OccurrenceConcrete getView(int depth) {
        if (!this.truncated) {
            return this;
        }
        this.truncated = false;
        Object returnValue = ObjectUtils.deepCopy(this.returnValue);
        Action onTruncate = () -> {
            this.truncated = true;
        };
        ObjectUtils.truncateObject(returnValue, depth, onTruncate);
        List<Parameter> parameters = new ArrayList<>();
        for (Parameter parameter : this.parameters) {
            Object parameterView = ObjectUtils.deepCopy(parameter.getData());
            ObjectUtils.truncateObject(parameterView, depth, onTruncate);
            parameters.add(new Parameter(parameter.getType(), parameterView));
        }
        OccurrenceConcrete occurrenceConcrete = new OccurrenceConcrete(getInstance(), returnValue, parameters, getStartTime(), getEndTime());
        return occurrenceConcrete;
    }

}