import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.stream.Collectors;

@Service
public class UserServiceImpl extends ServiceImpl<UserMapper, User> {

    private static final String SECRET_KEY = "your-encryption-key"; // 替換為你的加密密鑰

    /**
     * 根據(jù)明文手機號查詢單個用戶信息
     *
     * @param plainPhone 明文手機號
     * @return 匹配的用戶信息（數(shù)據(jù)庫密文解密后攜帶明文手機號信息）
     */
    public User getByPhone(String plainPhone) {
        // 1. 將明文手機號加密為密文
        String encryptedPhone = encryptPhone(plainPhone);

        // 2. 構(gòu)造查詢條件，通過密文匹配
        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("encrypted_phone", encryptedPhone);

        // 3. 查詢并返回結(jié)果
        User user = this.getOne(queryWrapper);

        // 4. 注入明文手機號回傳，以便客戶端知道原來的手機號
        if (user != null) {
            user.setPlainPhone(plainPhone); // 假設實體類 User 里有這個字段
        }

        return user;
    }

    /**
     * 根據(jù)明文手機號列表查詢對應的用戶信息
     *
     * @param plainPhones 明文手機號列表
     * @return 匹配的用戶信息列表（數(shù)據(jù)庫密文解密后攜帶明文手機號信息）
     */
    public List<User> listByPhoneIn(List<String> plainPhones) {
        // 1. 將明文手機號列表加密為密文列表
        List<String> encryptedPhones = plainPhones.stream()
                .map(this::encryptPhone) // 調(diào)用加密方法
                .collect(Collectors.toList());

        // 2. 構(gòu)造查詢條件，通過密文列表匹配
        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.in("encrypted_phone", encryptedPhones);

        // 3. 查詢并返回結(jié)果
        List<User> users = this.list(queryWrapper);

        // 4. 將原始明文手機號與對應的密文進行綁定，注入到返回結(jié)果中
        for (User user : users) {
            // 找到與當前密文匹配的明文手機號
            String plainPhone = plainPhones.get(encryptedPhones.indexOf(user.getEncryptedPhone()));
            user.setPlainPhone(plainPhone); // 假設實體類 User 里有這個字段
        }

        return users;
    }

    /**
     * 加密手機號的方法（對稱加密，AES）
     *
     * @param plainPhone 明文手機號
     * @return 加密密文手機號
     */
    private String encryptPhone(String plainPhone) {
        // 此處省略加密邏輯，可自行實現(xiàn) AES 加密對應方法
        // 假設這里有一工具類調(diào)用 AES 加密，示例傳入密鑰和數(shù)據(jù)進行加密
        return AESUtils.encrypt(plainPhone, SECRET_KEY); // 示例為偽代碼，請用實際工具實現(xiàn)
    }
}

-- 查詢特定手機號 
SELECT * FROM users WHERE phone_encrypted = AES_ENCRYPT(plainPhone, SECRET_KEY);