[Linux] python脚本模拟占用内存

女爷i 2023-07-12 11:42 4阅读 0赞

应用场景：测试系统在不同内存环境下的运行情况、低于设定阈值是否触发事件通知等。

思路：创建一个超长字符串，一个空格字符占1字节，1024个空格占1M，1024\*1024个空格占1G。

eatmem.py 内容如下，用法 python eatmem.py <block num>，其中block大小在脚本中设置为200M，block num 代表将要吃掉多少个200M的内存（实际用起来会有一定误差，可以结合free命令查看调整）。

#encoding=utf-8
    import sys, time
    if __name__ == '__main__':
        if(len(sys.argv) == 2):
            blocks = 0
            try:
                blocks = int(sys.argv[1])
            except ValueError:
                print("Usage: python eatmem.py <blocks to allocate,should be an integer> \n  Currently, 1 block represents 200M.")
    
            if(blocks > 0):
                eatFlag = True
         
                if(eatFlag):
                    mem_to_eat = str(blocks * 0.2) + "GB" if (blocks > 4) else str(blocks * 200) + "MB"
                    print("Will allocate about " + mem_to_eat + ". Please use 'Ctrl+C' to Release.")
                    try:
                        mem_list = []
                        for i in range(0, blocks):
                            mem_list[i] = ' ' * 1024 * 1024 * 200
                        while(True):
                            time.sleep(1)
                    except KeyboardInterrupt:
                        print "Will release the memory."
            else:
                print("Usage: python eatmem.py <blocks to eat,should be an integer> \n  The specified blocks should not be lower than 1.")
        else:
            print("Usage: python eatmem.py <blocks to eat,should be an integer> \n  eg: python eatmem.py 10 will allocate about 2G memory.")