c# - 需要帮助在 HotChocolate 过滤器中创建求和聚合操作

Question

此问题基于 https://github.com/ChilliCream/hotchocolate/issues/924 中的讨论- 这也是我获得灵感的地方。

我有一个系统，其中保存了一份员工名单。每个员工都有一个 WorkHours 属性，代表你每周工作多少小时。

我还有一组任务需要上述员工解决。

关联是通过 Allocation 类处理的。此类包含两个 unix 时间戳 Start 和 End，表示 Employee 正在处理特定任务的时间段。此外，他们有一个 HoursPerWeek 属性，表示员工每周必须在给定任务上花费多少小时，HoursPerWeek 是必需的，因为只要 Allocation.HoursPerWeek 的总和，员工就可以在同一时间段内与多个任务相关联不超过 Employee.WorkHours。

本质上，我想实现与此 LINQ 查询类似的东西

var ts = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
employees.Where(e =>
    e.Allocations
        .Where(a => a.Start < ts && a.End > ts)
        .Sum(a => a.HoursPerWeek)
    < e.WorkHours);

这会有效地给我在这个时间点有剩余工作时间的任何员工。

我无法在我的查询中直接引用 employee.WorkHours，但我正在尝试让它与 double 相比工作。这就是我到目前为止的进展

public class CustomFilterConventionExtension : FilterConventionExtension
{
    protected override void Configure(IFilterConventionDescriptor descriptor)
    {
        descriptor.Operation(CustomFilterOperations.Sum)
            .Name("sum");

        descriptor.Configure<ListFilterInputType<FilterInputType<Allocation>>>(descriptor =>
        {
            descriptor
                .Operation(CustomFilterOperations.Sum)
                .Type<ComparableOperationFilterInputType<double>>();
        });

        descriptor.AddProviderExtension(new QueryableFilterProviderExtension(
            y =>
            {
                y.AddFieldHandler<EmployeeAllocationSumOperationHandler>();
            }));
    }
}

public class EmployeeAllocationSumOperationHandler : FilterOperationHandler<QueryableFilterContext, Expression>
{
    public override bool CanHandle(ITypeCompletionContext context, IFilterInputTypeDefinition typeDefinition,
        IFilterFieldDefinition fieldDefinition)
    {
        return context.Type is IListFilterInputType &&
               fieldDefinition is FilterOperationFieldDefinition { Id: CustomFilterOperations.Sum };
    }

    public override bool TryHandleEnter(QueryableFilterContext context, IFilterField field, ObjectFieldNode node, [NotNullWhen(true)] out ISyntaxVisitorAction? action)
    {
        var ts = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
        var property = context.GetInstance();

        Expression<Func<ICollection<Allocation>, double>> expression = _ => _
            .Where(_ => _.Start < ts && _.End > ts)
            .Sum(_ => _.HoursPerWeek);
        var invoke = Expression.Invoke(expression, property);
        context.PushInstance(invoke);
        action = SyntaxVisitor.Continue;
        return true;
    }
}

services.AddGraphQLServer()
    .AddQueryType<EmployeeQuery>()
    .AddType<EmployeeType>()
    .AddProjections()
    .AddFiltering()
    .AddConvention<IFilterConvention, CustomFilterConventionExtension>()
    .AddSorting();

然后我尝试编写我的查询

employees (where: {allocations: {sum: {gt: 5}}}) {
  nodes{
    id, name
  }
}

然而，此实现不断抛出异常，因为它无法从 System.Obejct 转换为 System.Generic.IEnumerable。

但是在最终版本中，我希望能够查询的不是常量数字，而是像这样的员工工作时间

employees (where: {allocations: {sum: {gt: workHours}}}) {
  nodes{
    id, name
  }
}

有谁可以协助创建糟糕的 FilterOperation？也许您已经做过类似的事情，或者知道这实际上是不可能的。

如果有人愿意使用它，我已将所有代码放在 GitHub 存储库中 https://github.com/LordLyng/sum-filter-example

Answer 1

我仍然没有找到使用实际聚合的解决方案。但是我找到了一种在数据库级别而不是在 GraphQL 中解决我的特定问题的方法。

我最终向我的实体添加了两个 Prop public bool Available { get; set; }和 public double AvailableHours { get; set; } .添加这些 Prop 后，我为员工实体编辑了我的 EntityTypeConfiguration。在这里我添加了以下几行。请原谅我的 SQL，我远非流利；)

builder.Property(e => e.Available)
    .HasComputedColumnSql("CASE WHEN [WorkHours] > [dbo].[AllocSumForEmployee] ([Id]) THEN CAST(1 AS BIT) ELSE CAST(0 AS BIT) END", stored: false)
    .ValueGeneratedOnAddOrUpdate();
builder.Property(e => e.AvailableHours)
    .HasComputedColumnSql("IIF([WorkHours] - [dbo].[AllocSumForEmployee] ([Id]) > 0, [WorkHours] - [dbo].[AllocSumForEmployee] ([Id]), 0)", stored: false)
    .ValueGeneratedOnAddOrUpdate();

这里发生了几件事 - 首先，我们将计算列添加到我们的数据库架构中。其次，我们添加 ValueGeneratedOnAddOrUpdate方法来确保这些值不是由我们的应用程序设置的，而是留给数据库来处理。

计算列方法适用于单个表中的单个行的上下文，因此默认情况下现在可以查询其他表或行。这正是我解决问题所需要的。

眼尖的观察者可能已经注意到我在我的计算列中使用了一些看起来非常像方法的东西。事实上，仅此而已。

我知道我会根据 Start 进行大量分配查找和 End所以我索引了这些并通过运行 dotnet ef migrations add "<migration name"> 创建了我的迁移.要让魔术发挥作用，剩下要做的就是修改生成的迁移。

编辑后的迁移最终如下所示

protected override void Up(MigrationBuilder migrationBuilder)
{
    migrationBuilder.Sql(@"
        CREATE FUNCTION [dbo].[AllocSumForEmployee] (@id nvarchar(36))
        RETURNS Float
        AS 
        BEGIN
        RETURN 
            (SELECT COALESCE(SUM([HoursPerWeek]), 0) FROM [Allocations]
                WHERE 
                    [Start] < DATEDIFF_BIG(MILLISECOND,'1970-01-01 00:00:00.000', SYSUTCDATETIME()) AND 
                    [End] > DATEDIFF_BIG(MILLISECOND,'1970-01-01 00:00:00.000', SYSUTCDATETIME()) AND
                    [EmployeeId] = @id)
        END
    ");

    migrationBuilder.AddColumn<bool>(
        name: "Available",
        table: "Employees",
        type: "bit",
        nullable: false,
        computedColumnSql: "CASE WHEN [WorkHours] > [dbo].[AllocSumForEmployee] ([Id]) THEN CAST(1 AS BIT) ELSE CAST(0 AS BIT) END",
        stored: false);

    migrationBuilder.AddColumn<double>(
        name: "AvailableHours",
        table: "Employees",
        type: "float",
        nullable: false,
        computedColumnSql: "IIF([WorkHours] - [dbo].[AllocSumForEmployee] ([Id]) > 0, [WorkHours] - [dbo].[AllocSumForEmployee] ([Id]), 0)",
        stored: false);

    migrationBuilder.CreateIndex(
        name: "IX_Allocations_End",
        table: "Allocations",
        column: "End");

    migrationBuilder.CreateIndex(
        name: "IX_Allocations_Start",
        table: "Allocations",
        column: "Start");
}

protected override void Down(MigrationBuilder migrationBuilder)
{
    migrationBuilder.DropIndex(
        name: "IX_Allocations_End",
        table: "Allocations");

    migrationBuilder.DropIndex(
        name: "IX_Allocations_Start",
        table: "Allocations");

    migrationBuilder.DropColumn(
        name: "Available",
        table: "Employees");

    migrationBuilder.DropColumn(
        name: "AvailableHours",
        table: "Employees");

    migrationBuilder.Sql("DROP FUNCTION [dbo].[AllocSumForEmployee]");
}

我添加的“唯一”不是自动生成的东西是 Up 方法的第一部分和 Down 方法的最后部分。有效地在 Up 中注册用户定义的函数并在 Down 中删除该函数。

我的函数返回给定员工的 HoursPerWeek 所有分配的总和，其中 Start(存储为以毫秒为单位的 unix 时间戳)在现在之前(DATEDIFF_BIG(MILLISECOND,'1970-01-01 00:00:00.000', SYSUTCDATETIME()) 是一种在 T-SQL 中获取现在作为以毫秒为单位的 unix 时间戳的方法) ，以及 End 之后的位置(即主动分配)。

这实际上意味着计算属性的计算是在数据库上完成的，而 HotChocolte 一点也不聪明。它允许我做类似的查询

query {
  employees (where: {available: {eq: true}}) {
    nodes {
      id, name, available, availableHours
    }
  }
}

甚至

query {
  employees (where: {availableHours: {gt: 15}}) {
    nodes {
      id, name, available, availableHours
    }
  }
}

希望这可以帮助其他尝试使用聚合解决内联计算的人!

此处提供替代方法的代码 https://github.com/LordLyng/sum-filter-example/tree/computed-prop-on-entity