python 实现iotop

2321321132

实验过程
一、源码
[iyunv@real_server01 ~]# cat  iotop.py
#!/usr/bin/python

import sys, os, time, signal, re

class DiskIO:
    def __init__(self, pname=None, pid=None, reads=0, writes=0):
        self.pname = pname
        self.pid = pid
        self.reads = 0
        self.writes = 0

def main():
    argc = len(sys.argv)
    if argc != 1:
        print "usage: ./iotop"
        sys.exit(0)

    if os.getuid() != 0:
        print "must be run as root"
        sys.exit(0)

    signal.signal(signal.SIGINT, signal_handler)
    os.system('echo 1 > /proc/sys/vm/block_dump')
    print "TASK              PID       READ      WRITE"
    while True:
        os.system('dmesg -c > /tmp/diskio.log')
        l = []
        f = open('/tmp/diskio.log', 'r')
        line = f.readline()
        while line:
            m = re.match(\
                '^(\S+)\((\d+)\): (READ|WRITE) block (\d+) on (\S+)', line)
            if m != None:
                if not l:
                    l.append(DiskIO(m.group(1), m.group(2)))
                    line = f.readline()
                    continue
                found = False
                for item in l:
                    if item.pid == m.group(2):
                        found = True
                        if m.group(3) == "READ":
                            item.reads = item.reads + 1
                        elif m.group(3) == "WRITE":
                            item.writes = item.writes + 1
                if not found:
                    l.append(DiskIO(m.group(1), m.group(2)))
            line = f.readline()
        time.sleep(1)
        for item in l:
            print "%-10s %10s %10d %10d" % \
                (item.pname, item.pid, item.reads, item.writes)

def signal_handler(signal, frame):
    os.system('echo 0 > /proc/sys/vm/block_dump')
    sys.exit(0)

if __name__=="__main__":
    main()
二、赋予可执行权限
[iyunv@real_server01 ~]# chmod +x iotop.py

三、执行程序
[iyunv@real_server01 ~]# ./iotop.py
TASK              PID       READ      WRITE
dmesg           25559          0          0
dmesg           25566          0          0
dmesg           25568          0          0
dmesg           25570          0          0
dmesg           25572          0          0
dmesg           25574          0          0
jbd2/sda2-8        273          0          5
dmesg           25576          0          0
dmesg           25578          0          0
dmesg           25580          0          0
dmesg           25582          0          0
dmesg           25584          0          0
jbd2/sda2-8        273          0          5
dmesg           25586          0          0
dmesg           25588          0          0
dmesg           25590          0          0
dmesg           25592          0          0
dmesg           25594          0          0
jbd2/sda2-8        273          0          5
dmesg           25596          0          0
dmesg           25598          0          0
默认会一直输出 ，ctrl+c终止它

总结：