From a43dd7524614070d8ca93defaac13ed539a66294 Mon Sep 17 00:00:00 2001 From: Cool <747682928@qq.com> Date: Fri, 1 Dec 2023 20:57:59 +0800 Subject: [PATCH] =?UTF-8?q?2023.12.1=E7=89=88=E6=9C=AC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../chat/controller/LoginController.java | 9 ++++++++- .../WebSocketInterceptorController.java | 19 +++++++++++++++++++ src/main/java/com/example/chat/dao/Login.java | 10 ++++++++++ .../java/com/example/chat/dao/UserMapper.java | 2 ++ .../example/chat/jwt/InterceptorConfig.java | 2 +- .../com/example/chat/service/HeadOss.java | 2 +- src/main/resources/mapper/LoginMapper.xml | 5 +++++ 7 files changed, 46 insertions(+), 3 deletions(-) create mode 100644 src/main/java/com/example/chat/controller/WebSocketInterceptorController.java diff --git a/src/main/java/com/example/chat/controller/LoginController.java b/src/main/java/com/example/chat/controller/LoginController.java index 06fae90..c0377c4 100644 --- a/src/main/java/com/example/chat/controller/LoginController.java +++ b/src/main/java/com/example/chat/controller/LoginController.java @@ -1,6 +1,7 @@ package com.example.chat.controller; import com.example.chat.ChatApplication; +import com.example.chat.dao.UserMapper; import com.example.chat.dao.chat.ChatMapper; import com.example.chat.entity.Email; import com.example.chat.entity.Result; @@ -25,7 +26,8 @@ public class LoginController { CodeCheck codeCheck; @Resource MailUtil mailUtil; - + @Resource + UserMapper userMapper; @PostMapping("/login") @@ -63,4 +65,9 @@ public class LoginController { System.out.println(email.getEmail()); mailUtil.sendMail(email.getEmail()); } + + @PostMapping("/getHead") + public User getHead(@RequestBody User user) throws IOException { + return userMapper.getHead(user.getAccount()); + } } diff --git a/src/main/java/com/example/chat/controller/WebSocketInterceptorController.java b/src/main/java/com/example/chat/controller/WebSocketInterceptorController.java new file mode 100644 index 0000000..6033f7e --- /dev/null +++ b/src/main/java/com/example/chat/controller/WebSocketInterceptorController.java @@ -0,0 +1,19 @@ +package com.example.chat.controller; + +import lombok.extern.slf4j.Slf4j; +import org.springframework.web.bind.annotation.CrossOrigin; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.RestController; + +@CrossOrigin +@Slf4j +@RestController +public class WebSocketInterceptorController { + + + @GetMapping("/interceptor") + public void interceptor() { + log.info("ws进来咯"); + } + +} diff --git a/src/main/java/com/example/chat/dao/Login.java b/src/main/java/com/example/chat/dao/Login.java index e787e47..923b838 100644 --- a/src/main/java/com/example/chat/dao/Login.java +++ b/src/main/java/com/example/chat/dao/Login.java @@ -50,7 +50,17 @@ public class Login implements UserMapper { public void modifyHead(String pic, String account) throws IOException { SqlSession sqlSession=MybatisSingleton.getSqlSessionFactory().openSession(); sqlSession.update("dao.Login.updateHead",new User("","",account,pic,"")); + sqlSession.commit(); sqlSession.close(); } + @Override + public User getHead(String account) throws IOException { + + SqlSession sqlSession=MybatisSingleton.getSqlSessionFactory().openSession(); + User user=sqlSession.selectOne("dao.Login.getHead",account); + sqlSession.close(); + return user; + } + } diff --git a/src/main/java/com/example/chat/dao/UserMapper.java b/src/main/java/com/example/chat/dao/UserMapper.java index 9fda91f..c9bb28c 100644 --- a/src/main/java/com/example/chat/dao/UserMapper.java +++ b/src/main/java/com/example/chat/dao/UserMapper.java @@ -16,4 +16,6 @@ public interface UserMapper { public abstract User emailCheck(User user) throws IOException; public abstract void modifyHead(String Pic,String username) throws IOException; + public abstract User getHead(String account)throws IOException; + } diff --git a/src/main/java/com/example/chat/jwt/InterceptorConfig.java b/src/main/java/com/example/chat/jwt/InterceptorConfig.java index 48d97ab..2536e1f 100644 --- a/src/main/java/com/example/chat/jwt/InterceptorConfig.java +++ b/src/main/java/com/example/chat/jwt/InterceptorConfig.java @@ -24,7 +24,7 @@ public class InterceptorConfig implements WebMvcConfigurer { excludePath.add("/sendCode"); excludePath.add("/register");//注册 excludePath.add("/register"); - excludePath.add("/chat/get"); + registry.addInterceptor(tokenInterceptor)//注册拦截器 .addPathPatterns("/**")//拦截所有请求 .excludePathPatterns(excludePath);//添加拦截白名单 diff --git a/src/main/java/com/example/chat/service/HeadOss.java b/src/main/java/com/example/chat/service/HeadOss.java index 1c92c84..186cea6 100644 --- a/src/main/java/com/example/chat/service/HeadOss.java +++ b/src/main/java/com/example/chat/service/HeadOss.java @@ -43,7 +43,7 @@ public class HeadOss { } file.transferTo(targetFile); log.info("成功添加"); - String head=username+"/"+fileName; + String head="https://www.oss.coollh.cn/"+username+"/"+fileName; userMapper.modifyHead(head,username); Message message=new Message(username,head,"",""); chatMapper.updateMessage(message); diff --git a/src/main/resources/mapper/LoginMapper.xml b/src/main/resources/mapper/LoginMapper.xml index 52257d8..f987615 100644 --- a/src/main/resources/mapper/LoginMapper.xml +++ b/src/main/resources/mapper/LoginMapper.xml @@ -31,4 +31,9 @@ set pic=#{pic} where account = #{account} +