Examples with ResultSetLooping org.nutz.dao.pager.ResultSetLooping used on opensource projects

Example 6 with ResultSetLooping

use of org.nutz.dao.pager.ResultSetLooping in project nutz by nutzam.

the class PojoQueryRecordCallback method invoke.

public Object invoke(Connection conn, ResultSet rs, Pojo pojo, Statement stmt) throws SQLException {
    ResultSetLooping ing = new ResultSetLooping() {

        protected boolean createObject(int index, ResultSet rs, SqlContext context, int rowCount) {
            list.add(Record.create(rs));
            return true;
        }
    };
    ing.doLoop(rs, pojo.getContext());
    return ing.getList();
}

Example 7 with ResultSetLooping

use of org.nutz.dao.pager.ResultSetLooping in project nutz by nutzam.

the class PojoEachEntityCallback method invoke.

@SuppressWarnings("unchecked")
public Object invoke(Connection conn, ResultSet rs, Pojo pojo, Statement stmt) throws SQLException {
    // 得到回调
    final Each<Object> each = pojo.getContext().attr(Each.class);
    // 没有回调，什么都不用执行了
    if (null == each)
        return null;
    // 开始执行
    final Entity<?> en = pojo.getEntity();
    ResultSetLooping ing = new ResultSetLooping() {

        protected boolean createObject(int index, ResultSet rs, SqlContext context, int rowCount) {
            Object obj = en.getObject(rs, context.getFieldMatcher());
            try {
                each.invoke(index, obj, rowCount);
            } catch (ContinueLoop e) {
            } catch (LoopException e) {
                throw Lang.wrapThrow(e);
            }
            return false;
        }
    };
    try {
        // 循环开始
        if (each instanceof Loop)
            if (!((Loop<?>) each).begin())
                return 0;
        // 循环中
        ing.doLoop(rs, pojo.getContext());
        // 循环结束
        if (each instanceof Loop)
            ((Loop<?>) each).end();
    } catch (ExitLoop e) {
    } catch (LoopException e) {
        SQLException e2 = new SQLException();
        e2.initCause(e.getCause());
        throw e2;
    }
    // 返回数量
    return ing.getIndex() + 1;
}