javax.persistence.metamodel.Metamodel Java Examples

private void checkHibernateMappingConfig(Metamodel metamodel) {
    Set<String> mappedEntities = metamodel
        .getEntities()
        .stream()
        .map(EntityType::getJavaType)
        .map(Class::getName)
        .collect(Collectors.toSet());
    List<String> unMappedEntities = MCRConfiguration2.getString("MCR.Hibernate.Mappings")
        .map(MCRConfiguration2::splitValue)
        .orElseGet(Stream::empty)
        .filter(cName -> !mappedEntities.contains(cName))
        .collect(Collectors.toList());
    if (!unMappedEntities.isEmpty()) {
        throw new MCRException(
            "JPA Mapping is inclomplete. Could not find a mapping for these classes: " + unMappedEntities);
    }
}
 

@Override
public void startUp(ServletContext servletContext) {
    try {
        initializeJPA();
    } catch (PersistenceException e) {
        //fix for MCR-1236
        if (MCRConfiguration2.getBoolean("MCR.Persistence.Database.Enable").orElse(true)) {
            LogManager.getLogger()
                .error(() -> "Could not initialize JPA. Database access is disabled in this session.", e);
            MCRConfiguration2.set("MCR.Persistence.Database.Enable", String.valueOf(false));
        }
        MCREntityManagerProvider.init(e);
        return;
    }
    Metamodel metamodel = MCREntityManagerProvider.getEntityManagerFactory().getMetamodel();
    checkHibernateMappingConfig(metamodel);
    LogManager.getLogger()
        .info("Mapping these entities: {}", metamodel.getEntities()
            .stream()
            .map(EntityType::getJavaType)
            .map(Class::getName)
            .collect(Collectors.toList()));
    MCRShutdownHandler.getInstance().addCloseable(new MCRJPAShutdownProcessor());
}
 

private void addRootEntities(ModelStructure modelStructure) {
	Metamodel jpaMetamodel;
	Set<EntityType<?>> jpaEntities;

	String modelName = getDataSource().getConfiguration().getModelName();

	jpaMetamodel = getEntityManager().getMetamodel();
	jpaEntities = jpaMetamodel.getEntities();
	logger.debug("Jpa metamodel contains [" + jpaEntities.size() + "] entity types");

	for (EntityType<?> entityType : jpaEntities) {
		logger.debug("Adding entity type [" + entityType + "] to model structure");
		String entityTypeName = entityType.getJavaType().getName();
		addEntity(modelStructure, modelName, entityTypeName);
		logger.info("Entity type [" + entityType + "] succesfully added to model structure");
	}
}
 

public EntityType getTargetEntity(RegistryConfiguration registryConf, EntityManager entityManager) {

		EntityType targetEntity;

		String targetEntityName = getTargetEntityName(registryConf);

		Metamodel classMetadata = entityManager.getMetamodel();
		Iterator it = classMetadata.getEntities().iterator();

		targetEntity = null;
		while (it.hasNext()) {
			EntityType entity = (EntityType) it.next();
			String jpaEntityName = entity.getName();

			if (entity != null && jpaEntityName.equals(targetEntityName)) {
				targetEntity = entity;
				break;
			}
		}

		return targetEntity;
	}
 

private static EntityManager mockEntityManager() {
  EntityManager em = EasyMock.createMock(EntityManager.class);
  Metamodel mm = EasyMock.createMock(Metamodel.class);
  EasyMock.expect(em.getMetamodel()).andReturn(mm).anyTimes();
  Set<EntityType<?>> et = new HashSet<EntityType<?>>();
  EasyMock.expect(mm.getEntities()).andReturn(et).anyTimes();
  EasyMock.expect(em.isOpen()).andReturn(true).anyTimes();
  Query jpqlquery = EasyMock.createMock(Query.class);
  Capture<String> capturedArgument = new Capture<String>();
  EasyMock.expect(em.createQuery(EasyMock.capture(capturedArgument))).andReturn(jpqlquery).anyTimes();
  EasyMock.expect(jpqlquery.setParameter(EasyMock.anyInt(), EasyMock.anyObject()))
      .andReturn(jpqlquery).anyTimes();
  EasyMock.expect(jpqlquery.setParameter(EasyMock.anyInt(), (Calendar) EasyMock.anyObject(), 
      EasyMock.anyObject(TemporalType.TIMESTAMP.getClass()))).andReturn(jpqlquery).anyTimes();
  EasyMock.expect(jpqlquery.setParameter(EasyMock.anyInt(), (Time) EasyMock.anyObject(), 
      EasyMock.anyObject(TemporalType.TIME.getClass()))).andReturn(jpqlquery).anyTimes();
  List<Object> result = new ArrayList<Object>();
  result.add(5);
  EasyMock.expect(jpqlquery.getResultList()).andReturn(result).anyTimes();
  EasyMock.replay(em, mm, jpqlquery);
  return em;

}
 

private <T> String getTableName(EntityManager em, Class<T> entityClass){
	/*
	 * Check if the specified class is present in the metamodel.
	 * Throws IllegalArgumentException if not.
	 */
	Metamodel meta = em.getMetamodel();
	EntityType<T> entityType = meta.entity(entityClass);
	
	//Check whether @Table annotation is present on the class.
	Table t = entityClass.getAnnotation(Table.class);
	
	String tableName = (t == null) ? entityType.getName().toUpperCase() : t.name();
	return tableName;
}
 

private List<Class<?>> getAllManagedEntityTypes(EntityManagerFactory entityManagerFactory) {
	List<Class<?>> entityClasses = new ArrayList<>();
	Metamodel metamodel = entityManagerFactory.getMetamodel();

	for (ManagedType<?> managedType : metamodel.getManagedTypes())
		if (managedType.getJavaType().isAnnotationPresent(Entity.class))
			entityClasses.add(managedType.getJavaType());

	return entityClasses;
}
 

private void analyze(T aObject)
{
    if (aObject != null) {
        entityClass = HibernateProxyHelper.getClassWithoutInitializingProxy(aObject);

        String idProperty = null;
        Metamodel metamodel = getEntityManager().getMetamodel();
        EntityType entity = metamodel.entity(entityClass);
        Set<SingularAttribute> singularAttributes = entity.getSingularAttributes();
        for (SingularAttribute singularAttribute : singularAttributes) {
            if (singularAttribute.isId()) {
                idProperty = singularAttribute.getName();
                break;
            }
        }
        if (idProperty == null) {
            throw new RuntimeException("id field not found");
        }

        DirectFieldAccessor accessor = new DirectFieldAccessor(aObject);
        id = (Number) accessor.getPropertyValue(idProperty);
    }
    else {
        entityClass = null;
        id = null;
    }
}
 

public Metamodel getMetamodel() {
    final EntityManager entityManager = getEntityManager();
    try {
        final Timer timer = Op.getMetamodel.start(this.timer, this);
        try {
            return entityManager.getMetamodel();
        } finally {
            timer.stop();
        }
    } finally {
        closeIfNoTx(entityManager);
    }
}
 

public JPAEdmModel(final Metamodel metaModel, final String pUnitName) {
  super(metaModel, pUnitName);
}
 

public MetamodelQueryFinder(Metamodel metamodel) {
	this.metamodel = metamodel;
}
 

public Metamodel getMetamodel() {
    return delegate.getMetamodel();
}
 

public JPAEdmBaseViewImpl(final Metamodel metaModel, final String pUnitName) {
  this.metaModel = metaModel;
  this.pUnitName = pUnitName;
}
 

@Override
public Metamodel getMetamodel()
{
    return entityManager().getMetamodel();
}
 

@Override
public Metamodel getJPAMetaModel() {
  return new JPAEdmMetaModel();
}
 

@Override
public Metamodel getJPAMetaModel() {
  return new JPAMetaModelMock();
}
 

private Metamodel mockMetaModel() {
  Metamodel metaModel = EasyMock.createMock(Metamodel.class);
  EasyMock.replay(metaModel);
  return metaModel;
}
 

private Metamodel mockMetaModel() {
  Metamodel metaModel = EasyMock.createMock(Metamodel.class);
  EasyMock.expect(metaModel.getEntities()).andStubReturn(getLocalEntities());
  EasyMock.replay(metaModel);
  return metaModel;
}
 

@Override
public Metamodel getMetamodel()
{
    throw new IllegalStateException("not implemented");
}
 

@Test
public void testOdataJpaAccessFactory() {

  ODataJPAFactoryImpl oDataJPAFactoryImpl = new ODataJPAFactoryImpl();
  ODataJPAAccessFactory jpaAccessFactory = oDataJPAFactoryImpl
      .getODataJPAAccessFactory();
  ODataJPAContextImpl oDataJPAContextImpl = new ODataJPAContextImpl();

  EntityManagerFactory emf = new EntityManagerFactory() {

    @Override
    public boolean isOpen() {
      // TODO Auto-generated method stub
      return false;
    }

    @Override
    public Map<String, Object> getProperties() {
      // TODO Auto-generated method stub
      return null;
    }

    @Override
    public PersistenceUnitUtil getPersistenceUnitUtil() {
      // TODO Auto-generated method stub
      return null;
    }

    @Override
    public Metamodel getMetamodel() {
      // TODO Auto-generated method stub
      return null;
    }

    @Override
    public CriteriaBuilder getCriteriaBuilder() {
      // TODO Auto-generated method stub
      return null;
    }

    @Override
    public Cache getCache() {
      // TODO Auto-generated method stub
      return null;
    }

    @SuppressWarnings("rawtypes")
    @Override
    public EntityManager createEntityManager(final Map arg0) {
      // TODO Auto-generated method stub
      return null;
    }

    @Override
    public EntityManager createEntityManager() {
      // TODO Auto-generated method stub
      return null;
    }

    @Override
    public void close() {
      // TODO Auto-generated method stub

    }
  };
  oDataJPAContextImpl.setEntityManagerFactory(emf);
  oDataJPAContextImpl.setPersistenceUnitName("pUnit");

  assertNotNull(jpaAccessFactory.getODataJPAMessageService(new Locale(
      "en")));
  assertNotNull(jpaAccessFactory.createODataJPAContext());
  assertNotNull(jpaAccessFactory
      .createJPAEdmProvider(oDataJPAContextImpl));
  assertNotNull(jpaAccessFactory
      .createODataProcessor(oDataJPAContextImpl));

}
 

@Override
public Metamodel getMetamodel() {
    return delegate().getMetamodel();
}
 

@Override
public Metamodel getMetamodel() {
    return delegate.getMetamodel();
}
 

@Override
public Metamodel getMetamodel() {
	return emf().getMetamodel();
}
 

@Override
public Metamodel getMetamodel() {
	return em.getMetamodel();
}
 

@Override
public Metamodel getMetamodel() {
	return em().getMetamodel();
}
 

@Override
public Metamodel getMetamodel( )
{
    // TODO Auto-generated method stub
    return null;
}
 

@Override
public Metamodel getMetamodel( )
{
    // TODO Auto-generated method stub
    return null;
}
 

@Override
public Metamodel getMetamodel() {
	return entityManagers.get().getMetamodel();
}
 

private Metamodel mockMetaModel() {
  Metamodel metaModel = EasyMock.createMock(Metamodel.class);
  EasyMock.replay(metaModel);
  return metaModel;
}
 

@Override
public Metamodel getJPAMetaModel() {
  return new JPAMetaModelMock();
}