Fix data usage count

This commit is contained in:
Arturo Filastò
2018-09-12 13:48:18 +02:00
parent 54af7170d4
commit 4880d31120
2 changed files with 5 additions and 18 deletions
+4 -17
View File
@@ -63,12 +63,6 @@ func (c *Controller) SetInputIdxMap(inputIdxMap map[int64]int64) error {
return nil
}
type StatusEnd struct {
DownloadedKB float64 `json:"download_kb"`
UploadedKB float64 `json:"uploaded_kb"`
Failure string `json:"failure"`
}
// Init should be called once to initialise the nettest
func (c *Controller) Init(nt *mk.Nettest) error {
log.Debugf("Init: %v", nt)
@@ -287,19 +281,12 @@ func (c *Controller) Init(nt *mk.Nettest) error {
}
}
var endMsg StatusEnd
err := json.Unmarshal([]byte(e.Value.JSONStr), &endMsg)
if err != nil {
log.WithError(err).Errorf("failed to extract status.end message %s", e.Value.JSONStr)
return
if e.Value.Failure != "" {
log.Errorf("Failure in status.end: %s", e.Value.Failure)
}
if endMsg.Failure != "" {
log.Errorf("Failure in status.end: %s", endMsg.Failure)
}
c.res.DataUsageDown += endMsg.DownloadedKB
c.res.DataUsageDown += endMsg.UploadedKB
c.res.DataUsageDown += e.Value.DownloadedKB
c.res.DataUsageDown += e.Value.UploadedKB
})
log.Debugf("Registered all the handlers")