class MyElementQuery extends ElementQuery
{
    use NestedElementQueryTrait;

    protected function beforePrepare(): bool
    {
        // ...

        $this->applyNestedElementParams('elementable.fieldId', 'elementtable.primaryOwnerId');
    }
}

class MyElement extends Element
{
    public function afterSave(bool $isNew): void
    {
        // ...

        $this->saveOwnership($isNew, '{{%elementtable}}');
    }
}