-
Notifications
You must be signed in to change notification settings - Fork 111
/
telemetry.go
83 lines (73 loc) · 2.23 KB
/
telemetry.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package postgres
import (
"context"
"time"
"github.com/chef/automate/api/interservice/cfgmgmt/request"
"github.com/gofrs/uuid"
"github.com/pkg/errors"
)
type Telemetry struct {
ID string `db:"id" json:"id"`
LastTelemetryReportedAt time.Time `db:"last_telemetry_reported_at" json:"last_telemetry_reported_at"`
CreatedAt time.Time `db:"created_at" json:"created_at"`
}
//UpdateLastTelemetryReported Upsert the last client run telemetry reported date in postgres
func (pg *Postgres) UpdateTelemetryReported(ctx context.Context, req *request.UpdateTelemetryReportedRequest) error {
err := Transact(pg, func(tx *DBTrans) error {
lastTelemetryReportedAt, err := time.Parse(time.RFC3339, req.LastTelemetryReportedAt)
if err != nil {
return err
}
count, err := pg.mapper.SelectInt("SELECT COUNT(*) FROM telemetry;")
if err != nil {
return err
}
if count > 0 {
updatetelemetryDate := `update telemetry set last_telemetry_reported_at=$1`
_, err := tx.Exec(updatetelemetryDate, lastTelemetryReportedAt)
if err != nil {
return err
}
} else {
err = tx.StoreTelemetry(ctx, lastTelemetryReportedAt)
if err != nil {
return errors.Wrap(err, "Failed to insert telemetry reported date")
}
}
return nil
})
if err != nil {
return err
}
return nil
}
// StoreTelemetry Store last client run telemetry reported timestamp
func (trans *DBTrans) StoreTelemetry(ctx context.Context, lastTelemetryReportedAt time.Time) error {
telArr := make([]interface{}, 0)
tel := Telemetry{
ID: uuid.Must(uuid.NewV4()).String(),
LastTelemetryReportedAt: lastTelemetryReportedAt,
CreatedAt: time.Now(),
}
telArr = append(telArr, &tel)
err := trans.Insert(telArr...)
if err != nil {
return err
}
return nil
}
// GetTelemetry Get last client run telemetry reported timestamp
func (pg *Postgres) GetTelemetry(ctx context.Context) (Telemetry, error) {
var t Telemetry
rows, err := pg.db.Query(`SELECT id,last_telemetry_reported_at, created_at from telemetry`)
if err != nil {
return Telemetry{}, err
}
for rows.Next() {
err = rows.Scan(&t.ID, &t.LastTelemetryReportedAt, &t.CreatedAt)
if err != nil {
return Telemetry{}, err
}
}
return t, nil
}