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

我想从成功队列中获取一个流文件以从中读取count属性，然后我想获取count number流文件并将其放入流文件列表中，然后我将检查

如果流文件列表大小==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处理器？ 你能告诉我如何完成这项任务的其他想法吗