Aggregating by userId and creationTime

import { TableAggregate } from "@convex-dev/aggregate";
import { v } from "convex/values";
import { components } from "../_generated/api";
import { DataModel } from "../_generated/dataModel";
import { query } from "../_generated/server";
import { getDocumentOrThrow } from "../utils/db";

const durationAggregate = new TableAggregate<{
  Key: number;
  DataModel: DataModel;
  TableName: "tasks";
}>(components.aggregate, {
  sortKey: (doc) => doc._creationTime,
  sumValue: (doc) => doc.duration,
});

export const totalFocusTimeByUser = query({
  args: {
    userId: v.id("users"),
  },
  handler: async (ctx, args) => {
    await getDocumentOrThrow(ctx, "users", args.userId);
    const totalFocusTime = await durationAggregate.sum(ctx);
    return totalFocusTime;
  },
});

export const totalFocusTimeByUserForWeek = query({
  args: {
    userId: v.id("users"),
  },
  handler: async (ctx, args) => {
    await getDocumentOrThrow(ctx, "users", args.userId);
    const totalFocusTimeByWeek = await durationAggregate.sum(ctx);
    return totalFocusTimeByWeek;
  },
});