first

firstDesign.java

public class EntityServiceImpl implements EntityService {

    private EntityRepository repository;
    private ImageService imageService;

    @Override
    public void update(String id, UpdateData updateData) {
        Entity entity = repository.find(id);
        entity.setDescription(updateData.getDescription());
        entity.setTitle(updateData.getTitle());

        ImageEntity image = imageService.getImage(updateData.getImageUrl());
        ImageEntity previous = entity.getImage();
        entity.setImage(image);
        imageService.removeImage(previous);
        repository.save(entity);
    }

}