修正为一个函数来实现