Steve-Dee-Designs
/
woocommerce
mirror of https://github.com/woocommerce/woocommerce.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add column data type checks to ensure they match before running the download logs constraint.

This commit is contained in:
Gerhard Potgieter 2018-06-04 13:10:05 +02:00
parent 98c92f3246
commit 4fde0e9129
1 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
includes/class-wc-install.php
View File

@ -544,9 +544,28 @@ class WC_Install {
$wpdb->query( "ALTER TABLE {$wpdb->comments} ADD INDEX woo_idx_comment_type (comment_type)" );
}
// Add constraint to download logs.
// Get tables data types and check it matches before adding constraint.
$download_log_columns = $wpdb->get_results( "SHOW COLUMNS FROM {$wpdb->prefix}wc_download_log", ARRAY_A );
$download_log_column_type = '';
foreach ( $download_log_columns as $column ) {
if ( isset( $column['Field'], $column['Type'] ) && 'permission_id' === $column['Field'] ) {
$download_log_column_type = $column['Type'];
}
}
$download_permissions_columns = $wpdb->get_results( "SHOW COLUMNS FROM {$wpdb->prefix}woocommerce_downloadable_product_permissions", ARRAY_A );
$download_permissions_column_type = '';
foreach ( $download_permissions_columns as $column ) {
if ( isset( $column['Field'], $column['Type'] ) && 'permission_id' === $column['Field'] ) {
$download_permissions_column_type = $column['Type'];
}
}
// Add constraint to download logs if the columns matches.
if ( ! empty( $download_permissions_column_type ) && ! empty( $download_log_column_type ) && $download_permissions_column_type === $download_log_column_type ) {
$wpdb->query( "ALTER TABLE {$wpdb->prefix}wc_download_log ADD FOREIGN KEY (permission_id) REFERENCES {$wpdb->prefix}woocommerce_downloadable_product_permissions(permission_id) ON DELETE CASCADE" );
}
}
/**
* Get Table schema.