67 lines
2.2 KiB
Go
67 lines
2.2 KiB
Go
|
package store
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Sample is one ~1-second telemetry reading for an activity.
|
||
|
|
type Sample struct {
|
||
|
|
ElapsedSeconds float64
|
||
|
|
TimestampMs int64
|
||
|
|
HeartRate *float64
|
||
|
|
SpeedMps *float64
|
||
|
|
DistanceM *float64
|
||
|
|
ElevationM *float64
|
||
|
|
}
|
||
|
|
|
||
|
|
// ReplaceActivitySamples deletes any existing samples for activityID and
|
||
|
|
// bulk-inserts the given set, so re-syncing an activity's details is idempotent.
|
||
|
|
func (db *DB) ReplaceActivitySamples(ctx context.Context, activityID int64, samples []Sample) error {
|
||
|
|
tx, err := db.BeginTx(ctx, nil)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("begin replace samples tx: %w", err)
|
||
|
|
}
|
||
|
|
defer tx.Rollback()
|
||
|
|
|
||
|
|
if _, err := tx.ExecContext(ctx, `DELETE FROM activity_samples WHERE activity_id = ?`, activityID); err != nil {
|
||
|
|
return fmt.Errorf("delete existing samples for activity %d: %w", activityID, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
stmt, err := tx.PrepareContext(ctx, `
|
||
|
|
INSERT INTO activity_samples (activity_id, elapsed_seconds, timestamp_ms, heart_rate, speed_mps, distance_m, elevation_m)
|
||
|
|
VALUES (?,?,?,?,?,?,?)`)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("prepare insert sample: %w", err)
|
||
|
|
}
|
||
|
|
defer stmt.Close()
|
||
|
|
|
||
|
|
for _, s := range samples {
|
||
|
|
if _, err := stmt.ExecContext(ctx, activityID, s.ElapsedSeconds, s.TimestampMs, s.HeartRate, s.SpeedMps, s.DistanceM, s.ElevationM); err != nil {
|
||
|
|
return fmt.Errorf("insert sample for activity %d: %w", activityID, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return tx.Commit()
|
||
|
|
}
|
||
|
|
|
||
|
|
// SamplesForActivity returns all samples for an activity, ordered by elapsed_seconds.
|
||
|
|
func (db *DB) SamplesForActivity(ctx context.Context, activityID int64) ([]Sample, error) {
|
||
|
|
rows, err := db.QueryContext(ctx, `
|
||
|
|
SELECT elapsed_seconds, timestamp_ms, heart_rate, speed_mps, distance_m, elevation_m
|
||
|
|
FROM activity_samples WHERE activity_id = ? ORDER BY elapsed_seconds`, activityID)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("samples for activity %d: %w", activityID, err)
|
||
|
|
}
|
||
|
|
defer rows.Close()
|
||
|
|
|
||
|
|
samples := []Sample{}
|
||
|
|
for rows.Next() {
|
||
|
|
var s Sample
|
||
|
|
if err := rows.Scan(&s.ElapsedSeconds, &s.TimestampMs, &s.HeartRate, &s.SpeedMps, &s.DistanceM, &s.ElevationM); err != nil {
|
||
|
|
return nil, fmt.Errorf("scan sample row: %w", err)
|
||
|
|
}
|
||
|
|
samples = append(samples, s)
|
||
|
|
}
|
||
|
|
return samples, rows.Err()
|
||
|
|
}
|