Examples with IKafkaProducerService org.openkilda.floodlight.service.kafka.IKafkaProducerService used on opensource projects

Example 11 with IKafkaProducerService

use of org.openkilda.floodlight.service.kafka.IKafkaProducerService in project open-kilda by telstra.

the class RecordHandler method doInstallSwitchRules.

private void doInstallSwitchRules(final CommandMessage message) {
    SwitchRulesInstallRequest request = (SwitchRulesInstallRequest) message.getData();
    logger.info("Installing rules on '{}' switch: action={}", request.getSwitchId(), request.getInstallRulesAction());
    final IKafkaProducerService producerService = getKafkaProducer();
    final String replyToTopic = context.getKafkaSwitchManagerTopic();
    DatapathId dpid = DatapathId.of(request.getSwitchId().toLong());
    ISwitchManager switchManager = context.getSwitchManager();
    InstallRulesAction installAction = request.getInstallRulesAction();
    List<Long> installedRules = new ArrayList<>();
    try {
        if (installAction == InstallRulesAction.INSTALL_DROP) {
            installedRules.add(switchManager.installDropFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_BROADCAST) {
            installedRules.add(switchManager.installVerificationRule(dpid, true));
        } else if (installAction == InstallRulesAction.INSTALL_UNICAST) {
            // TODO: this isn't always added (ie if OF1.2). Is there a better response?
            installedRules.add(switchManager.installVerificationRule(dpid, false));
        } else if (installAction == InstallRulesAction.INSTALL_DROP_VERIFICATION_LOOP) {
            installedRules.add(switchManager.installDropLoopRule(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_BFD_CATCH) {
            // TODO: this isn't installed as well. Refactor this section
            installedRules.add(switchManager.installBfdCatchFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_ROUND_TRIP_LATENCY) {
            // TODO: this isn't installed as well. Refactor this section
            installedRules.add(switchManager.installRoundTripLatencyFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_UNICAST_VXLAN) {
            installedRules.add(switchManager.installUnicastVerificationRuleVxlan(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_MULTITABLE_PRE_INGRESS_PASS_THROUGH) {
            installedRules.add(switchManager.installPreIngressTablePassThroughDefaultRule(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_MULTITABLE_INGRESS_DROP) {
            installedRules.add(switchManager.installDropFlowForTable(dpid, INGRESS_TABLE_ID, MULTITABLE_INGRESS_DROP_COOKIE));
        } else if (installAction == InstallRulesAction.INSTALL_MULTITABLE_POST_INGRESS_DROP) {
            installedRules.add(switchManager.installDropFlowForTable(dpid, POST_INGRESS_TABLE_ID, MULTITABLE_POST_INGRESS_DROP_COOKIE));
        } else if (installAction == InstallRulesAction.INSTALL_MULTITABLE_EGRESS_PASS_THROUGH) {
            installedRules.add(switchManager.installEgressTablePassThroughDefaultRule(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_MULTITABLE_TRANSIT_DROP) {
            installedRules.add(switchManager.installDropFlowForTable(dpid, TRANSIT_TABLE_ID, MULTITABLE_TRANSIT_DROP_COOKIE));
        } else if (installAction == InstallRulesAction.INSTALL_LLDP_INPUT_PRE_DROP) {
            installedRules.add(switchManager.installLldpInputPreDropFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_LLDP_INGRESS) {
            installedRules.add(switchManager.installLldpIngressFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_LLDP_POST_INGRESS) {
            installedRules.add(switchManager.installLldpPostIngressFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_LLDP_POST_INGRESS_VXLAN) {
            installedRules.add(switchManager.installLldpPostIngressVxlanFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_LLDP_POST_INGRESS_ONE_SWITCH) {
            installedRules.add(switchManager.installLldpPostIngressOneSwitchFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_LLDP_TRANSIT) {
            installedRules.add(switchManager.installLldpTransitFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_ARP_INPUT_PRE_DROP) {
            installedRules.add(switchManager.installArpInputPreDropFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_ARP_INGRESS) {
            installedRules.add(switchManager.installArpIngressFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_ARP_POST_INGRESS) {
            installedRules.add(switchManager.installArpPostIngressFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_ARP_POST_INGRESS_VXLAN) {
            installedRules.add(switchManager.installArpPostIngressVxlanFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_ARP_POST_INGRESS_ONE_SWITCH) {
            installedRules.add(switchManager.installArpPostIngressOneSwitchFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_ARP_TRANSIT) {
            installedRules.add(switchManager.installArpTransitFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_SERVER_42_OUTPUT_VLAN || installAction == InstallRulesAction.INSTALL_SERVER_42_FLOW_RTT_OUTPUT_VLAN) {
            validateServer42Fields(request, installAction);
            installedRules.add(switchManager.installServer42FlowRttOutputVlanFlow(dpid, request.getServer42Port(), request.getServer42Vlan(), request.getServer42MacAddress()));
        } else if (installAction == InstallRulesAction.INSTALL_SERVER_42_OUTPUT_VXLAN || installAction == InstallRulesAction.INSTALL_SERVER_42_FLOW_RTT_OUTPUT_VXLAN) {
            validateServer42Fields(request, installAction);
            installedRules.add(switchManager.installServer42FlowRttOutputVxlanFlow(dpid, request.getServer42Port(), request.getServer42Vlan(), request.getServer42MacAddress()));
        } else if (installAction == InstallRulesAction.INSTALL_SERVER_42_TURNING || installAction == InstallRulesAction.INSTALL_SERVER_42_FLOW_RTT_TURNING) {
            installedRules.add(switchManager.installServer42FlowRttTurningFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_SERVER_42_FLOW_RTT_VXLAN_TURNING) {
            installedRules.add(switchManager.installServer42FlowRttVxlanTurningFlow(dpid));
        } else if (installAction == InstallRulesAction.INSTALL_SERVER_42_ISL_RTT_OUTPUT) {
            validateServer42Fields(request, installAction);
            installedRules.add(switchManager.installServer42IslRttOutputFlow(dpid, request.getServer42Port(), request.getServer42Vlan(), request.getServer42MacAddress()));
        } else if (installAction == InstallRulesAction.INSTALL_SERVER_42_ISL_RTT_TURNING) {
            installedRules.add(switchManager.installServer42IslRttTurningFlow(dpid));
        } else {
            installedRules.addAll(switchManager.installDefaultRules(dpid));
            if (request.isMultiTable()) {
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), MULTITABLE_PRE_INGRESS_PASS_THROUGH_COOKIE));
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), MULTITABLE_INGRESS_DROP_COOKIE));
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), MULTITABLE_POST_INGRESS_DROP_COOKIE));
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), MULTITABLE_EGRESS_PASS_THROUGH_COOKIE));
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), MULTITABLE_TRANSIT_DROP_COOKIE));
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), LLDP_POST_INGRESS_COOKIE));
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), LLDP_POST_INGRESS_VXLAN_COOKIE));
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), LLDP_POST_INGRESS_ONE_SWITCH_COOKIE));
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), ARP_POST_INGRESS_COOKIE));
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), ARP_POST_INGRESS_VXLAN_COOKIE));
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), ARP_POST_INGRESS_ONE_SWITCH_COOKIE));
                for (int port : request.getIslPorts()) {
                    installedRules.addAll(switchManager.installMultitableEndpointIslRules(dpid, port));
                }
                for (int port : request.getFlowPorts()) {
                    installedRules.add(switchManager.installIntermediateIngressRule(dpid, port));
                }
                for (Integer port : request.getFlowLldpPorts()) {
                    installedRules.add(switchManager.installLldpInputCustomerFlow(dpid, port));
                }
                for (Integer port : request.getFlowArpPorts()) {
                    installedRules.add(switchManager.installArpInputCustomerFlow(dpid, port));
                }
                if (request.isSwitchLldp()) {
                    installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), LLDP_INPUT_PRE_DROP_COOKIE));
                    installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), LLDP_TRANSIT_COOKIE));
                    installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), LLDP_INGRESS_COOKIE));
                }
                if (request.isSwitchArp()) {
                    installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), ARP_INPUT_PRE_DROP_COOKIE));
                    installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), ARP_TRANSIT_COOKIE));
                    installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), ARP_INGRESS_COOKIE));
                }
            }
            Integer server42Port = request.getServer42Port();
            Integer server42Vlan = request.getServer42Vlan();
            MacAddress server42MacAddress = request.getServer42MacAddress();
            if (request.isServer42FlowRttFeatureToggle()) {
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), SERVER_42_FLOW_RTT_TURNING_COOKIE));
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), SERVER_42_FLOW_RTT_VXLAN_TURNING_COOKIE));
                if (request.isServer42FlowRttSwitchProperty() && server42Port != null && server42Vlan != null && server42MacAddress != null) {
                    installedRules.add(switchManager.installServer42FlowRttOutputVlanFlow(dpid, server42Port, server42Vlan, server42MacAddress));
                    installedRules.add(switchManager.installServer42FlowRttOutputVxlanFlow(dpid, server42Port, server42Vlan, server42MacAddress));
                    for (Integer port : request.getServer42FlowRttPorts()) {
                        installedRules.add(switchManager.installServer42FlowRttInputFlow(dpid, server42Port, port, server42MacAddress));
                    }
                }
            }
            if (request.isServer42IslRttEnabled()) {
                installedRules.add(processInstallDefaultFlowByCookie(request.getSwitchId(), SERVER_42_ISL_RTT_TURNING_COOKIE));
                installedRules.add(switchManager.installServer42IslRttOutputFlow(dpid, request.getServer42Port(), request.getServer42Vlan(), request.getServer42MacAddress()));
                for (Integer port : request.getIslPorts()) {
                    installedRules.add(switchManager.installServer42IslRttInputFlow(dpid, server42Port, port));
                }
            }
        }
        SwitchRulesResponse response = new SwitchRulesResponse(installedRules.stream().filter(Objects::nonNull).collect(Collectors.toList()));
        InfoMessage infoMessage = new InfoMessage(response, System.currentTimeMillis(), message.getCorrelationId());
        producerService.sendMessageAndTrack(replyToTopic, record.key(), infoMessage);
    } catch (SwitchOperationException e) {
        logger.error("Failed to install rules on switch '{}'", request.getSwitchId(), e);
        anError(ErrorType.CREATION_FAILURE).withMessage(e.getMessage()).withDescription(request.getSwitchId().toString()).withCorrelationId(message.getCorrelationId()).withTopic(replyToTopic).withKey(record.key()).sendVia(producerService);
    }
}

Example 12 with IKafkaProducerService

use of org.openkilda.floodlight.service.kafka.IKafkaProducerService in project open-kilda by telstra.

the class RecordHandler method doReinstallDefaultFlowForSwitchManager.

/**
 * Reinstall default flow.
 *
 * @param message command message for flow deletion
 */
private void doReinstallDefaultFlowForSwitchManager(CommandMessage message) {
    ReinstallDefaultFlowForSwitchManagerRequest request = (ReinstallDefaultFlowForSwitchManagerRequest) message.getData();
    IKafkaProducerService producerService = getKafkaProducer();
    String replyToTopic = context.getKafkaSwitchManagerTopic();
    long cookie = request.getCookie();
    if (!Cookie.isDefaultRule(cookie)) {
        logger.warn("Failed to reinstall default switch rule for switch: '{}'. Rule {} is not default.", request.getSwitchId(), Long.toHexString(cookie));
        anError(ErrorType.DATA_INVALID).withMessage(format("Failed to reinstall default switch rule for switch %s. Rule %s is not default", request.getSwitchId(), Long.toHexString(cookie))).withDescription(request.getSwitchId().toString()).withCorrelationId(message.getCorrelationId()).withTopic(replyToTopic).sendVia(producerService);
    }
    SwitchId switchId = request.getSwitchId();
    DatapathId dpid = DatapathId.of(switchId.toLong());
    try {
        RemoveFlow command = RemoveFlow.builder().flowId("REMOVE_DEFAULT_FLOW").cookie(cookie).switchId(switchId).build();
        Set<Long> removedFlows = new HashSet<>(processDeleteFlow(command, dpid));
        for (Long removedFlow : removedFlows) {
            Long installedFlow;
            if (request instanceof ReinstallServer42FlowForSwitchManagerRequest) {
                installedFlow = processInstallServer42Rule((ReinstallServer42FlowForSwitchManagerRequest) request);
            } else {
                installedFlow = processInstallDefaultFlowByCookie(switchId, removedFlow);
            }
            InfoMessage response = new InfoMessage(new FlowReinstallResponse(removedFlow, installedFlow), System.currentTimeMillis(), message.getCorrelationId());
            producerService.sendMessageAndTrack(replyToTopic, message.getCorrelationId(), response);
        }
    } catch (SwitchOperationException e) {
        logger.error("Failed to reinstall switch rule for switch: '{}'", request.getSwitchId(), e);
        anError(ErrorType.INTERNAL_ERROR).withMessage(e.getMessage()).withDescription(request.getSwitchId().toString()).withCorrelationId(message.getCorrelationId()).withTopic(replyToTopic).sendVia(producerService);
    }
}