Question

我有一个我无法解决的问题，所以我来找你。

我需要编写一个程序来读取所有进程，程序必须按用户对它们进行排序，对于每个用户，它必须显示使用了多少内存。

例如：

user1：120MB
user2：300MB
user3：50MB
总计：470MB

我想用ps aux命令做这个，然后用awk命令输出pid和用户。然后使用pmap我只需要获得进程的总内存使用量。

Answer 1

试试这个可以解决问题的脚本：

#!/bin/bash
function mem_per_user {
    # take username as only parameter
    local user=$1
    # get all pid's of a specific user
    # you may elaborate the if statement in awk obey your own rules
    pids=`ps aux | awk -v username=$user '{if ($1 == username) {print $2}}'`

    local totalmem=0
    for pid in $pids
    do
        mem=`pmap $pid | tail -1 | \
            awk '{pos = match($2, /([0-9]*)K/, mem); if (pos > 0) print mem[1]}'`
        # when variable properly set
        if [ ! -z $mem ]
        then
            totalmem=$(( totalmem + $mem))
        fi
    done

    echo $totalmem
}

total_mem=0
for i in `seq 1 $#`
do
    per_user_memory=0
    eval username=\$$i
    per_user_memory=$(mem_per_user $username)
    total_mem=$(( $total_mem + $per_user_memory))

    echo "$username: $per_user_memory KB"
done
echo "Total: $total_mem KB"

祝你好运！

Answer 2

只是一点点更新，用户会自动选择

#!/bin/bash
function mem_per_user {
    # take username as only parameter
    local user=$1
    # get all pid's of a specific user
    # you may elaborate the if statement in awk obey your own rules
    pids=`ps aux | awk -v username=$user '{if ($1 == username) {print $2}}'`

    local totalmem=0
    for pid in $pids
    do
        mem=`pmap $pid | tail -1 | \
            awk '{pos = match($2, /([0-9]*)K/, mem); if (pos > 0) print mem[1]}'`
        # when variable properly set
        if [ ! -z $mem ]
        then
            totalmem=$(( totalmem + $mem))
        fi
    done

    echo $totalmem
}

total_mem=0
for username in `ps aux | awk '{ print $1 }' | tail -n +2 | sort | uniq`
do
    per_user_memory=0
    per_user_memory=$(mem_per_user $username)
    if [ "$per_user_memory" -gt 0 ]
    then
       total_mem=$(( $total_mem + $per_user_memory))

       echo "$username: $per_user_memory KB"
    fi
done
echo "Total: $total_mem KB"

Answer 3

您可以使用子进程模块在python中访问shell命令。它允许您生成子进程并连接到out / in / error。您可以执行ps -aux命令并在python中解析输出。

check out the docs here

Answer 4

这是我的版本。我认为Tim的版本工作不正常，KB中的值太大了。我认为应该使用pmap -x命令的RSS列来提供更准确的值。但请注意，您无法始终获得正确的值，因为进程可以共享内存。阅读此A way to determine a process's "real" memory usage, i.e. private dirty RSS?

#!/bin/bash
 if [ "$(id -u)" != "0" ]; then
 echo "WARNING: you have to run as root if you want to see all users"
 fi
echo "Printing only users that current memmory usage > 0 Kilobytes "
all=0
for username in `ps aux | awk '{ print $1 }' | tail -n +2 | sort | uniq`
do
 pids=`ps aux | grep $username | awk -F" " '{print $2}'`
 total_memory=0
 for pid in $pids
 do
  process_mem=`pmap -x $pid | tail -1 | awk -F" " '{print $4}'`

  if [ ! -z $process_mem ] 
  then #don't try to add if string has no length
   total_memory=$((total_memory+$process_mem))
  fi
 done
#print only those that use any memmory
if [ $total_memory -gt 0 ]
then
 total_memory=$((total_memory/(1024)))
 echo "$username : $total_memory MB"
 all=$((all+$total_memory))
 fi
done
echo "----------------------------------------"
echo "Total: $all MB"
echo "WARNING: Use at your own risk"

Bash - 内存使用情况

4 个答案: