[usage] Add usageAttributionID to WorksapceInstance model (in go)

Author: easyCZ

components/usage/pkg/db/workspace_instance.go

@@ -11,6 +11,7 @@ import (
 	"github.com/google/uuid"
 	"gorm.io/datatypes"
 	"gorm.io/gorm"
+	"strings"
 	"time"
 )
 
@@ -23,6 +24,7 @@ type WorkspaceInstance struct {
 	IdeURL             string         `gorm:"column:ideUrl;type:varchar;size:255;" json:"ideUrl"`
 	WorkspaceBaseImage string         `gorm:"column:workspaceBaseImage;type:varchar;size:255;" json:"workspaceBaseImage"`
 	WorkspaceImage     string         `gorm:"column:workspaceImage;type:varchar;size:255;" json:"workspaceImage"`
+	UsageAttributionID string         `gorm:"column:usageAttributionId;type:varchar;size:60;" json:"usageAttributionId"`
 
 	CreationTime VarcharTime `gorm:"column:creationTime;type:varchar;size:255;" json:"creationTime"`
 	StartedTime  VarcharTime `gorm:"column:startedTime;type:varchar;size:255;" json:"startedTime"`
@@ -87,3 +89,16 @@ func ListWorkspaceInstancesInRange(ctx context.Context, conn *gorm.DB, from, to
 
 	return instances, nil
 }
+
+// AttributionID consists of an entity, and an identifier in the form:
+// <entity>:<identifier>, e.g. team:a7dcf253-f05e-4dcf-9a47-cf8fccc74717
+type AttributionID string
+
+func (a AttributionID) Values() (entity string, identifier string) {
+	tokens := strings.Split(string(a), ":")
+	if len(tokens) != 2 {
+		return "", ""
+	}
+
+	return tokens[0], tokens[1]
+}