示例#1
0
  /**
   * Overrides the parent implementation to provide a more efficient mechanism for generating
   * primary keys, while generating the primary key support on the fly.
   *
   * @param count the batch size
   * @param entity the entity requesting primary keys
   * @param channel open JDBCChannel
   * @return NSArray of NSDictionary where each dictionary corresponds to a unique primary key value
   */
  public NSArray newPrimaryKeys(int count, EOEntity entity, JDBCChannel channel) {
    if (isPrimaryKeyGenerationNotSupported(entity)) {
      return null;
    }

    EOAttribute attribute = (EOAttribute) entity.primaryKeyAttributes().lastObject();
    String attrName = attribute.name();
    boolean isIntType = "i".equals(attribute.valueType());

    NSMutableArray results = new NSMutableArray(count);
    String sequenceName = sequenceNameForEntity(entity);
    DB2Expression expression = new DB2Expression(entity);

    boolean succeeded = false;
    for (int tries = 0; !succeeded && tries < 2; tries++) {
      while (results.count() < count) {
        try {
          StringBuffer sql = new StringBuffer();
          sql.append("SELECT ");
          sql.append("next value for " + sequenceName + " AS KEY");
          sql.append(" from sysibm.sysdummy1");
          expression.setStatement(sql.toString());
          channel.evaluateExpression(expression);
          try {
            NSDictionary row;
            while ((row = channel.fetchRow()) != null) {
              Enumeration pksEnum = row.allValues().objectEnumerator();
              while (pksEnum.hasMoreElements()) {
                Number pkObj = (Number) pksEnum.nextElement();
                Number pk;
                if (isIntType) {
                  pk = Integer.valueOf(pkObj.intValue());
                } else {
                  pk = Long.valueOf(pkObj.longValue());
                }
                results.addObject(new NSDictionary(pk, attrName));
              }
            }
          } finally {
            channel.cancelFetch();
          }
          succeeded = true;
        } catch (JDBCAdaptorException ex) {
          throw ex;
        }
      }
    }

    if (results.count() != count) {
      throw new IllegalStateException(
          "Unable to generate primary keys from the sequence for " + entity + ".");
    }

    return results;
  }
示例#2
0
 @MarshalsPointer
 public static long toNative(List<CVPixelFormatType> l, long flags) {
   if (l == null) {
     return 0L;
   }
   NSMutableArray<NSNumber> array = new NSMutableArray<>();
   for (CVPixelFormatType i : l) {
     array.add(NSNumber.valueOf(i.value()));
   }
   return NSObject.Marshaler.toNative(array, flags);
 }
示例#3
0
 @MarshalsPointer
 public static long toNative(List<CMVideoCodecType> l, long flags) {
   if (l == null) {
     return 0L;
   }
   NSMutableArray<NSString> array = new NSMutableArray<>();
   for (CMVideoCodecType i : l) {
     array.add(new NSString(i.asFourCharCode()));
   }
   return NSObject.Marshaler.toNative(array, flags);
 }
示例#4
0
 @MarshalsPointer
 public static long toNative(List<SCNMatrix4> l, long flags) {
   if (l == null) {
     return 0L;
   }
   NSMutableArray<NSValue> array = new NSMutableArray<>();
   for (SCNMatrix4 i : l) {
     array.add(NSValue.valueOf(i));
   }
   return NSObject.Marshaler.toNative(array, flags);
 }