A proc macro for designing loosely coupled Rust applications.
entrait
is used to generate an implemented trait from the definition of regular functions.
The emergent pattern that results from its use enable the following things:
* Zero-cost loose coupling and inversion of control
* Dependency graph as a compile time concept
* Mock library integrations
* Clean, readable, boilerplate-free code
The resulting pattern is referred to as the entrait pattern (see also: philosophy).
The macro looks like this:
```rust
fn my_function
which generates a new single-method trait named MyFunction
, with the method signature derived from the original function.
Entrait is a pure append-only macro: It will never alter the syntax of your function.
The new language items it generates will appear below the function.
In the first example, my_function
has a single parameter called deps
which is generic over a type D
, and represents dependencies injected into the function.
The dependency parameter is always the first parameter, which is analogous to the &self
parameter of the generated trait method.
To add a dependency, we just introduce a trait bound, now expressable as impl Trait
.
This is demonstrated by looking at one function calling another:
```rust
fn foo(deps: &impl Bar) { println!("{}", deps.bar(42)); }
fn bar
Other frameworks might represent multiple dependencies by having one value for each one, but entrait represents all dependencies within the same value. When the dependency parameter is generic, its trait bounds specifiy what methods we expect to be callable inside the function.
Multiple bounds can be expressed using the &(impl A + B)
syntax.
The single-value dependency design means that it is always the same reference that is passed around everywhere. But a reference to what, exactly? This is what we have managed to abstract away, which is the whole point.
When we want to compile a working application, we need an actual type to inject into the various entrait entrypoints. Two things will be important:
Entrait generates implemented traits, and the type to use for linking it all together is Impl<T>
:
```rust
fn foo(deps: &impl Bar) -> i32 { deps.bar() }
fn bar(_deps: &impl std::any::Any) -> i32 { 42 }
let app = Impl::new(()); assert_eq!(42, app.foo()); ```
π¬ Inspect the generated code π¬
The linking happens in the generated impl block for Impl<T>
, putting the entire impl under a where clause derived from the original dependency bounds:
rust
impl<T: Sync> Foo for Impl<T> where Self: Bar {
fn foo(&self) -> i32 {
foo(self) // <---- calls your function
}
}
Impl
is generic, so we can put whatever type we want into it.
Normally this would be some type that represents the global state/configuration of the running application.
But if dependencies can only be traits, and we always abstract away this type, how can this state ever be accessed?
So far we have only seen generic trait-based dependencies, but the dependency can also be a concrete type:
```rust struct Config(i32);
fn usetheconfig(config: &Config) -> i32 { config.0 }
fn doubleit(deps: &impl UseTheConfig) -> i32 { deps.usethe_config() * 2 }
asserteq!(42, Impl::new(Config(21)).doubleit()); ```
The parameter of use_the_config
is in the first position, so it represents the dependency.
We will notice two interesting things:
* Functions that depend on UseTheConfig
, either directly or indirectly, now have only one valid dependency type: Impl<Config>
1.
* Inside use_the_config
, we have a &Config
reference instead of &Impl<Config>
. This means we cannot call other entraited functions, because they are not implemented for Config
.
The last point means that a concrete dependency is the end of the line, a leaf in the dependency graph.
Typically, functions with a concrete dependency should be kept small and avoid extensive business logic. They ideally function as accessors, providing a loosely coupled abstraction layer over concrete application state.
To reduce the number of generated traits, entrait can be used as a mod
attribute.
When used in this mode, the macro will look for non-private functions directly within the module scope, to be represented as methods on the resulting trait.
This mode works mostly identically to the standalone function mode.
```rust
mod my_module {
pub fn foo(deps: &impl super::SomeTrait) {}
pub fn bar(deps: &impl super::OtherTrait) {}
}
``
This example generates a
MyModuletrait containing the methods
fooand
bar`.
Unimock
The whole point of entrait is to provide inversion of control, so that alternative dependency implementations can be used when unit testing function bodies. While test code can contain manual trait implementations, the most ergonomic way to test is to use a mocking library, which provides more features with less code.
Entrait works best together with unimock, as these two crates have been designed from the start with each other in mind.
Unimock exports a single mock struct which can be passed as argument to every function that accept a generic deps
parameter
(given that entrait is used with unimock support everywhere).
To enable mock configuration of entraited functions, supply the mock_api
option, e.g. mock_api=TraitMock
if the name of the trait is Trait
.
This works the same way for entraited modules, only that those already have a module to export from.
Unimock support for entrait is enabled by passing the unimock
option to entrait (#[entrait(Foo, unimock)]
), or turning on the unimock
feature, which makes all entraited functions mockable, even in upstream crates (as long as mock_api
is provided.).
```rust
fn foo
mod mymod {
pub fn bar
fn my_func(deps: &(impl Foo + MyMod)) -> i32 { deps.foo() + deps.bar() }
let mockeddeps = Unimock::new(( FooMock.eachcall(matching!()).returns(40), mymod::mock::bar.eachcall(matching!()).returns(2), ));
asserteq!(42, myfunc(&mocked_deps)); ```
Entrait with unimock supports un-mocking. This means that the test environment can be partially mocked!
```rust
fn sayhello(deps: &impl FetchPlanetName, planetid: u32) -> Result
fn fetchplanetname(deps: &impl FetchPlanet, planetid: u32) -> Result
pub struct Planet { name: String }
fn fetchplanet(deps: &(), planetid: u32) -> Result let hellostring = sayhello(
&Unimock::newpartial(
FetchPlanetMock
.somecall(matching!(123456))
.returns(Ok(Planet {
name: "World".to_string(),
}))
),
123456,
).unwrap(); asserteq!("Hello World!", hellostring);
``` This example used If you instead wish to use a more established mocking crate, there is also support for mockall.
Note that mockall has some limitations.
Multiple trait bounds are not supported, and deep tests will not work.
Also, mockall tends to generate a lot of code, often an order of magnitude more than unimock. Enabling mockall is done using the ```rust fn foo fn my_func(deps: &impl Foo) -> u32 {
deps.foo()
} fn main() {
let mut deps = MockFoo::new();
deps.expectfoo().returning(|| 42);
asserteq!(42, my_func(&deps));
}
``` A common technique for Rust application development is to choose a multi-crate architecture.
There are usually two main ways to go about it: The first option is how libraries are normally used: Its functions are just called, without any indirection. The second option can be referred to as a variant of the
dependency inversion principle.
This is usually a desirable architectural property, and achieving this with entrait is what this section is about. The main goal is to be able to express business logic centrally, and avoid depending directly on infrastructure details (onion architecture).
All of the examples in this section make some use of traits and trait delegation. Earlier it was mentioned that when concrete-type dependencies are used, the ```rust
pub struct Config {
foo: String,
} fn get_foo(config: &Config) -> &str {
&config.foo
}
``` Here we actually have a trait For making this work with any downstream application type, we just have to manually implement Using a concrete type like ```rust pub trait System {
fn current_time(&self) -> u128;
}
``` What the attribute does in this case, is just to generate the correct blanket implementations of the trait: delegation and mocks. To use with some Sometimes it might be desirable to have a delegation that involves dynamic dispatch.
Entrait has a ```rust trait ReadConfig: 'static {
fn read_config(&self) -> &str;
}
``` To use this together with some All cases up to this point have been leaf dependencies.
Leaf dependencies are delegations that exit from the To make your abstraction extendable and your dependency internal, we have to keep the ```rust pub trait Repository {
fn fetch(&self) -> i32;
}
``` This syntax introduces a total of three traits: This design makes it possible to separate concerns into three different crates, ordered from most-upstream to most-downstream:
1. Core logic: Depend on and call All delegation from In crate 2, we have to provide an implementation of ```rust
pub struct MyRepository; impl crate1::RepositoryImpl for MyRepository {
// this function has the now-familiar entrait-compatible signature:
fn fetch Entrait will split this trait implementation block in two: An inherent one containing the original code, and a proper trait implementation which performs the delegation. In the end, we just have to implement our A small variation of case 4: Use The implementation syntax is almost the same as in case 4, only that the entrait attribute must now be ```rust pub trait Repository {
fn fetch(&self) -> i32;
} pub struct MyRepository; impl RepositoryImpl for MyRepository {
fn fetch The app must now implement by default, entrait generates a trait that is module-private (no visibility keyword).
To change this, just put a visibility specifier before the trait name: ```rust
use entrait::*; fn foo Since Rust at the time of writing does not natively support async methods in traits, you may opt in to having ```rust async fn foo There is a cargo feature to automatically apply Entrait has experimental support for zero-cost futures. A nightly Rust compiler is needed for this feature. The entrait option is called ```rust use entrait::*; async fn foo There is a feature for turning this on everywhere: Some macros are used to transform the body of a function, or generate a body from scratch.
For example, we can use ```rust async fn fetch_thing(#[path] param: String) -> feignhttp::Result Here we had to use the Most often, you will only need to generate mock implementations for test code, and skip this for production code.
A notable exception to this is when building libraries.
When an application consists of several crates, downstream crates would likely want to mock out functionality from libraries. Entrait calls this exporting, and it unconditionally turns on autogeneration of mock implementations: ```rust fn bar(deps: &()) {}
fn foo(deps: &()) {}
``` It is also possible to reduce noise by doing | Feature | Implies | Description |
| ------------------- | --------------- | ------------------- |
| The To understand the entrait model and how to achieve Dependency Injection (DI) with it, we can compare it with a more widely used and classical alternative pattern:
Object-Oriented DI. In object-oriented DI, each named dependency is a separate object instance.
Each dependency exports a set of public methods, and internally points to a set of private dependencies.
A working application is built by fully instantiating such an object graph of interconnected dependencies. Entrait was built to address two drawbacks inherent to this design: This section lists known limitations of entrait: Cyclic dependency graphs are impossible with entrait.
In fact, this is not a limit of entrait itself, but with Rust's trait solver.
It is not able to prove that a type implements a trait if it needs to prove that it does in order to prove it. While this is a limitation, it is not necessarily a bad one.
One might say that a layered application architecture should never contain cycles.
If you do need recursive algorithms, you could model this as utility functions outside of the entraited APIs of the application.Unimock::new_partial
to create a mocker that works mostly like Impl
, except that the call graph can be short-circuited at arbitrary, run-time configurable points.
The example code goes through three layers (say_hello => fetch_planet_name => fetch_planet
), and only the deepest one gets mocked out.Alternative mocking: Mockall
mockall
entrait option.
There is no cargo feature to turn this on implicitly, because mockall doesn't work well when it's re-exported through another crate.[entrait(Foo, mockall)]
Multi-crate architecture
Case 1: Concrete leaf dependencies
T
in Impl<T>
, your application, and the type of the dependency have to match.
But this is only partially true.
It really comes down to which traits are implemented on what types:[entrait_export(pub GetFoo)]
π¬ Inspect the generated code π¬
rust
trait GetFoo {
fn get_foo(&self) -> &str;
}
impl<T: GetFoo> GetFoo for Impl<T> {
fn get_foo(&self) -> &str {
self.as_ref().get_foo()
}
}
impl GetFoo for Config {
fn get_foo(&self) -> &str {
get_foo(self)
}
}
GetFoo
that is implemented two times: for Impl<T> where T: GetFoo
and for Config
.
The first implementation is delegating to the other one.GetFoo
for that application:rust
struct App {
config: some_upstream_crate::Config,
}
impl some_upstream_crate::GetFoo for App {
fn get_foo(&self) -> &str {
self.config.get_foo()
}
}
Case 2: Hand-written trait as a leaf dependency
Config
from the first case can be contrived in many situations.
Sometimes a good old hand-written trait definition will do the job much better:[entrait]
π¬ Inspect the generated code π¬
rust
impl<T: System> System for Impl<T> {
fn current_time(&self) -> u128 {
self.as_ref().current_time()
}
}
App
, the app type itself should implement the trait.Case 3: Hand-written trait as a leaf dependency using dynamic dispatch
delegate_by =
option, where you can pass an alternative trait to use as part of the delegation strategy.
To enable dynamic dispatch, use ref
:[entrait(delegate_by=ref)]
π¬ Inspect the generated code π¬
rust
impl<T: ::core::convert::AsRef<dyn ReadConfig> + 'static> ReadConfig for Impl<T> {
fn read_config(&self) -> &str {
self.as_ref().as_ref().read_config()
}
}
App
, it should implement the AsRef<dyn ReadConfig>
trait.Case 4: Truly inverted internal dependencies - static dispatch
Impl<T>
layer, using delegation targets involving concete T
's.
This means that it is impossible to continue to use the entrait pattern and extend your application behind those abstractions.T
generic inside the [Impl] type.
To make this work, we have to make use of two helper traits:[entrait(RepositoryImpl, delegate_by = DelegateRepository)]
π¬ Inspect the generated code π¬
rust
pub trait RepositoryImpl<T> {
fn fetch(_impl: &Impl<T>) -> i32;
}
pub trait DelegateRepository<T> {
type Target: RepositoryImpl<T>;
}
impl<T: DelegateRepository<T>> Repository for Impl<T> {
fn fetch(&self) -> i32 {
<T as DelegateRepository<T>>::Target::fetch(self)
}
}
Repository
: The dependency, what the rest of the application directly calls.RepositoryImpl<T>
: The delegation target, a trait which needs to be implemented by some Target
type.DelegateRepository<T>
: The delegation selector, that selects the specific Target
type to be used for some specific App
.Repository
methods.
2. External system integration: Provide some implementation of the repository, by implementing RepositoryImpl<T>
.
3. Executable: Construct an App
that selects a specific repository implementation from crate 2.Repository
to RepositoryImpl<T>
goes via the DelegateRepository<T>
trait.
The method signatures in RepositoryImpl<T>
are static, and receives the &Impl<T>
via a normal parameter.
This allows us to continue using entrait patterns within those implementations!RepositoryImpl<T>
.
This can either be done manually, or by using the [entrait] attribute on an impl
block:[entrait]
π¬ Inspect the generated code π¬
rust
impl MyRepository {
fn fetch<D>(deps: &D) -> i32 {
unimplemented!()
}
}
impl<T> crate1::RepositoryImpl<T> for MyRepository {
#[inline]
fn fetch(_impl: &Impl<T>) -> i32 {
Self::fetch(_impl)
}
}
DelegateRepository<T>
:rust
// in crate3:
struct App;
impl crate1::DelegateRepository<Self> for App {
type Target = crate2::MyRepository;
}
fn main() { /* ... */ }
Case 5: Truly inverted internal dependencies - dynamic dispatch
delegate_by=ref
instead of a custom trait.
This makes the delegation happen using dynamic dispatch.#[entrait(ref)]
:[entrait(RepositoryImpl, delegate_by=ref)]
[entrait(ref)]
AsRef<dyn RepositoryImpl<Self>>
.Options and features
Trait visibility
[entrait(pub Foo)] // <-- public trait
async
support#[async_trait]
generated for your trait.
Enable the boxed-futures
cargo feature and pass the box_future
option like this:[entrait(Foo, box_future)]
#[async_trait]
to every generated async trait: use-boxed-futures
.Zero-cost async inversion of control - preview mode
associated_future
, and uses GATs and feature(type_alias_impl_trait)
.
This feature generates an associated future inside the trait, and the implementations use impl Trait
syntax to infer
the resulting type of the future:![feature(typealiasimpl_trait)]
[entrait(Foo, associated_future)]
use-associated-futures
.Integrating with other
fn
-targeting macros, and no_deps
feignhttp
to generate an HTTP client. Entrait will try as best as it
can to co-exist with macros like these. Since entrait
is a higher-level macro that does not touch fn bodies (it does not even try to parse them),
entrait should be processed after, which means it should be placed before lower level macros. Example:[entrait(FetchThing, no_deps)]
[feignhttp::get("https://my.api.org/api/{param}")]
no_deps
entrait option.
This is used to tell entrait that the function does not have a deps
parameter as its first input.
Instead, all the function's inputs get promoted to the generated trait method.Conditional compilation of mocks
[entrait_export(pub Bar)]
or
rust[entrait(pub Foo, export)]
use entrait::entrait_export as entrait
.Feature overview
unimock
| | Adds the [unimock] dependency, and turns on Unimock implementations for all traits. |
| use-boxed-futures
| boxed-futures
| Automatically applies the [asynctrait] macro to async trait methods. |
| use-associated-futures
| | Automatically transforms the return type of async trait methods into an associated future by using type-alias-impl-trait syntax. Requires a nightly compiler. |
| boxed-futures
| | Pulls in the [asynctrait] optional dependency, enabling the box_future
entrait option (macro parameter). |"Philosophy"
entrait
crate is central to the entrait pattern, an opinionated yet flexible and Rusty way to build testable applications/business logic.
DomainServices
.
There will typically be one such class per domain object, with a lot of methods in each.
This results in dependency graphs with fewer nodes overall, but the number of possible call graphs is much larger.
A common problem with this is that the actual dependenciesβthe functions actually getting calledβare encapsulated
and hidden away from public interfaces.
To construct valid dependency mocks in unit tests, a developer will have to read through full function bodies instead of looking at signatures.entrait
solves this by:
Limitations
Cyclic dependency graphs