コード例 #1
0
  @Test
  public void testArrayPrependFunction11() throws Exception {
    Object[] o =
        new Object[] {
          BigDecimal.valueOf(2345), BigDecimal.valueOf(-23.45), BigDecimal.valueOf(785)
        };
    Object[] o2 =
        new Object[] {
          BigDecimal.valueOf(-19),
          BigDecimal.valueOf(2345),
          BigDecimal.valueOf(-23.45),
          BigDecimal.valueOf(785)
        };
    Object element = BigDecimal.valueOf(-19);
    PDataType baseType = PDecimal.INSTANCE;

    PhoenixArray arr = new PhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        null,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.ASC);
  }
コード例 #2
0
  @Test
  public void testForNullsWith509NullsAtBeginning() throws Exception {
    // 509 nulls at the beginning
    Object[] o = new Object[511];
    for (int i = 0; i < o.length - 2; i++) o[i] = null;

    o[o.length - 2] = "1";
    o[o.length - 1] = "2";

    Object[] o2 = new Object[512];
    for (int i = 0; i < o2.length - 2; i++) o2[i] = null;

    o2[o2.length - 2] = "1";
    o2[o2.length - 1] = "2";

    Object element = null;
    PDataType baseType = PVarchar.INSTANCE;

    PhoenixArray arr = new PhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        1,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.DESC);
  }
コード例 #3
0
  @Test
  public void testArrayPrependFunction14() throws Exception {
    Calendar calendar = Calendar.getInstance();
    java.util.Date currentDate = calendar.getTime();
    java.sql.Timestamp timestamp = new java.sql.Timestamp(currentDate.getTime());

    Object[] o = new Object[] {timestamp, timestamp, timestamp};
    Object[] o2 = new Object[] {timestamp, timestamp, timestamp, timestamp};
    PDataType baseType = PTimestamp.INSTANCE;

    PhoenixArray arr = new PhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray(baseType, o2);
    test(
        arr,
        timestamp,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        null,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.ASC);
  }
コード例 #4
0
  @Test
  public void testArrayPrependFunction6() throws Exception {
    Object[] o = new Object[] {new Float(2.3), new Float(7.9), new Float(-9.6), new Float(2.3)};
    Object[] o2 =
        new Object[] {
          new Float(8.9), new Float(2.3), new Float(7.9), new Float(-9.6), new Float(2.3)
        };
    Object element = 8.9;
    PDataType baseType = PFloat.INSTANCE;

    PhoenixArray arr = new PhoenixArray.PrimitiveFloatPhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray.PrimitiveFloatPhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        null,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.ASC);
  }
コード例 #5
0
  @Test
  public void testArrayPrependFunction4() throws Exception {
    // offset array int
    Object[] o = new Object[Short.MAX_VALUE + 7];
    for (int i = 0; i < o.length; i++) {
      o[i] = "a";
    }
    Object[] o2 = new Object[Short.MAX_VALUE + 8];
    for (int i = 1; i < o2.length; i++) {
      o2[i] = "a";
    }
    Object element = "b";
    o2[0] = element;
    PDataType baseType = PVarchar.INSTANCE;

    PhoenixArray arr = new PhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        null,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.ASC);
  }
コード例 #6
0
  @Test
  public void testForNullsAllNulls() throws Exception {
    Object element = null;
    PDataType baseType = PChar.INSTANCE;

    PhoenixArray arr = null;
    PhoenixArray expected = null;
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        4,
        null,
        baseType,
        1,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.DESC);
  }
コード例 #7
0
  @Test
  public void testForNullsWithNullsInMiddle() throws Exception {
    Object[] o = new Object[] {"1   ", "2   ", null, "3   ", "4   "};
    Object[] o2 = new Object[] {null, "1   ", "2   ", null, "3   ", "4   "};
    Object element = null;
    PDataType baseType = PVarchar.INSTANCE;

    PhoenixArray arr = new PhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        1,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.DESC);
  }
コード例 #8
0
  @Test
  public void testArrayPrependFunction1() throws Exception {
    Object[] o = new Object[] {1, 2, -3, 4};
    Object[] o2 = new Object[] {5, 1, 2, -3, 4};
    Object element = 5;
    PDataType baseType = PInteger.INSTANCE;

    PhoenixArray arr = new PhoenixArray.PrimitiveIntPhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray.PrimitiveIntPhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        null,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.ASC);
  }
コード例 #9
0
  @Test
  public void testArrayPrependFunction21() throws Exception {
    Object[] o = new Object[] {4.78, 9.54, 2.34, -9.675, Double.MAX_VALUE};
    Object[] o2 = new Object[] {12.67, 4.78, 9.54, 2.34, -9.675, Double.MAX_VALUE};
    Object element = 12.67;
    PDataType baseType = PDouble.INSTANCE;

    PhoenixArray arr = new PhoenixArray.PrimitiveDoublePhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray.PrimitiveDoublePhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        null,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.DESC);
  }
コード例 #10
0
  @Test
  public void testArrayPrependFunction20() throws Exception {
    Object[] o = new Object[] {"1   ", "2   ", "3   ", "4   "};
    Object[] o2 = new Object[] {"5", "1", "2", "3", "4"};
    Object element = "5";
    PDataType baseType = PChar.INSTANCE;

    PhoenixArray arr = new PhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        4,
        null,
        baseType,
        1,
        null,
        expected,
        SortOrder.DESC,
        SortOrder.DESC);
  }
コード例 #11
0
  @Test
  public void testArrayPrependFunction10() throws Exception {
    Object[] o = new Object[] {(byte) 4, (byte) 8, (byte) 9};
    Object[] o2 = new Object[] {(byte) 6, (byte) 4, (byte) 8, (byte) 9};
    Object element = (byte) 6;
    PDataType baseType = PTinyint.INSTANCE;

    PhoenixArray arr = new PhoenixArray.PrimitiveBytePhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray.PrimitiveBytePhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        null,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.ASC);
  }
コード例 #12
0
  @Test
  public void testArrayPrependFunction8() throws Exception {
    Object[] o = new Object[] {123l, 677l, 98789l, -78989l, 66787l};
    Object[] o2 = new Object[] {543l, 123l, 677l, 98789l, -78989l, 66787l};
    Object element = 543l;
    PDataType baseType = PLong.INSTANCE;

    PhoenixArray arr = new PhoenixArray.PrimitiveLongPhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray.PrimitiveLongPhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        null,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.ASC);
  }
コード例 #13
0
  @Test
  public void testArrayPrependFunctionBoolean() throws Exception {
    Boolean[] o = new Boolean[] {true, false, false, true};
    Boolean[] o2 = new Boolean[] {false, true, false, false, true};
    Boolean element = false;
    PDataType baseType = PBoolean.INSTANCE;

    PhoenixArray arr = new PhoenixArray.PrimitiveBooleanPhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray.PrimitiveBooleanPhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        null,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.ASC);
  }
コード例 #14
0
  @Test
  public void testArrayPrependFunction9() throws Exception {
    Object[] o = new Object[] {(short) 34, (short) -23, (short) -89, (short) 999, (short) 34};
    Object[] o2 =
        new Object[] {(short) 7, (short) 34, (short) -23, (short) -89, (short) 999, (short) 34};
    Object element = (short) 7;
    PDataType baseType = PSmallint.INSTANCE;

    PhoenixArray arr = new PhoenixArray.PrimitiveShortPhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray.PrimitiveShortPhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        null,
        null,
        baseType,
        null,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.ASC);
  }
コード例 #15
0
  @Test
  public void testArrayPrependFunction24() throws Exception {
    byte[][] o =
        new byte[][] {
          new byte[] {2, 0},
          new byte[] {13, 3},
          new byte[] {5, 3},
          new byte[] {6, 3},
          new byte[] {2, 5}
        };
    byte[][] o2 =
        new byte[][] {
          new byte[] {5, 6},
          new byte[] {2, 0},
          new byte[] {13, 3},
          new byte[] {5, 3},
          new byte[] {6, 3},
          new byte[] {2, 5}
        };
    byte[] element = new byte[] {5, 6};
    PDataType baseType = PBinary.INSTANCE;

    PhoenixArray arr = new PhoenixArray(baseType, o);
    PhoenixArray expected = new PhoenixArray(baseType, o2);
    test(
        arr,
        element,
        PDataType.fromTypeId(baseType.getSqlType() + PDataType.ARRAY_TYPE_BASE),
        3,
        null,
        baseType,
        1,
        null,
        expected,
        SortOrder.ASC,
        SortOrder.DESC);
  }
コード例 #16
0
  @Override
  public void upsertEvents(List<Event> events) throws SQLException {
    Preconditions.checkNotNull(events);
    Preconditions.checkNotNull(connection);
    Preconditions.checkNotNull(this.upsertStatement);

    boolean wasAutoCommit = connection.getAutoCommit();
    connection.setAutoCommit(false);
    try (PreparedStatement colUpsert = connection.prepareStatement(upsertStatement)) {
      String value = null;
      Integer sqlType = null;
      for (Event event : events) {
        byte[] payloadBytes = event.getBody();
        if (payloadBytes == null || payloadBytes.length == 0) {
          continue;
        }
        String payload = new String(payloadBytes);
        Matcher m = inputPattern.matcher(payload.trim());

        if (!m.matches()) {
          logger.debug(
              "payload {} doesn't match the pattern {} ", payload, inputPattern.toString());
          continue;
        }
        if (m.groupCount() != colNames.size()) {
          logger.debug(
              "payload {} size doesn't match the pattern {} ", m.groupCount(), colNames.size());
          continue;
        }
        int index = 1;
        int offset = 0;
        for (int i = 0; i < colNames.size(); i++, offset++) {
          if (columnMetadata[offset] == null) {
            continue;
          }

          value = m.group(i + 1);
          sqlType = columnMetadata[offset].getSqlType();
          Object upsertValue = PDataType.fromTypeId(sqlType).toObject(value);
          if (upsertValue != null) {
            colUpsert.setObject(index++, upsertValue, sqlType);
          } else {
            colUpsert.setNull(index++, sqlType);
          }
        }

        // add headers if necessary
        Map<String, String> headerValues = event.getHeaders();
        for (int i = 0; i < headers.size(); i++, offset++) {

          String headerName = headers.get(i);
          String headerValue = headerValues.get(headerName);
          sqlType = columnMetadata[offset].getSqlType();
          Object upsertValue = PDataType.fromTypeId(sqlType).toObject(headerValue);
          if (upsertValue != null) {
            colUpsert.setObject(index++, upsertValue, sqlType);
          } else {
            colUpsert.setNull(index++, sqlType);
          }
        }

        if (autoGenerateKey) {
          sqlType = columnMetadata[offset].getSqlType();
          String generatedRowValue = this.keyGenerator.generate();
          Object rowkeyValue = PDataType.fromTypeId(sqlType).toObject(generatedRowValue);
          colUpsert.setObject(index++, rowkeyValue, sqlType);
        }
        colUpsert.execute();
      }
      connection.commit();
    } catch (Exception ex) {
      logger.error("An error {} occurred during persisting the event ", ex.getMessage());
      throw new SQLException(ex.getMessage());
    } finally {
      if (wasAutoCommit) {
        connection.setAutoCommit(true);
      }
    }
  }