修改代码，支持直接线上测试