package sw

import (
	"github.com/gaochao1/gosnmp"
	"time"
)

func ConnectionStat(ip, community string, timeout, retry int) (int, error) {
	vendor, err := SysVendor(ip, community, timeout)
	method := "get"
	var oid string
	switch vendor {
	case "Cisco_ASA", "Cisco_ASA_OLD":
		oid = "1.3.6.1.4.1.9.9.147.1.2.2.2.1.5.40.6"
	default:
		return 0, err
	}

	var snmpPDUs []gosnmp.SnmpPDU
	for i := 0; i < retry; i++ {
		snmpPDUs, err = RunSnmp(ip, community, oid, method, timeout)
		if len(snmpPDUs) > 0 {
			break
		}
		time.Sleep(100 * time.Millisecond)
	}

	if err == nil {
		for _, pdu := range snmpPDUs {
			return pdu.Value.(int), err
		}
	}

	return 0, err
}