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

import com.sitech.ismp.snmp.CommSnmpGet;
import org.apache.log4j.Logger;
import org.snmp4j.smi.VariableBinding;

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

/**
 * Created by mooker on 15/8/3.
 */
public class MultiHostCpuEntryWalker {

    private Logger logger = Logger.getLogger("COLL");

    public static final String ROOT_OID = "1.3.6.1.4.1.3375.2.1.7.5.2.1.1";

    public static final int ATTRIBCOUNT = 11;

    Vector mhcpuVector = new Vector();

    private void addEntry(MultiHostCpuEntry mhce) {
        mhcpuVector.add(mhce);
    }

    public Vector getEntry() {
        return mhcpuVector;
    }

    public void FillTable(String snmpip, int snmpport, String community) {
        CommSnmpGet snmpGet = new CommSnmpGet(snmpip, snmpport, community);
        snmpGet.init();
        snmpGet.setRootOid(ROOT_OID);
        try {
            snmpGet.workTable(ROOT_OID);
        } catch (IOException e) {
            System.out.println("[IP: " + snmpip + " 采集MultiHostCpuEntry失败]");
        }

        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);
            // 填充IfIndex
            MultiHostCpuEntry row = new MultiHostCpuEntry();
            row.setCpuIndex(key);
            row.setCpuId(value);
            this.addEntry(row);
        }

        for (int i = 0; i < mhcpuVector.size(); i++) {
            MultiHostCpuEntry row = (MultiHostCpuEntry) mhcpuVector.elementAt(i);
            String rowIndex = row.getCpuIndex();
            System.out.println("rowIndex: " + rowIndex);
            Vector oidsVector = getOtherOids(rowIndex, ATTRIBCOUNT);
            Vector resultVector = null;
            try {
                resultVector = snmpGet.getPDU(oidsVector);
            } catch (Exception e) {
                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.3375.2.1.7.5.2.1.11")) {
                        row.setCpuUsageRatio(var);
                        System.out.println(oid + " setCpuUsageRatio " + var);
                    }
                }
            }

        }
        snmpGet.release();
        System.out.println("[IP: " + snmpip + " MultiHostCpuEntry采集完毕]");
    }

    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);
            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;
    }
}