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

红太狼 2021-03-23 13:29 313阅读 0赞

Netty+SpringBoot+FastDFS+Html5实现聊天App，[项目github链接][github]。  
[本章完整代码][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包，[数据库建表详情][github]

创建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;
        }

[github]: https://github.com/ShimmerPig/PigChat