pure-python-adb/ppadb/plugins/device/cpustat.py at 6693e67ff878a93edfda894b2a10ebfa42bb1000 · Swind/pure-python-adb · GitHub

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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
import re
import time

from ppadb.plugins import Plugin
from ppadb.utils.logger import AdbLogging

logger = AdbLogging.get_logger(__name__)

class TotalCPUStat:
    def __init__(self, user, nice, system, idle, iowait, irq, softirq, stealstolen, guest, guest_nice):
        self.user = user
        self.nice = nice
        self.system = system
        self.idle = idle
        self.iowait = iowait
        self.irq = irq
        self.softirq = softirq
        self.stealstolen = stealstolen
        self.guest = guest
        self.guest_nice = guest_nice

    def total(self):
        return self.user + self.nice + self.system + self.idle + self.iowait + self.irq + self.softirq + self.stealstolen + self.guest + self.guest_nice

    def __add__(self, other):
        summary = TotalCPUStat(0, 0, 0, 0, 0, 0, 0, 0, 0, 0)

        summary.user = self.user + other.user
        summary.nice = self.nice + other.nice
        summary.system = self.system + other.system
        summary.idle = self.idle + other.idle
        summary.iowait = self.iowait + other.iowait
        summary.irq = self.irq + other.irq
        summary.softirq = self.softirq + other.softirq
        summary.stealstolen = self.stealstolen + other.stealstolen
        summary.guest = self.guest + other.guest
        summary.guest_nice = self.guest_nice + other.guest_nice

        return summary

    def __sub__(self, other):
        result = TotalCPUStat(0, 0, 0, 0, 0, 0, 0, 0, 0, 0)

        result.user = self.user - other.user
        result.nice = self.nice - other.nice
        result.system = self.system - other.system
        result.idle = self.idle - other.idle
        result.iowait = self.iowait - other.iowait
        result.irq = self.irq - other.irq
        result.softirq = self.softirq - other.softirq
        result.stealstolen = self.stealstolen - other.stealstolen
        result.guest = self.guest - other.guest
        result.guest_nice = self.guest_nice - other.guest_nice

        return result

    def __str__(self):
        attrs = vars(self)
        return ', '.join("%s: %s" % item for item in attrs.items())


class ProcessCPUStat:
    def __init__(self, name, utime, stime):
        self.name = name
        self.utime = utime
        self.stime = stime

    def __add__(self, other):
        summary = ProcessCPUStat(self.name, 0, 0)
        summary.utime = self.utime + other.utime
        summary.stime = self.stime + other.stime
        return summary

    def __sub__(self, other):
        result = ProcessCPUStat(self.name, 0, 0)
        result.utime = self.utime - other.utime
        result.stime = self.stime - other.stime
        return result

    def __str__(self):
        attrs = vars(self)
        return ', '.join("%s: %s" % item for item in attrs.items())

    def total(self):
        return self.utime + self.stime


class CPUStat(Plugin):
    total_cpu_pattern = re.compile(
        r"cpu\s+([\d]+)\s([\d]+)\s([\d]+)\s([\d]+)\s([\d]+)\s([\d]+)\s([\d]+)\s([\d]+)\s([\d]+)\s([\d]+)\s")

    def cpu_times(self):
        return self.get_total_cpu()

    def cpu_percent(self, interval=1):
        cpu_times_start = self.cpu_times()
        time.sleep(interval)
        cpu_times_end = self.cpu_times()

        diff = cpu_times_end - cpu_times_start
        return round(((diff.user + diff.system) / diff.total()) * 100, 2)

    def cpu_count(self):
        result = self.shell('ls /sys/devices/system/cpu')
        match = re.findall(r'cpu[0-9+]', result)
        return len(match)

    def get_total_cpu(self):
        result = self.shell('cat /proc/stat')
        match = self.total_cpu_pattern.search(result)
        if not match and len(match.groups()) != 10:
            logger.error("Can't get the total cpu usage from /proc/stat")
            return None

        return TotalCPUStat(*map(lambda x: int(x), match.groups()))

    def get_pid_cpu(self, pid):
        result = self.shell('cat /proc/{}/stat'.format(pid)).strip()

        if "No such file or directory" in result:
            return ProcessCPUStat("", 0, 0)
        else:
            items = result.split()
            return ProcessCPUStat(items[1], int(items[13]), int(items[14]))

    def get_all_thread_cpu(self, pid):
        result = self.shell("ls /proc/{}/task".format(pid))
        tids = list(map(lambda line: line.strip(), result.split("\n")))

        thread_result = {}
        for tid in tids:
            result = self.shell("cat /proc/{}/task/{}/stat".format(pid, tid))

            if "No such file or directory" not in result:
                items = result.split()
                thread_result[tid] = ProcessCPUStat(items[1], int(items[13]), int(items[14]))

        return thread_result