use of org.openkilda.messaging.command.yflow.SubFlowSharedEndpointEncapsulation in project open-kilda by telstra.
the class YFlowValidatorTest method failIfSubFlowHasNoId.
@Test(expected = InvalidFlowException.class)
public void failIfSubFlowHasNoId() throws InvalidFlowException, UnavailableFlowEndpointException {
YFlowRequest request = YFlowRequest.builder().yFlowId("test").sharedEndpoint(FlowEndpoint.builder().switchId(SWITCH_ID_1).portNumber(PORT_1).build()).subFlows(Arrays.asList(SubFlowDto.builder().sharedEndpoint(new SubFlowSharedEndpointEncapsulation(1, 0)).endpoint(FlowEndpoint.builder().switchId(SWITCH_ID_2).portNumber(PORT_2).build()).build(), SubFlowDto.builder().flowId("test_2").sharedEndpoint(new SubFlowSharedEndpointEncapsulation(2, 0)).endpoint(FlowEndpoint.builder().switchId(SWITCH_ID_3).portNumber(PORT_3).build()).build())).build();
yFlowValidator.validate(request);
}
use of org.openkilda.messaging.command.yflow.SubFlowSharedEndpointEncapsulation in project open-kilda by telstra.
the class YFlowValidatorTest method failIfOneSwitchFlowRequested.
@Test(expected = InvalidFlowException.class)
public void failIfOneSwitchFlowRequested() throws InvalidFlowException, UnavailableFlowEndpointException {
YFlowRequest request = YFlowRequest.builder().yFlowId("test").sharedEndpoint(FlowEndpoint.builder().switchId(SWITCH_ID_1).portNumber(PORT_1).build()).subFlows(Arrays.asList(SubFlowDto.builder().flowId("test_1").sharedEndpoint(new SubFlowSharedEndpointEncapsulation(1, 0)).endpoint(FlowEndpoint.builder().switchId(SWITCH_ID_1).portNumber(PORT_2).build()).build(), SubFlowDto.builder().flowId("test_2").sharedEndpoint(new SubFlowSharedEndpointEncapsulation(2, 0)).endpoint(FlowEndpoint.builder().switchId(SWITCH_ID_3).portNumber(PORT_3).build()).build())).build();
yFlowValidator.validate(request);
}
use of org.openkilda.messaging.command.yflow.SubFlowSharedEndpointEncapsulation in project open-kilda by telstra.
the class YFlowUpdateService method handlePartialUpdateRequest.
/**
* Handles request for y-flow patch updating.
*
* @param key command identifier.
* @param request request data.
*/
public void handlePartialUpdateRequest(@NonNull String key, @NonNull CommandContext commandContext, @NonNull YFlowPartialUpdateRequest request) throws DuplicateKeyException {
YFlowRequest target;
if (request.getYFlowId() != null) {
YFlow yFlow = yFlowRepository.findById(request.getYFlowId()).orElse(null);
target = YFlowRequestMapper.INSTANCE.toYFlowRequest(yFlow);
} else {
throw new FlowProcessingException(ErrorType.REQUEST_INVALID, "Need to specify the y-flow id");
}
if (target == null) {
throw new FlowProcessingException(ErrorType.NOT_FOUND, format("Y-flow was not found by the specified y-flow id: %s", request.getYFlowId()));
}
if (request.getSharedEndpoint() != null) {
if (target.getSharedEndpoint() == null) {
target.setSharedEndpoint(new FlowEndpoint(request.getSharedEndpoint().getSwitchId(), request.getSharedEndpoint().getPortNumber()));
} else {
SwitchId switchId = Optional.ofNullable(request.getSharedEndpoint().getSwitchId()).orElse(target.getSharedEndpoint().getSwitchId());
int portNumber = Optional.ofNullable(request.getSharedEndpoint().getPortNumber()).orElse(target.getSharedEndpoint().getPortNumber());
target.setSharedEndpoint(new FlowEndpoint(switchId, portNumber));
}
}
Optional.ofNullable(request.getMaximumBandwidth()).ifPresent(target::setMaximumBandwidth);
Optional.ofNullable(request.getPathComputationStrategy()).ifPresent(target::setPathComputationStrategy);
Optional.ofNullable(request.getEncapsulationType()).ifPresent(target::setEncapsulationType);
Optional.ofNullable(request.getMaxLatency()).ifPresent(target::setMaxLatency);
Optional.ofNullable(request.getMaxLatencyTier2()).ifPresent(target::setMaxLatencyTier2);
Optional.ofNullable(request.getIgnoreBandwidth()).ifPresent(target::setIgnoreBandwidth);
Optional.ofNullable(request.getPeriodicPings()).ifPresent(target::setPeriodicPings);
Optional.ofNullable(request.getPinned()).ifPresent(target::setPinned);
Optional.ofNullable(request.getPriority()).ifPresent(target::setPriority);
Optional.ofNullable(request.getStrictBandwidth()).ifPresent(target::setStrictBandwidth);
Optional.ofNullable(request.getDescription()).ifPresent(target::setDescription);
Optional.ofNullable(request.getAllocateProtectedPath()).ifPresent(target::setAllocateProtectedPath);
Optional.ofNullable(request.getDiverseFlowId()).ifPresent(target::setDiverseFlowId);
if (request.getSubFlows() != null && !request.getSubFlows().isEmpty()) {
Map<String, SubFlowDto> stringSubFlowDtoMap;
if (target.getSubFlows() != null) {
stringSubFlowDtoMap = target.getSubFlows().stream().collect(Collectors.toMap(SubFlowDto::getFlowId, Function.identity()));
} else {
throw new FlowProcessingException(ErrorType.INTERNAL_ERROR, format("Sub-flows for y-flow %s not found", target.getYFlowId()));
}
List<SubFlowDto> subFlows = new ArrayList<>();
for (SubFlowPartialUpdateDto subFlowPartialUpdate : request.getSubFlows()) {
SubFlowDto subFlow = stringSubFlowDtoMap.get(subFlowPartialUpdate.getFlowId());
if (subFlow != null) {
if (subFlowPartialUpdate.getEndpoint() != null) {
if (subFlow.getEndpoint() == null) {
subFlow.setEndpoint(new FlowEndpoint(subFlowPartialUpdate.getEndpoint().getSwitchId(), subFlowPartialUpdate.getEndpoint().getPortNumber(), subFlowPartialUpdate.getEndpoint().getVlanId(), subFlowPartialUpdate.getEndpoint().getInnerVlanId()));
} else {
SwitchId switchId = Optional.ofNullable(subFlowPartialUpdate.getEndpoint().getSwitchId()).orElse(subFlow.getEndpoint().getSwitchId());
int portNumber = Optional.ofNullable(subFlowPartialUpdate.getEndpoint().getPortNumber()).orElse(subFlow.getEndpoint().getPortNumber());
int vlanId = Optional.ofNullable(subFlowPartialUpdate.getEndpoint().getVlanId()).orElse(subFlow.getEndpoint().getOuterVlanId());
int innerVlanId = Optional.ofNullable(subFlowPartialUpdate.getEndpoint().getInnerVlanId()).orElse(subFlow.getEndpoint().getInnerVlanId());
subFlow.setEndpoint(new FlowEndpoint(switchId, portNumber, vlanId, innerVlanId));
}
}
if (subFlowPartialUpdate.getSharedEndpoint() != null) {
if (subFlow.getSharedEndpoint() == null) {
subFlow.setSharedEndpoint(new SubFlowSharedEndpointEncapsulation(subFlowPartialUpdate.getSharedEndpoint().getVlanId(), subFlowPartialUpdate.getSharedEndpoint().getInnerVlanId()));
} else {
int vlanId = Optional.ofNullable(subFlowPartialUpdate.getSharedEndpoint().getVlanId()).orElse(subFlow.getSharedEndpoint().getVlanId());
int innerVlanId = Optional.ofNullable(subFlowPartialUpdate.getSharedEndpoint().getInnerVlanId()).orElse(subFlow.getSharedEndpoint().getInnerVlanId());
subFlow.setSharedEndpoint(new SubFlowSharedEndpointEncapsulation(vlanId, innerVlanId));
}
}
Optional.ofNullable(subFlowPartialUpdate.getDescription()).ifPresent(subFlow::setDescription);
subFlows.add(subFlow);
} else {
throw new FlowProcessingException(ErrorType.REQUEST_INVALID, format("There is no sub-flows with sub-flow id: %s", subFlowPartialUpdate.getFlowId()));
}
}
target.setSubFlows(subFlows);
}
target.setType(Type.UPDATE);
handleRequest(key, commandContext, target);
}
use of org.openkilda.messaging.command.yflow.SubFlowSharedEndpointEncapsulation in project open-kilda by telstra.
the class OnSubFlowAllocatedAction method performWithResponse.
@Override
protected Optional<Message> performWithResponse(State from, State to, Event event, YFlowCreateContext context, YFlowCreateFsm stateMachine) {
String subFlowId = context.getSubFlowId();
if (!stateMachine.isCreatingSubFlow(subFlowId)) {
throw new IllegalStateException("Received an event for non-pending sub-flow " + subFlowId);
}
String yFlowId = stateMachine.getYFlowId();
stateMachine.saveActionToHistory("Creating a sub-flow", format("Allocated resources for sub-flow %s of y-flow %s", subFlowId, yFlowId));
stateMachine.addAllocatedSubFlow(subFlowId);
SubFlowDto subFlowDto = stateMachine.getTargetFlow().getSubFlows().stream().filter(f -> f.getFlowId().equals(subFlowId)).findAny().orElseThrow(() -> new FlowProcessingException(ErrorType.INTERNAL_ERROR, format("Can't find definition of created sub-flow %s", subFlowId)));
SubFlowSharedEndpointEncapsulation sharedEndpoint = subFlowDto.getSharedEndpoint();
FlowEndpoint endpoint = subFlowDto.getEndpoint();
log.debug("Start creating sub-flow references from {} to y-flow {}", subFlowId, stateMachine.getYFlowId());
YFlow result = transactionManager.doInTransaction(() -> {
YFlow yFlow = yFlowRepository.findById(yFlowId).orElseThrow(() -> new FlowProcessingException(ErrorType.INTERNAL_ERROR, format("Y-flow %s not found", yFlowId)));
Flow flow = flowRepository.findById(subFlowId).orElseThrow(() -> new FlowProcessingException(ErrorType.INTERNAL_ERROR, format("Flow %s not found", subFlowId)));
YSubFlow subFlow = YSubFlow.builder().yFlow(yFlow).flow(flow).sharedEndpointVlan(sharedEndpoint.getVlanId()).sharedEndpointInnerVlan(sharedEndpoint.getInnerVlanId()).endpointSwitchId(endpoint.getSwitchId()).endpointPort(endpoint.getPortNumber()).endpointVlan(endpoint.getOuterVlanId()).endpointInnerVlan(endpoint.getInnerVlanId()).build();
yFlow.addSubFlow(subFlow);
return yFlow;
});
if (subFlowId.equals(stateMachine.getMainAffinityFlowId())) {
stateMachine.getRequestedFlows().forEach(requestedFlow -> {
String requestedFlowId = requestedFlow.getFlowId();
if (!requestedFlowId.equals(subFlowId)) {
stateMachine.addSubFlow(requestedFlowId);
stateMachine.addCreatingSubFlow(requestedFlowId);
stateMachine.notifyEventListeners(listener -> listener.onSubFlowProcessingStart(yFlowId, requestedFlowId));
CommandContext flowContext = stateMachine.getCommandContext().fork(requestedFlowId);
requestedFlow.setAffinityFlowId(stateMachine.getMainAffinityFlowId());
flowCreateService.startFlowCreation(flowContext, requestedFlow, yFlowId);
}
});
}
if (stateMachine.getAllocatedSubFlows().size() == stateMachine.getSubFlows().size()) {
return Optional.of(buildResponseMessage(result, stateMachine.getCommandContext()));
} else {
return Optional.empty();
}
}
use of org.openkilda.messaging.command.yflow.SubFlowSharedEndpointEncapsulation in project open-kilda by telstra.
the class YFlowValidatorTest method failIfSubFlowHasNoEndpointProvided.
@Test(expected = InvalidFlowException.class)
public void failIfSubFlowHasNoEndpointProvided() throws InvalidFlowException, UnavailableFlowEndpointException {
YFlowRequest request = YFlowRequest.builder().yFlowId("test").sharedEndpoint(FlowEndpoint.builder().switchId(SWITCH_ID_1).portNumber(PORT_1).build()).subFlows(Arrays.asList(SubFlowDto.builder().flowId("test_1").sharedEndpoint(new SubFlowSharedEndpointEncapsulation(1, 0)).build(), SubFlowDto.builder().flowId("test_2").sharedEndpoint(new SubFlowSharedEndpointEncapsulation(2, 0)).endpoint(FlowEndpoint.builder().switchId(SWITCH_ID_3).portNumber(PORT_3).build()).build())).build();
yFlowValidator.validate(request);
}
Aggregations