@Test
 public void toDate() throws Exception {
   List<Expression> args =
       Lists.newArrayList(getInvertedLiteral("2001-11-30 00:00:00:0", PDataType.VARCHAR));
   evaluateAndAssertResult(
       new ToDateFunction(args, null, DateUtil.getDateParser("yyyy-MM-dd HH:mm:ss:S")),
       date(11, 30, 2001));
 }
 @Test
 public void toChar() throws Exception {
   List<Expression> args =
       Lists.newArrayList(getInvertedLiteral(date(12, 11, 2001), PDataType.DATE));
   evaluateAndAssertResult(
       new ToCharFunction(
           args, FunctionArgumentType.TEMPORAL, "", DateUtil.getDateFormatter("MM/dd/yy hh:mm a")),
       "12/11/01 12:00 AM");
 }
Exemplo n.º 3
0
  @Override
  public boolean evaluate(Tuple tuple, ImmutableBytesWritable ptr) {
    BigDecimal finalResult = BigDecimal.ZERO;

    for (int i = 0; i < children.size(); i++) {
      if (!children.get(i).evaluate(tuple, ptr)) {
        return false;
      }
      if (ptr.getLength() == 0) {
        return true;
      }
      BigDecimal value;
      PDataType type = children.get(i).getDataType();
      SortOrder sortOrder = children.get(i).getSortOrder();
      if (type == PTimestamp.INSTANCE || type == PUnsignedTimestamp.INSTANCE) {
        value = (BigDecimal) (PDecimal.INSTANCE.toObject(ptr, type, sortOrder));
      } else if (type.isCoercibleTo(PDecimal.INSTANCE)) {
        value =
            (((BigDecimal) PDecimal.INSTANCE.toObject(ptr, sortOrder))
                    .multiply(QueryConstants.BD_MILLIS_IN_DAY))
                .setScale(6, RoundingMode.HALF_UP);
      } else if (type.isCoercibleTo(PDouble.INSTANCE)) {
        value =
            ((BigDecimal.valueOf(type.getCodec().decodeDouble(ptr, sortOrder)))
                    .multiply(QueryConstants.BD_MILLIS_IN_DAY))
                .setScale(6, RoundingMode.HALF_UP);
      } else {
        value = BigDecimal.valueOf(type.getCodec().decodeLong(ptr, sortOrder));
      }
      finalResult = finalResult.add(value);
    }
    Timestamp ts = DateUtil.getTimestamp(finalResult);
    byte[] resultPtr = new byte[getDataType().getByteSize()];
    PTimestamp.INSTANCE.toBytes(ts, resultPtr, 0);
    ptr.set(resultPtr);
    return true;
  }
public class TruncateFunctionTest extends BaseClientManagedTimeTest {
  private static Format format = DateUtil.getDateParser(DateUtil.DEFAULT_MS_DATE_FORMAT);
  private static final String DS1 = "1970-01-10 00:58:01.587";
  private static final String DS2 = "1970-01-20 01:02:45.906";
  private static final String DS3 = "1970-01-30 01:30:24.353";

  private static Date toDate(String s) throws ParseException {
    return (Date) (format.parseObject(s));
  }

  private static Timestamp toTimestamp(String s) throws ParseException {
    return new Timestamp(((Date) (format.parseObject(s))).getTime());
  }

  @Test
  public void testTruncate() throws Exception {
    long ts = nextTimestamp();
    String tenantId = getOrganizationId();
    ensureTableCreated(getUrl(), ATABLE_NAME, ts - 5);
    Properties props = new Properties();
    props.setProperty(CURRENT_SCN_ATTRIB, Long.toString(ts - 3));
    Connection conn = DriverManager.getConnection(getUrl(), props);
    try {
      PreparedStatement stmt =
          conn.prepareStatement(
              "upsert into "
                  + "ATABLE("
                  + "    ORGANIZATION_ID, "
                  + "    ENTITY_ID, "
                  + "    A_DATE, "
                  + "    A_TIMESTAMP)"
                  + "VALUES (?, ?, ?, ?)");
      stmt.setString(1, tenantId);
      stmt.setString(2, ROW1);
      stmt.setDate(3, toDate(DS1));
      stmt.setTimestamp(4, toTimestamp(DS1));
      stmt.execute();
      stmt.setString(1, tenantId);
      stmt.setString(2, ROW2);
      stmt.setDate(3, toDate(DS2));
      stmt.setTimestamp(4, toTimestamp(DS2));
      stmt.execute();
      stmt.setString(1, tenantId);
      stmt.setString(2, ROW3);
      stmt.setDate(3, toDate(DS3));
      stmt.setTimestamp(4, toTimestamp(DS3));
      stmt.execute();
      conn.commit();
      conn.close();

      props.setProperty(CURRENT_SCN_ATTRIB, Long.toString(ts + 1));
      conn = DriverManager.getConnection(getUrl(), props);
      String query =
          "SELECT entity_id, trunc(a_date, 'day', 7), trunc(a_timestamp, 'second', 10) FROM ATABLE WHERE organization_id = ?";
      PreparedStatement statement = conn.prepareStatement(query);
      statement.setString(1, tenantId);
      ResultSet rs = statement.executeQuery();

      assertTrue(rs.next());
      assertEquals(ROW1, rs.getString(1));
      assertEquals(new Date((long) 7 * QueryConstants.MILLIS_IN_DAY), rs.getDate(2));
      assertEquals(toTimestamp("1970-01-10 00:58:00.000"), rs.getTimestamp(3));

      assertTrue(rs.next());
      assertEquals(ROW2, rs.getString(1));
      assertEquals(new Date((long) 14 * QueryConstants.MILLIS_IN_DAY), rs.getDate(2));
      assertEquals(toTimestamp("1970-01-20 01:02:40.000"), rs.getTimestamp(3));

      assertTrue(rs.next());
      assertEquals(ROW3, rs.getString(1));
      assertEquals(new Date((long) 28 * QueryConstants.MILLIS_IN_DAY), rs.getDate(2));
      assertEquals(toTimestamp("1970-01-30 01:30:20.000"), rs.getTimestamp(3));

      assertFalse(rs.next());
    } finally {
      conn.close();
    }
  }
}