Using new Input System in ECS

The first approach that came into my mind for using new input system with ECS is to create a mono behaviour that listens to Input System callbacks, and modifies entity data accordingly.
Is there a better way of doing this?

 

I tried a few different methods, but ultimately gave up due to weird bugs in the new input system.

You can see my post here:
https://forum.unity.com/threads/usage-pattern-from-ecs-jobs.648058/#post-4344538