Describe the feature you would like

We would like to reuse this code in our custom block builder but we need to pass in a wrapped version of State<DB>. Currently this is not possible. The easiest solution I see is to create a new trait StateDB which encapsulates some of the required functionality.