Search in sources :

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);
    }
}
Also used : InvalidProtocolBufferException(com.google.protobuf.InvalidProtocolBufferException) ListFlowsResponse(org.openkilda.server42.control.messaging.flowrtt.ListFlowsResponse) CommandPacketResponse(org.openkilda.server42.control.messaging.Control.CommandPacketResponse) CommandPacket(org.openkilda.server42.control.messaging.Control.CommandPacket) Any(com.google.protobuf.Any) HashSet(java.util.HashSet) KafkaHandler(org.springframework.kafka.annotation.KafkaHandler)

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());
}
Also used : FlowDirection(org.openkilda.server42.messaging.FlowDirection) Control(org.openkilda.server42.control.messaging.Control) FlowRttControl(org.openkilda.server42.control.messaging.flowrtt.FlowRttControl) ClearIsls(org.openkilda.server42.control.messaging.islrtt.ClearIsls) ArgumentMatchers.argThat(org.mockito.ArgumentMatchers.argThat) CommandPacketResponse(org.openkilda.server42.control.messaging.Control.CommandPacketResponse) ArgumentMatchers.eq(org.mockito.ArgumentMatchers.eq) AddFlow(org.openkilda.server42.control.messaging.flowrtt.AddFlow) Assertions.assertThat(org.assertj.core.api.Assertions.assertThat) RunWith(org.junit.runner.RunWith) Autowired(org.springframework.beans.factory.annotation.Autowired) Type(org.openkilda.server42.control.messaging.Control.CommandPacket.Type) Value(org.springframework.beans.factory.annotation.Value) ZeroMqClient(org.openkilda.server42.control.zeromq.ZeroMqClient) ListIslsRequest(org.openkilda.server42.control.messaging.islrtt.ListIslsRequest) ArgumentCaptor(org.mockito.ArgumentCaptor) ClearFlows(org.openkilda.server42.control.messaging.flowrtt.ClearFlows) KafkaTemplate(org.springframework.kafka.core.KafkaTemplate) ListFlowsResponse(org.openkilda.server42.control.messaging.flowrtt.ListFlowsResponse) Map(java.util.Map) RemoveIsl(org.openkilda.server42.control.messaging.islrtt.RemoveIsl) SpringRunner(org.springframework.test.context.junit4.SpringRunner) Headers(org.openkilda.server42.control.messaging.flowrtt.Headers) IslRttControl(org.openkilda.server42.control.messaging.islrtt.IslRttControl) Flow(org.openkilda.server42.control.messaging.flowrtt.FlowRttControl.Flow) MockBean(org.springframework.boot.test.mock.mockito.MockBean) RemoveFlow(org.openkilda.server42.control.messaging.flowrtt.RemoveFlow) InvalidProtocolBufferException(com.google.protobuf.InvalidProtocolBufferException) ListFlowsRequest(org.openkilda.server42.control.messaging.flowrtt.ListFlowsRequest) Test(org.junit.Test) TestPropertySource(org.springframework.test.context.TestPropertySource) Mockito.when(org.mockito.Mockito.when) Builder(org.openkilda.server42.control.messaging.Control.CommandPacketResponse.Builder) Mockito.verify(org.mockito.Mockito.verify) CommandPacket(org.openkilda.server42.control.messaging.Control.CommandPacket) List(java.util.List) AddIsl(org.openkilda.server42.control.messaging.islrtt.AddIsl) SwitchId(org.openkilda.model.SwitchId) SwitchToVlanMapping(org.openkilda.server42.control.config.SwitchToVlanMapping) SpringBootTest(org.springframework.boot.test.context.SpringBootTest) PushSettings(org.openkilda.server42.control.messaging.flowrtt.PushSettings) ContextConfiguration(org.springframework.test.context.ContextConfiguration) Any(com.google.protobuf.Any) IslEndpoint(org.openkilda.server42.control.messaging.islrtt.IslRttControl.IslEndpoint) ListIslsResponse(org.openkilda.server42.control.messaging.islrtt.ListIslsResponse) Headers(org.openkilda.server42.control.messaging.flowrtt.Headers) Builder(org.openkilda.server42.control.messaging.Control.CommandPacketResponse.Builder) ListFlowsRequest(org.openkilda.server42.control.messaging.flowrtt.ListFlowsRequest) ListFlowsResponse(org.openkilda.server42.control.messaging.flowrtt.ListFlowsResponse) CommandPacketResponse(org.openkilda.server42.control.messaging.Control.CommandPacketResponse) AddFlow(org.openkilda.server42.control.messaging.flowrtt.AddFlow) Flow(org.openkilda.server42.control.messaging.flowrtt.FlowRttControl.Flow) RemoveFlow(org.openkilda.server42.control.messaging.flowrtt.RemoveFlow) Test(org.junit.Test) SpringBootTest(org.springframework.boot.test.context.SpringBootTest)

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;
}
Also used : ResponseEntity(org.springframework.http.ResponseEntity) Headers(org.openkilda.server42.control.messaging.flowrtt.Headers) ListFlowsRequest(org.openkilda.server42.control.messaging.flowrtt.ListFlowsRequest) DeferredResult(org.springframework.web.context.request.async.DeferredResult) GetMapping(org.springframework.web.bind.annotation.GetMapping) ResponseBody(org.springframework.web.bind.annotation.ResponseBody)

Aggregations

Any (com.google.protobuf.Any)2 InvalidProtocolBufferException (com.google.protobuf.InvalidProtocolBufferException)2 CommandPacket (org.openkilda.server42.control.messaging.Control.CommandPacket)2 CommandPacketResponse (org.openkilda.server42.control.messaging.Control.CommandPacketResponse)2 Headers (org.openkilda.server42.control.messaging.flowrtt.Headers)2 ListFlowsRequest (org.openkilda.server42.control.messaging.flowrtt.ListFlowsRequest)2 ListFlowsResponse (org.openkilda.server42.control.messaging.flowrtt.ListFlowsResponse)2 HashSet (java.util.HashSet)1 List (java.util.List)1 Map (java.util.Map)1 Assertions.assertThat (org.assertj.core.api.Assertions.assertThat)1 Test (org.junit.Test)1 RunWith (org.junit.runner.RunWith)1 ArgumentCaptor (org.mockito.ArgumentCaptor)1 ArgumentMatchers.argThat (org.mockito.ArgumentMatchers.argThat)1 ArgumentMatchers.eq (org.mockito.ArgumentMatchers.eq)1 Mockito.verify (org.mockito.Mockito.verify)1 Mockito.when (org.mockito.Mockito.when)1 SwitchId (org.openkilda.model.SwitchId)1 SwitchToVlanMapping (org.openkilda.server42.control.config.SwitchToVlanMapping)1