之前的功能的一个更好的写法

代码越写越少越好哈 o(╯□╰)o 不过之前的已经push上去了