代码重构实战

背景

用户头像上传功能中，服务端上传流程如下：

1. base64解码成字符串
2. 图片写入服务器本地临时目录
3. 上传到阿里云OSS/七牛云存储
4. 图片审核

代码如下

type Parms struct {
    Head string //客户端发送base64字符串
}
func Upload(p Params){
    if p.Head == ""{
        return errors.New("img is empty")
    }
    
    str,err := Base64Decode(p.Head)
    if err != nil{
        return err
    }
    
    filename := makeUniqueName()
    if err := FileWrite(filename,str);err != nil{
         return err
    }
    
    err := UploadAliOss(filename)
    if err != nil{
        return err
    }
    
    err := VerifyImg(filename)
    if err != nil{
        return err
    }

}

问题

以前多个项目通过jenkins发布不同的上线指令，每新建一个项目，运维需要单独写一套shell命令，无法复用之前的，维护成本比较高。后来我们升级部署方式，只需要修改项目部署目录配置文件，具体后续再介绍。

一直以来稳定运行，周末收到反馈出现问题，用户无法上传头像。

• 检查代码的git历史提交记录，近期没有修改代码，排除代码bug
• 报错提示只在调用FileWrite和UploadAliOss两个方法时出现
• 在打包镜像的时候没有在容器中创建临时目录，所以无法写入文件，用户头像无法上传

临时解决

手动在容器中创建一个临时目录，保存图片文件。

完全解决

我认为，图片文件全部保存在第三方对象存储中，服务器保存一份是完全没必要的，只会占用磁盘空间。

我们可以创建Reader对象把图片文件读取到缓冲区，再上传到第三方对象存储。

代码如下

func Upload(p Params){

    if p.Head == ""{
        return errors.New("img is empty")
    }
    
    str,err := Base64Decode(p.Head)
    if err != nil{
        return err
    }
    
    filename := makeUniqueName()
    buf := bytes.NewReader([]byte(str))
    
    err := UploadAliOssBuff(filename,buf)
    if err != nil{
        return err
    }
    
    err := VerifyImg(filename)
    if err != nil{
        return err
    }

}

总结

在接手维护老项目时候，避免不了遇到坑。除了吐槽前人写的糟糕外，我们可以尽我所能提升代码维护质量，让后来者更容易维护。

一片空地，可能开始只有几片垃圾。如果一直不清理，你来丢，他也丢，很快就会变成垃圾堆。

微信公众号