NCGEARS/cheesy-arena-lite
1
0
Fork 0
mirror of https://github.com/Team254/cheesy-arena-lite.git synced 2026-03-09 21:56:50 -04:00
Code Issues Actions Packages Projects Releases 1 Wiki Activity

Fix bandwidth monitoring to work with new switch.

Browse Source
This commit is contained in:
Patrick Fairbank
2017-09-02 11:56:10 -07:00
parent 346912572f
commit c4728de33f
1 changed files with 49 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
field/bandwidth_monitor.go
View File

@@ -16,19 +16,19 @@ const (
monitoringIntervalMs = 1000
toRobotBytesOid = ".1.3.6.1.2.1.2.2.1.10"
fromRobotBytesOid = ".1.3.6.1.2.1.2.2.1.16"
red1Port = 2
red2Port = 4
red3Port = 6
blue1Port = 8
blue2Port = 10
blue3Port = 12
red1Port = 6
red2Port = 8
red3Port = 10
blue1Port = 12
blue2Port = 14
blue3Port = 16
)
type BandwidthMonitor struct {
allianceStations *map[string]*AllianceStation
snmpClient *wapsnmp.WapSNMP
toRobotOid wapsnmp.Oid
fromRobotOid wapsnmp.Oid
toRobotOids []wapsnmp.Oid
fromRobotOids []wapsnmp.Oid
lastToRobotBytes map[string]interface{}
lastFromRobotBytes map[string]interface{}
lastBytesTime time.Time
@@ -36,8 +36,15 @@ type BandwidthMonitor struct {
// Loops indefinitely to query the managed switch via SNMP (Simple Network Management Protocol).
func (arena *Arena) monitorBandwidth() {
monitor := BandwidthMonitor{allianceStations: &arena.AllianceStations,
toRobotOid: wapsnmp.MustParseOid(toRobotBytesOid), fromRobotOid: wapsnmp.MustParseOid(fromRobotBytesOid)}
monitor := BandwidthMonitor{allianceStations: &arena.AllianceStations}
for _, port := range []int{red1Port, red2Port, red3Port, blue1Port, blue2Port, blue3Port} {
toOid := fmt.Sprintf("%s.%d", toRobotBytesOid, 10100+port)
fromOid := fmt.Sprintf("%s.%d", fromRobotBytesOid, 10100+port)
monitor.toRobotOids = append(monitor.toRobotOids, wapsnmp.MustParseOid(toOid))
monitor.fromRobotOids = append(monitor.fromRobotOids, wapsnmp.MustParseOid(fromOid))
}
for {
if monitor.snmpClient != nil && monitor.snmpClient.Target != arena.EventSettings.SwitchAddress {
// Switch address has changed; must re-create the SNMP client.
@@ -51,6 +58,7 @@ func (arena *Arena) monitorBandwidth() {
arena.EventSettings.SwitchPassword, wapsnmp.SNMPv2c, 2*time.Second, 0)
if err != nil {
log.Printf("Error starting bandwidth monitoring: %v", err)
continue
}
}
@@ -66,22 +74,22 @@ func (arena *Arena) monitorBandwidth() {
func (monitor *BandwidthMonitor) updateBandwidth() error {
// Retrieve total number of bytes sent/received per port.
toRobotBytes, err := monitor.snmpClient.GetTable(monitor.toRobotOid)
toRobotBytes, err := monitor.snmpClient.GetMultiple(monitor.toRobotOids)
if err != nil {
return err
}
fromRobotBytes, err := monitor.snmpClient.GetTable(monitor.fromRobotOid)
fromRobotBytes, err := monitor.snmpClient.GetMultiple(monitor.fromRobotOids)
if err != nil {
return err
}
// Calculate the bandwidth usage over time.
monitor.updateStationBandwidth("R1", red1Port, toRobotBytes, fromRobotBytes)
monitor.updateStationBandwidth("R2", red2Port, toRobotBytes, fromRobotBytes)
monitor.updateStationBandwidth("R3", red3Port, toRobotBytes, fromRobotBytes)
monitor.updateStationBandwidth("B1", blue1Port, toRobotBytes, fromRobotBytes)
monitor.updateStationBandwidth("B2", blue2Port, toRobotBytes, fromRobotBytes)
monitor.updateStationBandwidth("B3", blue3Port, toRobotBytes, fromRobotBytes)
monitor.updateStationBandwidth("R1", 0, toRobotBytes, fromRobotBytes)
monitor.updateStationBandwidth("R2", 1, toRobotBytes, fromRobotBytes)
monitor.updateStationBandwidth("R3", 2, toRobotBytes, fromRobotBytes)
monitor.updateStationBandwidth("B1", 3, toRobotBytes, fromRobotBytes)
monitor.updateStationBandwidth("B2", 4, toRobotBytes, fromRobotBytes)
monitor.updateStationBandwidth("B3", 5, toRobotBytes, fromRobotBytes)
monitor.lastToRobotBytes = toRobotBytes
monitor.lastFromRobotBytes = fromRobotBytes
@@ -89,7 +97,7 @@ func (monitor *BandwidthMonitor) updateBandwidth() error {
return nil
}
func (monitor *BandwidthMonitor) updateStationBandwidth(station string, port int, toRobotBytes map[string]interface{},
func (monitor *BandwidthMonitor) updateStationBandwidth(station string, oidIndex int, toRobotBytes map[string]interface{},
fromRobotBytes map[string]interface{}) {
dsConn := (*monitor.allianceStations)[station].DsConn
if dsConn == nil {
@@ -98,11 +106,29 @@ func (monitor *BandwidthMonitor) updateStationBandwidth(station string, port int
}
secondsSinceLast := time.Now().Sub(monitor.lastBytesTime).Seconds()
toRobotBytesForPort := uint32(toRobotBytes[fmt.Sprintf("%s.%d", toRobotBytesOid, port)].(wapsnmp.Counter))
lastToRobotBytesForPort := uint32(monitor.lastToRobotBytes[fmt.Sprintf("%s.%d", toRobotBytesOid, port)].(wapsnmp.Counter))
toOid := monitor.toRobotOids[oidIndex].String()
if _, ok := toRobotBytes[toOid]; !ok {
log.Printf("Error: OID %s not present in new to-robot stats %v.", toOid, toRobotBytes)
return
}
toRobotBytesForPort := uint32(toRobotBytes[toOid].(wapsnmp.Counter))
if _, ok := monitor.lastToRobotBytes[toOid]; !ok {
// This may be the first time reading.
return
}
lastToRobotBytesForPort := uint32(monitor.lastToRobotBytes[toOid].(wapsnmp.Counter))
dsConn.MBpsToRobot = float64(toRobotBytesForPort-lastToRobotBytesForPort) / 1024 / 1024 / secondsSinceLast
fromRobotBytesForPort := uint32(fromRobotBytes[fmt.Sprintf("%s.%d", fromRobotBytesOid, port)].(wapsnmp.Counter))
lastFromRobotBytesForPort := uint32(monitor.lastFromRobotBytes[fmt.Sprintf("%s.%d", fromRobotBytesOid, port)].(wapsnmp.Counter))
fromOid := monitor.fromRobotOids[oidIndex].String()
if _, ok := fromRobotBytes[fromOid]; !ok {
log.Printf("Error: OID %s not present in new from-robot stats %v.", fromOid, fromRobotBytes)
return
}
fromRobotBytesForPort := uint32(fromRobotBytes[fromOid].(wapsnmp.Counter))
if _, ok := monitor.lastFromRobotBytes[fromOid]; !ok {
// This may be the first time reading.
return
}
lastFromRobotBytesForPort := uint32(monitor.lastFromRobotBytes[fromOid].(wapsnmp.Counter))
dsConn.MBpsFromRobot = float64(fromRobotBytesForPort-lastFromRobotBytesForPort) / 1024 / 1024 / secondsSinceLast
}