Examples with ListFlowsRequest org.openkilda.server42.control.messaging.flowrtt.ListFlowsRequest used on opensource projects

Example 1 with ListFlowsRequest

use of org.openkilda.server42.control.messaging.flowrtt.ListFlowsRequest in project open-kilda by telstra.

the class Gate method listen.

@KafkaHandler
void listen(ListFlowsRequest data, @Header(KafkaHeaders.RECEIVED_MESSAGE_KEY) String switchIdKey) {
    CommandPacket commandPacket = getFlowListCommandPacket(switchIdKey);
    try {
        CommandPacketResponse serverResponse = zeroMqClient.send(commandPacket);
        if (serverResponse == null) {
            log.error("No response from server on {}", data.getHeaders().getCorrelationId());
            return;
        }
        HashSet<String> flowList = new HashSet<>();
        for (Any any : serverResponse.getResponseList()) {
            flowList.add(any.unpack(Flow.class).getFlowId());
        }
        ListFlowsResponse response = ListFlowsResponse.builder().headers(data.getHeaders()).flowIds(flowList).build();
        template.send(toStorm, response);
    } catch (InvalidProtocolBufferException e) {
        log.error("Marshalling error on {}", data);
    }
}

Example 2 with ListFlowsRequest

use of org.openkilda.server42.control.messaging.flowrtt.ListFlowsRequest in project open-kilda by telstra.

the class GateTest method listFlowsTest.

@Test
public void listFlowsTest() throws Exception {
    Builder commandPacketResponseBuilded = CommandPacketResponse.newBuilder();
    Flow flow1 = Flow.newBuilder().setFlowId("some-flow-id-01").build();
    Flow flow2 = Flow.newBuilder().setFlowId("some-flow-id-02").build();
    commandPacketResponseBuilded.addResponse(Any.pack(flow1));
    commandPacketResponseBuilded.addResponse(Any.pack(flow2));
    CommandPacketResponse commandPacketResponse = commandPacketResponseBuilded.build();
    when(zeroMqClient.send(argThat(commandPacket -> commandPacket.getType() == Type.LIST_FLOWS))).thenReturn(commandPacketResponse);
    String switchId = "00:00:1b:45:18:d6:71:5a";
    Headers headers = Headers.builder().correlationId("some-correlation-id").build();
    gate.listen(new ListFlowsRequest(headers), switchId);
    ArgumentCaptor<ListFlowsResponse> argument = ArgumentCaptor.forClass(ListFlowsResponse.class);
    verify(template).send(eq(toStorm), argument.capture());
    ListFlowsResponse response = argument.getValue();
    assertThat(response.getFlowIds()).contains(flow1.getFlowId(), flow2.getFlowId());
}

Example 3 with ListFlowsRequest

use of org.openkilda.server42.control.messaging.flowrtt.ListFlowsRequest in project open-kilda by telstra.

the class KafkaController method getFlowList.

@GetMapping(value = "/flow/")
@ResponseBody
private DeferredResult<ResponseEntity<?>> getFlowList(@RequestParam String switchId) throws InterruptedException, ExecutionException, TimeoutException {
    Headers headers = buildHeader();
    String correlationId = headers.getCorrelationId();
    send(switchId, new ListFlowsRequest(headers));
    DeferredResult<ResponseEntity<?>> deferredResult = new DeferredResult<>(500L);
    deferredResult.onTimeout(() -> {
        deferredResult.setErrorResult(ResponseEntity.status(HttpStatus.REQUEST_TIMEOUT).body("Request timeout occurred."));
        deferredResultConcurrentHashMap.remove(correlationId);
    });
    deferredResult.onCompletion(() -> deferredResultConcurrentHashMap.remove(correlationId));
    deferredResult.onError((Throwable throwable) -> {
        deferredResult.setErrorResult(ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).body(throwable));
        deferredResultConcurrentHashMap.remove(correlationId);
    });
    deferredResultConcurrentHashMap.put(correlationId, deferredResult);
    return deferredResult;
}