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

39 lines
817 B
Go
Raw Normal View History

2019-05-05 23:36:49 +03:00
package services
import (
"time"
2019-05-05 23:36:49 +03:00
2019-05-11 18:49:56 +03:00
"github.com/jinzhu/gorm"
2019-05-05 23:36:49 +03:00
"github.com/n1try/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
)
2019-05-06 01:40:41 +03:00
const TableHeartbeat = "heartbeat"
2019-05-05 23:36:49 +03:00
type HeartbeatService struct {
2019-05-11 18:49:56 +03:00
Db *gorm.DB
2019-05-05 23:36:49 +03:00
}
2019-05-11 18:49:56 +03:00
func (srv *HeartbeatService) InsertBatch(heartbeats *[]models.Heartbeat) error {
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
}
2019-05-11 18:49:56 +03:00
func (srv *HeartbeatService) GetAllFrom(date 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 > ?", date).
Find(&heartbeats).Error; err != nil {
return nil, err
}
2019-05-11 18:49:56 +03:00
return &heartbeats, nil
}