Examples with OpenejbJarType org.apache.openejb.jee.oejb2.OpenejbJarType used on opensource projects

Example 6 with OpenejbJarType

use of org.apache.openejb.jee.oejb2.OpenejbJarType in project tomee by apache.

the class OpenEjb2Conversion method mergeEntityMappings.

public final void mergeEntityMappings(final String moduleId, final EntityMappings entityMappings, final OpenejbJar openejbJar, final OpenejbJarType openejbJarType) {
    final Map<String, EntityData> entities = new TreeMap<String, EntityData>();
    if (entityMappings != null) {
        for (final Entity entity : entityMappings.getEntity()) {
            try {
                entities.put(entity.getDescription(), new EntityData(entity));
            } catch (final IllegalArgumentException e) {
                LoggerFactory.getLogger(this.getClass()).error(e.getMessage(), e);
            }
        }
    }
    for (final org.apache.openejb.jee.oejb2.EnterpriseBean enterpriseBean : openejbJarType.getEnterpriseBeans()) {
        if (!(enterpriseBean instanceof EntityBeanType)) {
            continue;
        }
        final EntityBeanType bean = (EntityBeanType) enterpriseBean;
        final EntityData entityData = entities.get(moduleId + "#" + bean.getEjbName());
        if (entityData == null) {
            // todo warn no such ejb in the ejb-jar.xml
            continue;
        }
        final Table table = new Table();
        table.setName(bean.getTableName());
        entityData.entity.setTable(table);
        for (final EntityBeanType.CmpFieldMapping cmpFieldMapping : bean.getCmpFieldMapping()) {
            final String cmpFieldName = cmpFieldMapping.getCmpFieldName();
            final Field field = entityData.fields.get(cmpFieldName);
            if (field == null) {
                // todo warn no such cmp-field in the ejb-jar.xml
                continue;
            }
            final Column column = new Column();
            column.setName(cmpFieldMapping.getTableColumn());
            field.setColumn(column);
        }
        if (bean.getKeyGenerator() != null) {
            // todo support complex primary keys
            final Attributes attributes = entityData.entity.getAttributes();
            if (attributes != null && attributes.getId().size() == 1) {
                final Id id = attributes.getId().get(0);
                // todo detect specific generation strategy
                id.setGeneratedValue(new GeneratedValue(GenerationType.IDENTITY));
            }
        }
        for (final QueryType query : bean.getQuery()) {
            final NamedQuery namedQuery = new NamedQuery();
            final QueryType.QueryMethod queryMethod = query.getQueryMethod();
            // todo deployment id could change in one of the later conversions... use entity name instead, but we need to save it off
            final StringBuilder name = new StringBuilder();
            name.append(entityData.entity.getName()).append(".").append(queryMethod.getMethodName());
            if (queryMethod.getMethodParams() != null && !queryMethod.getMethodParams().getMethodParam().isEmpty()) {
                name.append('(');
                boolean first = true;
                for (final String methodParam : queryMethod.getMethodParams().getMethodParam()) {
                    if (!first) {
                        name.append(",");
                    }
                    name.append(methodParam);
                    first = false;
                }
                name.append(')');
            }
            namedQuery.setName(name.toString());
            namedQuery.setQuery(query.getEjbQl());
            entityData.entity.getNamedQuery().add(namedQuery);
        }
    }
    for (final EjbRelationType relation : openejbJarType.getEjbRelation()) {
        final List<EjbRelationshipRoleType> roles = relation.getEjbRelationshipRole();
        if (roles.isEmpty()) {
            continue;
        }
        if (relation.getManyToManyTableName() == null) {
            final EjbRelationshipRoleType leftRole = roles.get(0);
            final EjbRelationshipRoleType.RelationshipRoleSource leftRoleSource = leftRole.getRelationshipRoleSource();
            final String leftEjbName = leftRoleSource == null ? null : leftRoleSource.getEjbName();
            final EntityData leftEntityData = entities.get(moduleId + "#" + leftEjbName);
            final EjbRelationshipRoleType.CmrField cmrField = leftRole.getCmrField();
            final String leftFieldName = null != cmrField ? cmrField.getCmrFieldName() : null;
            RelationField field;
            if (leftRole.isForeignKeyColumnOnSource()) {
                field = null != leftFieldName && null != leftEntityData ? leftEntityData.relations.get(leftFieldName) : null;
                // todo warn field not found
                if (field == null) {
                    continue;
                }
            } else {
                final RelationField other = null != leftFieldName && null != leftEntityData ? leftEntityData.relations.get(leftFieldName) : null;
                // todo warn field not found
                if (other == null) {
                    continue;
                }
                field = other.getRelatedField();
                // todo warn field not found
                if (field == null) {
                    if (other instanceof OneToMany) {
                        // for a unidirectional oneToMany, the join column declaration
                        // is placed on the oneToMany element instead of manyToOne
                        field = other;
                    } else {
                        continue;
                    }
                }
            }
            // is marked as the owning field
            if (field instanceof OneToOne) {
                final OneToOne left = (OneToOne) field;
                final OneToOne right = (OneToOne) left.getRelatedField();
                if (right != null) {
                    left.setMappedBy(null);
                    right.setMappedBy(left.getName());
                }
            }
            final EjbRelationshipRoleType.RoleMapping roleMapping = leftRole.getRoleMapping();
            for (final EjbRelationshipRoleType.RoleMapping.CmrFieldMapping cmrFieldMapping : roleMapping.getCmrFieldMapping()) {
                final JoinColumn joinColumn = new JoinColumn();
                joinColumn.setName(cmrFieldMapping.getForeignKeyColumn());
                joinColumn.setReferencedColumnName(cmrFieldMapping.getKeyColumn());
                field.getJoinColumn().add(joinColumn);
            }
        } else {
            final JoinTable joinTable = new JoinTable();
            joinTable.setName(relation.getManyToManyTableName());
            // 
            // left
            final EjbRelationshipRoleType leftRole = roles.get(0);
            RelationField left = null;
            if (leftRole.getRelationshipRoleSource() != null) {
                final String leftEjbName = leftRole.getRelationshipRoleSource().getEjbName();
                final EntityData leftEntityData = entities.get(moduleId + "#" + leftEjbName);
                if (leftEntityData == null) {
                    // todo warn no such entity in ejb-jar.xml
                    continue;
                }
                final EjbRelationshipRoleType.CmrField lcf = leftRole.getCmrField();
                left = (null != lcf ? leftEntityData.relations.get(lcf.getCmrFieldName()) : null);
            }
            if (left != null) {
                left.setJoinTable(joinTable);
                final EjbRelationshipRoleType.RoleMapping roleMapping = leftRole.getRoleMapping();
                for (final EjbRelationshipRoleType.RoleMapping.CmrFieldMapping cmrFieldMapping : roleMapping.getCmrFieldMapping()) {
                    final JoinColumn joinColumn = new JoinColumn();
                    joinColumn.setName(cmrFieldMapping.getForeignKeyColumn());
                    joinColumn.setReferencedColumnName(cmrFieldMapping.getKeyColumn());
                    joinTable.getJoinColumn().add(joinColumn);
                }
            }
            // right
            if (roles.size() > 1) {
                final EjbRelationshipRoleType rightRole = roles.get(1);
                // if there wasn't a left cmr field, find the field for the right, so we can add the join table to it
                if (left == null) {
                    final EjbRelationshipRoleType.CmrField rcf = rightRole.getCmrField();
                    if (rcf == null) {
                        // todo warn no cmr field declared for either role
                        continue;
                    } else if (rightRole.getRelationshipRoleSource() != null) {
                        final String rightEjbName = rightRole.getRelationshipRoleSource().getEjbName();
                        final EntityData rightEntityData = entities.get(moduleId + "#" + rightEjbName);
                        if (rightEntityData == null) {
                            // todo warn no such entity in ejb-jar.xml
                            continue;
                        }
                        final RelationField right = rightEntityData.relations.get(rcf.getCmrFieldName());
                        right.setJoinTable(joinTable);
                    }
                }
                final EjbRelationshipRoleType.RoleMapping roleMapping = rightRole.getRoleMapping();
                for (final EjbRelationshipRoleType.RoleMapping.CmrFieldMapping cmrFieldMapping : roleMapping.getCmrFieldMapping()) {
                    final JoinColumn joinColumn = new JoinColumn();
                    joinColumn.setName(cmrFieldMapping.getForeignKeyColumn());
                    joinColumn.setReferencedColumnName(cmrFieldMapping.getKeyColumn());
                    joinTable.getInverseJoinColumn().add(joinColumn);
                }
            }
        }
    }
}

Example 7 with OpenejbJarType

use of org.apache.openejb.jee.oejb2.OpenejbJarType in project tomee by apache.

the class ReadDescriptors method readOpenejbJar.

private void readOpenejbJar(final EjbModule ejbModule) throws OpenEJBException {
    final Source source = getSource(ejbModule.getAltDDs().get("openejb-jar.xml"));
    if (source != null) {
        try {
            // Attempt to parse it first as a v3 descriptor
            final OpenejbJar openejbJar = JaxbOpenejbJar3.unmarshal(OpenejbJar.class, source.get()).postRead();
            ejbModule.setOpenejbJar(openejbJar);
        } catch (final Exception v3ParsingException) {
            // Attempt to parse it second as a v2 descriptor
            final OpenejbJar openejbJar = new OpenejbJar();
            ejbModule.setOpenejbJar(openejbJar);
            try {
                final JAXBElement element = (JAXBElement) JaxbOpenejbJar2.unmarshal(OpenejbJarType.class, source.get());
                final OpenejbJarType o2 = (OpenejbJarType) element.getValue();
                ejbModule.getAltDDs().put("openejb-jar.xml", o2);
                final GeronimoEjbJarType g2 = OpenEjb2Conversion.convertToGeronimoOpenejbXml(o2);
                ejbModule.getAltDDs().put("geronimo-openejb.xml", g2);
            } catch (final Exception v2ParsingException) {
                // Now we have to determine which error to throw; the v3 file exception or the fallback v2 file exception.
                final Exception[] realIssue = { v3ParsingException };
                try {
                    final SAXParserFactory factory = Saxs.namespaceAwareFactory();
                    final SAXParser parser = factory.newSAXParser();
                    parser.parse(source.get(), new DefaultHandler() {

                        public void startElement(final String uri, final String localName, final String qName, final Attributes attributes) throws SAXException {
                            if (localName.equals("environment")) {
                                realIssue[0] = v2ParsingException;
                                throw new SAXException("Throw exception to stop parsing");
                            }
                            if (uri == null) {
                                return;
                            }
                            if (uri.contains("openejb-jar-2.") || uri.contains("geronimo.apache.org/xml/ns")) {
                                realIssue[0] = v2ParsingException;
                                throw new SAXException("Throw exception to stop parsing");
                            }
                        }
                    });
                } catch (final Exception dontCare) {
                // no-op
                }
                String filePath = "<error: could not be written>";
                try {
                    File tempFile;
                    try {
                        tempFile = File.createTempFile("openejb-jar-", ".xml");
                    } catch (final Throwable e) {
                        final File tmp = new File("tmp");
                        if (!tmp.exists() && !tmp.mkdirs()) {
                            throw new IOException("Failed to create local tmp directory: " + tmp.getAbsolutePath());
                        }
                        tempFile = File.createTempFile("openejb-jar-", ".xml", tmp);
                    }
                    try {
                        IO.copy(source.get(), tempFile);
                    } catch (final IOException e) {
                    // no-op
                    }
                    filePath = tempFile.getAbsolutePath();
                } catch (final IOException e) {
                // no-op
                }
                final Exception e = realIssue[0];
                if (e instanceof SAXException) {
                    throw new OpenEJBException("Cannot parse the openejb-jar.xml. Xml content written to: " + filePath, e);
                } else if (e instanceof JAXBException) {
                    throw new OpenEJBException("Cannot unmarshall the openejb-jar.xml. Xml content written to: " + filePath, e);
                } else if (e instanceof IOException) {
                    throw new OpenEJBException("Cannot read the openejb-jar.xml.", e);
                } else {
                    throw new OpenEJBException("Encountered unknown error parsing the openejb-jar.xml.", e);
                }
            }
        }
    }
    final Source source1 = getSource(ejbModule.getAltDDs().get("geronimo-openejb.xml"));
    if (source1 != null) {
        try {
            GeronimoEjbJarType geronimoEjbJarType = null;
            final Object o = JaxbOpenejbJar2.unmarshal(GeronimoEjbJarType.class, source1.get());
            if (o instanceof GeronimoEjbJarType) {
                geronimoEjbJarType = (GeronimoEjbJarType) o;
            } else if (o instanceof JAXBElement) {
                final JAXBElement element = (JAXBElement) o;
                geronimoEjbJarType = (GeronimoEjbJarType) element.getValue();
            }
            if (geronimoEjbJarType != null) {
                final Object nested = geronimoEjbJarType.getOpenejbJar();
                if (nested != null && nested instanceof OpenejbJar) {
                    final OpenejbJar existingOpenejbJar = ejbModule.getOpenejbJar();
                    if (existingOpenejbJar == null || existingOpenejbJar.getEjbDeploymentCount() <= 0) {
                        final OpenejbJar openejbJar = (OpenejbJar) nested;
                        ejbModule.getAltDDs().put("openejb-jar.xml", openejbJar);
                        ejbModule.setOpenejbJar(openejbJar);
                    }
                }
                ejbModule.getAltDDs().put("geronimo-openejb.xml", geronimoEjbJarType);
            }
        } catch (final Exception e) {
            throw new OpenEJBException("Failed parsing geronimo-openejb.xml", e);
        }
    }
}