Exemple #1
0
  private void loadLoadGroups(JDBCEntityMetaData metadata) {
    loadGroupMasks = new HashMap();

    // load optimistic locking mask and add it to all the load group masks
    JDBCOptimisticLockingMetaData olMD = metadata.getOptimisticLocking();
    if (olMD != null) {
      if (versionField != null) {
        defaultLockGroupMask = new boolean[tableFields.length];
        defaultLockGroupMask[versionField.getTableIndex()] = true;
        versionField.setLockingStrategy(LockingStrategy.VERSION);
      } else if (olMD.getGroupName() != null) {
        defaultLockGroupMask = loadGroupMask(olMD.getGroupName(), null);
        for (int i = 0; i < tableFields.length; ++i) {
          if (defaultLockGroupMask[i]) {
            JDBCCMPFieldBridge tableField = tableFields[i];
            tableField.setLockingStrategy(LockingStrategy.GROUP);
            tableField.addDefaultFlag(ADD_TO_WHERE_ON_UPDATE);
          }
        }
      } else { // read or modified strategy
        LockingStrategy strategy =
            (olMD.getLockingStrategy()
                    == JDBCOptimisticLockingMetaData.LockingStrategy.READ_STRATEGY
                ? LockingStrategy.READ
                : LockingStrategy.MODIFIED);
        for (int i = 0; i < tableFields.length; ++i) {
          JDBCCMPFieldBridge field = tableFields[i];
          if (!field.isPrimaryKeyMember()) field.setLockingStrategy(strategy);
        }
      }
    }

    // add the * load group
    boolean[] defaultLoadGroup = new boolean[tableFields.length];
    Arrays.fill(defaultLoadGroup, true);
    for (int i = 0; i < primaryKeyFields.length; ++i) {
      int tableIndex = primaryKeyFields[i].getTableIndex();
      defaultLoadGroup[tableIndex] = false;
    }
    loadGroupMasks.put(DEFAULT_LOADGROUP_NAME, defaultLoadGroup);

    // put each group in the load groups map by group name
    Iterator groupNames = metadata.getLoadGroups().keySet().iterator();
    while (groupNames.hasNext()) {
      // get the group name
      String groupName = (String) groupNames.next();
      boolean[] loadGroup = loadGroupMask(groupName, defaultLockGroupMask);
      loadGroupMasks.put(groupName, loadGroup);
    }
    loadGroupMasks = Collections.unmodifiableMap(loadGroupMasks);
  }
Exemple #2
0
  public void init() throws Exception {
    dataSource = manager.getDataSource(metadata.getDataSourceName());
    qualifiedTableName = SQLUtil.fixTableName(metadata.getDefaultTableName(), dataSource);
    int dotIndex = qualifiedTableName.indexOf('.');
    tableName = dotIndex == -1 ? qualifiedTableName : qualifiedTableName.substring(dotIndex + 1);

    // CMP fields
    loadCMPFields(metadata);

    // CMR fields
    loadCMRFields(metadata);

    // create locking field
    JDBCOptimisticLockingMetaData lockMetaData = metadata.getOptimisticLocking();
    if (lockMetaData != null && lockMetaData.getLockingField() != null) {
      JDBCOptimisticLockingMetaData.LockingStrategy strategy = lockMetaData.getLockingStrategy();
      JDBCCMPFieldMetaData versionMD = lockMetaData.getLockingField();

      versionField = getCMPFieldByName(versionMD.getFieldName());
      boolean hidden = versionField == null;
      switch (strategy) {
        case VERSION_COLUMN_STRATEGY:
          {
            if (hidden) versionField = new JDBCLongVersionFieldBridge(manager, versionMD);
            else versionField = new JDBCLongVersionFieldBridge((JDBCCMP2xFieldBridge) versionField);
            break;
          }
        case TIMESTAMP_COLUMN_STRATEGY:
          {
            if (hidden) versionField = new JDBCTimestampVersionFieldBridge(manager, versionMD);
            else
              versionField =
                  new JDBCTimestampVersionFieldBridge((JDBCCMP2xFieldBridge) versionField);
            break;
          }
        case KEYGENERATOR_COLUMN_STRATEGY:
          {
            if (hidden)
              versionField =
                  new JDBCKeyGenVersionFieldBridge(
                      manager, versionMD, lockMetaData.getKeyGeneratorFactory());
            else
              versionField =
                  new JDBCKeyGenVersionFieldBridge(
                      (JDBCCMP2xFieldBridge) versionField, lockMetaData.getKeyGeneratorFactory());
            break;
          }
      }
      if (hidden) addCMPField(versionField);
      else tableFields[versionField.getTableIndex()] = versionField;
    }

    // audit fields
    JDBCAuditMetaData auditMetaData = metadata.getAudit();
    if (auditMetaData != null) {
      JDBCCMPFieldMetaData auditField = auditMetaData.getCreatedPrincipalField();
      if (auditField != null) {
        createdPrincipalField = getCMPFieldByName(auditField.getFieldName());
        if (createdPrincipalField == null) {
          createdPrincipalField = new JDBCCMP2xFieldBridge(manager, auditField);
          addCMPField(createdPrincipalField);
        }
      } else {
        createdPrincipalField = null;
      }

      auditField = auditMetaData.getCreatedTimeField();
      if (auditField != null) {
        createdTimeField = getCMPFieldByName(auditField.getFieldName());
        if (createdTimeField == null) {
          createdTimeField =
              new JDBCCMP2xFieldBridge(manager, auditField, JDBCTypeFactory.EQUALS, false);
          addCMPField(createdTimeField);
        } else {
          // just to override state factory and check-dirty-after-get
          createdTimeField =
              new JDBCCMP2xFieldBridge(
                  (JDBCCMP2xFieldBridge) createdTimeField, JDBCTypeFactory.EQUALS, false);
          tableFields[createdTimeField.getTableIndex()] = createdTimeField;
        }
      } else {
        createdTimeField = null;
      }

      auditField = auditMetaData.getUpdatedPrincipalField();
      if (auditField != null) {
        updatedPrincipalField = getCMPFieldByName(auditField.getFieldName());
        if (updatedPrincipalField == null) {
          updatedPrincipalField = new JDBCCMP2xUpdatedPrincipalFieldBridge(manager, auditField);
          addCMPField(updatedPrincipalField);
        } else {
          updatedPrincipalField =
              new JDBCCMP2xUpdatedPrincipalFieldBridge(
                  (JDBCCMP2xFieldBridge) updatedPrincipalField);
          tableFields[updatedPrincipalField.getTableIndex()] = updatedPrincipalField;
        }
      } else {
        updatedPrincipalField = null;
      }

      auditField = auditMetaData.getUpdatedTimeField();
      if (auditField != null) {
        updatedTimeField = getCMPFieldByName(auditField.getFieldName());
        if (updatedTimeField == null) {
          updatedTimeField = new JDBCCMP2xUpdatedTimeFieldBridge(manager, auditField);
          addCMPField(updatedTimeField);
        } else {
          updatedTimeField =
              new JDBCCMP2xUpdatedTimeFieldBridge((JDBCCMP2xFieldBridge) updatedTimeField);
          tableFields[updatedTimeField.getTableIndex()] = updatedTimeField;
        }
      } else {
        updatedTimeField = null;
      }
    }

    // ejbSelect methods
    loadSelectors(metadata);
  }