结组作业5

今天写调用chat接口的service的实现层代码

package org.test.tongyuzhe.Service;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.test.tongyuzhe.Mapper.ChatMapper;
import org.test.tongyuzhe.Pojo.Chat;

import java.util.List;

@Service
public class ChatService {
    @Autowired
    ChatMapper chatMapper;

    /**
     * 通过userId来获取当前用户的历史问问题记录
     * @param userId 用户id
     * @return 问题和回答列表信息
     */
    public List<Chat> getMessage(Integer userId) {
        List<Chat> list=chatMapper.getMessage(userId);
        return list;
    }

    /**
     * 添加message（也就是保存用户提出的问题和答案等信息）
     * @param chat
     */
    public void insertMessage(Chat chat) {
       chatMapper.insertMessage(chat);
    }
}

主要是一些方法的定义

还写了一部分在用户注册和登录时的一部分代码

package org.test.tongyuzhe.Controller;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.test.tongyuzhe.Pojo.Result;
import org.test.tongyuzhe.Pojo.User;
import org.test.tongyuzhe.Service.UserService;
import org.test.tongyuzhe.utils.JwtUtil;
import org.test.tongyuzhe.utils.Md5Util;

import java.util.HashMap;
import java.util.Map;

@RestController
@Validated
@RequestMapping("/user")
public class UserController {

    @Autowired
    private UserService userService;

    @PostMapping("/login")
    public Result<?> login(@RequestBody User user){

        //查询用户
        User loginUser = userService.findByUserName(user.getUsername());

        //判断用户是否存在
        if (loginUser == null){
            return Result.error("用户名错误");
        }
        //判断密码是否正确 loginUser对象中password是密文
        if(Md5Util.getMD5String(user.getPassword()).equals(loginUser.getPassword())){
            //登录成功,生成Token
            Map<String,Object> claims = new HashMap<>();
            claims.put("id",loginUser.getUserId());
            claims.put("username",loginUser.getUsername());
            String token = JwtUtil.genToken(claims);
            // System.out.println(token);
            return Result.success(String.valueOf(loginUser.getRole()),token);
        }

        return Result.error("密码错误");
    }


    @PostMapping("/register")
    public Result<?> childRegister(@RequestBody User user){
        //查询用户
        User userQuery = userService.findByUserName(user.getUsername());
        if (userQuery == null){
            //注册
            userService.register(user);
            return Result.success();
        }else {
            //占用
            return Result.error("用户名已被注册");
        }

    }

}