app/src/main/java/com/wireguard/android/backend/Statistics.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

/*
 * Copyright © 2020 WireGuard LLC. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0
 */

package com.wireguard.android.backend;

import android.os.SystemClock;
import android.util.Pair;

import com.wireguard.crypto.Key;

import java.util.HashMap;
import java.util.Map;

public class Statistics {
    private long lastTouched = SystemClock.elapsedRealtime();
    private final Map<Key, Pair<Long, Long>> peerBytes = new HashMap<>();

    Statistics() { }

    void add(final Key key, final long rx, final long tx) {
        peerBytes.put(key, Pair.create(rx, tx));
        lastTouched = SystemClock.elapsedRealtime();
    }

    public boolean isStale() {
        return SystemClock.elapsedRealtime() - lastTouched > 900;
    }

    public Key[] peers() {
        return peerBytes.keySet().toArray(new Key[0]);
    }

    public long peerRx(final Key peer) {
        if (!peerBytes.containsKey(peer))
            return 0;
        return peerBytes.get(peer).first;
    }

    public long peerTx(final Key peer) {
        if (!peerBytes.containsKey(peer))
            return 0;
        return peerBytes.get(peer).second;
    }

    public long totalRx() {
        long rx = 0;
        for (final Pair<Long, Long> val : peerBytes.values()) {
            rx += val.first;
        }
        return rx;
    }

    public long totalTx() {
        long tx = 0;
        for (final Pair<Long, Long> val : peerBytes.values()) {
            tx += val.second;
        }
        return tx;
    }
}