1
0
mirror of https://github.com/muety/wakapi.git synced 2023-08-10 21:12:56 +03:00
wakapi/services/heartbeat.go

90 lines
2.1 KiB
Go
Raw Normal View History

2019-05-05 23:36:49 +03:00
package services
import (
2020-03-09 19:30:23 +03:00
"github.com/jasonlvhit/gocron"
2020-03-31 13:22:17 +03:00
"github.com/muety/wakapi/utils"
2020-03-09 19:30:23 +03:00
"log"
"time"
2019-05-05 23:36:49 +03:00
2019-05-11 18:49:56 +03:00
"github.com/jinzhu/gorm"
2020-03-31 13:22:17 +03:00
"github.com/muety/wakapi/models"
2019-05-11 18:49:56 +03:00
gormbulk "github.com/t-tiger/gorm-bulk-insert"
2019-05-05 23:36:49 +03:00
)
2020-03-09 19:30:23 +03:00
const (
TableHeartbeat = "heartbeat"
cleanUpInterval = time.Duration(aggregateIntervalDays) * 2 * 24 * time.Hour
)
2019-05-05 23:36:49 +03:00
type HeartbeatService struct {
2019-05-21 18:16:46 +03:00
Config *models.Config
2019-05-16 23:53:03 +03:00
Db *gorm.DB
2019-05-05 23:36:49 +03:00
}
2020-02-20 16:28:55 +03:00
func (srv *HeartbeatService) Init() {}
func (srv *HeartbeatService) InsertBatch(heartbeats []*models.Heartbeat) error {
2019-05-11 18:49:56 +03:00
var batch []interface{}
for _, h := range heartbeats {
batch = append(batch, *h)
2019-05-05 23:36:49 +03:00
}
2019-05-11 18:49:56 +03:00
if err := gormbulk.BulkInsert(srv.Db, batch, 3000); err != nil {
2019-05-05 23:36:49 +03:00
return err
}
return nil
}
func (srv *HeartbeatService) GetAllWithin(from, to time.Time, user *models.User) ([]*models.Heartbeat, error) {
var heartbeats []*models.Heartbeat
2019-05-11 18:49:56 +03:00
if err := srv.Db.
Where(&models.Heartbeat{UserID: user.ID}).
Where("time >= ?", from).
Where("time <= ?", to).
2019-05-20 20:46:41 +03:00
Order("time asc").
Find(&heartbeats).Error; err != nil {
2019-05-11 18:49:56 +03:00
return nil, err
}
return heartbeats, nil
}
// Will return *models.Heartbeat object with only user_id and time fields filled
func (srv *HeartbeatService) GetFirstUserHeartbeats(userIds []string) ([]*models.Heartbeat, error) {
var heartbeats []*models.Heartbeat
if err := srv.Db.
Table("heartbeats").
Select("user_id, min(time) as time").
Where("user_id IN (?)", userIds).
Group("user_id").
Scan(&heartbeats).Error; err != nil {
return nil, err
}
return heartbeats, nil
}
2020-03-09 19:30:23 +03:00
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()
}