|
1 | 1 | package com.liskovsoft.smartyoutubetv2.common.misc; |
2 | 2 |
|
3 | | -import android.util.Pair; |
| 3 | +import androidx.annotation.NonNull; |
4 | 4 |
|
5 | 5 | import com.liskovsoft.smartyoutubetv2.common.utils.Utils; |
6 | 6 |
|
7 | 7 | public class BufferingDetector { |
8 | | - private static final long BUFFERING_THRESHOLD_MS = 3_000; |
9 | 8 | private static final long BUFFERING_WINDOW_MS = 60_000; |
10 | | - private static final long BUFFERING_RECURRENCE_COUNT = 2; |
11 | | - private static final long BUFFERING_CONTINUATION_MS = 20_000; |
12 | | - |
13 | | - private Pair<Integer, Long> mBufferingCount; |
14 | | - private final Runnable mOnLongBuffering = this::updateBufferingCountIfNeeded; |
| 9 | + private static final long BUFFERING_DURATION_MS = 20_000; |
| 10 | + |
| 11 | + private long mBeginTimeMs; |
| 12 | + private long mStartTimeMs; |
| 13 | + private long mTotalDurationMs; |
| 14 | + private final Runnable mOnLongBuffering = this::onLongBuffering; |
15 | 15 | private final OnLongBuffering mCallback; |
16 | 16 |
|
17 | 17 | public interface OnLongBuffering { |
18 | 18 | void onLongBuffering(); |
19 | 19 | } |
20 | 20 |
|
21 | | - public BufferingDetector(OnLongBuffering callback) { |
| 21 | + public BufferingDetector(@NonNull OnLongBuffering callback) { |
22 | 22 | mCallback = callback; |
23 | 23 | } |
24 | 24 |
|
25 | 25 | public void onStartBuffering() { |
26 | | - Utils.postDelayed(mOnLongBuffering, BUFFERING_THRESHOLD_MS); |
| 26 | + long currentTimeMs = System.currentTimeMillis(); |
| 27 | + if (currentTimeMs - mBeginTimeMs > BUFFERING_WINDOW_MS) { |
| 28 | + mBeginTimeMs = currentTimeMs; |
| 29 | + mTotalDurationMs = 0; |
| 30 | + } |
| 31 | + mStartTimeMs = currentTimeMs; |
| 32 | + Utils.postDelayed(mOnLongBuffering, BUFFERING_DURATION_MS - mTotalDurationMs); |
27 | 33 | } |
28 | 34 |
|
29 | 35 | public void onStopBuffering() { |
30 | 36 | Utils.removeCallbacks(mOnLongBuffering); |
| 37 | + long stopTimeMs = System.currentTimeMillis(); |
| 38 | + mTotalDurationMs += (stopTimeMs - mStartTimeMs); |
31 | 39 | } |
32 | 40 |
|
33 | 41 | /** |
34 | 42 | * Reset buffering stats |
35 | 43 | */ |
36 | 44 | public void reset() { |
37 | | - mBufferingCount = null; |
38 | | - onStopBuffering(); |
39 | | - } |
40 | | - |
41 | | - private void updateBufferingCountIfNeeded() { |
42 | | - updateBufferingCount(); |
43 | | - if (isBufferingRecurrent()) { |
44 | | - mBufferingCount = null; |
45 | | - mCallback.onLongBuffering(); |
46 | | - } else { |
47 | | - // Count continuous buffering as a new occurrences.... |
48 | | - Utils.postDelayed(mOnLongBuffering, BUFFERING_CONTINUATION_MS); |
49 | | - } |
50 | | - } |
51 | | - |
52 | | - private void updateBufferingCount() { |
53 | | - final long currentTimeMs = System.currentTimeMillis(); |
54 | | - int bufferingCount = 0; |
55 | | - long previousTimeMs = 0; |
56 | | - |
57 | | - if (mBufferingCount != null) { |
58 | | - bufferingCount = mBufferingCount.first; |
59 | | - previousTimeMs = mBufferingCount.second; |
60 | | - } |
61 | | - |
62 | | - if (currentTimeMs - previousTimeMs < BUFFERING_WINDOW_MS) { |
63 | | - bufferingCount++; |
64 | | - } else { |
65 | | - bufferingCount = 1; |
66 | | - } |
67 | | - |
68 | | - mBufferingCount = new Pair<>(bufferingCount, currentTimeMs); |
| 45 | + mBeginTimeMs = mStartTimeMs = mTotalDurationMs = 0; |
| 46 | + Utils.removeCallbacks(mOnLongBuffering); |
69 | 47 | } |
70 | 48 |
|
71 | | - private boolean isBufferingRecurrent() { |
72 | | - return mBufferingCount != null && mBufferingCount.first > BUFFERING_RECURRENCE_COUNT; |
| 49 | + private void onLongBuffering() { |
| 50 | + reset(); |
| 51 | + mCallback.onLongBuffering(); |
73 | 52 | } |
74 | 53 | } |
0 commit comments