添加登录

pom.xml

@@ -5,7 +5,7 @@
 	<parent>
 		<groupId>org.springframework.boot</groupId>
 		<artifactId>spring-boot-starter-parent</artifactId>
-		<version>3.3.1</version>
+		<version>2.7.16</version>
 		<relativePath/> <!-- lookup parent from repository -->
 	</parent>
 	<groupId>com.example</groupId>
@@ -34,12 +34,25 @@
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter</artifactId>
 		</dependency>
-
+		<dependency>
+			<groupId>com.auth0</groupId>
+			<artifactId>java-jwt</artifactId>
+			<version>3.8.3</version>
+		</dependency>
+		<dependency>
+			<groupId>org.mybatis.spring.boot</groupId>
+			<artifactId>mybatis-spring-boot-starter</artifactId>
+			<version>2.3.0</version>
+		</dependency>
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter-test</artifactId>
 			<scope>test</scope>
 		</dependency>
+		<dependency>
+			<groupId>org.projectlombok</groupId>
+			<artifactId>lombok</artifactId>
+		</dependency>
 		<!--        Json-->
 		<dependency>
 			<groupId>com.alibaba</groupId>
@@ -51,6 +64,15 @@
 			<artifactId>httpclient</artifactId>
 			<version>4.5.6</version>
 		</dependency>
+		<dependency>
+			<groupId>mysql</groupId>
+			<artifactId>mysql-connector-java</artifactId>
+			<version>8.0.33</version>
+		</dependency>
+		<dependency>
+			<groupId>org.springframework.boot</groupId>
+			<artifactId>spring-boot-starter-data-redis</artifactId>
+		</dependency>
 		<dependency>
 			<groupId>org.springframework.boot</groupId>
 			<artifactId>spring-boot-starter-web</artifactId>

src/main/java/com/example/aitest/controller/ApiController.java

@@ -1,24 +1,14 @@
 package com.example.aitest.controller;
 
-import com.alibaba.fastjson.JSON;
+
 import com.alibaba.fastjson.JSONObject;
 import com.example.aitest.dto.ApiDto;
 import com.example.aitest.service.ApiService;
-import jakarta.annotation.Resource;
-import org.apache.http.client.methods.CloseableHttpResponse;
-import org.apache.http.client.methods.HttpPost;
-import org.apache.http.entity.ContentType;
-import org.apache.http.entity.StringEntity;
-import org.apache.http.impl.client.CloseableHttpClient;
-import org.apache.http.impl.client.HttpClients;
-import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.CrossOrigin;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
-
+import javax.annotation.Resource;
-import java.io.File;
-import java.io.FileOutputStream;
 import java.io.IOException;
 import java.nio.file.Files;
 import java.nio.file.Path;

src/main/java/com/example/aitest/controller/UserController.java

@@ -0,0 +1,28 @@
+package com.example.aitest.controller;
+
+import com.example.aitest.pojo.User;
+import com.example.aitest.service.UserService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RestController;
+
+import javax.annotation.Resource;
+import java.security.NoSuchAlgorithmException;
+
+@RestController
+public class UserController {
+
+    @Autowired
+    public UserService userService;
+    @PostMapping("/login")
+    public String login(@RequestBody User user) throws NoSuchAlgorithmException {
+        return userService.login(user);
+    }
+
+    @PostMapping("/register")
+    public String register(@RequestBody User user) throws NoSuchAlgorithmException {
+        return userService.register(user);
+    }
+
+}

src/main/java/com/example/aitest/dao/redis/CodeRedis.java

@@ -0,0 +1,32 @@
+package com.example.aitest.dao.redis;
+
+import org.springframework.data.redis.core.StringRedisTemplate;
+import org.springframework.data.redis.core.ValueOperations;
+import org.springframework.stereotype.Repository;
+
+import javax.annotation.Resource;
+import java.util.concurrent.TimeUnit;
+
+@Repository
+public class CodeRedis implements RedisUtil {
+    @Resource
+    StringRedisTemplate stringRedisTemplate;
+
+    @Override
+    public void addRedis(String account, String code) {
+        if (account != null && code != null) {
+            ValueOperations<String, String> valueOperations = stringRedisTemplate.opsForValue();
+            valueOperations.set(account, code, 60, TimeUnit.SECONDS);
+        }
+    }
+
+    @Override
+    public String getRedis(String account) {
+        return stringRedisTemplate.opsForValue().get(account);
+    }
+
+    @Override
+    public void delRedis(String account) {
+
+    }
+}

src/main/java/com/example/aitest/dao/redis/RedisUtil.java

@@ -0,0 +1,13 @@
+package com.example.aitest.dao.redis;
+
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface RedisUtil {
+    public abstract void addRedis(String key, String value);
+
+    public abstract void delRedis(String key);
+
+    public abstract String getRedis(String key);
+
+}

src/main/java/com/example/aitest/dao/redis/TokenRedis.java

@@ -0,0 +1,31 @@
+package com.example.aitest.dao.redis;
+
+import org.springframework.data.redis.core.StringRedisTemplate;
+import org.springframework.data.redis.core.ValueOperations;
+import org.springframework.stereotype.Repository;
+
+import javax.annotation.Resource;
+import java.util.concurrent.TimeUnit;
+
+@Repository
+public class TokenRedis implements RedisUtil {
+
+    @Resource
+    StringRedisTemplate stringRedisTemplate;
+
+    @Override
+    public void addRedis(String key, String value) {
+        ValueOperations<String, String> valueOperations = stringRedisTemplate.opsForValue();
+        valueOperations.set(key, value, 60, TimeUnit.MINUTES);
+    }
+
+    @Override
+    public void delRedis(String key) {
+
+    }
+
+    @Override
+    public String getRedis(String key) {
+        return stringRedisTemplate.opsForValue().get(key);
+    }
+}

src/main/java/com/example/aitest/jwt/InterceptorConfig.java

@@ -0,0 +1,41 @@
+package com.example.aitest.jwt;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.servlet.config.annotation.CorsRegistry;
+import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
+import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
+
+import java.util.ArrayList;
+
+@Configuration
+public class InterceptorConfig implements WebMvcConfigurer {
+    TokenInterceptor tokenInterceptor;
+
+    public InterceptorConfig(TokenInterceptor tokenInterceptor) {//构造函数
+        this.tokenInterceptor = tokenInterceptor;
+    }
+
+    @Override
+    public void addInterceptors(InterceptorRegistry registry) {//配置拦截器
+        System.out.println("拦截器被配置了");
+        ArrayList<String> excludePath = new ArrayList<>();
+        excludePath.add("/login");//登录
+        excludePath.add("/checkCode");
+        excludePath.add("/sendCode");
+        excludePath.add("/register");//注册
+        excludePath.add("/register");
+        excludePath.add("/index/modifyHead");
+        registry.addInterceptor(tokenInterceptor)//注册拦截器
+                .addPathPatterns("/**")//拦截所有请求
+                .excludePathPatterns(excludePath);//添加拦截白名单
+        WebMvcConfigurer.super.addInterceptors(registry);//调用父接口
+    }
+
+    @Override
+    public void addCorsMappings(CorsRegistry registry) {
+        registry.addMapping("/**").allowCredentials(true)//允许携带cookie
+                .allowedMethods("GET", "POST", "DELETE", "PUT", "PATCH", "OPTIONS", "HEAD")//允许访问的方法
+                .allowedOriginPatterns("*")//允许的跨域访问地址
+                .maxAge(3600 * 24);//options缓存时间
+    }
+}

src/main/java/com/example/aitest/jwt/JWTUtil.java

@@ -0,0 +1,34 @@
+package com.example.aitest.jwt;
+
+import com.auth0.jwt.JWT;
+import com.auth0.jwt.JWTVerifier;
+import com.auth0.jwt.algorithms.Algorithm;
+import com.auth0.jwt.interfaces.DecodedJWT;
+
+
+public class JWTUtil {
+    private static final String tokenPassword = "uziCjb";
+
+    public static String sign(String username) {//用用户名作为被加密的对象
+        String token;
+
+        token = JWT.create()//生成jwt令牌，加密过程
+                .withIssuer("llh")
+                .withClaim("username", username)
+                .sign(Algorithm.HMAC256(tokenPassword));
+        return token;//返回加密后的token
+    }
+
+    public static String verify(String token) {
+        JWTVerifier jwtVerifier = JWT.require(Algorithm.HMAC256(tokenPassword)).withIssuer("llh").build();//构建一个jwt解码器
+        DecodedJWT jwtToken = jwtVerifier.verify(token);//解码
+        if (token.isEmpty()) {//若token为空则返回false拦截
+            return null;
+        } else {
+            System.out.println("认证通过：");
+            System.out.println("issuer: " + jwtToken.getIssuer());
+            System.out.println("username: " + jwtToken.getClaim("username").asString());
+            return jwtToken.getClaim("username").asString();
+        }
+    }
+}

src/main/java/com/example/aitest/jwt/TokenInterceptor.java

@@ -0,0 +1,57 @@
+package com.example.aitest.jwt;
+
+import com.example.aitest.dao.redis.RedisUtil;
+import org.springframework.stereotype.Component;
+import org.springframework.web.servlet.HandlerInterceptor;
+
+import javax.annotation.Resource;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+@Component
+public class TokenInterceptor implements HandlerInterceptor {
+    @Resource(name = "tokenRedis")
+    RedisUtil redisUtil;//注册服务
+
+    @Override
+    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
+        System.out.println("触发拦截");
+        String token = request.getHeader("Token");
+//        System.out.println(token + "检查一下token是啥");
+//        Enumeration<String> headerNames = request.getHeaderNames();
+//        while (headerNames.hasMoreElements()) {
+//            String headerName = headerNames.nextElement();
+//            String headerValue = request.getHeader(headerName);
+//            System.out.println(headerName + ": " + headerValue);
+//        }
+        try {
+            if (request.getMethod().equals("OPTIONS")) {//检查是否为跨域请求
+                return true;
+            }
+
+            response.setCharacterEncoding("utf-8");
+
+            if (token != null) {
+                String verifyToken = JWTUtil.verify(token);
+
+                if (redisUtil.getRedis(verifyToken) != null) {
+                    response.setStatus(200);//设置状态码为正常，即通过登录验证
+                    System.out.println("通过拦截器");
+                    return true;
+                } else {
+                    response.setStatus(401);
+                    return false;//redis里没有token，即登录超时
+                }
+            } else {
+                response.setStatus(402);
+                return false;//请求头不带token
+            }
+
+        } catch (Exception exception) {
+            response.setStatus(500);//发生了不可预测的错误
+            throw new RuntimeException();
+        }
+
+
+    }
+}

src/main/java/com/example/aitest/mapper/UserMapper.java

@@ -0,0 +1,14 @@
+package com.example.aitest.mapper;
+
+import com.example.aitest.pojo.User;
+import org.apache.ibatis.annotations.Mapper;
+
+@Mapper
+public interface UserMapper {
+
+    User login(User user);
+
+    void register(User user);
+
+    User getUser(User user);
+}

src/main/java/com/example/aitest/pojo/User.java

@@ -0,0 +1,10 @@
+package com.example.aitest.pojo;
+
+import lombok.Data;
+
+@Data
+public class User {
+    private String username;
+    private String password;
+    private String phone;
+}

src/main/java/com/example/aitest/service/UserService.java

@@ -0,0 +1,16 @@
+package com.example.aitest.service;
+
+import com.example.aitest.pojo.User;
+import org.springframework.stereotype.Service;
+
+import java.security.NoSuchAlgorithmException;
+
+public interface UserService {
+
+    String login(User user) throws NoSuchAlgorithmException;
+
+    String register(User user) throws NoSuchAlgorithmException;
+
+
+
+}

src/main/java/com/example/aitest/service/impl/UserServiceImpl.java

@@ -0,0 +1,60 @@
+package com.example.aitest.service.impl;
+
+import com.example.aitest.dao.redis.RedisUtil;
+import com.example.aitest.jwt.JWTUtil;
+import com.example.aitest.mapper.UserMapper;
+import com.example.aitest.pojo.User;
+import com.example.aitest.service.UserService;
+import org.springframework.stereotype.Service;
+
+import javax.annotation.Resource;
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+@Service
+public class UserServiceImpl implements UserService {
+
+    @Resource
+    public UserMapper userMapper;
+    @Resource(name = "tokenRedis")
+    public RedisUtil tokenRedis;
+    @Override
+    public String login(User user) throws NoSuchAlgorithmException {
+        MessageDigest md5 = MessageDigest.getInstance("MD5");
+        md5.update(user.getPassword().getBytes());
+        byte[] digest = md5.digest();
+        user.setPassword(parseByte2HexStr(digest));
+        User loginUser = userMapper.login(user);
+        System.out.println(loginUser);
+        if (loginUser != null) {
+            String token = JWTUtil.sign(user.getPhone());
+            tokenRedis.addRedis(user.getPhone(), token);
+            return token;
+        }
+        return "登录失败";
+    }
+
+    @Override
+    public String register(User user) throws NoSuchAlgorithmException {
+        if (userMapper.getUser(user) != null) {
+            return "用户已存在";
+        }
+        MessageDigest md5 = MessageDigest.getInstance("MD5");
+        md5.update(user.getPassword().getBytes());
+        byte[] digest = md5.digest();
+        user.setPassword(parseByte2HexStr(digest));
+        userMapper.register(user);
+        return "注册成功";
+    }
+
+    private String parseByte2HexStr(byte[] buf) {
+        StringBuilder sb = new StringBuilder();
+        for (byte b : buf) {
+            String hex = Integer.toHexString(b & 0xFF);
+            if (hex.length() == 1) {
+                hex = '0' + hex;
+            }
+            sb.append(hex.toUpperCase());
+        }
+        return sb.toString();
+    }
+}

src/main/resources/application.properties

@@ -1,2 +0,0 @@
-spring.application.name=AITest
-server.port=8081

src/main/resources/application.yml

@@ -0,0 +1,16 @@
+spring:
+  application:
+    name: AITest
+  redis:
+    port: 6379
+#    password: ob666666
+    host: localhost
+  datasource:
+    username: root
+    password: ob666666
+    url: jdbc:mysql://localhost:3306/AITest
+    driver-class-name: com.mysql.cj.jdbc.Driver
+mybatis:
+  mapper-locations: classpath:mapper/*.xml
+server:
+  port: 8081

src/main/resources/mapper/UserMapper.xml

@@ -0,0 +1,15 @@
+<?xml version="1.0" encoding="utf-8" ?>
+<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd" >
+<mapper namespace="com.example.aitest.mapper.UserMapper">
+
+    <select id="login" resultType="com.example.aitest.pojo.User">
+        select * from user where phone=#{phone} and password=#{password}
+    </select>
+
+    <select id="getUser" resultType="com.example.aitest.pojo.User">
+        select * from user where phone=#{phone}
+    </select>
+    <insert id="register" parameterType="com.example.aitest.pojo.User">
+        insert into user (phone, password, username)values(#{phone}, #{password}, #{username})
+    </insert>
+</mapper>

src/test/java/com/example/aitest/MD5Test.java

@@ -0,0 +1,32 @@
+package com.example.aitest;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.boot.test.context.SpringBootTest;
+
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+
+@SpringBootTest
+public class MD5Test {
+
+    @Test
+    public void test() throws NoSuchAlgorithmException {
+        MessageDigest md5 = MessageDigest.getInstance("MD5");
+        md5.update("ob666666".getBytes());
+        byte[] digest = md5.digest();
+        String s = parseByte2HexStr(digest);
+        System.out.println(s);
+    }
+    private String parseByte2HexStr(byte buf[]) {
+        StringBuffer sb = new StringBuffer();
+        for (int i = 0; i < buf.length; i++) {
+            String hex = Integer.toHexString(buf[i] & 0xFF);
+            if (hex.length() == 1) {
+                hex = '0' + hex;
+            }
+            sb.append(hex.toUpperCase());
+        }
+        return sb.toString();
+    }
+
+}