perl - 使用 Parallel::ForkManager 并行执行命令

Question

我想知道我对以下脚本/逻辑的理解是否正确。

我有节点列表，我需要通过对服务器执行 SSH 来利用我拥有的服务器数量在每个节点上运行某个命令，这意味着该过程应该并行发生。

我有 node_list.txt包含节点列表的文件:

node1
node2
.
.
node49
node50

我已经定义了一个数组中的服务器数量 @hosts我应该在哪里执行 SSH 并通过拆分 node_file.txt 对每个节点执行命令在可用服务器中分成相等数量的部分(称为 $node_list_X.txt)。

一旦我有了这些文件( node_list_1.txt ， node_list_2.txt ， node_list_3.txt ， node_list_4.txt )，我将登录到每个已经定义的服务器并通过传递 node_list_X.txt 在每个主机上执行某些命令并行文件。

为了并行执行，我使用 Parallel::ForkManager Perl 模块。

所以，让我们在每个主机中说 -

192.168.0.1 -> node_list_1.txt (13 nodes)
192.168.0.2 -> node_list_2.txt (13 nodes)
192.168.0.3 -> node_list_3.txt (12 nodes)
192.168.0.4 -> node_list_4.txt (12 nodes)

将并行运行。

脚本如下:

...
my @hosts = ("192.168.0.1", "192.168.0.2", "192.168.0.3","192.168.0.4");

open(my $node_fh, '<', $node_file)
        or die "can't open $node_file: $!";

my @lines =  <$node_fh>;

my %Files;

my $num_buckets = scalar @hosts;

my $per_bucket = int( @lines / $num_buckets );
my $num_extras =      @lines % $num_buckets;
my $path = "/home/user/vinod/test/";

for my $bucket_num (0..$num_buckets-1) {
   my $num_lines = $per_bucket;
   if ($num_extras) {
      ++$num_lines;
      --$num_extras;
   }

   last if($num_lines == 0);
   my $qfn = $path."node_list_${bucket_num}.txt";
   open(my $fh, '>', $qfn)
      or die("Can't create \"$qfn\": $!\n");

   $fh->print(splice(@lines, 0, $num_lines));
   $Files{$bucket_num} = $qfn;
}
print Dumper(\%Files);

my $command = #"defining my command here";

my $pm = Parallel::ForkManager->new(5);
my $ssh;

DATA_LOOP:
foreach my $n (0..$num_buckets-1) {
    if( exists $Files{$n} ) {
        my $pid = $pm->start and next DATA_LOOP;

        $command_to_execute = $command." ".$Files{$n};
        $ssh = SSH_Connection( $hosts[$n-1], "user", "password" );
        $result = $ssh->capture($command_to_execute);
      
        $pm->finish;       
    }
}
$pm->wait_all_children;
undef $ssh;

#SSH Connect
sub SSH_Connection {
    my ( $host, $user, $passwd ) = @_;
    my $ssh = Net::OpenSSH->new($host,
                                user => $user,
                                password => $passwd,
                                master_opts => [-o => "StrictHostKeyChecking=no"]
    );
    $ssh->error and die "Couldn't establish SSH connection: ". $ssh->error;
    return $ssh;
}

这里一切正常。

当我定义 $pm 时对象，并行进程设置为 5。

my $pm = Parallel::ForkManager->new(5);

这是否意味着在特定服务器(例如:192.168.0.1)中一次应该运行 5 个并行进程。意味着它应该从 node_list_1.txt 中获取 5 个节点(共 13 个)文件并执行命令？

我的理解正确吗？如果不是，那么在每个服务器中使用多线程并行运行命令的可能解决方案是什么？

Answer 1

Does this means at a time in a perticular server (Ex:192.168.0.1) it should run 5 parallel process.

没有。 P::FM 对服务器一无所知。它管理进程，->new(5) 意味着 ->start 将等待它创建的进程之一完成，然后再创建一个新进程，如果其中有 5 个是仍在执行。

what could be the possible solution to run the command in each server parallelly with multi-threading?

假设您通常指的是多任务而不是专门的多线程(因为您没有使用线程)，则可以按如下方式为每个主机创建一个进程:

my %children;
my $error = 0;
for my $host (@hosts) {
    my $pid = fork();
    if (!defined($pid)) {
       warn("Can't execute on $host: Can't fork: $!\n");
       next;
    }

    if ($pid) {
       ++$children{$pid};
       next;
    }

    if (!eval {
       do_it($host);
       return 1;  # No exception
    }) {
       warn("Error executing commands on $host: $@");
    }
}

while (%children) {
   ( my $pid = wait() ) >= 0
      or die("Can't wait: $!\n");

   delete($children{$pid});   
}