mirror of https://github.com/muety/wakapi.git
84 lines
1.8 KiB
Go
84 lines
1.8 KiB
Go
package services
|
|
|
|
import (
|
|
"github.com/muety/wakapi/config"
|
|
"github.com/muety/wakapi/models"
|
|
"time"
|
|
)
|
|
|
|
const HeartbeatDiffThreshold = 2 * time.Minute
|
|
|
|
type DurationService struct {
|
|
config *config.Config
|
|
heartbeatService IHeartbeatService
|
|
}
|
|
|
|
func NewDurationService(heartbeatService IHeartbeatService) *DurationService {
|
|
srv := &DurationService{
|
|
config: config.Get(),
|
|
heartbeatService: heartbeatService,
|
|
}
|
|
return srv
|
|
}
|
|
|
|
func (srv *DurationService) Get(from, to time.Time, user *models.User, filters *models.Filters) (models.Durations, error) {
|
|
heartbeats, err := srv.heartbeatService.GetAllWithin(from, to, user)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Aggregation
|
|
var count int
|
|
var latest *models.Duration
|
|
|
|
mapping := make(map[string][]*models.Duration)
|
|
|
|
for _, h := range heartbeats {
|
|
if filters != nil && !filters.Match(h) {
|
|
continue
|
|
}
|
|
|
|
d1 := models.NewDurationFromHeartbeat(h)
|
|
|
|
if list, ok := mapping[d1.GroupHash]; !ok || len(list) < 1 {
|
|
mapping[d1.GroupHash] = []*models.Duration{d1}
|
|
}
|
|
|
|
if latest == nil {
|
|
latest = d1
|
|
continue
|
|
}
|
|
|
|
dur := d1.Time.T().Sub(latest.Time.T().Add(latest.Duration))
|
|
if dur > HeartbeatDiffThreshold {
|
|
dur = HeartbeatDiffThreshold
|
|
}
|
|
latest.Duration += dur
|
|
|
|
if dur >= HeartbeatDiffThreshold || latest.GroupHash != d1.GroupHash {
|
|
list := mapping[d1.GroupHash]
|
|
if d0 := list[len(list)-1]; d0 != d1 {
|
|
mapping[d1.GroupHash] = append(mapping[d1.GroupHash], d1)
|
|
}
|
|
latest = d1
|
|
} else {
|
|
latest.NumHeartbeats++
|
|
}
|
|
|
|
count++
|
|
}
|
|
|
|
durations := make(models.Durations, 0, count)
|
|
|
|
for _, list := range mapping {
|
|
for _, d := range list {
|
|
if d.Duration == 0 {
|
|
d.Duration = HeartbeatDiffThreshold
|
|
}
|
|
durations = append(durations, d)
|
|
}
|
|
}
|
|
|
|
return durations.Sorted(), nil
|
|
}
|