/*******************************************************************************
* This file is part of OpenNMS(R).
*
* Copyright (C) 2008-2011 The OpenNMS Group, Inc.
* OpenNMS(R) is Copyright (C) 1999-2011 The OpenNMS Group, Inc.
*
* OpenNMS(R) is a registered trademark of The OpenNMS Group, Inc.
*
* OpenNMS(R) is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published
* by the Free Software Foundation, either version 3 of the License,
* or (at your option) any later version.
*
* OpenNMS(R) is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with OpenNMS(R). If not, see:
* http://www.gnu.org/licenses/
*
* For more information contact:
* OpenNMS(R) Licensing <license@opennms.org>
* http://www.opennms.org/
* http://www.opennms.com/
*******************************************************************************/
package org.opennms.netmgt.provision.scan.snmp;
import java.net.InetAddress;
import java.util.ArrayList;
import java.util.List;
import org.opennms.netmgt.config.SnmpAgentConfigFactory;
import org.opennms.netmgt.provision.ScanContext;
import org.opennms.netmgt.provision.Scanner;
import org.opennms.netmgt.snmp.AggregateTracker;
import org.opennms.netmgt.snmp.Collectable;
import org.opennms.netmgt.snmp.CollectionTracker;
import org.opennms.netmgt.snmp.SingleInstanceTracker;
import org.opennms.netmgt.snmp.SnmpAgentConfig;
import org.opennms.netmgt.snmp.SnmpResult;
import org.opennms.netmgt.snmp.SnmpUtils;
import org.opennms.netmgt.snmp.SnmpWalker;
import org.springframework.util.Assert;
/**
* <p>AbstractSnmpScanner class.</p>
*
* @author brozow
* @version $Id: $
*/
public class AbstractSnmpScanner implements Scanner {
private String m_name = null;
private SnmpAgentConfigFactory m_snmpAgentConfigFactory = null;
private List<SnmpExchange> m_exchangeCollection = null;
/**
* <p>Constructor for AbstractSnmpScanner.</p>
*
* @param name a {@link java.lang.String} object.
*/
protected AbstractSnmpScanner(String name) {
m_name = name;
m_exchangeCollection = new ArrayList<SnmpExchange>();
}
/**
* <p>getName</p>
*
* @return a {@link java.lang.String} object.
*/
public String getName() {
return m_name;
}
/**
* <p>setSnmpAgentConfigFactory</p>
*
* @param snmpPeerFactory a {@link org.opennms.netmgt.config.SnmpAgentConfigFactory} object.
*/
public void setSnmpAgentConfigFactory(SnmpAgentConfigFactory snmpPeerFactory) {
m_snmpAgentConfigFactory = snmpPeerFactory;
}
/* (non-Javadoc)
* @see org.opennms.netmgt.provision.Scanner#init()
*/
/**
* <p>init</p>
*/
public void init() {
Assert.notNull(m_snmpAgentConfigFactory, "snmpAgentConfigFactory must be set");
onInit();
}
/**
* <p>onInit</p>
*/
protected void onInit() {
}
/* (non-Javadoc)
* @see org.opennms.netmgt.provision.Scanner#scan(org.opennms.netmgt.provision.ScanContext)
*/
/** {@inheritDoc} */
public void scan(ScanContext context) throws InterruptedException {
InetAddress agentAddress = context.getAgentAddress("SNMP");
if (agentAddress == null) {
return;
}
SnmpAgentConfig agentConfig = m_snmpAgentConfigFactory.getAgentConfig(agentAddress);
SnmpWalker walker = SnmpUtils.createWalker(agentConfig, getName(), createCollectionTracker(context));
walker.start();
walker.waitFor();
}
/**
* @param context
* @return
*/
private CollectionTracker createCollectionTracker(final ScanContext scanContext) {
List<Collectable> trackers = new ArrayList<Collectable>();
for(SnmpExchange exchange : m_exchangeCollection) {
trackers.add(exchange.createTracker(scanContext));
}
return new AggregateTracker(trackers);
}
public static interface Storer {
public void storeResult(ScanContext scanContext, SnmpResult res);
}
public interface SnmpExchange {
public CollectionTracker createTracker(ScanContext context);
public void andStoreIn(Storer storer);
}
/**
* <p>getSingleInstance</p>
*
* @param base a {@link java.lang.String} object.
* @param inst a {@link java.lang.String} object.
* @return a {@link org.opennms.netmgt.provision.scan.snmp.AbstractSnmpScanner.SnmpExchange} object.
*/
protected SnmpExchange getSingleInstance(final String base, final String inst) {
SnmpExchange exchange = new SnmpExchange() {
Storer m_storer;
public CollectionTracker createTracker(final ScanContext scanContext) {
return new SingleInstanceTracker(base, inst) {
@Override
protected void storeResult(SnmpResult res) {
m_storer.storeResult(scanContext, res);
}
};
}
public void andStoreIn(Storer storer) {
m_storer = storer;
}
};
m_exchangeCollection.add(exchange);
return exchange;
}
}