修改 message 方法
整理一下代码