package com.dao.dataobject;
 
import lombok.Data;
 
/**
 * 一個(gè)標(biāo)記接口，用于標(biāo)識需要加密/解密的字段
 */
@Data
public class Encrypt {
    private String value;
 
   
    public Encrypt(byte[] decrypt) {
    }
 
 
    public Encrypt(String value) {
        this.value = value;
    }
}

package com.dao.handler;
 
import cn.hutool.crypto.SecureUtil;
import cn.hutool.crypto.symmetric.AES;
import com.bit.lotterySystem.dao.dataobject.Encrypt;
import org.apache.ibatis.type.JdbcType;
import org.apache.ibatis.type.MappedJdbcTypes;
import org.apache.ibatis.type.MappedTypes;
import org.apache.ibatis.type.TypeHandler;
 
import java.nio.charset.StandardCharsets;
import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
 
/**
 * 自定義的 TypeHandler，用于處理 Encrypt 類型與 VARCHAR 類型之間的轉(zhuǎn)換（包含加密解密）
 */
@MappedTypes(Encrypt.class)//告訴 MyBatis，這個(gè) Handler 是處理 Java 中的 Encrypt 類型的。
@MappedJdbcTypes(JdbcType.VARCHAR)//處理后的jdbc類型
public class EncryptTypeHandler implements TypeHandler<Encrypt> {
 
    // 注意：在生產(chǎn)環(huán)境中，密鑰絕不能硬編碼在代碼里！應(yīng)從配置文件或環(huán)境變量中讀取。
    private final byte[] key = "123456789abcdefg".getBytes(StandardCharsets.UTF_8); // 16位密鑰，AES-128
 
    /**
     * 設(shè)置參數(shù)
     * 將 Encrypt 對象中的明文加密后，以 String 形式存入數(shù)據(jù)庫。
     *
     * @param ps SQL預(yù)編譯對象
     * @param i 需要賦值的索引位置
     * @param parameter 原本位置i需要賦的值
     * @param jdbcType jdbc的類型
     * @throws SQLException
     */
    @Override
    public void setParameter(PreparedStatement ps, int i, Encrypt parameter, JdbcType jdbcType) throws SQLException {
        if (parameter == null || parameter.getValue() == null) {
            ps.setString(i, null);
            return;
        }
        // 1. 獲取需要加密的明文
        String plaintext = parameter.getValue();
        // 2. 使用 Hutool 的 SecureUtil 進(jìn)行 AES 加密
        AES aes = SecureUtil.aes(key);
        String ciphertext = aes.encryptHex(plaintext); // encryptHex 返回十六進(jìn)制字符串
        // 3. 將加密后的密文設(shè)置到 PreparedStatement
        ps.setString(i, ciphertext);
    }
 
   
    /**
     * 獲取值
     * 將從數(shù)據(jù)庫讀出的密文 String 解密后，封裝成 Encrypt 對象。
     *
     * @param rs 結(jié)果集
     * @param columnName 索引名
     * @return
     * @throws SQLException
     */
     */
    @Override
    public Encrypt getResult(ResultSet rs, String columnName) throws SQLException {
        String ciphertext = rs.getString(columnName);
        return decrypt(ciphertext);
    }
 
   
    /**
     * 獲取值,從 ResultSet 中根據(jù)列索引獲取值（從數(shù)據(jù)庫讀取）
     * @param rs 結(jié)果集
     * @param columnIndex 索引位置
     * @return
     * @throws SQLException
     */
    @Override
    public Encrypt getResult(ResultSet rs, int columnIndex) throws SQLException {
        String ciphertext = rs.getString(columnIndex);
        return decrypt(ciphertext);
    }
 
 
     /**
     * 獲取值,從 CallableStatement 中獲取值（存儲過程相關(guān)）
     * @param cs 結(jié)果集
     * @param columnIndex 索引位置
     * @return
     * @throws SQLException
     */
    @Override
    public Encrypt getResult(CallableStatement cs, int columnIndex) throws SQLException {
        String ciphertext = cs.getString(columnIndex);
        return decrypt(ciphertext);
    }
    
    /**
     * 通用的解密方法
     */
    private Encrypt decrypt(String ciphertext) {
        if (ciphertext == null) {
            return null;
        }
        // 1. 使用 Hutool 的 SecureUtil 進(jìn)行 AES 解密
        AES aes = SecureUtil.aes(key);
        String plaintext = aes.decryptStr(ciphertext, StandardCharsets.UTF_8);
        // 2. 將解密后的明文封裝成 Encrypt 對象返回
        return new Encrypt(plaintext);
    }
}

<dependency>
    <groupId>cn.hutool</groupId>
    <artifactId>hutool-all</artifactId>
    <version>5.8.41</version>
</dependency>

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE configuration
        PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
    <typeHandlers>
        <package name="com.dao.handler"/>
    </typeHandlers>
</configuration>

# application.yml
mybatis:
  type-handlers-package: com.dao.handler
  # 其他配置...

package com.dao.mapper;
 
import com.bit.lotterySystem.dao.dataobject.Encrypt;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
 
@Mapper
public interface UserMapper {
 
    /**
     * 根據(jù)手機(jī)號統(tǒng)計(jì)用戶數(shù)量
     * @param phoneNumber 注意這里的參數(shù)類型是 Encrypt
     * @return 用戶數(shù)量
     */
    @Select("SELECT count(*) FROM user WHERE phone_number = #{phoneNumber}")
    int countPhoneNumber(@Param("phoneNumber") Encrypt phoneNumber);
 
}

@Service
public class UserService {
 
    @Autowired
    private UserMapper userMapper;
 
    /**
     * 校驗(yàn)手機(jī)號是否已被使用
     * @param phoneNumber 原始的手機(jī)號明文
     * @return 是否被使用
     */
    private boolean checkPhoneNumberUsed(String phoneNumber) {
        // 我們只需要將明文手機(jī)號封裝進(jìn) Encrypt 對象即可
        // MyBatis 會在后臺自動(dòng)調(diào)用 EncryptTypeHandler 進(jìn)行加密
        int count = userMapper.countPhoneNumber(new Encrypt(phoneNumber));
        return count > 0;
    }
    
    // ...
}