[work] Linux Shell多进程并发以及并发数控制

红太狼 2022-04-02 03:46 168阅读 0赞

# 1. 基础知识准备 #

## 1.1. linux后台进程 ##

Unix是一个多任务系统，允许多用户同时运行多个程序。shell的元字符`&`提供了在后台运行不需要键盘输入的程序的方法。输入命令后，其后紧跟`&`字符，该命令就会被送往到linux后台执行，而终端又可以继续输入下一个命令了。  
比如：

sh a.sh &
    sh b.sh &
    sh c.sh &

这三个命令就会被**同时**送往linux后台执行，在这个程度上，认为这三个命令**并发执行**了。

## 1.2. linux文件描述符 ##

文件描述符（缩写fd）在形式上是一个非负整数。实际上，它是一个索引值，指向内核为每一个进程所维护的该进程打开文件的记录表。当程序打开一个现有文件或者创建一个新文件时，内核向进程返回一个文件描述符。每一个unix进程，都会拥有三个标准的文件描述符，来对应三种不同的流：

<table> 
 <thead> 
  <tr> 
   <th>文件描述符</th> 
   <th>名称</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>0</td> 
   <td>Standard Input</td> 
  </tr> 
  <tr> 
   <td>1</td> 
   <td>Standard Output</td> 
  </tr> 
  <tr> 
   <td>2</td> 
   <td>Standard Error</td> 
  </tr> 
 </tbody> 
</table>

每一个文件描述符会对应一个打开文件，同时，不同的文件描述符也可以对应同一个打开文件；同一个文件可以被不同的进程打开，也可以被同一个进程多次打开。

在`/proc/PID/fd`中，列举了进程`PID`所拥有的文件描述符，例如

#!/bin/bash
    source /etc/profile;
    
    # $$表示当前进程的PID
    PID=$$
    
    # 查看当前进程的文件描述符指向
    ll /proc/$PID/fd
    echo "-------------------";echo
    
    # 文件描述符1与文件tempfd1进行绑定
    ( [ -e ./tempfd1 ] || touch ./tempfd1 ) && exec 1<>./tempfd1
    
    # 查看当前进程的文件描述符指向
    ll /proc/$PID/fd
    echo "-------------------";echo;

[ouyangyewei@localhost learn_linux]$ sh learn_redirect.sh 
    total 0
    lrwx------. 1 ouyangyewei ouyangyewei 64 Jan  4 22:17 0 -> /dev/pts/0
    lrwx------. 1 ouyangyewei ouyangyewei 64 Jan  4 22:17 1 -> /dev/pts/0
    lrwx------. 1 ouyangyewei ouyangyewei 64 Jan  4 22:17 2 -> /dev/pts/0
    lr-x------. 1 ouyangyewei ouyangyewei 64 Jan  4 22:17 255 -> /home/ouyangyewei/workspace/learn_linux/learn_redirect.sh
    -------------------
    
    [ouyangyewei@localhost learn_linux]$ cat tempfd1 
    total 0
    lrwx------. 1 ouyangyewei ouyangyewei 64 Jan  4 22:17 0 -> /dev/pts/0
    lrwx------. 1 ouyangyewei ouyangyewei 64 Jan  4 22:17 1 -> /home/ouyangyewei/workspace/learn_linux/tempfd1
    lrwx------. 1 ouyangyewei ouyangyewei 64 Jan  4 22:17 2 -> /dev/pts/0
    lr-x------. 1 ouyangyewei ouyangyewei 64 Jan  4 22:17 255 -> /home/ouyangyewei/workspace/learn_linux/learn_redirect.sh
    -------------------

上述的例子中第12行，将文件描述符1与文件`tempfile`进行了绑定，此后，文件描述符1指向了`tempfile`文件，标准输出被重定向到了文件`tempfile`中。

## 1.3. linux管道 ##

在Unix或类Unix操作系统中，管道是一个由标准输入输出链接起来的进程集合，因此，每一个进程的输出将直接作为下一个进程的输入，

linux管道包含两种：

*  匿名管道
 *  命名管道

***管道有一个特点，如果管道中没有数据，那么取管道数据的操作就会滞留，直到管道内进入数据，然后读出后才会终止这一操作；同理，写入管道的操作如果没有读取管道的操作，这一动作就会滞留。***

### 1.3.1. 匿名管道 ###

在Unix或类Unix操作系统的命令行中，匿名管道使用ASCII中垂直线`|`作为匿名管道符，匿名管道的两端是两个普通的，匿名的，打开的文件描述符：一个**只读端**和一个**只写端**，这就让其它进程无法连接到该匿名管道。

例如：

cat file | less

为了执行上面的指令，Shell创建了两个进程来分别执行`cat`和`less`。下图展示了这两个进程是如何使用管道的：  
![unix\_unnamed\_pipe][unix_unnamed_pipe]  
有一点值得注意的是两个进程都连接到了管道上，这样写入进程`cat`就将其标准输出（文件描述符为`fd 1`）连接到了管道的写入端，读取进程`less`就将其标准输入（文件描述符为`fd 0`）连接到了管道的读入端。实际上，这两个进程并不知道管道的存在，它们只是从标准文件描述符中读取数据和写入数据。shell必须要完成相关的工作。

### 1.3.2. 命名管道（FIFO，First In First Out） ###

命名管道也称FIFO，从语义上来讲，FIFO其实与匿名管道类似，但值得注意：

*  在文件系统中，FIFO拥有名称，并且是以设备特俗文件的形式存在的；
 *  任何进程都可以通过FIFO共享数据；
 *  除非FIFO两端同时有读与写的进程，否则FIFO的数据流通将会阻塞；
 *  匿名管道是由shell自动创建的，存在于内核中；而FIFO则是由程序创建的（比如`mkfifo`命令），存在于文件系统中；
 *  匿名管道是单向的字节流，而FIFO则是双向的字节流；

比如，可以利用FIFO实现单服务器、多客户端的应用程序:  
![unix\_named\_pipe][unix_named_pipe]

--------------------

有了上面的知识准备，现在可以开始讲述，**linux多进程并发时，如何控制每次并发的进程数。**

# 2. linux多进程并发数控制 #

最近小A需要生产2015年全年的KPI数据报表，现在小A已经将生产脚本写好了，生产脚本一次只能生产指定一天的KPI数据，假设跑一次生产脚本需要5分钟，那么：  
\* 如果是循环顺序执行，那么需要时间：5 \* 365 = 1825 分钟，约等于 6 天  
\* 如果是一次性放到linux后台并发执行，365个后台任务，系统可承受不住哦！

既然不能一次性把365个任务放到linux后台执行，那么，能不能实现自动地每次将N个任务放到后台并发执行呢？当然是可以的啦。

#! /bin/bash
    source /etc/profile;
    
    # -----------------------------
    
    tempfifo=$$.fifo        # $$表示当前执行文件的PID
    begin_date=$1           # 开始时间
    end_date=$2             # 结束时间
    
    if [ $# -eq 2 ] 
    then
        if [ "$begin_date" \> "$end_date" ]
        then
            echo "Error! $begin_date is greater than $end_date"
            exit 1;
        fi
    else
        echo "Error! Not enough params."
        echo "Sample: sh loop_kpi 2015-12-01 2015-12-07"
        exit 2;
    fi
    
    # -----------------------------
    
    trap "exec 1000>&-;exec 1000<&-;exit 0" 2
    mkfifo $tempfifo
    exec 1000<>$tempfifo
    rm -rf $tempfifo
    
    for ((i=1; i<=8; i++))
    do
        echo >&1000
    done
    
    while [ $begin_date != $end_date ]
    do
        read -u1000
        {
            echo $begin_date
            hive -f kpi_report.sql --hivevar date=$begin_date
            echo >&1000
        } &
    
        begin_date=`date -d "+1 day $begin_date" +"%Y-%m-%d"`
    done
    
    wait
    echo "done!!!!!!!!!!"

*  第6～22行：比如：`sh loop_kpi_report.sh 2015-01-01 2015-12-01`：
    
     *  `$1`表示脚本入参的第一个参数，等于2015-01-01
     *  `$2`表示脚本入参的第二个参数，等于2015-12-01
     *  `$#`表示脚本入参的个数，等于2
     *  第13行用于比较传入的两个日期的大小，`\>`是转义
 *  第26行：表示在脚本运行过程中，如果接收到`Ctrl+C`中断命令，则关闭文件描述符1000的读写，并正常退出
    
     *  `exec 1000>&-;`表示关闭文件描述符1000的写
     *  `exec 1000<&-;`表示关闭文件描述符1000的读
     *  trap是捕获中断命令
 *  第27～29行：
    
     *  第27行，创建一个管道文件
     *  第28行，将文件描述符1000与FIFO进行绑定，`<`读的绑定，`>`写的绑定，`<>`则标识对文件描述符1000的所有操作等同于对管道文件`$tempfifo`的操作
     *  第29行，可能会有这样的疑问：为什么不直接使用管道文件呢？事实上这并非多此一举，管道的一个重要特性，就是读写必须同时存在，缺失某一个操作，另一个操作就是滞留，而第28行的绑定文件描述符（读、写绑定）正好解决了这个问题
 *  第31～34行：对文件描述符1000进行写入操作。通过循环写入8个空行，这个8就是我们要定义的后台并发的线程数。为什么是写空行而不是写其它字符？因为管道文件的读取，是以行为单位的
 *  第37～42行：
    
     *  第37行，`read -u1000`的作用就是读取管道中的一行，在这里就是读取一个空行；每次读取管道就会减少一个空行
     *  第39～41行，注意到第42行结尾的`&`吗？它表示进程放到linux后台中执行
     *  第41行，执行完后台任务之后，往文件描述符1000中写入一个空行。这是关键所在了，由于`read -u1000`每次操作，都会导致管道减少一个空行，当linux后台放入了8个任务之后，由于文件描述符1000没有可读取的空行，将导致`read -u1000`一直处于等待。

--------------------

# 3. 参考资料 #

*  Unix Power Tools
 *  UNIX系统编程手册
 *  UNIX管道：[https://zh.wikipedia.org/wiki/%E7%AE%A1%E9%81%93\_(Unix)][https_zh.wikipedia.org_wiki_E7_AE_A1_E9_81_93_Unix]

[unix_unnamed_pipe]: /images/20220402/6fdef662197b43a58398b1108eba5f7a.png
[unix_named_pipe]: /images/20220402/d9308e5f33144b7ba1c5570ab3d7de50.png
[https_zh.wikipedia.org_wiki_E7_AE_A1_E9_81_93_Unix]: https://zh.wikipedia.org/wiki/%E7%AE%A1%E9%81%93_%28Unix%29