简单重构一下