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 PhysicalEntryWalker {
	public static final String ROOT_OID = "1.3.6.1.2.1.47.1.1.1.1.2";

	public static final int ATTRIBCOUNT = 8;

	private String IP;

	Vector physicalVecotor = new Vector();

	private void addEntry(PhysicalEntry physical) {
		physicalVecotor.add(physical);
	}
	public Vector getEntry(){
		return physicalVecotor;
	}
	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
			PhysicalEntry physical = new PhysicalEntry();
			physical.setRowIndex(key);
			physical.setPhysicalIndex(key);
			physical.setPhysicalDescr(value);
			this.physicalVecotor.add(physical);
		}

		// try {
		for (int i = 0; i < physicalVecotor.size(); i++) {
			PhysicalEntry physical = (PhysicalEntry) physicalVecotor.elementAt(i);
			String rowIndex = physical.getPhysicalIndex();
			//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.2.1.47.1.1.1.1.2")) {
						physical.setPhysicalDescr(var);
					} else if(oid.startsWith("1.3.6.1.2.1.47.1.1.1.1.3")){
						physical.setPhysicalVendorType(var);
					}else if(oid.startsWith("1.3.6.1.2.1.47.1.1.1.1.5")){
						physical.setPhysicalClass(var);
					}else if(oid.startsWith("1.3.6.1.2.1.47.1.1.1.1.7")){
						physical.setPhysicalName(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;
	}
}