Method TrackUsage

internal/services/usage.go:118–141

TrackUsage increments cost for a user/project in hourly, weekly, and lifetime buckets. Uses upsert to create or update the usage records atomically. The success parameter indicates whether the request completed successfully. We will be charging only for successful requests, but we track failed reque

(ctx context.Context, userID bson.ObjectID, projectID string, cost float64, success bool)

// The success parameter indicates whether the request completed successfully.
// We will be charging only for successful requests, but we track failed requests for monitoring.
func (s *UsageService) TrackUsage(ctx context.Context, userID bson.ObjectID, projectID string, cost float64, success bool) error {
	if cost == 0 {
		return nil
	}

	now := time.Now()

	// Track hourly usage
	if err := s.trackHourlyUsage(ctx, userID, projectID, cost, success, now); err != nil {
		return err
	}

	// Track weekly usage
	if err := s.trackWeeklyUsage(ctx, userID, projectID, cost, success, now); err != nil {
		return err
	}

	// Track lifetime usage
	if err := s.trackLifetimeUsage(ctx, userID, projectID, cost, success, now); err != nil {
		return err
	}

	return nil
}

func (s *UsageService) upsertUsage(ctx context.Context, collection *mongo.Collection, filter bson.M, cost float64, success bool, now time.Time) error {
	costField := "failed_cost"