8859c36234
https://storjlabs.atlassian.net/browse/V3-2545 Change-Id: I1dd54a0c77cb4905bb1f350beeb82c6f7700ee70
52 lines
1.8 KiB
Go
52 lines
1.8 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package satellitedb
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"storj.io/common/storj"
|
|
dbx "storj.io/storj/satellite/satellitedb/dbx"
|
|
)
|
|
|
|
type downtimeTrackingDB struct {
|
|
db *satelliteDB
|
|
}
|
|
|
|
// Add adds a record for a particular node ID with the amount of time it has been offline.
|
|
func (db *downtimeTrackingDB) Add(ctx context.Context, nodeID storj.NodeID, trackedTime time.Time, timeOffline time.Duration) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
_, err = db.db.Create_NodesOfflineTime(ctx,
|
|
dbx.NodesOfflineTime_NodeId(nodeID.Bytes()),
|
|
dbx.NodesOfflineTime_TrackedAt(trackedTime),
|
|
dbx.NodesOfflineTime_Seconds(int(timeOffline.Seconds())),
|
|
)
|
|
|
|
return Error.Wrap(err)
|
|
}
|
|
|
|
// GetOfflineTime gets the total amount of offline time for a node within a certain timeframe.
|
|
// "total offline time" is defined as the sum of all offline time windows that begin inside the provided time window.
|
|
// An offline time window that began before `begin` but that overlaps with the provided time window is not included.
|
|
// An offline time window that begins within the provided time window, but that extends beyond `end` is included.
|
|
func (db *downtimeTrackingDB) GetOfflineTime(ctx context.Context, nodeID storj.NodeID, begin, end time.Time) (time.Duration, error) {
|
|
offlineEntries, err := db.db.All_NodesOfflineTime_By_NodeId_And_TrackedAt_Greater_And_TrackedAt_LessOrEqual(ctx,
|
|
dbx.NodesOfflineTime_NodeId(nodeID.Bytes()),
|
|
dbx.NodesOfflineTime_TrackedAt(begin),
|
|
dbx.NodesOfflineTime_TrackedAt(end),
|
|
)
|
|
if err != nil {
|
|
return time.Duration(0), Error.Wrap(err)
|
|
}
|
|
|
|
totalSeconds := 0
|
|
for _, entry := range offlineEntries {
|
|
totalSeconds += entry.Seconds
|
|
}
|
|
duration := time.Duration(totalSeconds) * time.Second
|
|
return duration, nil
|
|
}
|