Total Complexity | 43 |
Total Lines | 330 |
Duplicated Lines | 0 % |
Changes | 3 | ||
Bugs | 0 | Features | 0 |
Complex classes like limiter often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
While breaking up the class, it is a good idea to analyze how other classes use limiter, and based on these observations, apply Extract Interface, too.
1 | <?php |
||
21 | class limiter |
||
22 | { |
||
23 | /** |
||
24 | * [$capacity Bucket volume] |
||
25 | * @var integer |
||
26 | */ |
||
27 | private $capacity = 100; |
||
28 | |||
29 | /** |
||
30 | * [$leakRate Constant rate at which the bucket will leak] |
||
31 | * @var integer |
||
32 | */ |
||
33 | private $leakRate = 1; |
||
34 | |||
35 | /** |
||
36 | * [$unpacked] |
||
37 | */ |
||
38 | private $unpacked; |
||
39 | |||
40 | /** |
||
41 | * [$bucket] |
||
42 | */ |
||
43 | private $bucket; |
||
44 | |||
45 | /** |
||
46 | * [$timeframe Durations are in seconds] |
||
47 | * @var array |
||
48 | */ |
||
49 | private static $timeframe = [ |
||
50 | 'SECOND' => 1, |
||
51 | 'MINUTE' => 60, |
||
52 | 'HOUR' => 3600, |
||
53 | 'DAY' => 86400, |
||
54 | 'CUSTOM' => 0, |
||
55 | ]; |
||
56 | |||
57 | /** |
||
58 | * [$window Timeframe window] |
||
59 | * @var integer |
||
60 | */ |
||
61 | private $window; |
||
62 | |||
63 | /** |
||
64 | * [$unlimited Rate limiter bypass if true] |
||
65 | * @var boolean |
||
66 | */ |
||
67 | private $unlimited = false; |
||
68 | |||
69 | /** |
||
70 | * [$scope Set the default scope] |
||
71 | * @var string |
||
72 | */ |
||
73 | private $scope = 'private'; |
||
74 | |||
75 | /** |
||
76 | * [setupOptions Set any Responsible API options] |
||
77 | * @return self |
||
78 | */ |
||
79 | public function setupOptions() |
||
80 | { |
||
81 | $options = $this->getOptions(); |
||
82 | |||
83 | if (isset($options['rateLimit'])) { |
||
84 | $this->setCapacity($options['rateLimit']); |
||
85 | } |
||
86 | |||
87 | if (isset($options['rateWindow'])) { |
||
88 | $this->setTimeframe($options['rateWindow']); |
||
89 | } |
||
90 | |||
91 | if (isset($options['leak']) && !$options['leak']) { |
||
92 | $options['leakRate'] = 0; |
||
93 | } |
||
94 | |||
95 | if (isset($options['leakRate'])) { |
||
96 | $this->setLeakRate($options['leakRate']); |
||
97 | } |
||
98 | |||
99 | if (isset($options['unlimited']) && ($options['unlimited'] == 1 || $options['unlimited'] == true)) { |
||
100 | $this->setUnlimited(); |
||
101 | } |
||
102 | |||
103 | if (isset($options['requestType']) && $options['requestType'] == 'debug') { |
||
104 | $this->setUnlimited(); |
||
105 | } |
||
106 | |||
107 | if( isset($this->account->scope) && |
||
108 | ($this->account->scope == 'anonymous' || $this->account->scope == 'public') |
||
109 | ) { |
||
110 | $this->scope = $this->account->scope; |
||
111 | } |
||
112 | |||
113 | return $this; |
||
114 | } |
||
115 | |||
116 | /** |
||
117 | * [throttleRequest Build the Responsible API throttle] |
||
118 | * @return void |
||
119 | */ |
||
120 | public function throttleRequest() |
||
121 | { |
||
122 | if ($this->isUnlimited() || $this->scope !== 'private') { |
||
123 | return true; |
||
|
|||
124 | } |
||
125 | |||
126 | /** |
||
127 | * [$unpack Unpack the account bucket data] |
||
128 | */ |
||
129 | $this->unpacked = (new throttle\tokenPack)->unpack( |
||
130 | $this->getAccount()->bucket |
||
131 | ); |
||
132 | if (empty($this->unpacked)) { |
||
133 | $this->unpacked = array( |
||
134 | 'drops' => 1, |
||
135 | 'time' => $this->getAccount()->access, |
||
136 | ); |
||
137 | } |
||
138 | |||
139 | $this->bucket = (new throttle\tokenBucket()) |
||
140 | ->setTimeframe($this->getTimeframe()) |
||
141 | ->setCapacity($this->getCapacity()) |
||
142 | ->setLeakRate($this->getLeakRate()) |
||
143 | ->pour($this->unpacked['drops'], $this->unpacked['time']) |
||
144 | ; |
||
145 | |||
146 | /** |
||
147 | * Check if the bucket still has capacity to fill |
||
148 | */ |
||
149 | if ($this->bucket->capacity()) { |
||
150 | $this->bucket->pause(false); |
||
151 | $this->bucket->fill(); |
||
152 | } else { |
||
153 | if ($this->getLeakRate() <= 0) { |
||
154 | if ($this->unpacked['pauseAccess'] == false) { |
||
155 | $this->bucket->pause(true); |
||
156 | $this->save(); |
||
157 | } |
||
158 | |||
159 | if ($this->bucket->refill($this->getAccount()->access)) { |
||
160 | $this->save(); |
||
161 | } |
||
162 | } |
||
163 | |||
164 | (new exception\errorException)->error('TOO_MANY_REQUESTS'); |
||
165 | } |
||
166 | |||
167 | $this->save(); |
||
168 | } |
||
169 | |||
170 | /** |
||
171 | * [updateBucket Store the buckets token data and user access time] |
||
172 | * @return void |
||
173 | */ |
||
174 | private function save() |
||
175 | { |
||
176 | $this->packed = (new throttle\tokenPack)->pack( |
||
177 | $this->bucket->getTokenData() |
||
178 | ); |
||
179 | |||
180 | /** |
||
181 | * [Update account access] |
||
182 | */ |
||
183 | /*$user = (new user\account($this->getAccount()->account_id)) |
||
184 | ->setBucketToken($this->packed) |
||
185 | ->updateAccountAccess();*/ |
||
186 | |||
187 | $user = (new user\user) |
||
188 | ->setAccountID($this->getAccount()->account_id) |
||
189 | ->setBucketToken($this->packed) |
||
190 | ->updateAccountAccess() |
||
191 | ; |
||
192 | } |
||
193 | |||
194 | /** |
||
195 | * [getThrottle Return a list of the throttled results] |
||
196 | * @return array |
||
197 | */ |
||
198 | public function getThrottle() |
||
199 | { |
||
200 | if ($this->isUnlimited() || $this->scope !== 'private') { |
||
201 | return array( |
||
202 | 'unlimited' => true, |
||
203 | ); |
||
204 | } |
||
205 | |||
206 | $windowFrame = (is_string($this->getTimeframe())) |
||
207 | ? $this->getTimeframe() |
||
208 | : $this->getTimeframe() . 'secs' |
||
209 | ; |
||
210 | |||
211 | return array( |
||
212 | 'limit' => $this->getCapacity(), |
||
213 | 'leakRate' => $this->getLeakRate(), |
||
214 | 'leak' => $this->bucket->getLeakage(), |
||
215 | 'lastAccess' => $this->getLastAccessDate(), |
||
216 | 'description' => $this->getCapacity() . ' requests per ' . $windowFrame, |
||
217 | 'bucket' => $this->bucket->getTokenData(), |
||
218 | ); |
||
219 | } |
||
220 | |||
221 | /** |
||
222 | * [getLastAccessDate Get the last recorded access in date format] |
||
223 | * @return string |
||
224 | */ |
||
225 | private function getLastAccessDate() |
||
226 | { |
||
227 | if (isset($this->bucket->getTokenData()['time'])) { |
||
228 | return date('m/d/y h:i:sa', $this->bucket->getTokenData()['time']); |
||
229 | } |
||
230 | |||
231 | return 'Can\'t be converted'; |
||
232 | } |
||
233 | |||
234 | /** |
||
235 | * [setAccount Set the requests account] |
||
236 | */ |
||
237 | public function setAccount($account) |
||
238 | { |
||
239 | $this->account = $account; |
||
240 | return $this; |
||
241 | } |
||
242 | |||
243 | /** |
||
244 | * [getAccount Get the requests account] |
||
245 | */ |
||
246 | public function getAccount() |
||
247 | { |
||
248 | return $this->account; |
||
249 | } |
||
250 | |||
251 | /** |
||
252 | * [options Responsible API options] |
||
253 | * @param array $options |
||
254 | */ |
||
255 | public function options($options) |
||
256 | { |
||
257 | $this->options = $options; |
||
258 | return $this; |
||
259 | } |
||
260 | |||
261 | /** |
||
262 | * [getOptions Get the stored Responsible API options] |
||
263 | * @return array |
||
264 | */ |
||
265 | private function getOptions() |
||
266 | { |
||
267 | return $this->options; |
||
268 | } |
||
269 | |||
270 | /** |
||
271 | * [setCapacity Set the buckets capacity] |
||
272 | * @param integer $capacity |
||
273 | */ |
||
274 | public function setCapacity($capacity) |
||
275 | { |
||
276 | $this->capacity = $capacity; |
||
277 | } |
||
278 | |||
279 | /** |
||
280 | * [getCapacity Get the buckets capacity] |
||
281 | * @return integer |
||
282 | */ |
||
283 | public function getCapacity() |
||
284 | { |
||
285 | return $this->capacity; |
||
286 | } |
||
287 | |||
288 | /** |
||
289 | * [setTimeframe Set the window timeframe] |
||
290 | * @param string|integer $timeframe |
||
291 | */ |
||
292 | public function setTimeframe($timeframe) |
||
293 | { |
||
294 | if (is_numeric($timeframe)) { |
||
295 | self::$timeframe['CUSTOM'] = $timeframe; |
||
296 | $this->window = self::$timeframe['CUSTOM']; |
||
297 | return; |
||
298 | } |
||
299 | |||
300 | if (isset(self::$timeframe[$timeframe])) { |
||
301 | $this->window = self::$timeframe[$timeframe]; |
||
302 | return; |
||
303 | } |
||
304 | |||
305 | $this->window = self::$timeframe['MINUTE']; |
||
306 | } |
||
307 | |||
308 | /** |
||
309 | * [getTimeframe Get the timeframe window] |
||
310 | * @return integer |
||
311 | */ |
||
312 | public function getTimeframe() |
||
313 | { |
||
314 | return $this->window; |
||
315 | } |
||
316 | |||
317 | /** |
||
318 | * [setLeakRate Set the buckets leak rate] |
||
319 | * Options: slow, medium, normal, default, fast or custom positive integer |
||
320 | * @param string|integer $leakRate |
||
321 | */ |
||
322 | private function setLeakRate($leakRate) |
||
323 | { |
||
324 | $this->leakRate = $leakRate; |
||
325 | } |
||
326 | |||
327 | /** |
||
328 | * [getLeakRate Get the buckets leak rate] |
||
329 | * @return string|integer |
||
330 | */ |
||
331 | private function getLeakRate() |
||
332 | { |
||
333 | return $this->leakRate; |
||
334 | } |
||
335 | |||
336 | /** |
||
337 | * [setUnlimited Rate limiter bypass] |
||
338 | */ |
||
339 | private function setUnlimited() |
||
340 | { |
||
341 | $this->unlimited = true; |
||
342 | } |
||
343 | |||
344 | /** |
||
345 | * [isUnlimited Check if the Responsible API is set to unlimited] |
||
346 | * @return boolean |
||
347 | */ |
||
348 | private function isUnlimited() |
||
351 | } |
||
352 | } |
||
353 |