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/5.
 */
public class PoolMemberEntryWalker {

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

    public static final String ROOT_OID = "1.3.6.1.4.1.3375.2.2.5.3.2.1.1";

    public static final int ATTRIBCOUNT = 18;

    Vector pmeVector = new Vector();

    private void addEntry(PoolMemberEntry vse) {
        pmeVector.add(vse);
    }

    public Vector getEntry() {
        return pmeVector;
    }

    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 + " 采集PoolMemberEntry失败]");
        }

        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
            PoolMemberEntry row = new PoolMemberEntry();
            row.setRowIndex(key);
            row.setPoolName(value);
            this.addEntry(row);
        }

        for (int i = 0; i < pmeVector.size(); i++) {
            PoolMemberEntry row = (PoolMemberEntry) pmeVector.elementAt(i);
            String rowIndex = row.getRowIndex();
            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.2.5.3.2.1.3.")) {
                        row.setAddr(var);
                    }
                }
            }

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

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