FlowPathConverter.java

Go to the documentation of this file.

package org.openkilda.integration.converter;

import java.util.List;
import java.util.Map;

import org.openkilda.integration.model.response.FlowPathNode;
import org.openkilda.integration.model.response.FlowPayload;
import org.openkilda.integration.service.SwitchIntegrationService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

@Component
public class FlowPathConverter {

    @Autowired
    SwitchIntegrationService switchIntegrationService;

    public FlowPayload getFlowPath(final String flowId, final FlowPayload flowPayload) {
        Map<String, String> csNames = switchIntegrationService.getCustomSwitchNameFromFile();
        setSwitchName(flowPayload.getForward(), csNames);
        setSwitchName(flowPayload.getReverse(), csNames);
        return flowPayload;
    }
    
    private void setSwitchName(List<FlowPathNode> pathNodes, Map<String, String> csNames) {
        pathNodes.parallelStream().forEach((pathNode) -> {
            pathNode.setSwitchName(
                    switchIntegrationService.customSwitchName(csNames, pathNode.getSwitchId()));
        });
    }

}