Skip to content

Commit c90db68

Browse files
author
Andrew Farries
committed
Update refs to column name in usage component
1 parent 219ad1a commit c90db68

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

components/usage/pkg/controller/reconciler.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ func usageReportToUsageRecords(report UsageReport) []db.WorkspaceInstanceUsage {
243243
}
244244

245245
usageRecords = append(usageRecords, db.WorkspaceInstanceUsage{
246-
WorkspaceID: instance.ID,
246+
InstanceID: instance.ID,
247247
AttributionID: attributionId,
248248
StartedAt: instance.CreationTime.Time(),
249249
StoppedAt: stoppedAt,

components/usage/pkg/db/workspace_instance_usage.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ import (
1515
)
1616

1717
type WorkspaceInstanceUsage struct {
18-
WorkspaceID uuid.UUID `gorm:"primary_key;column:workspaceId;type:char;size:36;" json:"workspaceId"`
18+
InstanceID uuid.UUID `gorm:"primary_key;column:instanceId;type:char;size:36;" json:"instanceId"`
1919
AttributionID AttributionID `gorm:"column:attributionId;type:varchar;size:255;" json:"attributionId"`
2020
StartedAt time.Time `gorm:"column:startedAt;type:timestamp;default:CURRENT_TIMESTAMP(6);" json:"startedAt"`
2121
StoppedAt sql.NullTime `gorm:"column:stoppedAt;type:timestamp;" json:"stoppedAt"`
@@ -31,7 +31,7 @@ func (u *WorkspaceInstanceUsage) TableName() string {
3131

3232
func CreateUsageRecords(ctx context.Context, conn *gorm.DB, records []WorkspaceInstanceUsage) error {
3333
db := conn.WithContext(ctx).Clauses(clause.OnConflict{
34-
Columns: []clause.Column{{Name: "workspaceId"}},
34+
Columns: []clause.Column{{Name: "instanceId"}},
3535
UpdateAll: true,
3636
})
3737

components/usage/pkg/db/workspace_instance_usage_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ func TestCanCreateUsageRecords(t *testing.T) {
2929
{
3030
Name: "One workspace instance",
3131
UsageRecords: []db.WorkspaceInstanceUsage{{
32-
WorkspaceID: instanceID,
32+
InstanceID: instanceID,
3333
AttributionID: teamAttributionID,
3434
StartedAt: time.Now(),
3535
StoppedAt: sql.NullTime{},
@@ -67,7 +67,7 @@ func TestCanHandleDuplicateRecords(t *testing.T) {
6767
Name: "The same instance twice",
6868
UsageRecords: []db.WorkspaceInstanceUsage{
6969
{
70-
WorkspaceID: instanceID,
70+
InstanceID: instanceID,
7171
AttributionID: teamAttributionID,
7272
StartedAt: instanceStartTime,
7373
StoppedAt: sql.NullTime{},
@@ -76,7 +76,7 @@ func TestCanHandleDuplicateRecords(t *testing.T) {
7676
Deleted: false,
7777
},
7878
{
79-
WorkspaceID: instanceID,
79+
InstanceID: instanceID,
8080
AttributionID: teamAttributionID,
8181
StartedAt: instanceStartTime,
8282
StoppedAt: sql.NullTime{},

0 commit comments

Comments
 (0)