Jooq批量Upsert时，如何用MySQLDSL处理TableRecord的byte[]加密字段？

你遇到的核心问题是：MyTableRecord需要的是实际的byte[]值，但MySQLDSL.aesEncrypt()返回的是SQL表达式（Field<byte[]>对象），直接赋值会把表达式的字符串形式（比如cast(aes_encrypt(...) as binary)）存入数据库，而非加密后的字节数组。

下面提供两种可行的解决方案，根据你的安全需求选择：

方案1：数据库端执行加密（推荐，更安全） #

这种方式让MySQL服务器直接执行AES加密，密钥不会暴露在应用端，适合敏感数据场景。需要放弃loadInto直接插入记录的方式，改用批量INSERT ... ON DUPLICATE KEY UPDATE语句，将加密表达式写入SQL中：

List<DataDTO> dtoList = // 你的DTO列表

// 构造批量插入的SQL，包含加密逻辑
dslContext.insertInto(MYTABLE)
    // 列出需要插入/更新的字段
    .columns(MYTABLE.ID, MYTABLE.SECURESTRING, MYTABLE.OTHER_FIELD)
    .select(
        select(
            field(name("t.id")),
            // 在SQL中直接调用AES加密函数
            MySQLDSL.aesEncrypt(field(name("t.secure_str")), field(name("t.encrypt_key"))),
            field(name("t.other_field"))
        )
        .from(
            // 将DTO列表转为临时表数据
            values(
                dtoList.stream()
                    .map(dto -> row(
                        dto.getId(),
                        dto.getSecureString(),
                        dto.getKey(),
                        dto.getOtherField()
                    ))
                    .toArray(Field[]::new)
            ).as("t", "id", "secure_str", "encrypt_key", "other_field")
        )
    )
    // 处理重复键时的更新逻辑
    .onDuplicateKeyUpdate()
    .set(MYTABLE.SECURESTRING, MySQLDSL.aesEncrypt(field(name("t.secure_str")), field(name("t.encrypt_key"))))
    .set(MYTABLE.OTHER_FIELD, field(name("t.other_field")))
    .execute();

方案2：应用端生成加密字节数组 #

如果必须使用loadInto，需要在应用端实现和MySQL兼容的AES加密逻辑，生成实际的byte[]后再赋值给MyTableRecord：

第一步：实现MySQL兼容的AES加密方法 #

MySQL的aes_encrypt默认使用ECB模式、PKCS5填充，且会将密钥补全/截断到16字节（AES-128），字符集默认是latin1，所以需要对齐这些参数：

import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.StandardCharsets;

public static byte[] mysqlAesEncrypt(String input, String key) throws Exception {
    // 对齐MySQL的密钥处理：截断/补全到16字节，用latin1编码
    byte[] keyBytes = key.getBytes(StandardCharsets.ISO_8859_1);
    byte[] adjustedKey = new byte[16];
    System.arraycopy(keyBytes, 0, adjustedKey, 0, Math.min(keyBytes.length, 16));

    // 初始化AES加密器，匹配MySQL默认配置
    Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
    SecretKeySpec secretKey = new SecretKeySpec(adjustedKey, "AES");
    cipher.init(Cipher.ENCRYPT_MODE, secretKey);

    // 加密输入字符串（同样用latin1编码）
    return cipher.doFinal(input.getBytes(StandardCharsets.ISO_8859_1));
}

第二步：构建记录列表并执行批量Upsert #

import java.util.stream.Collectors;

List<DataDTO> dtoList = // 你的DTO列表

// 将DTO转为MyTableRecord，提前加密字段
List<MyTableRecord> records = dtoList.stream().map(dto -> {
    MyTableRecord record = new MyTableRecord();
    record.setId(dto.getId());
    
    try {
        // 应用端加密后赋值给byte[]字段
        record.setSecureString(mysqlAesEncrypt(dto.getSecureString(), dto.getKey()));
    } catch (Exception e) {
        throw new RuntimeException("AES加密失败", e);
    }
    
    // 设置其他字段
    record.setOtherField(dto.getOtherField());
    return record;
}).collect(Collectors.toList());

// 使用loadInto执行批量Upsert
dslContext.loadInto(MYTABLE)
    .loadRecords(records)
    .onDuplicateKeyUpdate()
    // 指定重复时需要更新的字段
    .fields(MYTABLE.SECURESTRING, MYTABLE.OTHER_FIELD)
    .execute();

两种方案对比 #

内容的提问来源于stack exchange，提问作者introvertkernel