NestJS+TypeORM+PostgreSQL下，如何基于动态计算的用户「时长」过滤用户表？

针对你的需求，我整理了两种实用的实现思路，结合你已经写好的TS工作日计算函数，我们可以从数据库层和应用层两个方向来落地：

方案一：数据库层面实现（推荐，大场景更高效） #

如果你的用户表数据量不小，优先考虑在数据库层完成计算和过滤，避免把大量数据拉到应用内存处理。

步骤1：在PostgreSQL中创建自定义工作日计算函数 #

先写一个PostgreSQL函数，用来计算两个日期之间的有效工作日（排除周末和节假日），逻辑和你TS函数一致：

CREATE OR REPLACE FUNCTION calculate_working_days(start_date DATE, end_date DATE)
RETURNS INTEGER AS $$
DECLARE
    total_working_days INTEGER;
    holiday_count INTEGER;
BEGIN
    -- 计算两个日期之间的总工作日（排除周六、周日）
    total_working_days := (
        SELECT COUNT(*)
        FROM generate_series(start_date, end_date, '1 day'::INTERVAL) AS days
        WHERE EXTRACT(DOW FROM days) NOT IN (0, 6) -- 0代表周日，6代表周六
    );

    -- 减去这段时间内的节假日数量（避免重复扣除周末节假日）
    SELECT COUNT(*) INTO holiday_count
    FROM holidays
    WHERE holiday_date BETWEEN start_date AND end_date
    AND EXTRACT(DOW FROM holiday_date) NOT IN (0, 6);

    RETURN total_working_days - holiday_count;
END;
$$ LANGUAGE plpgsql;

注意：如果你的holidays表只存工作日的节假日，可以去掉最后一行的周末判断。

步骤2：在TypeORM中用QueryBuilder实现动态过滤 #

在你的UserRepository里，通过QueryBuilder结合自定义函数，动态判断用户状态来计算时长，再添加过滤条件：

import { Repository, EntityRepository } from 'typeorm';
import { User } from './user.entity';

@EntityRepository(User)
export class UserRepository extends Repository<User> {
  async filterUsersByAge(minAge?: number, maxAge?: number) {
    const queryBuilder = this.createQueryBuilder('user');

    // 动态生成用户时长的计算逻辑：根据状态选择结束日期
    const ageCalculation = `
      CASE
        WHEN user.status = 'Active' THEN calculate_working_days(user.createdDate, CURRENT_DATE)
        WHEN user.status = 'Deactivated' THEN calculate_working_days(user.createdDate, user.deactivatedDate)
        ELSE 0
      END AS user_age
    `;

    // 添加计算字段到查询结果
    queryBuilder.addSelect(ageCalculation, 'user_age');

    // 应用过滤条件（用HAVING而不是WHERE，因为是对查询生成的别名过滤）
    if (minAge) {
      queryBuilder.having('user_age >= :minAge', { minAge });
    }
    if (maxAge) {
      queryBuilder.having('user_age <= :maxAge', { maxAge });
    }

    // 必须GROUP BY用户主键，确保每个用户的计算结果唯一
    queryBuilder.groupBy('user.id');

    return queryBuilder.getMany();
  }
}

这样查询时，数据库会直接返回符合时长条件的用户，性能比应用层过滤好很多。

方案二：应用层过滤（适合小数据量场景） #

如果你的用户表数据量不大，也可以直接用你已经写好的TS函数来处理，不用修改数据库：

async filterUsersByAge(minAge?: number, maxAge?: number) {
  // 先拉取所有用户（也可以先加基础过滤，比如按status筛选）
  const users = await this.find();

  // 遍历用户，计算时长并过滤
  return users.filter(user => {
    let endDate: Date;
    const startDate = user.createdDate;

    // 根据状态选择结束日期
    if (user.status === 'Active') {
      endDate = new Date();
    } else if (user.status === 'Deactivated') {
      endDate = user.deactivatedDate;
    } else {
      return false; // 其他状态按需求处理，比如直接排除
    }

    // 调用你已有的TS工作日计算函数
    const userAge = yourWorkingDaysCalculator(startDate, endDate);

    // 判断是否符合过滤条件
    const meetsMin = minAge ? userAge >= minAge : true;
    const meetsMax = maxAge ? userAge <= maxAge : true;

    return meetsMin && meetsMax;
  });
}

这个方案的优点是快速落地，不用改数据库，但数据量大的时候会占用较多应用内存，性能会下降。

一些注意事项 #

• 索引优化：给holidays.holiday_date加索引，能加快数据库函数里的节假日计数查询。
• 时区一致性：确保数据库和NestJS应用的时区设置一致，避免日期计算出现偏差。
• 测试验证：一定要测试不同场景的用户，比如刚创建的Active用户、跨多个节假日的Deactivated用户，确保时长计算正确。

备注：内容来源于stack exchange，提问作者Rajeesh