/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.apache.phoenix.schema.types; import java.math.BigDecimal; import java.sql.Date; import java.sql.Types; import java.text.Format; import org.apache.hadoop.hbase.io.ImmutableBytesWritable; import org.apache.hadoop.hbase.util.Bytes; import org.apache.phoenix.schema.SortOrder; import org.apache.phoenix.util.DateUtil; import org.apache.phoenix.util.StringUtil; public class PDate extends PDataType<Date> { public static final PDate INSTANCE = new PDate(); private PDate() { super("DATE", Types.DATE, Date.class, new DateCodec(), 11); // After TIMESTAMP and DATE to ensure toLiteral finds those first } @Override public byte[] toBytes(Object object) { if (object == null) { throw newIllegalDataException(this + " may not be null"); } byte[] bytes = new byte[getByteSize()]; toBytes(object, bytes, 0); return bytes; } @Override public int toBytes(Object object, byte[] bytes, int offset) { if (object == null) { throw newIllegalDataException(this + " may not be null"); } getCodec().encodeLong(((java.util.Date) object).getTime(), bytes, offset); return this.getByteSize(); } @Override public Object toObject(Object object, PDataType actualType) { if (object == null) { return null; } if (equalsAny(actualType, PTime.INSTANCE, PUnsignedTime.INSTANCE)) { return new Date(((java.sql.Time) object).getTime()); } else if (equalsAny(actualType, PTimestamp.INSTANCE, PUnsignedTimestamp.INSTANCE)) { return new Date(((java.sql.Timestamp) object).getTime()); } else if (equalsAny(actualType, PDate.INSTANCE, PUnsignedDate.INSTANCE)) { return object; } else if (equalsAny(actualType, PLong.INSTANCE, PUnsignedLong.INSTANCE)) { return new Date((Long) object); } else if (actualType == PDecimal.INSTANCE) { return new Date(((BigDecimal) object).longValueExact()); } else if (actualType == PVarchar.INSTANCE) { return DateUtil.parseDate((String) object); } return throwConstraintViolationException(actualType, this); } @Override public Date toObject(byte[] b, int o, int l, PDataType actualType, SortOrder sortOrder, Integer maxLength, Integer scale) { if (l == 0) { return null; } if (equalsAny(actualType, PTimestamp.INSTANCE, PUnsignedTimestamp.INSTANCE, PDate.INSTANCE, PUnsignedDate.INSTANCE, PTime.INSTANCE, PUnsignedTime.INSTANCE, PLong.INSTANCE, PUnsignedLong.INSTANCE)) { return new Date(actualType.getCodec().decodeLong(b, o, sortOrder)); } else if (actualType == PDecimal.INSTANCE) { BigDecimal bd = (BigDecimal) actualType.toObject(b, o, l, actualType, sortOrder); return new Date(bd.longValueExact()); } throwConstraintViolationException(actualType, this); return null; } @Override public boolean isCastableTo(PDataType targetType) { return super.isCastableTo(targetType) || equalsAny(targetType, PDecimal.INSTANCE, PLong.INSTANCE, PUnsignedLong.INSTANCE); } @Override public boolean isCoercibleTo(PDataType targetType) { return equalsAny(targetType, PDate.INSTANCE, PTime.INSTANCE, PTimestamp.INSTANCE, PVarbinary.INSTANCE, PBinary.INSTANCE); } @Override public boolean isCoercibleTo(PDataType targetType, Object value) { if (value != null) { if (equalsAny(targetType, PUnsignedTimestamp.INSTANCE, PUnsignedDate.INSTANCE, PUnsignedTime.INSTANCE)) { return ((java.util.Date) value).getTime() >= 0; } } return super.isCoercibleTo(targetType, value); } @Override public boolean isFixedWidth() { return true; } @Override public Integer getByteSize() { return Bytes.SIZEOF_LONG; } @Override public int compareTo(Object lhs, Object rhs, PDataType rhsType) { if (rhsType == PTimestamp.INSTANCE || rhsType == PUnsignedTimestamp.INSTANCE) { return -rhsType.compareTo(rhs, lhs, PTime.INSTANCE); } return ((java.util.Date) rhs).compareTo((java.util.Date) lhs); } @Override public Object toObject(String value) { if (value == null || value.length() == 0) { return null; } return DateUtil.parseDate(value); } @Override public boolean isBytesComparableWith(PDataType otherType) { return super.isBytesComparableWith(otherType) || otherType == PTime.INSTANCE; } @Override public String toStringLiteral(Object o, Format formatter) { if (formatter == null) { // If default formatter has not been overridden, // use default one. formatter = DateUtil.DEFAULT_DATE_FORMATTER; } return "'" + StringUtil.escapeStringConstant(super.toStringLiteral(o, formatter)) + "'"; } @Override public void coerceBytes(ImmutableBytesWritable ptr, Object object, PDataType actualType, Integer maxLength, Integer scale, SortOrder actualModifier, Integer desiredMaxLength, Integer desiredScale, SortOrder expectedModifier) { if (ptr.getLength() > 0 && actualType == PTimestamp.INSTANCE && actualModifier == expectedModifier) { ptr.set(ptr.get(), ptr.getOffset(), getByteSize()); return; } super.coerceBytes(ptr, object, actualType, maxLength, scale, actualModifier, desiredMaxLength, desiredScale, expectedModifier); } @Override public Object getSampleValue(Integer maxLength, Integer arrayLength) { return new Date((Long) PLong.INSTANCE.getSampleValue(maxLength, arrayLength)); } static class DateCodec extends PLong.LongCodec { @Override public int decodeInt(byte[] b, int o, SortOrder sortOrder) { throw new UnsupportedOperationException(); } @Override public PhoenixArrayFactory getPhoenixArrayFactory() { return new PhoenixArrayFactory() { @Override public PhoenixArray newArray(PDataType type, Object[] elements) { return new PhoenixArray(type, elements); } }; } } }