Netty+SpringBoot+FastDFS+Html5实现聊天App详解(二)

Netty+SpringBoot+FastDFS+Html5实现聊天App，项目介绍。

Netty+SpringBoot+FastDFS+Html5实现聊天App， 项目github链接 。

本章完整代码

本节主要讲解聊天App PigChat中关于用户信息处理，以及文件服务器FastDFS的相关操作。

包含以下内容：

（1）注册与登录功能

（2）文件服务器的配置

（3）上传用户头像

（4）设置用户昵称

（5）用户二维码的生成与上传

注册与登录功能

自定义一个工具类IMoocJSONResult，是后端响应前端的数据结构。

包含下面三个属性：

// 响应业务状态
    private Integer status;
    // 响应消息
    private String msg;
    // 响应中的数据
    private Object data;

提供错误响应与正常响应的方法：

public static IMoocJSONResult ok(Object data) {
        return new IMoocJSONResult(data);
    }

    public static IMoocJSONResult ok() {
        return new IMoocJSONResult(null);
    }
    
    public static IMoocJSONResult errorMsg(String msg) {
        return new IMoocJSONResult(500, msg, null);
    }
    
    public static IMoocJSONResult errorMap(Object data) {
        return new IMoocJSONResult(501, "error", data);
    }
    
    public static IMoocJSONResult errorTokenMsg(String msg) {
        return new IMoocJSONResult(502, msg, null);
    }
    
    public static IMoocJSONResult errorException(String msg) {
        return new IMoocJSONResult(555, msg, null);
    }

根据数据库所建的表创建对应的pojo包与mapper包， 数据库建表详情

创建UserController方法，写入进行注册于登录处理的registOrLogin接口。

【0】前端传入Users对象，首先判断前端传入的Users对象是否为空。

【1】然后通过userService的queryUsernameIsExist方法根据传入的用户名在数据库中进行查询。

【1.1】若该用户存在则进行登录，通过userService的queryUserForLogin方法判断前端传入的用户名与密码试凑匹配，若匹配则登录成功，否则登录失败。

【1.2】若该用户不存在则记性注册，根据前端传入的信息构建Users对象，通过userService的saveUser将其保存入数据库中。

【2】最后构造UsersVO对象，返回给前端。

注意：密码需要使用MD5工具类进行加密后再保存到数据库中。

/**
     * @Description: 用户注册/登录
     */
    @PostMapping("/registOrLogin")
    public IMoocJSONResult registOrLogin(@RequestBody Users user) throws Exception {
        
        // 0. 判断用户名和密码不能为空
        if (StringUtils.isBlank(user.getUsername()) 
                || StringUtils.isBlank(user.getPassword())) {
            return IMoocJSONResult.errorMsg("用户名或密码不能为空...");
        }
        
        // 1. 判断用户名是否存在，如果存在就登录，如果不存在则注册
        boolean usernameIsExist = userService.queryUsernameIsExist(user.getUsername());
        Users userResult = null;
        if (usernameIsExist) {
            // 1.1 登录
            userResult = userService.queryUserForLogin(user.getUsername(), 
                                    MD5Utils.getMD5Str(user.getPassword()));
            if (userResult == null) {
                return IMoocJSONResult.errorMsg("用户名或密码不正确..."); 
            }
        } else {
            // 1.2 注册
            user.setNickname(user.getUsername());
            user.setFaceImage("");
            user.setFaceImageBig("");
            user.setPassword(MD5Utils.getMD5Str(user.getPassword()));
            userResult = userService.saveUser(user);
        }
        // 2.构造UsersVO对象
        UsersVO userVO = new UsersVO();
        BeanUtils.copyProperties(userResult, userVO);
        
        return IMoocJSONResult.ok(userVO);
    }

文件服务器的配置

在linux中配置好文件服务器FastDFS后，需要在项目中添加如下配置：

（1）在Application同目录下创建FastdfsImporter

package com.imooc;

import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.EnableMBeanExport;
import org.springframework.context.annotation.Import;
import org.springframework.jmx.support.RegistrationPolicy;

import com.github.tobato.fastdfs.FdfsClientConfig;

/**
 * 导入FastDFS-Client组件
 * 
 * @author tobato
 *
 */
@Configuration
@Import(FdfsClientConfig.class)
// 解决jmx重复注册bean的问题
@EnableMBeanExport(registration = RegistrationPolicy.IGNORE_EXISTING)
public class FastdfsImporter {
    // 导入依赖组件
}

（2）在application.properties中添加如下配置：

fdfs.soTimeout=1501
fdfs.connectTimeout=601
fdfs.thumbImage.width=80
fdfs.thumbImage.height=80
# 192.168.1.70为Linux虚拟机的ip地址
fdfs.trackerList[0]=192.168.1.70:22122

启动服务命令：

/usr/bin/fdfs_trackerd /etc/fdfs/tracker.conf
/usr/bin/fdfs_storaged /etc/fdfs/storage.conf

#查看服务启动情况(23000/22122端口)
netstat -lnp |grep fdfs
cd /usr/local/nginx/sbin
./nginx

上传用户头像

在UserController中添加上传用户头像的uploadFaceBase64接口。

【1】前端传入UserBO对象，首先获取前端传来的base64字符串，并通过文件工具类FileUtils的base64ToFile方法将其转换成文件对象保存在本地。

【2】将文件对象转换成MultipartFile，并通过fastDFSClient的uploadBase64方法将其上传到文件服务器fastDFS中，打印出服务器返回的路径，我们可以通过这个路径访问这张图片。

【3】对返回的路径进行切割后得到缩略图的路径。

【4】更新数据库中用户头像信息。

/**
     * @Description: 上传用户头像
     */
    @PostMapping("/uploadFaceBase64")
    public IMoocJSONResult uploadFaceBase64(@RequestBody UsersBO userBO) throws Exception {
        
        // 1. 获取前端传过来的base64字符串, 然后转换为文件对象再上传
        String base64Data = userBO.getFaceData();
            // 在本地存储图片的路径  
        String userFacePath = "C://" + userBO.getUserId() + "userface64.png";
        FileUtils.base64ToFile(userFacePath, base64Data);
        
        // 2.上传文件到fastdfs
        MultipartFile faceFile = FileUtils.fileToMultipart(userFacePath);
        String url = fastDFSClient.uploadBase64(faceFile);
        System.out.println(url);
        
//        "dhawuidhwaiuh3u89u98432.png"
//        "dhawuidhwaiuh3u89u98432_80x80.png"
        
        // 3.获取缩略图的url
        String thump = "_80x80.";
        String arr[] = url.split("//.");
        String thumpImgUrl = arr[0] + thump + arr[1];
        
        // 4.更新用户头像
        Users user = new Users();
        user.setId(userBO.getUserId());
        user.setFaceImage(thumpImgUrl);
        user.setFaceImageBig(url);
        
        Users result = userService.updateUserInfo(user);
        
        return IMoocJSONResult.ok(result);
    }

设置用户昵称

在UserController添加设置用户昵称的setNickname接口。

/**
     * @Description: 设置用户昵称
     */
    @PostMapping("/setNickname")
    public IMoocJSONResult setNickname(@RequestBody UsersBO userBO) throws Exception {
        
        Users user = new Users();
        user.setId(userBO.getUserId());
        user.setNickname(userBO.getNickname());
        
        Users result = userService.updateUserInfo(user);
        
        return IMoocJSONResult.ok(result);
    }

用户二维码的生成与上传

在UserServiceImpl中引入相关工具类与组件

//二维码工具类
     @Autowired
    private QRCodeUtils qrCodeUtils;
    //上传文件到fsatDFS需要的组件
    @Autowired
    private FastDFSClient fastDFSClient;

在UserServiceImpl保存用户信息的saveUser方法中需要为每一个用户生成一个唯一的二维码。

【1】通过二维码工具类qrCodeUtils的createQRCode方法为每个用户生成一个唯一的二维码，第一个参数为生成的二维码存储的路径，第二个参数为二维码中保存的信息，然后将文件转成MultipartFile对象，方便上传操作。

【2】通过fastDFSClient的uploadQRCode方法将二维码图片上传到文件服务器中。

@Transactional(propagation = Propagation.REQUIRED)
    @Override
    public Users saveUser(Users user) {
        //生成唯一的id
        String userId = sid.nextShort();
        
        // 1.为每个用户生成一个唯一的二维码
            //本地用来存储生成的二维码图片的路径  
        String qrCodePath = "C://user" + userId + "qrcode.png";
        // 扫描二维码后得到的信息:zhuzhu_qrcode:[username]
        qrCodeUtils.createQRCode(qrCodePath, "zhuzhu_qrcode:" + user.getUsername());
        MultipartFile qrCodeFile = FileUtils.fileToMultipart(qrCodePath);
        
           //2.上传文件
        String qrCodeUrl = "";
        try {
            qrCodeUrl = fastDFSClient.uploadQRCode(qrCodeFile);
        } catch (IOException e) {
            e.printStackTrace();
        }
        user.setQrcode(qrCodeUrl);
        
        user.setId(userId);
        userMapper.insert(user);
        
        return user;
    }