pueue/ARCHITECTURE.md
2022-12-13 11:56:12 +01:00

5.7 KiB

Architecture Guide

This document is supposed to give you a short introduction to the project.
It explains the project structure, so you can get a rough overview of the overall architecture.

Feel free to expand this document!

Overall Structure

This project is divided into two modules, the client (pueue) and the daemon (pueued).
Pueue also depends on pueue-lib. Pueue-lib contains everything that is shared between the daemon and the client.

This includes:

  • The protocol used for communicating.
  • Settings, since they're parsed by both binaries.
  • All data structs, namely state, task and message.
  • Helper to interact with task's logs.

Daemon

The daemon is composed of two main components.

  1. Request handling in daemon/network/. This is the code responsible for communicating with clients. In daemon/network/message_handler/ you can find neatly separated handlers for all of Pueue's subcommands.
  2. The TaskHandler in daemon/task_handler/. It's responsible for everything regarding process interaction.

All information that's not sub-process specific, is stored in the State (pueue-lib/state.rs) struct.
Both components share a reference to the State, a Arc<Mutex<State>>. That way we can guarantee a single source of truth and a consistent state.

It's also important to know, that there's a mpsc channel.
This channel is used to send on-demand messages from the network request handler to the the TaskHandler. This includes stuff like "Start/Pause/Kill" sub-processes or "Reset everything".

Request handling

The daemon/network/socket.rs module contains the logic for accepting client connections and receiving payloads. The request accept and handle logic is a single async-await loop run by the main thread.

The payload is then deserialized to Message (pueue-lib/message.rs) and handled by its respective function. All functions used for handling these messages can be found in daemon/network/message_handler.

Many messages can be instantly handled by simply modifying or reading the state. \ However, sometimes the TaskHandler has to be notified, if something involves modifying actual system processes (start/pause/kill tasks). That's when the mpsc channel to the TaskHandler comes into play.

TaskHandler

The TaskHandler is responsible for actually starting and managing system processes.
It's further important to note, that it runs in its own thread.

The TaskHandler runs a never ending loop, which checks a few times each second, if

  • there are new instructions in the mpsc channel.
  • a new task can be started.
  • tasks finished and can be finalized.
  • delayed tasks can be enqueued (-d flag on pueue add)
  • A few other things. Check the TaskHandler::run function in daemon/task_handler/mod.rs.

The TaskHandler is by far the most complex piece of code in this project, but there is also a lot of documentation.

Shared State

Whenever you're writing some core-logic in Pueue, please make sure to understand how mutexes work.

Try to be conservative with your state.lock() calls, since this also blocks the request handler! Only use the state, if you absolutely have to.

At the same time, you should also lock early enough to prevent inconsistent states. Operations should generally be atomic.
Anyhow, working with mutexes is usually straight-forward, but can sometimes be a little bit tricky.

Code Style

This is a result of tokei ./client ./daemon on commit eed378e at the 2022-12-13.

===============================================================================
 Language            Files        Lines         Code     Comments       Blanks
===============================================================================
 JSON                    1          119          119            0            0
 Markdown                2          310            0          192          118
 Pest                    1           69           43           12           14
 TOML                    1           78           62            7            9
 YAML                    1           27           27            0            0
-------------------------------------------------------------------------------
 Rust                   90         9403         7116          808         1479
 |- Markdown            87         1367            0         1258          109
 (Total)                          10770         7116         2066         1588
===============================================================================
 Total                  96        10006         7367         1019         1620
===============================================================================

Format and Clippy

cargo format and cargo clean && cargo clippy should never return any warnings on the current stable Rust version!

PR's are automatically checked for these two and won't be accepted unless everything looks fine.

Comments

  1. All functions must have a doc block.
  2. All non-trivial structs must have a doc block.
  3. Rather too many inline comments than too few.
  4. Non-trivial code should be well documented!

In general, please add a lot of comments. It makes maintenance, collaboration and reviews MUCH easier.