<?php

$entityManager = $args->getEntityManager();
$unitOfWork = $entityManager->getUnitOfWork();
$updatedEntities = $unitOfWork->getScheduledEntityUpdates();

if ($entity->getId()) {
    $modelState = new ModelState();
    $modelState->setDictionaryState($entity);
    $modelState->setModel($updatedEntity);
    $modelState->setUser($user);
    $updatedEntity->addModelState($modelState);
    $entityManager->persist($modelState);

    $classModelStateMetadata = $entityManager->getClassMetadata(ModelState::class);
    $unitOfWork->computeChangeSet($classModelStateMetadata, $modelState);

    $classModelMetadata = $entityManager->getClassMetadata(Model::class);
    $unitOfWork->computeChangeSet($classModelMetadata, $updatedEntity);
}