NIFI：如何传输一定数量的流文件

Question

我想从它获取一个flowFile（从成功队列中）读取count属性，然后我想获取计数数量流文件并将其放入流文件列表中之后我将检查if flowfile list size == count我将全部转移收到的流文件我试过这样的代码：

import org.apache.commons.io.IOUtils
import java.nio.charset.StandardCharsets

def flowFile = session.get()
if (!flowFile) return
int numb=(flowFile.getAttribute("count") as Double).round()
int  m=numb;
session.rollback();
def flowFiles = session.get(m)

if (flowFiles.size()==m){

 //transfer first and other files to success

    session.transfer(flowFiles, REL_SUCCESS)

}else{
   //rollback everything we got from incoming queue
    session.rollback()
}

但是如果我使用wait / notify处理器或routeonattribute处理器，这个逻辑对我不起作用？ 你能告诉我如何完成这项任务吗？