mirror/wakapi
1
0
Fork 0
mirror of https://github.com/muety/wakapi.git synced 2023-08-10 21:12:56 +03:00
Code Issues Projects Releases Wiki Activity

Add database cleanup functionality.

Browse Source
This commit is contained in:
Ferdinand Mütsch
2020-03-09 17:30:23 +01:00
parent c7556fa9a5
commit 33ebb8a771
5 changed files with 48 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
services/aggregation.go
View File

@ -11,7 +11,7 @@ import (
)
const (
intervalDays int = 1 // TODO: Make configurable
aggregateIntervalDays int = 1 // TODO: Make configurable
)
type AggregationService struct {
@ -126,7 +126,7 @@ func generateUserJobs(userId string, lastAggregation time.Time, jobs chan<- *Agg
from = time.Date(
lastAggregation.Year(),
lastAggregation.Month(),
lastAggregation.Day()+intervalDays,
lastAggregation.Day()+aggregateIntervalDays,
0, 0, 0, 0,
lastAggregation.Location(),
)
@ -136,7 +136,7 @@ func generateUserJobs(userId string, lastAggregation time.Time, jobs chan<- *Agg
to = time.Date(
from.Year(),
from.Month(),
from.Day()+intervalDays,
from.Day()+aggregateIntervalDays,
0, 0, 0, 0,
from.Location(),
)

34
services/heartbeat.go
View File

@ -1,6 +1,9 @@
package services
import (
"github.com/jasonlvhit/gocron"
"github.com/n1try/wakapi/utils"
"log"
"time"
"github.com/jinzhu/gorm"
@ -8,7 +11,10 @@ import (
gormbulk "github.com/t-tiger/gorm-bulk-insert"
)
const TableHeartbeat = "heartbeat"
const (
TableHeartbeat = "heartbeat"
cleanUpInterval = time.Duration(aggregateIntervalDays) * 2 * 24 * time.Hour
)
type HeartbeatService struct {
Config *models.Config
@ -55,3 +61,29 @@ func (srv *HeartbeatService) GetFirstUserHeartbeats(userIds []string) ([]*models
}
return heartbeats, nil
}
func (srv *HeartbeatService) DeleteBefore(t time.Time) error {
if err := srv.Db.
Where("time <= ?", t).
Delete(models.Heartbeat{}).Error; err != nil {
return err
}
return nil
}
func (srv *HeartbeatService) CleanUp() error {
refTime := utils.StartOfDay().Add(-cleanUpInterval)
if err := srv.DeleteBefore(refTime); err != nil {
log.Printf("Failed to clean up heartbeats older than %v – %v\n", refTime, err)
return err
}
log.Printf("Successfully cleaned up heartbeats older than %v\n", refTime)
return nil
}
func (srv *HeartbeatService) ScheduleCleanUp() {
srv.CleanUp()
gocron.Every(1).Day().At("02:30").Do(srv.CleanUp)
<-gocron.Start()
}