mafredri / zsh-async

Because your terminal should be able to perform tasks asynchronously without external tools!
MIT License
770 stars 34 forks source link
async asynchronous shell zsh

zsh-async

Test

Because your terminal should be able to perform tasks asynchronously without external tools!

Intro (TL;DR)

With zsh-async you can run multiple asynchronous jobs, enforce unique jobs (multiple instances of the same job will not run), flush all currently running jobs and create multiple workers (each with their own jobs). For each worker you can register a callback-function through which you will be notified about the job results (job name, return code, output and execution time).

Overview

zsh-async is a small library for running asynchronous tasks in zsh without requiring any external tools. It utilizes zsh/zpty to launch a pseudo-terminal in which all commands get executed without blocking any other processes. Checking for completed tasks can be done manually, by polling, or better yet, automatically whenever a process has finished executing by notifying through a SIGWINCH kill-signal.

This library bridges the gap between spawning child processes and disowning them. Child processes launched by normal means clutter the terminal with output about their state, and disowned processes become separate entities, no longer under control of the parent. Now you can have both!

Usage

The async worker is a separate environment (think web worker). You send it a job (command + parameters) to execute and it returns the result of that execution through a callback function. If you find that you need to stop/start a worker to update global state (variables) you should consider refactoring so that state is passed during the async_job call (e.g. async_job my_worker my_function $state1 $state2).

Note that because the worker is a separate forked environment, any functions you want to use as jobs in the worker need to be defined before the worker is started, otherwise you will get a command not found error when you try to launch the job.

Installation

Manual

You can either source the async.zsh script directly or insert under your $fpath as async and autoload it through autoload -Uz async && async.

Integration

zplug
zplug "mafredri/zsh-async", from:"github", use:"async.zsh"

Functions

The zsh-async library has a bunch of functions that need to be used to perform async actions:

async_init

Initializes the async library (not required if using async from $fpath with autoload.)

async_start_worker <worker_name> [-u] [-n] [-p <pid>]

Start a new async worker with optional parameters, a worker can be told to only run unique tasks and to notify a process when tasks are complete.

async_stop_worker <worker_name_1> [<worker_name_2>]

Simply stops a worker and all active jobs will be terminated immediately.

async_job <worker_name> <my_function> [<function_params>]

Start a new asynchronous job on specified worker, assumes the worker is running. Note if you are using a function for the job, it must have been defined before the worker was started or you will get a command not found error.

async_worker_eval <worker_name> <my_function> [<function_params>]

Evaluate a command (like async_job) inside the async worker, then worker environment can be manipulated. For example, issuing a cd command will change the PWD of the worker which will then be inherited by all future async jobs.

Output will be returned via callback, job name will be [async/eval].

async_process_results <worker_name> <callback_function>

Get results from finished jobs and pass it to the to callback function. This is the only way to reliably return the job name, return code, output and execution time and with minimal effort.

If the async process buffer becomes corrupt, the callback will be invoked with the first argument being [async] (job name), non-zero return code and fifth argument describing the error (stderr).

The callback_function is called with the following parameters:

Possible error return codes for the job name [async]:

async_register_callback <worker_name> <callback_function>

Register a callback for completed jobs. As soon as a job is finished, async_process_results will be called with the specified callback function. This requires that a worker is initialized with the -n (notify) option.

async_unregister_callback <worker_name>

Unregister the callback for a specific worker.

async_flush_jobs <worker_name>

Flush all current jobs running on a worker. This will terminate any and all running processes under the worker by sending a SIGTERM to the entire process group, use with caution.

Example code

#!/usr/bin/env zsh
source ./async.zsh
async_init

# Initialize a new worker (with notify option)
async_start_worker my_worker -n

# Create a callback function to process results
COMPLETED=0
completed_callback() {
    COMPLETED=$(( COMPLETED + 1 ))
    print $@
}

# Register callback function for the workers completed jobs
async_register_callback my_worker completed_callback

# Give the worker some tasks to perform
async_job my_worker print hello
async_job my_worker sleep 0.3

# Wait for the two tasks to be completed
while (( COMPLETED < 2 )); do
    print "Waiting..."
    sleep 0.1
done

print "Completed $COMPLETED tasks!"

# Output:
#   Waiting...
#   print 0 hello 0.001583099365234375
#   Waiting...
#   Waiting...
#   sleep 0 0.30631208419799805
#   Completed 2 tasks!

Testing

Tests are located in *_test.zsh and can be run by executing the test runner: ./test.zsh.

Example:

$ ./test.zsh
ok  ./async_test.zsh    2.334s

The test suite can also run specific tasks that match a pattern, for example:

$ ./test.zsh -v -run zle
=== RUN   test_zle_watcher
--- PASS: test_zle_watcher (0.07s)
PASS
ok  ./async_test.zsh    0.070s

Limitations

Tips

If you do not wish to use the notify feature, you can couple zsh-async with zsh/sched or the zsh periodic function for scheduling the worker results to be processed.

Why did I make this?

I found a great theme for zsh, Pure by Sindre Sorhus. After using it for a while I noticed some graphical glitches due to the terminal being updated by a disowned process. Thus, I became inspired to get my hands dirty and find a solution. I tried many things, coprocesses (seemed too limited by themselves), different combinations of trapping kill-signals, etc. I also had problems with the zsh process ending up in a deadlock due to some zsh bug. After working out the kinks, I ended up with this and thought, hey, why not make it a library.