package com.sitech.ismp.coll.net.cisco;

import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Set;
import java.util.Vector;

import org.snmp4j.smi.VariableBinding;

import com.sitech.ismp.snmp.CommSnmpGet;

public class ConnectionEntryWalker {
	
	//public static final String ROOT_OID = "1.3.6.1.4.1.9.9.147.1.2.2.2.1.1";
	//1.3.6.1.4.1.9.9.147.1.2.2.2.1.4.40
	public static final String ROOT_OID = "1.3.6.1.4.1.9.9.147.1.2.2.2.1.4.40.";
	public static final int ATTRIBCOUNT = 5;

	private String IP;

	Vector connectionVecotor = new Vector();

	private void addEntry(ConnectionEntry conn) {
		connectionVecotor.add(conn);
	}
	public Vector getEntry(){
		return connectionVecotor;
	}
	public void walker(String ip, int port, String community){
		this.IP = ip;
		CommSnmpGet snmpGet = new CommSnmpGet(ip, port, community);
		snmpGet.init();
		snmpGet.setRootOid(ROOT_OID);
		try {
			snmpGet.workTable(ROOT_OID);
		} catch (IOException e) {
			// TODO Auto-generated catch block
			System.out.println("[IP: " + IP + " 采集RunEntry失败]");
		}

		HashMap hm = snmpGet.getWalkHashMap();
		Set set = hm.keySet();
		Iterator it = set.iterator();
		while (it.hasNext()) {
			String key = (String) it.next();
			String value = (String) hm.get(key);
			// 填充Index
			ConnectionEntry conn = new ConnectionEntry();
			conn.setConnectionStatService(key);
			
			this.connectionVecotor.add(conn);
		}

		// try {
		for (int i = 0; i < connectionVecotor.size(); i++) {
			ConnectionEntry conn = (ConnectionEntry) connectionVecotor.elementAt(i);
			String rowIndex = conn.getConnectionStatService();
			//System.out.println("rowIndex: " + rowIndex);
			Vector oidsVector = getOtherOids(rowIndex, ATTRIBCOUNT);
			Vector resultVector = null;
			
			try {
				resultVector = snmpGet.getPDU(oidsVector);
			} catch (Exception e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}

			if (resultVector != null) {
				for (int j = 0; j < resultVector.size(); j++) {
					VariableBinding vb = (VariableBinding) resultVector.get(j);
					String oid = vb.getOid().toString();
					String var = vb.getVariable().toString();
					 System.out.println(oid + " = " + var);
					if (oid.startsWith("1.3.6.1.4.1.9.9.147.1.2.2.2.1.1")) {
						conn.setConnectionStatService(rowIndex);
					} else if(oid.startsWith("1.3.6.1.4.1.9.9.147.1.2.2.2.1.2")){
						conn.setConnectionStatType(var);
					}else if(oid.startsWith("1.3.6.1.4.1.9.9.147.1.2.2.2.1.3")){
						conn.setConnectionStatDescription(var);
					}else if(oid.startsWith("1.3.6.1.4.1.9.9.147.1.2.2.2.1.4")){
						conn.setConnectionStatCount(var);
					}else if(oid.startsWith("1.3.6.1.4.1.9.9.147.1.2.2.2.1.5")){
						conn.setConnectionStatValue(var);
					}
				}
			}
		}
		snmpGet.release();
	} 
	
	private Vector getOtherOids(String rowIndex, int attribCount) {
		Vector vector = new Vector();
		for (int i = 2; i <= attribCount; i++) {
			String oid = releaseRootOidLast(ROOT_OID, rowIndex, i);
			 //System.out.println("rowIndex change oid: " + oid);
			vector.add(oid);
		}
		return vector;
	}
	private String releaseRootOidLast(String rootOid, String rowIndex, int int_i) {
		String result = "";
		String rowIndexLast = rowIndex.substring(rootOid.length(), rowIndex
				.length());
		String rootOidStart = rootOid.substring(0, rootOid.length() - 1);

		result = rootOidStart + int_i + rowIndexLast;
		return result;
	}
}