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

Include user ID in report export status key.

This commit is contained in:
Jeff Stieler 2019-06-26 10:24:06 -06:00
parent b76d7b61fc
commit 1650794f2b
2 changed files with 25 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
plugins/woocommerce-admin/includes/api/class-wc-admin-rest-reports-export-controller.php
View File

@ -164,7 +164,7 @@ class WC_Admin_REST_Reports_Export_Controller extends WC_Admin_REST_Reports_Cont
$export_id = str_replace( '.', '', microtime( true ) ); $export_id = str_replace( '.', '', microtime( true ) );
WC_Admin_Report_Exporter::queue_report_export( $user_id, $export_id, $report_type, $report_args ); WC_Admin_Report_Exporter::queue_report_export( $user_id, $export_id, $report_type, $report_args );
WC_Admin_Report_Exporter::update_export_percentage_complete( $report_type, $export_id, 0 ); WC_Admin_Report_Exporter::update_export_percentage_complete( $user_id, $report_type, $export_id, 0 );
// @todo: handle error case? // @todo: handle error case?
$result = array( $result = array(
@ -197,7 +197,8 @@ class WC_Admin_REST_Reports_Export_Controller extends WC_Admin_REST_Reports_Cont
public function export_status( $request ) { public function export_status( $request ) {
$report_type = $request['type']; $report_type = $request['type'];
$export_id = $request['export_id']; $export_id = $request['export_id'];
$percentage = WC_Admin_Report_Exporter::get_export_percentage_complete( $report_type, $export_id ); $user_id = get_current_user_id();
$percentage = WC_Admin_Report_Exporter::get_export_percentage_complete( $user_id, $report_type, $export_id );
if ( false === $percentage ) { if ( false === $percentage ) {
return new WP_Error( return new WP_Error(

35
plugins/woocommerce-admin/includes/export/class-wc-admin-report-exporter.php
View File

@ -115,25 +115,35 @@ class WC_Admin_Report_Exporter {
$exporter->set_filename( "wc-{$report_type}-report-export-{$user_id}-{$export_id}" ); $exporter->set_filename( "wc-{$report_type}-report-export-{$user_id}-{$export_id}" );
$exporter->generate_file(); $exporter->generate_file();
self::update_export_percentage_complete( $report_type, $export_id, $exporter->get_percent_complete() ); self::update_export_percentage_complete( $user_id, $report_type, $export_id, $exporter->get_percent_complete() );
}
/**
* Generate a key to reference an export status.
*
* @param int $user_id User ID who requested the export.
* @param string $report_type Report type. E.g. 'customers'.
* @param string $export_id Unique ID for report (timestamp expected).
* @return string Status key.
*/
protected static function get_status_key( $user_id, $report_type, $export_id ) {
return implode( ':', array( $user_id, $report_type, $export_id ) );
} }
/** /**
* Update the completion percentage of a report export. * Update the completion percentage of a report export.
* *
* @param int $user_id User ID who requested the export.
* @param string $report_type Report type. E.g. 'customers'. * @param string $report_type Report type. E.g. 'customers'.
* @param string $export_id Unique ID for report (timestamp expected). * @param string $export_id Unique ID for report (timestamp expected).
* @param int $percentage Completion percentage. * @param int $percentage Completion percentage.
* @return void * @return void
*/ */
public static function update_export_percentage_complete( $report_type, $export_id, $percentage ) { public static function update_export_percentage_complete( $user_id, $report_type, $export_id, $percentage ) {
$exports_status = get_option( self::EXPORT_STATUS_OPTION, array() ); $exports_status = get_option( self::EXPORT_STATUS_OPTION, array() );
$status_key = self::get_status_key( $user_id, $report_type, $export_id );
if ( ! isset( $exports_status[ $report_type ] ) ) { $exports_status[ $status_key ] = $percentage;
$exports_status[ $report_type ] = array();
}
$exports_status[ $report_type ][ $export_id ] = $percentage;
update_option( self::EXPORT_STATUS_OPTION, $exports_status ); update_option( self::EXPORT_STATUS_OPTION, $exports_status );
} }
@ -141,18 +151,17 @@ class WC_Admin_Report_Exporter {
/** /**
* Get the completion percentage of a report export. * Get the completion percentage of a report export.
* *
* @param int $user_id User ID who requested the export.
* @param string $report_type Report type. E.g. 'customers'. * @param string $report_type Report type. E.g. 'customers'.
* @param string $export_id Unique ID for report (timestamp expected). * @param string $export_id Unique ID for report (timestamp expected).
* @return bool|int Completion percentage, or false if export not found. * @return bool|int Completion percentage, or false if export not found.
*/ */
public static function get_export_percentage_complete( $report_type, $export_id ) { public static function get_export_percentage_complete( $user_id, $report_type, $export_id ) {
$exports_status = get_option( self::EXPORT_STATUS_OPTION, array() ); $exports_status = get_option( self::EXPORT_STATUS_OPTION, array() );
$status_key = self::get_status_key( $user_id, $report_type, $export_id );
if ( if ( isset( $exports_status[ $status_key ] ) ) {
isset( $exports_status[ $report_type ] ) && return $exports_status[ $status_key ];
isset( $exports_status[ $report_type ][ $export_id ] )
) {
return $exports_status[ $report_type ][ $export_id ];
} }
return false; return false;