获取MemberPress非订阅交易的企业账户ID：一种弹性处理方案

add_action('mepr-event-transaction-completed', 'my_custom_corporate_logic');

function my_custom_corporate_logic($event) {
  $transaction = $event->get_data();
  $membership_type_ids = array(1, 2, 4); // 示例会员类型ID

  if (in_array($transaction->product_id, $membership_type_ids) && $transaction->txn_type == 'payment') {
    $org_id = $transaction->corporate_account_id;

    // 期望此处能获取到正确的企业ID
    // my_custom_function($org_id); 

    // 问题：对于非订阅交易，即使数据库有记录，$org_id 仍可能为 0
  }
}

add_action('mepr-event-transaction-completed', 'handle_corporate_id_for_transaction');

function handle_corporate_id_for_transaction($event) {
  $transaction = $event->get_data();
  $membership_type_ids = array(1, 2, 4); // 示例：需要处理的会员类型ID

  // 确保是支付类型的交易且属于我们关注的会员类型
  if (in_array($transaction->product_id, $membership_type_ids) && $transaction->txn_type == 'payment') {

    // 检查 corporate_account_id 是否有效
    if ($transaction->corporate_account_id !== "0" && $transaction->corporate_account_id !== 0) {
      // 如果企业ID有效，则直接执行你的自定义逻辑
      // rc_setup_new_org($transaction->corporate_account_id); // 示例：执行有效ID的逻辑
      // write_log('Corporate ID found immediately for transaction: ' . $transaction->trans_num . ' ID: ' . $transaction->corporate_account_id);
    } else {
      // 如果企业ID为0，则调度一个延迟任务
      // 记录日志以便调试
      error_log('MemberPress: Corporate ID returned as 0 for transaction ' . $transaction->trans_num . '. Scheduling delayed fix.');

      // 调度一个在2分钟后执行的单次事件
      // 'send_fix_for_zero_transaction' 是自定义的钩子名称
      // array($transaction) 是传递给延迟函数的参数
      wp_schedule_single_event(
        strtotime("+2 minutes"),
        'send_fix_for_zero_transaction',
        array($transaction)
      );
      return; // 立即返回，等待延迟任务处理
    }
  }
}

add_action('send_fix_for_zero_transaction', 'single_transaction_create_corporate_delayed');

function single_transaction_create_corporate_delayed($transaction_data) {
  // 注意：此函数仅在MemberPress事件未及时提供corporate_id时触发
  // 目标是重新获取完整的交易数据，此时corporate_id应该已可用。

  $trans_num = $transaction_data->trans_num;

  // 重新从数据库中获取完整的交易对象
  $full_trans = MeprTransaction::get_one_by_trans_num($trans_num); 

  if ($full_trans && $full_trans->corporate_account_id !== "0" && $full_trans->corporate_account_id !== 0) {
    // 此时，我们应该能获取到正确的企业ID
    $corporate_id = $full_trans->corporate_account_id;

    // 在这里执行你的自定义逻辑，例如：
    // my_custom_function_with_corporate_id($corporate_id, $full_trans);
    error_log('MemberPress: Successfully retrieved corporate ID ' . $corporate_id . ' for transaction ' . $trans_num . ' via delayed cron.');
  } else {
    // 如果延迟后仍然无法获取，可能需要进一步的错误处理或通知
    error_log('MemberPress: Failed to retrieve corporate ID for transaction ' . $trans_num . ' even after delayed cron. Manual intervention might be needed.');
  }
}