org.snmp4j.util.TreeEvent Java Examples

private PortNumber getAddDropPort(int channel, boolean isAddPort) {
    OID oid = new OID(CTRL_CHANNEL_ADD_DROP_PORT_INDEX + (isAddPort ? "1" : "2"));

    for (TreeEvent event : snmp.get(oid)) {
        if (event == null) {
            return null;
        }

        VariableBinding[] varBindings = event.getVariableBindings();

        for (VariableBinding varBinding : varBindings) {
            if (varBinding.getOid().last() == channel) {
                int port = varBinding.getVariable().toInt();
                if (!isAddPort) {
                    port += DROP_PORT_OFFSET;
                }
                return PortNumber.portNumber(port);

            }
        }

    }

    return null;
}
 

/**
 * Perform a SNMP walk and returns the list of {@link TreeEvent}
 * @return the list of {@link TreeEvent}
 */
public List<TreeEvent> walk() {
    TreeUtils treeUtils = new TreeUtils(this.snmp, new DefaultPDUFactory());
    @SuppressWarnings("unchecked")
    List<TreeEvent> events = treeUtils.getSubtree(this.target, this.oid);
    return events;
}
 

private int getAlarmId(TreeEvent treeEvents) {
    VariableBinding[] varBindings = treeEvents.getVariableBindings();
    for (VariableBinding varBinding : varBindings) {
        return varBinding.getVariable().toInt();
    }
    return -1;
}
 

/**
 * Perform a SNMP walk and returns the list of {@link TreeEvent}
 * @return the list of {@link TreeEvent}
 */
public List<TreeEvent> walk() {
    TreeUtils treeUtils = new TreeUtils(this.snmp, new DefaultPDUFactory());
    @SuppressWarnings("unchecked")
    List<TreeEvent> events = treeUtils.getSubtree(this.target, this.oid);
    return events;
}
 

private List<FlowRule> fetchRules(OID oid, boolean isAdd, PortNumber linePort) {
    List<FlowRule> rules = new LinkedList<>();

    for (TreeEvent event : snmp.get(oid)) {
        if (event == null) {
            continue;
        }

        VariableBinding[] varBindings = event.getVariableBindings();
        for (VariableBinding varBinding : varBindings) {
            CrossConnectCache cache = this.handler().get(CrossConnectCache.class);

            if (varBinding.getVariable().toInt() == IN_SERVICE) {
                int channel = varBinding.getOid().removeLast();

                PortNumber addDropPort = getAddDropPort(channel, isAdd);
                if (addDropPort == null) {
                    continue;
                }

                TrafficSelector selector = DefaultTrafficSelector.builder()
                        .matchInPort(isAdd ? addDropPort : linePort)
                        .add(Criteria.matchOchSignalType(OchSignalType.FIXED_GRID))
                        .add(Criteria.matchLambda(toOchSignal(channel)))
                        .build();
                TrafficTreatment treatment = DefaultTrafficTreatment.builder()
                        .setOutput(isAdd ? linePort : addDropPort)
                        .build();

                // Lookup flow ID and priority
                int hash = Objects.hash(data().deviceId(), selector, treatment);
                Pair<FlowId, Integer> lookup = cache.get(hash);
                if (lookup == null) {
                    continue;
                }

                FlowRule fr = DefaultFlowRule.builder()
                        .forDevice(data().deviceId())
                        .makePermanent()
                        .withSelector(selector)
                        .withTreatment(treatment)
                        .withPriority(lookup.getRight())
                        .withCookie(lookup.getLeft().value())
                        .build();
                rules.add(fr);
            }
        }
    }

    return rules;
}
 

public List<TreeEvent> get(OID oid) {
    TreeUtils treeUtils = new TreeUtils(snmp, new DefaultPDUFactory());
    treeUtils.setMaxRepetitions(MAX_REPETITIONS);
    return treeUtils.getSubtree(target, oid);
}
 

private List<PortDescription> getPorts() {
    try {
        snmp = new LumentumSnmpDevice(handler().data().deviceId());
    } catch (IOException e) {
        log.error("Failed to connect to device: ", e);

        return Collections.emptyList();
    }

    List<PortDescription> ports = Lists.newLinkedList();

    OID[] oids = {
            new OID(CTRL_PORT_STATE + "1"),
            new OID(CTRL_PORT_STATE + "2")
    };

    for (OID oid : oids) {

        for (TreeEvent event : snmp.get(oid)) {
            if (event != null) {
                VariableBinding[] varBindings = event.getVariableBindings();
                for (VariableBinding varBinding : varBindings) {
                    if (varBinding.getVariable().toInt() == 1) {
                        int portNumber = varBinding.getOid().removeLast();
                        int portDirection = varBinding.getOid().removeLast();
                        SparseAnnotations ann = DefaultAnnotations.builder()
                                .set(AnnotationKeys.PORT_NAME, portDirection + "-" + portNumber)
                                .build();
                        PortDescription p = omsPortDescription(
                                PortNumber.portNumber(ports.size() + 1L),
                                true,
                                LumentumSnmpDevice.START_CENTER_FREQ,
                                LumentumSnmpDevice.END_CENTER_FREQ,
                                LumentumSnmpDevice.CHANNEL_SPACING.frequency(),
                                ann);
                        ports.add(p);
                    }
                }
            }
        }
    }

    // Create LINE IN and LINE OUT ports as these are not reported through SNMP
    SparseAnnotations annLineIn = DefaultAnnotations.builder()
            .set(AnnotationKeys.PORT_NAME, "LINE IN")
            .build();
    ports.add(omsPortDescription(
            PortNumber.portNumber(ports.size() + 1L),
            true,
            LumentumSnmpDevice.START_CENTER_FREQ,
            LumentumSnmpDevice.END_CENTER_FREQ,
            LumentumSnmpDevice.CHANNEL_SPACING.frequency(),
            annLineIn
    ));

    SparseAnnotations annLineOut = DefaultAnnotations.builder()
            .set(AnnotationKeys.PORT_NAME, "LINE OUT")
            .build();
    ports.add(omsPortDescription(
            PortNumber.portNumber(ports.size() + 1L),
            true,
            LumentumSnmpDevice.START_CENTER_FREQ,
            LumentumSnmpDevice.END_CENTER_FREQ,
            LumentumSnmpDevice.CHANNEL_SPACING.frequency(),
            annLineOut
    ));

    return ports;
}
 

/**
 * Method to construct {@link FlowFile} attributes from a {@link TreeEvent}
 * @param treeEvent a {@link TreeEvent}
 * @param flowFile instance of the {@link FlowFile} to update
 * @param processSession instance of {@link ProcessSession}
 * @return updated {@link FlowFile}
 */
public static FlowFile updateFlowFileAttributesWithTreeEventProperties(TreeEvent treeEvent, FlowFile flowFile, ProcessSession processSession) {
    Map<String, String> attributes = new HashMap<String, String>();
    addWalkOidValues(attributes, treeEvent.getVariableBindings());
    flowFile = processSession.putAllAttributes(flowFile, attributes);
    return flowFile;
}
 

/**
 * Method to construct {@link FlowFile} attributes from a {@link TreeEvent}
 * @param treeEvent a {@link TreeEvent}
 * @param flowFile instance of the {@link FlowFile} to update
 * @param processSession instance of {@link ProcessSession}
 * @return updated {@link FlowFile}
 */
public static FlowFile updateFlowFileAttributesWithTreeEventProperties(TreeEvent treeEvent, FlowFile flowFile, ProcessSession processSession) {
    Map<String, String> attributes = new HashMap<String, String>();
    addWalkOidValues(attributes, treeEvent.getVariableBindings());
    flowFile = processSession.putAllAttributes(flowFile, attributes);
    return flowFile;
}