分享个抢占资源的无耻脚本

万里云

其实一开始要我写这么个脚本我是拒绝的，因为不能让我写我就写，写完后独占机器，这样别人一定会骂我“你怎么这么无耻”。

后来发现不写不行，和那些每隔十几分钟就查看服务器资源的人相比，不花点心思，就一辈子都别想抢到资源了。

1. #! /bin/bash
   
2. 
   
3. function runjob()
   
4. {
   
5.     local ncpu=$1
   
6.     # add your jobscripts here
   
7.     echo "There are $ncpu idle CPU"
   
8.     wait
   
9. }
   
10. 
    
11. function main()
    
12. {
    
13.     # programs to watch
    
14.     local proglist="pw yambo vasp"
    
15. 
    
16.     # toal number of cpus on this machine
    
17.     # and min/max number of cpus for the job
    
18.     local ncputot=16
    
19.     local ncpumin=16
    
20.     local ncpumax=16
    
21. 
    
22.     # time step and maximum number of loop
    
23.     local dt=10
    
24.     local nmax=8640
    
25.    
    
26. 
    
27.     # main loop
    
28.     for i in $(seq 1 $nmax); do
    
29.         # initialize ncpuocc and ncpuemp
    
30.         local ncpuocc=0
    
31.         local ncpuemp=0
    
32.         # determine ncpuocc and ncpuemp
    
33.         for prog in $proglist; do
    
34.             local ncpui=$(ps -e | grep "$prog" | wc -l)
    
35.             ncpuocc=$(echo $ncpui $ncpuocc | awk '{print $1+$2}')
    
36.         done
    
37.         ncpuemp=$(echo $ncputot $ncpuocc | awk '{print $1-$2}')
    
38.         # if ncpuemp >= ncpumin submit the job;
    
39.         # else, wait for the next cycle.
    
40.         if [ $ncpuemp -ge $ncpumin  ]; then
    
41.             # if ncpuemp <= ncpumax, use ncpuemp cpus;
    
42.             # else, use ncpumax cpus;
    
43.             if [ $ncpuemp -le $ncpumax ]; then
    
44.                 runjob $ncpuemp
    
45.                 break
    
46.             else
    
47.                 runjob $ncpumax
    
48.                 break
    
49.             fi
    
50.         else
    
51.             sleep $dt
    
52.         fi
    
53.     done
    
54.    
    
55. 
    
56.     # exit
    
57.     return 0
    
58. }
    
59. 
    
60. main
    

复制代码

算法不复杂，就是每隔一定时间根据当前机器上总核数和要监视的程序列表计算出可用核数，如果可用核数满足要求则运行任务。

需要修改的地方有这么几处：

runjob这个函数，把计算用的命令添加进去。这个函数已经从main函数接收了当前可用核数作为参数。

ncputot是该机器上CPU总核数，ncpumin是自己运行的任务所需最小核数（如程序算出来的可用核数不满足此条件则继续等待直至满足，为资源需求量大的程序设计），ncpumax是自己任务所用的最大核数（如果可用核数小于这个数，则使用可用核数，否则使用最大核数，避免占用太多资源）。

proglist是待监视的程序列表，main函数会统计由该列表中的程序占用的核数，与ncputot作差后得到可用核数。

dt是每隔多少秒检查一次资源，nmax是总共检查多少次。dt=10，nmax=8640就是在一天之内每隔十秒查看一次资源，如果运气不是太差，没人抢得过你。
此脚本适用于多人共用同一服务器的情形，不适用于集群。

wull

果然很无耻 哈哈
可见  会写脚本是多么重要

smutao

如果有队列系统 怎么办

万里云

smutao 发表于 2015-7-10 04:24
如果有队列系统 怎么办

这个就是针对无队列系统的。

有队列的不会出现抢资源的情况，所以用不到。

qczgzly

收藏备用～～

bay618

也分享一个自己写的吧, 实际上定义了两条命令loadint和waitload

1. loadint () { echo " `uptime 2> /dev/null | awk -F ':|,' '{print($(NF-2))}'` / 0.99 " | bc ; }
   
2. waitload () { while [ `loadint` -gt $1 ] ; do sleep 180 ; done ; }

复制代码

loadint 可以输出当前的负载, 以整数输出.
waitload 后面跟一个整数, 每隔一段时间检测一次, 当负载小于这个整数时候结束. 配合shell 的用法可以提交了.
比如下面的用法提交g09任务

1. waitload 4 && g09 aaa.gjf &

复制代码

exity

If同时有两个人在用这个脚本会发生什么？

pika02

exity 发表于 2019-12-8 10:38
If同时有两个人在用这个脚本会发生什么？

你可以想象一下几万人同时用抢票软件的场景

小范范1989

这个无耻的脚本——————————我好喜欢。

pinpo

拷贝一份备用

欢乐多

还是有自己的服务器好啊