JVM-Sandbox-Repeater中Lambda表达式操作MyBatis时Mock数据返回异常问题解析

问题背景

在使用JVM-Sandbox-Repeater进行流量回放时，开发人员遇到了一个特殊场景：当代码中使用Lambda表达式结合PageHelper进行MyBatis分页查询时，虽然MyBatis的Mock数据配置成功，但实际流程却中断，Mock数据未能正常返回。

典型代码示例如下：

PageHelper.startPage(1,15).doSelectPageInfo(() -> dao.select(arg1,arg2));

问题根源分析

经过深入排查，发现问题的根本原因在于PageHelper的实现机制。PageHelper内部使用了线程变量(ThreadLocal)来存储分页信息，包括总记录数(total)等关键数据。在流量回放过程中，虽然MyBatis的DAO层方法被成功Mock，但PageHelper相关的线程变量未被正确初始化，导致total值为0，进而使得整个分页流程提前终止。

解决方案

要解决这个问题，需要在JVM-Sandbox-Repeater中增加对PageHelper关键方法的录制和回放支持。具体实现方案如下：

1. 创建一个专门的PageHelper插件，用于拦截和录制PageHelper的核心方法调用
2. 重点捕获影响分页结果的关键方法，如getTotal、getPageNum等
3. 在回放阶段正确恢复这些方法的返回值

实现代码示例：

@MetaInfServices(InvokePlugin.class)
public class PageHelperPlugin extends AbstractInvokePluginAdapter {

    @Override
    protected List<EnhanceModel> getEnhanceModelList() {
        EnhanceModel enhanceModel = EnhanceModel.builder()
                .classPattern("com.github.pagehelper.Page")
                .methodPatterns(
                        MethodPattern.transform("getTotal", "getCountColumn", 
                                "getDialectClass", "getEndRow", "getOrderBy", 
                                "getPageNum", "getPages", "getPageSize", 
                                "getPageSizeZero", "getReasonable", "getStartRow"))
                .watchTypes(EnhanceModel.RECORD_EVENT_TYPE).build();

        return CollectionUtil.newArrayList(enhanceModel);
    }

    @Override
    protected InvocationProcessor getInvocationProcessor() {
        return new PageHelperInvocationProcessor(getType());
    }

    @Override
    public InvokeType getType() {
        return InvokeType.PAGE_HELPER;
    }

    @Override
    public boolean isEntrance() {
        return false;
    }
}

技术要点

1. 线程变量问题：在流量录制和回放过程中，需要特别注意线程局部变量(ThreadLocal)的处理，这类变量不会自动跨线程传递。

2. Lambda表达式处理：Lambda表达式在JVM层面会被编译为匿名类，JVM-Sandbox-Repeater需要能够正确处理这种调用方式。

3. 分页组件集成：对于常用的第三方组件如PageHelper，需要专门适配以确保其内部状态能够被正确录制和回放。

最佳实践建议

1. 在项目中使用JVM-Sandbox-Repeater时，应提前识别所有使用的第三方组件，特别是那些依赖线程变量的组件。

2. 对于复杂的调用链，特别是涉及Lambda表达式和函数式编程的场景，建议进行充分的测试验证。

3. 可以建立常见组件的插件库，如MyBatis、PageHelper等，便于团队共享使用。

4. 在录制阶段，注意检查是否完整捕获了所有必要的调用信息，包括看似"不重要"的getter方法。

通过以上分析和解决方案，我们成功解决了Lambda表达式结合PageHelper进行MyBatis操作时Mock数据无法正常返回的问题。这为类似场景下的流量回放提供了可靠的技术保障。