Method RecordUsage

enterprise/aiseats/tracker.go:75–116

RecordUsage will record the AI seat usage for the user. There is a race condition between checking if the user should be recorded or throttled and actually recording. This is fine, as it just means we record the usage twice. The throttle just exists to prevent excessive database queries.

(ctx context.Context, userID uuid.UUID, reason agplaiseats.Reason)

// it just means we record the usage twice.
// The throttle just exists to prevent excessive database queries.
func (t *SeatTracker) RecordUsage(ctx context.Context, userID uuid.UUID, reason agplaiseats.Reason) {
	now := t.clock.Now()
	if t.skipRecord(userID, now) {
		return
	}

	isNew, err := t.db.UpsertAISeatState(ctx, database.UpsertAISeatStateParams{
		UserID:               userID,
		FirstUsedAt:          now,
		LastEventType:        reason.EventType,
		LastEventDescription: reason.Description,
	})
	if err != nil {
		t.logger.Warn(ctx, "upsert AI seat state", slog.Error(err), slog.F("user_id", userID), slog.F("event_type", reason.EventType))
		t.recordThrottle(userID, now, failedThrottleInterval)
		return
	}

	t.recordThrottle(userID, now, throttleInterval)
	if isNew && t.auditor != nil {
		// Record an audit log for the first time a user uses an AI seat.
		auditor := t.auditor.Load()
		if auditor == nil || *auditor == nil {
			return
		}
		audit.BackgroundAudit[database.AiSeatState](ctx, &audit.BackgroundAuditParams[database.AiSeatState]{
			Audit:  *auditor,
			Log:    t.logger,
			UserID: userID,
			Time:   now,
			Action: database.AuditActionCreate,
			New: database.AiSeatState{
				UserID:               userID,
				FirstUsedAt:          now,
				LastUsedAt:           now,
				LastEventType:        reason.EventType,
				LastEventDescription: reason.Description,
				UpdatedAt:            now,
			},
		})
	}
}