SpringBoot整合MD5加密完成注册和登录方式

<dependencies>
    <!-- Spring Boot Starter Web -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    
    <!-- Spring Boot Starter Data JPA -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-data-jpa</artifactId>
    </dependency>
    
    <!-- H2 Database -->
    <dependency>
        <groupId>com.h2database</groupId>
        <artifactId>h2</artifactId>
        <scope>runtime</scope>
    </dependency>

    <!-- Spring Boot Starter Test (可选) -->
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-test</artifactId>
        <scope>test</scope>
    </dependency>
</dependencies>

spring:
  h2:
    console:
      enabled: true # 启用 H2 控制台
  datasource:
    url: jdbc:h2:mem:testdb # H2 内存数据库连接 URL
    driver-class-name: org.h2.Driver # H2 驱动类
    username: sa # 数据库用户名
    password: '' # 数据库密码（默认为空）
  jpa:
    database-platform: org.hibernate.dialect.H2Dialect # 使用 H2 方言

import java.security.MessageDigest;

public class MD5Util {

    // MD5 加密方法
    public static String md5(String input) {
        try {
            MessageDigest md = MessageDigest.getInstance("MD5"); // 创建 MD5 实例
            byte[] messageDigest = md.digest(input.getBytes()); // 获取加密后的字节数组
            StringBuilder hexString = new StringBuilder();
            for (byte b : messageDigest) {
                String hex = Integer.toHexString(0xff & b); // 转换为十六进制字符串
                if (hex.length() == 1) hexString.append('0'); // 补零
                hexString.append(hex);
            }
            return hexString.toString(); // 返回十六进制字符串
        } catch (Exception e) {
            throw new RuntimeException("MD5 encryption error", e); // 抛出运行时异常
        }
    }
}

package com.example.model;

import javax.persistence.Entity;
import javax.persistence.GeneratedValue;
import javax.persistence.GenerationType;
import javax.persistence.Id;

@Entity // 标记该类为 JPA 实体类
public class User {
    @Id // 标记为主键
    @GeneratedValue(strategy = GenerationType.IDENTITY) // 主键自增策略
    private Long id; // 用户 ID
    private String username; // 用户名
    private String password; // 密码

    // Getter 和 Setter 方法
    public Long getId() {
        return id;
    }

    public void setId(Long id) {
        this.id = id;
    }

    public String getUsername() {
        return username;
    }

    public void setUsername(String username) {
        this.username = username;
    }

    public String getPassword() {
        return password;
    }

    public void setPassword(String password) {
        this.password = password;
    }
}

package com.example.repository;

import com.example.model.User;
import org.springframework.data.jpa.repository.JpaRepository;

// UserRepository 接口，继承 JpaRepository 提供基本的 CRUD 操作
public interface UserRepository extends JpaRepository<User, Long> {
    User findByUsername(String username); // 根据用户名查找用户
}

@Service // 标记为服务类，供 Spring 管理
public class UserService {
    @Autowired // 自动注入 UserRepository
    private UserRepository userRepository;

    // 用户注册方法
    public User register(String username, String password) {
        // 检查用户名是否已存在
        if (userRepository.findByUsername(username) != null) {
            throw new RuntimeException("用户名已存在"); // 抛出异常或返回错误信息
        }

        User user = new User(); // 创建新的用户实例
        user.setUsername(username); // 设置用户名
        user.setPassword(MD5Util.md5(password)); // 使用工具类进行 MD5 加密
        return userRepository.save(user); // 保存用户到数据库
    }

    // 用户登录方法
    public boolean login(String username, String password) {
        User user = userRepository.findByUsername(username); // 查找用户
        // 检查用户是否存在及密码是否匹配
        return user != null && user.getPassword().equals(MD5Util.md5(password));
    }
}

package com.example.controller;

import com.example.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

// 控制器类，处理用户相关请求
@RestController
@RequestMapping("/api/users") // 设置请求路径前缀
public class UserController {
    @Autowired // 自动注入 UserService
    private UserService userService;

    // 用户注册接口
    @PostMapping("/register")
    public String register(@RequestParam String username, @RequestParam String password) {
        userService.register(username, password); // 调用注册服务
        return "User registered!"; // 返回注册成功信息
    }

    // 用户登录接口
    @PostMapping("/login")
    public String login(@RequestParam String username, @RequestParam String password) {
        // 调用登录服务并返回结果
        return userService.login(username, password) ? "Login successful!" : "Login failed!";
    }
}