use of org.openkilda.messaging.command.yflow.YFlowPartialUpdateRequest in project open-kilda by telstra.
the class YFlowUpdateServiceTest method shouldPatchFlowWithTransitSwitches.
@Test
public void shouldPatchFlowWithTransitSwitches() throws UnroutableFlowException, RecoverableException, DuplicateKeyException {
// given
createYFlow();
List<SubFlowPartialUpdateDto> subFlowPartialUpdateDtos = new ArrayList<>();
subFlowPartialUpdateDtos.add(SubFlowPartialUpdateDto.builder().flowId("test_flow_1").endpoint(FlowPartialUpdateEndpoint.builder().switchId(SWITCH_NEW_FIRST_EP).portNumber(2).vlanId(103).build()).build());
subFlowPartialUpdateDtos.add(SubFlowPartialUpdateDto.builder().flowId("test_flow_2").endpoint(FlowPartialUpdateEndpoint.builder().switchId(SWITCH_NEW_SECOND_EP).portNumber(3).vlanId(104).build()).build());
YFlowPartialUpdateRequest request = YFlowPartialUpdateRequest.builder().yFlowId("test_successful_yflow").maximumBandwidth(2000L).subFlows(subFlowPartialUpdateDtos).build();
preparePathComputationForUpdate("test_flow_1", buildNewFirstSubFlowPathPair());
preparePathComputationForUpdate("test_flow_2", buildNewSecondSubFlowPathPair());
prepareYPointComputation(SWITCH_SHARED, SWITCH_NEW_FIRST_EP, SWITCH_NEW_SECOND_EP, SWITCH_TRANSIT);
// when
processUpdateRequestAndSpeakerCommands(request);
verifyNorthboundSuccessResponse(yFlowUpdateHubCarrier, YFlowResponse.class);
verifyYFlowStatus(request.getYFlowId(), FlowStatus.UP);
YFlow flow = getYFlow(request.getYFlowId());
assertEquals(2000L, flow.getMaximumBandwidth());
Set<SwitchId> expectedEndpointSwitchIds = Stream.of(SWITCH_NEW_FIRST_EP, SWITCH_NEW_SECOND_EP).collect(Collectors.toSet());
Set<SwitchId> actualEndpointSwitchIds = flow.getSubFlows().stream().map(YSubFlow::getEndpointSwitchId).collect(Collectors.toSet());
assertEquals(expectedEndpointSwitchIds, actualEndpointSwitchIds);
}
use of org.openkilda.messaging.command.yflow.YFlowPartialUpdateRequest in project open-kilda by telstra.
the class YFlowServiceImpl method patchYFlow.
@Override
public CompletableFuture<YFlow> patchYFlow(String yFlowId, YFlowPatchPayload patchPayload) {
log.debug("Processing y-flow patch: {}", yFlowId);
String correlationId = RequestCorrelationId.getId();
YFlowPartialUpdateRequest yFlowPartialUpdateRequest;
try {
yFlowPartialUpdateRequest = flowMapper.toYFlowPatchRequest(yFlowId, patchPayload);
} catch (IllegalArgumentException e) {
throw new MessageException(correlationId, System.currentTimeMillis(), ErrorType.DATA_INVALID, e.getMessage(), "Can not parse arguments of the flow patch request");
}
CommandMessage request = new CommandMessage(yFlowPartialUpdateRequest, System.currentTimeMillis(), correlationId);
return messagingChannel.sendAndGet(flowHsTopic, request).thenApply(YFlowResponse.class::cast).thenApply(YFlowResponse::getYFlow).thenApply(flowMapper::toYFlow);
}
use of org.openkilda.messaging.command.yflow.YFlowPartialUpdateRequest in project open-kilda by telstra.
the class RouterBolt method handleInput.
@Override
protected void handleInput(Tuple input) {
if (active) {
String key = input.getStringByField(FIELD_ID_KEY);
if (StringUtils.isBlank(key)) {
// TODO: the key must be unique, but the correlationId comes in from outside and we can't guarantee that.
// IMPORTANT: Storm may initiate reprocessing of the same tuple (e.g. in the case of timeout) and
// cause creating multiple FSMs for the same tuple. This must be avoided.
// As for now tuples are routed by the key field, and services can check FSM uniqueness.
key = getCommandContext().getCorrelationId();
}
CommandMessage message = (CommandMessage) input.getValueByField(FIELD_ID_PAYLOAD);
MessageData data = message.getData();
if (data instanceof FlowRequest) {
FlowRequest request = (FlowRequest) data;
log.debug("Received request {} with key {}", request, key);
Values values = new Values(key, request.getFlowId(), request);
switch(request.getType()) {
case CREATE:
emitWithContext(ROUTER_TO_FLOW_CREATE_HUB.name(), input, values);
break;
case UPDATE:
emitWithContext(ROUTER_TO_FLOW_UPDATE_HUB.name(), input, values);
break;
default:
throw new UnsupportedOperationException(format("Flow operation %s is not supported", request.getType()));
}
} else if (data instanceof FlowRerouteRequest) {
FlowRerouteRequest rerouteRequest = (FlowRerouteRequest) data;
log.debug("Received a reroute request {}/{} with key {}. MessageId {}", rerouteRequest.getFlowId(), rerouteRequest.getAffectedIsl(), key, input.getMessageId());
Values values = new Values(key, rerouteRequest.getFlowId(), data);
emitWithContext(ROUTER_TO_FLOW_REROUTE_HUB.name(), input, values);
} else if (data instanceof FlowDeleteRequest) {
FlowDeleteRequest deleteRequest = (FlowDeleteRequest) data;
log.debug("Received a delete request {} with key {}. MessageId {}", deleteRequest.getFlowId(), key, input.getMessageId());
Values values = new Values(key, deleteRequest.getFlowId(), data);
emitWithContext(ROUTER_TO_FLOW_DELETE_HUB.name(), input, values);
} else if (data instanceof FlowPathSwapRequest) {
FlowPathSwapRequest pathSwapRequest = (FlowPathSwapRequest) data;
log.debug("Received a path swap request {} with key {}. MessageId {}", pathSwapRequest.getFlowId(), key, input.getMessageId());
Values values = new Values(key, pathSwapRequest.getFlowId(), data);
emitWithContext(ROUTER_TO_FLOW_PATH_SWAP_HUB.name(), input, values);
} else if (data instanceof SwapFlowEndpointRequest) {
log.debug("Received a swap flow endpoints request with key {}. MessageId {}", key, input.getMessageId());
emitWithContext(ROUTER_TO_FLOW_SWAP_ENDPOINTS_HUB.name(), input, new Values(key, data));
} else if (data instanceof CreateFlowLoopRequest) {
log.debug("Received a create flow loop request with key {}. MessageId {}", key, input.getMessageId());
CreateFlowLoopRequest request = (CreateFlowLoopRequest) data;
emitWithContext(ROUTER_TO_FLOW_UPDATE_HUB.name(), input, new Values(key, request.getFlowId(), data));
} else if (data instanceof DeleteFlowLoopRequest) {
log.debug("Received a delete flow loop request with key {}. MessageId {}", key, input.getMessageId());
DeleteFlowLoopRequest request = (DeleteFlowLoopRequest) data;
emitWithContext(ROUTER_TO_FLOW_UPDATE_HUB.name(), input, new Values(key, request.getFlowId(), data));
} else if (data instanceof FlowMirrorPointCreateRequest) {
log.debug("Received a flow mirror point create request with key {}. MessageId {}", key, input.getMessageId());
FlowMirrorPointCreateRequest request = (FlowMirrorPointCreateRequest) data;
emitWithContext(ROUTER_TO_FLOW_CREATE_MIRROR_POINT_HUB.name(), input, new Values(key, request.getFlowId(), data));
} else if (data instanceof FlowMirrorPointDeleteRequest) {
log.debug("Received a flow mirror point delete request with key {}. MessageId {}", key, input.getMessageId());
FlowMirrorPointDeleteRequest request = (FlowMirrorPointDeleteRequest) data;
emitWithContext(ROUTER_TO_FLOW_DELETE_MIRROR_POINT_HUB.name(), input, new Values(key, request.getFlowId(), data));
} else if (data instanceof FlowValidationRequest) {
log.debug("Received a flow validation request with key {}. MessageId {}", key, input.getMessageId());
FlowValidationRequest request = (FlowValidationRequest) data;
emitWithContext(ROUTER_TO_FLOW_VALIDATION_HUB.name(), input, new Values(key, request.getFlowId(), data));
} else if (data instanceof YFlowRequest) {
YFlowRequest request = (YFlowRequest) data;
log.debug("Received request {} with key {}", request, key);
Values values = new Values(key, request.getYFlowId(), request);
switch(request.getType()) {
case CREATE:
emitWithContext(ROUTER_TO_YFLOW_CREATE_HUB.name(), input, values);
break;
case UPDATE:
emitWithContext(ROUTER_TO_YFLOW_UPDATE_HUB.name(), input, values);
break;
default:
throw new UnsupportedOperationException(format("Y-flow operation %s is not supported", request.getType()));
}
} else if (data instanceof YFlowPartialUpdateRequest) {
YFlowPartialUpdateRequest request = (YFlowPartialUpdateRequest) data;
log.debug("Received a y-flow partial update request {} with key {}", request, key);
emitWithContext(ROUTER_TO_YFLOW_UPDATE_HUB.name(), input, new Values(key, request.getYFlowId(), data));
} else if (data instanceof YFlowRerouteRequest) {
YFlowRerouteRequest request = (YFlowRerouteRequest) data;
log.debug("Received a y-flow reroute request {} with key {}", data, key);
emitWithContext(ROUTER_TO_YFLOW_REROUTE_HUB.name(), input, new Values(key, request.getYFlowId(), data));
} else if (data instanceof YFlowDeleteRequest) {
YFlowDeleteRequest request = (YFlowDeleteRequest) data;
log.debug("Received a y-flow delete request {} with key {}", request, key);
emitWithContext(ROUTER_TO_YFLOW_DELETE_HUB.name(), input, new Values(key, request.getYFlowId(), data));
} else if (data instanceof YFlowsDumpRequest) {
log.debug("Received a y-flow dump request {} with key {}", data, key);
emitWithContext(ROUTER_TO_YFLOW_READ.name(), input, new Values(key, data));
} else if (data instanceof YFlowReadRequest) {
log.debug("Received a y-flow read request {} with key {}", data, key);
emitWithContext(ROUTER_TO_YFLOW_READ.name(), input, new Values(key, data));
} else if (data instanceof YFlowPathsReadRequest) {
log.debug("Received a y-flow read path request {} with key {}", data, key);
emitWithContext(ROUTER_TO_YFLOW_READ.name(), input, new Values(key, data));
} else if (data instanceof SubFlowsReadRequest) {
log.debug("Received a y-flow sub-flows request {} with key {}", data, key);
emitWithContext(ROUTER_TO_YFLOW_READ.name(), input, new Values(key, data));
} else if (data instanceof YFlowValidationRequest) {
YFlowValidationRequest request = (YFlowValidationRequest) data;
log.debug("Received a y-flow validation request {} with key {}", request, key);
emitWithContext(ROUTER_TO_YFLOW_VALIDATION_HUB.name(), input, new Values(key, request.getYFlowId(), data));
} else if (data instanceof YFlowSyncRequest) {
YFlowSyncRequest request = (YFlowSyncRequest) data;
log.debug("Received a y-flow synchronization request {} with key {}", request, key);
YFlowRerouteRequest rerouteRequest = new YFlowRerouteRequest(request.getYFlowId(), emptySet(), true, "initiated via synchronization request", false);
emitWithContext(ROUTER_TO_YFLOW_REROUTE_HUB.name(), input, new Values(key, rerouteRequest.getYFlowId(), rerouteRequest));
} else {
unhandledInput(input);
}
}
}
use of org.openkilda.messaging.command.yflow.YFlowPartialUpdateRequest in project open-kilda by telstra.
the class YFlowUpdateHubBolt method onRequest.
@Override
protected void onRequest(Tuple input) throws PipelineException {
currentKey = pullKey(input);
CommandData payload = pullValue(input, FIELD_ID_PAYLOAD, CommandData.class);
try {
if (payload instanceof YFlowRequest) {
yflowUpdateService.handleRequest(currentKey, pullContext(input), (YFlowRequest) payload);
} else if (payload instanceof YFlowPartialUpdateRequest) {
yflowUpdateService.handlePartialUpdateRequest(currentKey, pullContext(input), (YFlowPartialUpdateRequest) payload);
} else {
unhandledInput(input);
}
} catch (DuplicateKeyException e) {
log.error("Failed to handle a request with key {}. {}", currentKey, e.getMessage());
} catch (FlowProcessingException e) {
ErrorData errorData = new ErrorData(e.getErrorType(), "Y-flow update error", e.getMessage());
CommandContext commandContext = getCommandContext();
ErrorMessage errorMessage = new ErrorMessage(errorData, commandContext.getCreateTime(), commandContext.getCorrelationId());
sendNorthboundResponse(errorMessage);
}
}
Aggregations