Edit File by line

<?php

[0] Fix | Delete

[1] Fix | Delete

namespace Smush\Core\Api;

[2] Fix | Delete

[3] Fix | Delete

class Backoff {

[4] Fix | Delete

private $max_attempts = 5;

[5] Fix | Delete

[6] Fix | Delete

private $wait = 1;

[7] Fix | Delete

[8] Fix | Delete

private $use_jitter = true;

[9] Fix | Delete

[10] Fix | Delete

private $decider;

[11] Fix | Delete

[12] Fix | Delete

public function __construct() {

[13] Fix | Delete

$this->set_decider( $this->get_default_decider() );

[14] Fix | Delete

}

[15] Fix | Delete

[16] Fix | Delete

public function run( $callback ) {

[17] Fix | Delete

$attempt = 0;

[18] Fix | Delete

$try = true;

[19] Fix | Delete

$result = null;

[20] Fix | Delete

$max_attempts = $this->get_max_attempts();

[21] Fix | Delete

[22] Fix | Delete

while ( $try ) {

[23] Fix | Delete

$this->wait( $attempt );

[24] Fix | Delete

$result = call_user_func( $callback );

[25] Fix | Delete

[26] Fix | Delete

$attempt ++;

[27] Fix | Delete

[28] Fix | Delete

if ( $attempt >= $max_attempts ) {

[29] Fix | Delete

$try = false;

[30] Fix | Delete

} else {

[31] Fix | Delete

$try = call_user_func( $this->get_decider(), $result );

[32] Fix | Delete

}

[33] Fix | Delete

}

[34] Fix | Delete

[35] Fix | Delete

return $result;

[36] Fix | Delete

}

[37] Fix | Delete

[38] Fix | Delete

private function wait( $attempt ) {

[39] Fix | Delete

if ( $attempt == 0 ) {

[40] Fix | Delete

return;

[41] Fix | Delete

}

[42] Fix | Delete

[43] Fix | Delete

usleep( $this->get_wait_time( $attempt ) * 1000 );

[44] Fix | Delete

}

[45] Fix | Delete

[46] Fix | Delete

/**

[47] Fix | Delete

* @return mixed

[48] Fix | Delete

[49] Fix | Delete

private function get_max_attempts() {

[50] Fix | Delete

return $this->max_attempts;

[51] Fix | Delete

}

[52] Fix | Delete

[53] Fix | Delete

/**

[54] Fix | Delete

* @param mixed $max_attempts

[55] Fix | Delete

[56] Fix | Delete

* @return Backoff

[57] Fix | Delete

[58] Fix | Delete

public function set_max_attempts( $max_attempts ) {

[59] Fix | Delete

$this->max_attempts = max( (int) $max_attempts, 0 );

[60] Fix | Delete

[61] Fix | Delete

return $this;

[62] Fix | Delete

}

[63] Fix | Delete

[64] Fix | Delete

/**

[65] Fix | Delete

* @return mixed

[66] Fix | Delete

[67] Fix | Delete

private function get_wait_time( $attempt ) {

[68] Fix | Delete

$wait_time = $attempt == 1

[69] Fix | Delete

? $this->wait

[70] Fix | Delete

: pow( 2, $attempt ) * $this->wait;

[71] Fix | Delete

[72] Fix | Delete

return $this->jitter( (int) $wait_time );

[73] Fix | Delete

}

[74] Fix | Delete

[75] Fix | Delete

/**

[76] Fix | Delete

* @return mixed

[77] Fix | Delete

[78] Fix | Delete

private function get_initial_wait() {

[79] Fix | Delete

return $this->wait;

[80] Fix | Delete

}

[81] Fix | Delete

[82] Fix | Delete

/**

[83] Fix | Delete

* @param mixed $wait

[84] Fix | Delete

[85] Fix | Delete

* @return Backoff

[86] Fix | Delete

[87] Fix | Delete

public function set_wait( $wait ) {

[88] Fix | Delete

$this->wait = $wait;

[89] Fix | Delete

[90] Fix | Delete

return $this;

[91] Fix | Delete

}

[92] Fix | Delete

[93] Fix | Delete

/**

[94] Fix | Delete

* @return mixed

[95] Fix | Delete

[96] Fix | Delete

private function get_decider() {

[97] Fix | Delete

return $this->decider;

[98] Fix | Delete

}

[99] Fix | Delete

[100] Fix | Delete

/**

[101] Fix | Delete

* @param mixed $decider

[102] Fix | Delete

[103] Fix | Delete

* @return Backoff

[104] Fix | Delete

[105] Fix | Delete

public function set_decider( $decider ) {

[106] Fix | Delete

$this->decider = $decider;

[107] Fix | Delete

[108] Fix | Delete

return $this;

[109] Fix | Delete

}

[110] Fix | Delete

[111] Fix | Delete

private function get_default_decider() {

[112] Fix | Delete

return function ( $result ) {

[113] Fix | Delete

return is_wp_error( $result );

[114] Fix | Delete

};

[115] Fix | Delete

}

[116] Fix | Delete

[117] Fix | Delete

private function set_jitter( $useJitter ) {

[118] Fix | Delete

$this->use_jitter = $useJitter;

[119] Fix | Delete

}

[120] Fix | Delete

[121] Fix | Delete

public function enable_jitter() {

[122] Fix | Delete

$this->set_jitter( true );

[123] Fix | Delete

[124] Fix | Delete

return $this;

[125] Fix | Delete

}

[126] Fix | Delete

[127] Fix | Delete

public function disable_jitter() {

[128] Fix | Delete

$this->set_jitter( false );

[129] Fix | Delete

[130] Fix | Delete

return $this;

[131] Fix | Delete

}

[132] Fix | Delete

[133] Fix | Delete

private function jitter_enabled() {

[134] Fix | Delete

return $this->use_jitter;

[135] Fix | Delete

}

[136] Fix | Delete

[137] Fix | Delete

private function jitter( $wait_time ) {

[138] Fix | Delete

if ( ! $this->jitter_enabled() ) {

[139] Fix | Delete

return $wait_time;

[140] Fix | Delete

}

[141] Fix | Delete

[142] Fix | Delete

$jitter_percentage = mt_rand( 1, 20 );

[143] Fix | Delete

$add_or_subtract = array_rand( array(

[144] Fix | Delete

- 1 => - 1,

[145] Fix | Delete

+ 1 => + 1,

[146] Fix | Delete

) );

[147] Fix | Delete

$jitter = ( $wait_time * $jitter_percentage / 100 ) * $add_or_subtract;

[148] Fix | Delete

[149] Fix | Delete

return $wait_time + $jitter;

[150] Fix | Delete

}

[151] Fix | Delete

}

[152] Fix | Delete

[153] Fix | Delete