elasticsearch collapse 折叠查询后如何保证总数

我们都知道，使用 collapse 折叠查询，目的就是通过某一字段去重。
假设：from=0，size=600，通过 spu_type 进行折叠，inner_hits_size=5
结果：一般情况下得到的结果少于 600 条
问题：想咨询有没有什么办法可以按照 spu_type 折叠且 inner_hits_size=5，而且最终得到的总数量也为 600 呢？
 
我的想法：
es 目前还不支持直接通过 collapse 来实现，那么是否可以通过一个plugins 来实现，且是需要在截取 topHits 前就将聚合结果聚合出来并返回我要求的总数据量。以及有没有相关 plugins 实现思路？
aggs 性能有问题，数据量大概 6 亿，所以不考虑 aggs 方案。