yigitcolakoglu
/
fr1nge.xyz
1
0
Fork 0
Code Issues 0 Pull Requests 0 Projects 0 Releases 0 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 Commits
1 Branch
8.0 MiB
 
 
 
 
 
Tree: 6d1a366f0d
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '6d1a366f0d'
${ noResults }
fr1nge.xyz/public/posts/index.xml

343 lines
16 KiB
Raw Blame History

<?xml version="1.0" encoding="utf-8" standalone="yes"?>
<rss version="2.0" xmlns:atom="http://www.w3.org/2005/Atom">
<channel>
<title>Posts on Fr1nge&#39;s Personal Blog</title>
<link>http://fr1nge.xyz/posts/</link>
<description>Recent content in Posts on Fr1nge&#39;s Personal Blog</description>
<generator>Hugo -- gohugo.io</generator>
<language>en-us</language>
<copyright>Yigit Colakoglu</copyright>
<lastBuildDate>Wed, 05 May 2021 17:08:12 +0300</lastBuildDate><atom:link href="http://fr1nge.xyz/posts/index.xml" rel="self" type="application/rss+xml" />
<item>
<title>Supercharge Your Bash Scripts with Multiprocessing</title>
<link>http://fr1nge.xyz/posts/supercharge-your-bash-scripts-with-multiprocessing/</link>
<pubDate>Wed, 05 May 2021 17:08:12 +0300</pubDate>
<guid>http://fr1nge.xyz/posts/supercharge-your-bash-scripts-with-multiprocessing/</guid>
<description>Bash is a great tool for automating tasks and improving you work flow. However, it is SLOW. Adding multiprocessing to the scripts you write can improve the performance greatly.
What is multiprocessing? In the simplest terms, multiprocessing is the principle of splitting the computations or jobs that a script has to do and running them on different processes. In even simpler terms however, multiprocessing is the computer science equivalent of hiring more than one worker when you are constructing a building.</description>
<content>&lt;p&gt;Bash is a great tool for automating tasks and improving you work flow. However,
it is &lt;em&gt;&lt;strong&gt;SLOW&lt;/strong&gt;&lt;/em&gt;. Adding multiprocessing to the scripts you write can improve
the performance greatly.&lt;/p&gt;
&lt;h2 id=&#34;what-is-multiprocessing&#34;&gt;What is multiprocessing?&lt;/h2&gt;
&lt;p&gt;In the simplest terms, multiprocessing is the principle of splitting the
computations or jobs that a script has to do and running them on different
processes. In even simpler terms however, multiprocessing is the computer
science equivalent of hiring more than one
worker when you are constructing a building.&lt;/p&gt;
&lt;h3 id=&#34;introducing-&#34;&gt;Introducing &amp;ldquo;&amp;amp;&amp;rdquo;&lt;/h3&gt;
&lt;p&gt;While implementing multiprocessing the sign &lt;code&gt;&amp;amp;&lt;/code&gt; is going to be our greatest
friend. It is an essential sign if you are writing bash scripts and a very
useful tool in general when you are in the terminal. What &lt;code&gt;&amp;amp;&lt;/code&gt; does is that it
makes the command you added it to the end of run in the background and allows
the rest of the script to continue running as the command runs in the
background. One thing to keep in mind is that since it creates a fork of the
process you ran the command on, if you change a variable that the command in the
background uses while it runs, it will not be affected. Here is a simple
example:&lt;/p&gt;
&lt;div class=&#34;collapsable-code&#34;&gt;
&lt;input id=&#34;1&#34; type=&#34;checkbox&#34; /&gt;
&lt;label for=&#34;1&#34;&gt;
&lt;span class=&#34;collapsable-code__language&#34;&gt;bash&lt;/span&gt;
&lt;span class=&#34;collapsable-code__toggle&#34; data-label-expand=&#34;Show&#34; data-label-collapse=&#34;Hide&#34;&gt;&lt;/span&gt;
&lt;/label&gt;
&lt;pre class=&#34;language-bash&#34; &gt;&lt;code&gt;
foo=&amp;#34;yeet&amp;#34;
function run_in_background(){
sleep 0.5
echo &amp;#34;The value of foo in the function run_in_background is $foo&amp;#34;
}
run_in_background &amp;amp; # Spawn the function run_in_background in the background
foo=&amp;#34;YEET&amp;#34;
echo &amp;#34;The value of foo changed to $foo.&amp;#34;
wait # wait for the background process to finish
&lt;/code&gt;&lt;/pre&gt;
&lt;/div&gt;
&lt;p&gt;This should output:&lt;/p&gt;
&lt;pre&gt;&lt;code&gt;The value of foo changed to YEET.
The value of foo in here is yeet
&lt;/code&gt;&lt;/pre&gt;&lt;p&gt;As you can see, the value of &lt;code&gt;foo&lt;/code&gt; did not change in the background process even though
we changed it in the main function.&lt;/p&gt;
&lt;h2 id=&#34;baby-steps&#34;&gt;Baby steps&amp;hellip;&lt;/h2&gt;
&lt;p&gt;Just like anything related to computer science, there is more than one way of
achieving our goal. We are going to take the easier, less intimidating but less
efficient route first before moving on to the big boy implementation. Let&amp;rsquo;s open up vim and get to scripting!
First of all, let&amp;rsquo;s write a very simple function that allows us to easily test
our implementation:&lt;/p&gt;
&lt;div class=&#34;collapsable-code&#34;&gt;
&lt;input id=&#34;1&#34; type=&#34;checkbox&#34; /&gt;
&lt;label for=&#34;1&#34;&gt;
&lt;span class=&#34;collapsable-code__language&#34;&gt;bash&lt;/span&gt;
&lt;span class=&#34;collapsable-code__toggle&#34; data-label-expand=&#34;Show&#34; data-label-collapse=&#34;Hide&#34;&gt;&lt;/span&gt;
&lt;/label&gt;
&lt;pre class=&#34;language-bash&#34; &gt;&lt;code&gt;
function tester(){
# A function that takes an int as a parameter and sleeps
echo &amp;#34;$1&amp;#34;
sleep &amp;#34;$1&amp;#34;
echo &amp;#34;ENDED $1&amp;#34;
}
&lt;/code&gt;&lt;/pre&gt;
&lt;/div&gt;
&lt;p&gt;Now that we have something to run in our processes, we now need to spawn several
of them in controlled manner. Controlled being the keyword here. That&amp;rsquo;s because
each system has a maximum number of processes that can be spawned (You can find
that out with the command &lt;code&gt;ulimit -u&lt;/code&gt;). In our case, we want to limit the
processes being ran to the variable &lt;code&gt;num_processes&lt;/code&gt;. Here is the implementation:&lt;/p&gt;
&lt;div class=&#34;collapsable-code&#34;&gt;
&lt;input id=&#34;1&#34; type=&#34;checkbox&#34; /&gt;
&lt;label for=&#34;1&#34;&gt;
&lt;span class=&#34;collapsable-code__language&#34;&gt;bash&lt;/span&gt;
&lt;span class=&#34;collapsable-code__toggle&#34; data-label-expand=&#34;Show&#34; data-label-collapse=&#34;Hide&#34;&gt;&lt;/span&gt;
&lt;/label&gt;
&lt;pre class=&#34;language-bash&#34; &gt;&lt;code&gt;
num_processes=$1
pcount=0
for i in {1..10}; do
((pcount=pcount%num_processes));
((pcount&amp;#43;&amp;#43;==0)) &amp;amp;&amp;amp; wait
tester $i &amp;amp;
done
&lt;/code&gt;&lt;/pre&gt;
&lt;/div&gt;
&lt;p&gt;What this loop does is that it takes the number of processes you would like to
spawn as an argument and runs &lt;code&gt;tester&lt;/code&gt; in that many processes. Go ahead and test it out!
You might notice however that the processes are run int batches. And the size of
batches is the &lt;code&gt;num_processes&lt;/code&gt; variable. The reason this happens is because
every time we spawn &lt;code&gt;num_processes&lt;/code&gt; processes, we &lt;code&gt;wait&lt;/code&gt; for all the processes
to end. This implementation is not a problem in itself, there are many cases
where you can use this implementation and it works perfectly fine. However, if
you don&amp;rsquo;t want this to happen, we have to dump this naive approach all together
and improve our tool belt.&lt;/p&gt;
&lt;h2 id=&#34;real-chads-use-job-pools&#34;&gt;Real Chads use Job Pools&lt;/h2&gt;
&lt;p&gt;The solution to the bottleneck that was introduced in our previous approach lies
in using job pools. Job pools are where jobs created by a main process get sent
and wait to get executed. This approach solves our problems because instead of
spawning a new process for every copy and waiting for all the processes to
finish we instead only create a set number of processes(workers) which
continuously pick up jobs from the job pool not waiting for any other process to finish.
Here is the implementation that uses job pools. Brace yourselves, because it is
kind of complicated.&lt;/p&gt;
&lt;div class=&#34;collapsable-code&#34;&gt;
&lt;input id=&#34;1&#34; type=&#34;checkbox&#34; /&gt;
&lt;label for=&#34;1&#34;&gt;
&lt;span class=&#34;collapsable-code__language&#34;&gt;bash&lt;/span&gt;
&lt;span class=&#34;collapsable-code__toggle&#34; data-label-expand=&#34;Show&#34; data-label-collapse=&#34;Hide&#34;&gt;&lt;/span&gt;
&lt;/label&gt;
&lt;pre class=&#34;language-bash&#34; &gt;&lt;code&gt;
job_pool_end_of_jobs=&amp;#34;NO_JOB_LEFT&amp;#34;
job_pool_job_queue=/tmp/job_pool_job_queue_$$
job_pool_progress=/tmp/job_pool_progress_$$
job_pool_pool_size=-1
job_pool_nerrors=0
function job_pool_cleanup()
{
rm -f ${job_pool_job_queue}
rm -f ${job_pool_progress}
}
function job_pool_exit_handler()
{
job_pool_stop_workers
job_pool_cleanup
}
function job_pool_worker()
{
local id=$1
local job_queue=$2
local cmd=
local args=
exec 7&amp;lt;&amp;gt; ${job_queue}
while [[ &amp;#34;${cmd}&amp;#34; != &amp;#34;${job_pool_end_of_jobs}&amp;#34; &amp;amp;&amp;amp; -e &amp;#34;${job_queue}&amp;#34; ]]; do
flock --exclusive 7
IFS=$&amp;#39;\v&amp;#39;
read cmd args &amp;lt;${job_queue}
set -- ${args}
unset IFS
flock --unlock 7
if [[ &amp;#34;${cmd}&amp;#34; == &amp;#34;${job_pool_end_of_jobs}&amp;#34; ]]; then
echo &amp;#34;${cmd}&amp;#34; &amp;gt;&amp;amp;7
else
{ ${cmd} &amp;#34;$@&amp;#34; ; }
fi
done
exec 7&amp;gt;&amp;amp;-
}
function job_pool_stop_workers()
{
echo ${job_pool_end_of_jobs} &amp;gt;&amp;gt; ${job_pool_job_queue}
wait
}
function job_pool_start_workers()
{
local job_queue=$1
for ((i=0; i&amp;lt;${job_pool_pool_size}; i&amp;#43;&amp;#43;)); do
job_pool_worker ${i} ${job_queue} &amp;amp;
done
}
function job_pool_init()
{
local pool_size=$1
job_pool_pool_size=${pool_size:=1}
rm -rf ${job_pool_job_queue}
rm -rf ${job_pool_progress}
touch ${job_pool_progress}
mkfifo ${job_pool_job_queue}
echo 0 &amp;gt;${job_pool_progress} &amp;amp;
job_pool_start_workers ${job_pool_job_queue}
}
function job_pool_shutdown()
{
job_pool_stop_workers
job_pool_cleanup
}
function job_pool_run()
{
if [[ &amp;#34;${job_pool_pool_size}&amp;#34; == &amp;#34;-1&amp;#34; ]]; then
job_pool_init
fi
printf &amp;#34;%s\v&amp;#34; &amp;#34;$@&amp;#34; &amp;gt;&amp;gt; ${job_pool_job_queue}
echo &amp;gt;&amp;gt; ${job_pool_job_queue}
}
function job_pool_wait()
{
job_pool_stop_workers
job_pool_start_workers ${job_pool_job_queue}
}
&lt;/code&gt;&lt;/pre&gt;
&lt;/div&gt;
&lt;p&gt;Ok&amp;hellip; But that the actual fuck is going in here???&lt;/p&gt;
&lt;h3 id=&#34;fifo-and-flock&#34;&gt;fifo and flock&lt;/h3&gt;
&lt;p&gt;In order to understand what this code is doing, you first need to understand two
key commands that we are using, &lt;code&gt;fifo&lt;/code&gt; and &lt;code&gt;flock&lt;/code&gt;. Despite their complicated
names, they are actually quite simple. Let&amp;rsquo;s check their man pages to figure out
their purposes, shall we?&lt;/p&gt;
&lt;h4 id=&#34;man-fifo&#34;&gt;man fifo&lt;/h4&gt;
&lt;p&gt;fifo&amp;rsquo;s man page tells us that:&lt;/p&gt;
&lt;pre&gt;&lt;code&gt;NAME
fifo - first-in first-out special file, named pipe
DESCRIPTION
A FIFO special file (a named pipe) is similar to a pipe, except that
it is accessed as part of the filesystem. It can be opened by multiple
processes for reading or writing. When processes are exchanging data
via the FIFO, the kernel passes all data internally without writing it
to the filesystem. Thus, the FIFO special file has no contents on the
filesystem; the filesystem entry merely serves as a reference point so
that processes can access the pipe using a name in the filesystem.
&lt;/code&gt;&lt;/pre&gt;&lt;p&gt;So put in &lt;strong&gt;very&lt;/strong&gt; simple terms, a fifo is a named pipe that can allows
communication between processes. Using a fifo allows us to loop through the jobs
in the pool without having to delete them manually, because once we read them
with &lt;code&gt;read cmd args &amp;lt; ${job_queue}&lt;/code&gt;, the job is out of the pipe and the next
read outputs the next job in the pool. However the fact that we have multiple
processes introduces one caveat, what if two processes access the pipe at the
same time? They would run the same command and we don&amp;rsquo;t want that. So we resort
to using &lt;code&gt;flock&lt;/code&gt;.&lt;/p&gt;
&lt;h4 id=&#34;man-flock&#34;&gt;man flock&lt;/h4&gt;
&lt;p&gt;flock&amp;rsquo;s man page defines it as:&lt;/p&gt;
&lt;pre&gt;&lt;code&gt; SYNOPSIS
flock [options] file|directory command [arguments]
flock [options] file|directory -c command
flock [options] number
DESCRIPTION
This utility manages flock(2) locks from within shell scripts or from
the command line.
The first and second of the above forms wrap the lock around the
execution of a command, in a manner similar to su(1) or newgrp(1).
They lock a specified file or directory, which is created (assuming
appropriate permissions) if it does not already exist. By default, if
the lock cannot be immediately acquired, flock waits until the lock is
available.
The third form uses an open file by its file descriptor number. See
the examples below for how that can be used.
&lt;/code&gt;&lt;/pre&gt;&lt;p&gt;Cool, translated to modern English that us regular folks use, &lt;code&gt;flock&lt;/code&gt; is a thin
wrapper around the C standard function &lt;code&gt;flock&lt;/code&gt; (see &lt;code&gt;man 2 flock&lt;/code&gt; if you are
interested). It is used to manage locks and has several forms. The one we are
interested in is the third one. According to the man page, it uses and open file
by its &lt;strong&gt;file descriptor number&lt;/strong&gt;. Aha! so that was the purpose of the &lt;code&gt;exec 7&amp;lt;&amp;gt; ${job_queue}&lt;/code&gt; calls in the &lt;code&gt;job_pool_worker&lt;/code&gt; function. It would essentially
assign the file descriptor 7 to the fifo &lt;code&gt;job_queue&lt;/code&gt; and afterwards lock it with
&lt;code&gt;flock --exclusive 7&lt;/code&gt;. Cool. This way only one process at a time can read from
the fifo &lt;code&gt;job_queue&lt;/code&gt;&lt;/p&gt;
&lt;h2 id=&#34;great-but-how-do-i-use-this&#34;&gt;Great! But how do I use this?&lt;/h2&gt;
&lt;p&gt;It depends on your preference, you can either save this in a file(e.g.
job_pool.sh) and source it in your bash script. Or you can simply paste it
inside an existing bash script. Whatever tickles your fancy. I have also
provided an example that replicates our first implementation. Just paste the
below code under our &amp;ldquo;chad&amp;rdquo; job pool script.&lt;/p&gt;
&lt;div class=&#34;collapsable-code&#34;&gt;
&lt;input id=&#34;1&#34; type=&#34;checkbox&#34; /&gt;
&lt;label for=&#34;1&#34;&gt;
&lt;span class=&#34;collapsable-code__language&#34;&gt;bash&lt;/span&gt;
&lt;span class=&#34;collapsable-code__toggle&#34; data-label-expand=&#34;Show&#34; data-label-collapse=&#34;Hide&#34;&gt;&lt;/span&gt;
&lt;/label&gt;
&lt;pre class=&#34;language-bash&#34; &gt;&lt;code&gt;
function tester(){
# A function that takes an int as a parameter and sleeps
echo &amp;#34;$1&amp;#34;
sleep &amp;#34;$1&amp;#34;
echo &amp;#34;ENDED $1&amp;#34;
}
num_workers=$1
job_pool_init $num_workers
pcount=0
for i in {1..10}; do
job_pool_run tester &amp;#34;$i&amp;#34;
done
job_pool_wait
job_pool_shutdown
&lt;/code&gt;&lt;/pre&gt;
&lt;/div&gt;
&lt;p&gt;Hopefully this article was(or will be) helpful to you. From now on, you don&amp;rsquo;t
ever have to write single threaded bash scripts like normies :)&lt;/p&gt;
</content>
</item>
</channel>
</rss>