#!/usr/bin/env python3
import os
import time
import psutil
from datetime import datetime
import subprocess
import argparse
from loguru import logger
import vvdutils as vvdef clear_page_cache():
"""清空页面缓存（安全操作）"""
try:
with open('/proc/sys/vm/drop_caches', 'w') as f:
f.write('1\n') # 1表示清空页缓存
logger.info(f"[{datetime.now()}] 已清理页面缓存")
except Exception as e:
logger.info(f"清理缓存失败: {str(e)}")
def should_clear_swap():
"""判断是否满足SWAP清理条件"""
mem = psutil.virtual_memory()
swap = psutil.swap_memory()
# 条件1：物理内存剩余>30%且SWAP使用>10%
condition1 = (mem.percent < 70) and (swap.percent > 10)
# 条件2：SWAP使用量超过1GB 且物理内存剩余>1.5GB
condition2 = swap.used > 1024**3 and mem.free > 1024**3 * 1.5 # 1GB
return condition1 or condition2
def clear_swap():
"""安全清空SWAP空间"""
try:
subprocess.run(['swapoff', '-a'], check=True)
subprocess.run(['swapon', '-a'], check=True)
logger.info(f"[{datetime.now()}] 已成功重置SWAP空间")
except subprocess.CalledProcessError as e:
logger.info(f"SWAP操作失败: {str(e)}")
def main(interval=3600):
while True:
# 定时清理页面缓存
clear_page_cache()
# 条件性清理SWAP
if should_clear_swap():
clear_swap()
else:
logger.info(f"[{datetime.now()}] 不满足SWAP清理条件")
time.sleep(interval)
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--interval', type=int, default=3600,
help='检测间隔(秒)')
args = parser.parse_args()
vv.UruLogConfig.config("mem_clean", [], "mem_clean.log", log_root="/home/vvd/")
logger.info(f"""启动内存管理服务:
检测间隔: {args.interval}秒
""")
main(args.interval)
pass
