wordpress WooCommerce 添加 立即购买 按钮

// 添加立即结账按钮

function add_content_after_addtocart() {
// get the current post/product ID
$current_product_id = get_the_ID();
// get the product based on the ID
$product = wc_get_product( $current_product_id );
// get the "Checkout Page" URL
$checkout_url = WC()->cart->get_checkout_url();
// run only on simple products
if( $product->is_type( 'simple' ) ){
echo '<a class="single_add_to_cart_button button alt" href="'.$checkout_url.'?add-to-cart='.$current_product_id.'">立即购买</a>';
}
}
add_action( 'woocommerce_after_add_to_cart_button', 'add_content_after_addtocart' );