java 體驗AI代碼助手 代碼解讀復制代碼// 判斷用戶 ID 是否有效
public void checkUserExistence(String userId) {
    User user = userDao.findById(userId);
    if (user == null) {
        throw new RuntimeException("用戶ID無效");
    }
}

// 判斷部門 ID 是否有效
public void checkDeptExistence(String deptId) {
    Dept dept = deptDao.findById(deptId);
    if (dept == null) {
        throw new RuntimeException("部門ID無效");
    }   
}

java 體驗AI代碼助手 代碼解讀復制代碼/**
 * 確認數(shù)據(jù)庫字段值有效（通用）
 * 
 * @param <V> 待驗證值的類型
 * @param valueToCheck 待驗證的值
 * @param columnExtractor 實體類屬性提取函數(shù)
 * @param queryExecutor 單條數(shù)據(jù)查詢執(zhí)行器
 * @param errorMessage 異常提示信息模板
 */
public static <T, R, V> void ensureColumnValueValid(V valueToCheck, SFunction<T, R> columnExtractor, SFunction<LambdaQueryWrapper<T>, T> queryExecutor, String errorMessage) {
    if (valueToCheck == null) return;
    
    LambdaQueryWrapper<T> wrapper = new LambdaQueryWrapper<>();
    wrapper.select(columnExtractor);
    wrapper.eq(columnExtractor, valueToCheck);
    wrapper.last("LIMIT 1");

    T entity = queryExecutor.apply(wrapper);
    R columnValue = columnExtractor.apply(entity);
    if (entity == null || columnValue == null)
        throw new DataValidationException(String.format(errorMessage, valueToCheck));
}

java 體驗AI代碼助手 代碼解讀復制代碼// 判斷用戶 ID 是否有效
public void checkUserExistence(String userId) {
    User user = userDao.findById(userId);
    if (user == null) {
        throw new RuntimeException("用戶ID無效");
    }
}

// 判斷部門 ID 是否有效
public void checkDeptExistence(String deptId) {
    Dept dept = deptDao.findById(deptId);
    if (dept == null) {
        throw new RuntimeException("部門ID無效");
    }   
}

java 體驗AI代碼助手 代碼解讀復制代碼public void assignTaskToUser(AddOrderDTO dto) {
    ensureColumnValueValid(dto.getUserId(), User::getId, userDao::getOne, "用戶ID無效");
    ensureColumnValueValid(dto.getDeptId(), Dept::getId, deptDao::getOne, "部門ID無效");    
    ensureColumnValueValid(dto.getCustomerId(), Customer::getId, customerDao::getOne, "客戶ID無效");
    ensureColumnValueValid(dto.getDeptId(), Dept::getId, deptDao::getOne, "部門ID無效");
    ensureColumnValueValid(dto.getSupplieId(), Supplie::getId, supplierDao::getOne, "供應商ID無效");

    // 現(xiàn)在可以確信客戶存在
    Customer cus = customerDao.findById(dto.getCustomerId());     
    
    // 創(chuàng)建訂單的邏輯...
}

java 體驗AI代碼助手 代碼解讀復制代碼/**
 * 驗證查詢結(jié)果中指定列的值是否與預期值匹配
 *
 * @param <T>             實體類型
 * @param <R>             目標列值的類型
 * @param <C>             查詢條件列值的類型
 * @param targetColumn    目標列的提取函數(shù)，用于獲取想要驗證的列值
 * @param expectedValue   期望的列值
 * @param conditionColumn 條件列的提取函數(shù)，用于設置查詢條件
 * @param conditionValue  條件列對應的值
 * @param queryMethod     執(zhí)行查詢的方法引用，返回單個實體對象
 * @param errorMessage    驗證失敗時拋出異常的錯誤信息模板
 * @throws RuntimeException 當查詢結(jié)果中目標列的值與預期值不匹配時拋出異常
 */
public static <T, R, C> void validateColumnValueMatchesExpected(
      SFunction<T, R> targetColumn, R expectedValue,
      SFunction<T, C> conditionColumn, C conditionValue,
      SFunction<LambdaQueryWrapper<T>, T> queryMethod,
      String errorMessage) {

   // 創(chuàng)建查詢包裝器，選擇目標列并設置查詢條件
   LambdaQueryWrapper<T> wrapper = new LambdaQueryWrapper<>();
   wrapper.select(targetColumn);
   wrapper.eq(conditionColumn, conditionValue);

   // 執(zhí)行查詢方法
   T one = queryMethod.apply(wrapper);
   // 如果查詢結(jié)果為空，則直接返回，視為驗證通過（或忽略）
   if (one == null) return;

   // 獲取查詢結(jié)果中目標列的實際值
   R actualValue = targetColumn.apply(one);

   // 比較實際值與預期值是否匹配，這里假設notMatch是一個自定義方法用于比較不匹配情況
   boolean doesNotMatch = notMatch(actualValue, expectedValue);
   if (doesNotMatch) {
      // 若不匹配，則根據(jù)錯誤信息模板拋出異常
      throw new RuntimeException(String.format(errorMessage, expectedValue, actualValue));
   }
}

// 假設的輔助方法，用于比較值是否不匹配，根據(jù)實際需要實現(xiàn)
private static <R> boolean notMatch(R actual, R expected) {
    // 示例簡單實現(xiàn)為不相等判斷，實際情況可能更復雜
    return !Objects.equals(actual, expected);
}

java 體驗AI代碼助手 代碼解讀復制代碼// 當用戶角色不是 “普通用戶” 時拋異常
validateColumnValueMatchesExpected(User::getRoleType, "普通用戶", User::getId, userId, userMapper::getOne, "用戶角色不是普通用戶，無法升級為管理員！");

java 體驗AI代碼助手 代碼解讀復制代碼/**
 * 驗證查詢結(jié)果中指定列的值是否位于預期值列表內(nèi)
 *
 * @param <T>             實體類型
 * @param <R>             目標列值的類型
 * @param <C>             查詢條件列值的類型
 * @param targetColumn    目標列的提取函數(shù)，用于獲取想要驗證的列值
 * @param expectedValueList 期望值的列表
 * @param conditionColumn 條件列的提取函數(shù)，用于設置查詢條件
 * @param conditionValue  條件列對應的值
 * @param queryMethod     執(zhí)行查詢的方法引用，返回單個實體對象
 * @param errorMessage    驗證失敗時拋出異常的錯誤信息模板
 * @throws RuntimeException 當查詢結(jié)果中目標列的值不在預期值列表內(nèi)時拋出異常
 */
public static <T, R, C> void validateColumnValueInExpectedList(
        SFunction<T, R> targetColumn, List<R> expectedValueList,
        SFunction<T, C> conditionColumn, C conditionValue,
        SFunction<LambdaQueryWrapper<T>, T> queryMethod,
        String errorMessage) {

    LambdaQueryWrapper<T> wrapper = new LambdaQueryWrapper<>();
    wrapper.select(targetColumn);
    wrapper.eq(conditionColumn, conditionValue);

    T one = queryMethod.apply(wrapper);
    if (one == null) return;

    R actualValue = targetColumn.apply(one);
    if (actualValue == null) throw new RuntimeException("列查詢結(jié)果為空");

    if (!expectedValueList.contains(actualValue)) {        
        throw new RuntimeException(errorMessage);
    }
}

// 假設 OrderStatusEnum 枚舉了所有可能的訂單狀態(tài)，cancelableStatuses 包含可取消的狀態(tài)
List<String> cancelableStatuses = Arrays.asList(OrderStatusEnum.WAITING_PAYMENT.getValue(), OrderStatusEnum.WAITING_DELIVERY.getValue());

// 驗證訂單狀態(tài)是否在可取消狀態(tài)列表內(nèi)
validateColumnValueInExpectedList(Order::getStatus, cancelableStatuses, Order::getOrderId, orderId, orderMapper::selectOne, "訂單當前狀態(tài)不允許取消！");