Examples with ColumnTracker org.opennms.netmgt.snmp.ColumnTracker used on opensource projects

Example 1 with ColumnTracker

use of org.opennms.netmgt.snmp.ColumnTracker in project opennms by OpenNMS.

the class AggregateTrackerProxyTest method setUp.

@Before
public void setUp() {
    // Create a hierarchy of aggregated trackers
    columnTrackers = new ColumnTracker[] { new ColumnTracker(baseOids[0]), new ColumnTracker(baseOids[1]), new ColumnTracker(baseOids[2]) };
    singleInstanceTracker = new SingleInstanceTracker(baseOids[3], SnmpInstId.INST_ZERO);
    childAggregateTracker = new AggregateTracker(columnTrackers);
    parentAggregateTracker = new AggregateTracker(Lists.newArrayList(childAggregateTracker, singleInstanceTracker), gatherer);
    // Verify the generated requests
    List<WalkRequest> expectedRequests = new ArrayList<>();
    // Column tracker requests
    WalkRequest expectedRequest = new WalkRequest(baseOids[0]);
    expectedRequest.setCorrelationId("0-0");
    expectedRequest.setMaxRepetitions(2);
    expectedRequests.add(expectedRequest);
    expectedRequest = new WalkRequest(baseOids[1]);
    expectedRequest.setCorrelationId("0-1");
    expectedRequest.setMaxRepetitions(2);
    expectedRequests.add(expectedRequest);
    expectedRequest = new WalkRequest(baseOids[2]);
    expectedRequest.setCorrelationId("0-2");
    expectedRequest.setMaxRepetitions(2);
    expectedRequests.add(expectedRequest);
    // Single instance tracker request
    expectedRequest = new WalkRequest(baseOids[3]);
    expectedRequest.setCorrelationId("1");
    expectedRequest.setInstance(SnmpInstId.INST_ZERO);
    expectedRequest.setMaxRepetitions(1);
    expectedRequests.add(expectedRequest);
    assertThat(parentAggregateTracker.getWalkRequests(), hasSize(4));
    assertThat(parentAggregateTracker.getWalkRequests(), contains(expectedRequests.toArray()));
    // We shouldn't be finished yet
    assertThat(parentAggregateTracker.isFinished(), equalTo(false));
    assertThat(gatherer.getResults(), hasSize(0));
}

Example 2 with ColumnTracker

use of org.opennms.netmgt.snmp.ColumnTracker in project opennms by OpenNMS.

the class SnmpProxyRpcModule method walk.

private CompletableFuture<Collection<SnmpResponseDTO>> walk(SnmpRequestDTO request, List<SnmpWalkRequestDTO> walks) {
    final CompletableFuture<Collection<SnmpResponseDTO>> future = new CompletableFuture<>();
    final Map<String, SnmpResponseDTO> responsesByCorrelationId = new LinkedHashMap<>();
    final List<Collectable> trackers = new ArrayList<>(walks.size());
    for (final SnmpWalkRequestDTO walk : walks) {
        CollectionTracker tracker;
        if (walk.getInstance() != null) {
            if (walk.getOids().size() != 1) {
                future.completeExceptionally(new IllegalArgumentException("Single instance requests must have a single OID."));
                return future;
            }
            final SnmpObjId oid = walk.getOids().get(0);
            tracker = new SingleInstanceTracker(oid, new SnmpInstId(walk.getInstance())) {

                @Override
                protected void storeResult(SnmpResult res) {
                    addResult(res, walk.getCorrelationId(), responsesByCorrelationId);
                }
            };
        } else {
            final Collection<Collectable> columnTrackers = walk.getOids().stream().map(oid -> SnmpObjId.get(oid)).map(objId -> new ColumnTracker(objId)).collect(Collectors.toList());
            tracker = new AggregateTracker(columnTrackers) {

                @Override
                protected void storeResult(SnmpResult res) {
                    addResult(res, walk.getCorrelationId(), responsesByCorrelationId);
                }
            };
        }
        if (walk.getMaxRepetitions() != null) {
            tracker.setMaxRepetitions(walk.getMaxRepetitions());
        }
        trackers.add(tracker);
    }
    AggregateTracker aggregate = new AggregateTracker(trackers);
    final SnmpWalker walker = SnmpUtils.createWalker(request.getAgent(), request.getDescription(), aggregate);
    walker.setCallback(new SnmpWalkCallback() {

        @Override
        public void complete(SnmpWalker tracker, Throwable t) {
            try {
                if (t != null) {
                    future.completeExceptionally(t);
                } else {
                    future.complete(responsesByCorrelationId.values());
                }
            } finally {
                // Close the tracker using a separate thread
                // This allows the SnmpWalker to clean up properly instead
                // of interrupting execution as it's executing the callback
                REAPER_EXECUTOR.submit(new Runnable() {

                    @Override
                    public void run() {
                        tracker.close();
                    }
                });
            }
        }
    });
    walker.start();
    return future;
}