public DateScalarSubtractDateColumn(long value, int colNum, int outputColumn) { this.colNum = colNum; this.value = new Timestamp(0); this.value.setTime(DateWritable.daysToMillis((int) value)); this.outputColumn = outputColumn; scratchTimestamp2 = new Timestamp(0); }
/* copied over from VectorUDFTimestampFieldLong */ private TimestampWritable toTimestampWritable(long daysSinceEpoch) { Timestamp ts = new Timestamp(DateWritable.daysToMillis((int) daysSinceEpoch)); return new TimestampWritable(ts); }
@Override /** * Method to evaluate scalar-column operation in vectorized fashion. * * @batch a package of rows with each column stored in a vector */ public void evaluate(VectorizedRowBatch batch) { if (childExpressions != null) { super.evaluateChildren(batch); } // Input #2 is type date (epochDays). LongColumnVector inputColVector2 = (LongColumnVector) batch.cols[colNum]; // Output is type HiveIntervalDayTime. IntervalDayTimeColumnVector outputColVector = (IntervalDayTimeColumnVector) batch.cols[outputColumn]; int[] sel = batch.selected; boolean[] inputIsNull = inputColVector2.isNull; boolean[] outputIsNull = outputColVector.isNull; outputColVector.noNulls = inputColVector2.noNulls; outputColVector.isRepeating = inputColVector2.isRepeating; int n = batch.size; long[] vector2 = inputColVector2.vector; // return immediately if batch is empty if (n == 0) { return; } if (inputColVector2.isRepeating) { scratchTimestamp2.setTime(DateWritable.daysToMillis((int) vector2[0])); dtm.subtract(value, scratchTimestamp2, outputColVector.getScratchIntervalDayTime()); outputColVector.setFromScratchIntervalDayTime(0); // Even if there are no nulls, we always copy over entry 0. Simplifies code. outputIsNull[0] = inputIsNull[0]; } else if (inputColVector2.noNulls) { if (batch.selectedInUse) { for (int j = 0; j != n; j++) { int i = sel[j]; scratchTimestamp2.setTime(DateWritable.daysToMillis((int) vector2[i])); dtm.subtract(value, scratchTimestamp2, outputColVector.getScratchIntervalDayTime()); outputColVector.setFromScratchIntervalDayTime(i); } } else { for (int i = 0; i != n; i++) { scratchTimestamp2.setTime(DateWritable.daysToMillis((int) vector2[i])); dtm.subtract(value, scratchTimestamp2, outputColVector.getScratchIntervalDayTime()); outputColVector.setFromScratchIntervalDayTime(i); } } } else { /* there are nulls */ if (batch.selectedInUse) { for (int j = 0; j != n; j++) { int i = sel[j]; scratchTimestamp2.setTime(DateWritable.daysToMillis((int) vector2[i])); dtm.subtract(value, scratchTimestamp2, outputColVector.getScratchIntervalDayTime()); outputColVector.setFromScratchIntervalDayTime(i); outputIsNull[i] = inputIsNull[i]; } } else { for (int i = 0; i != n; i++) { scratchTimestamp2.setTime(DateWritable.daysToMillis((int) vector2[i])); dtm.subtract(value, scratchTimestamp2, outputColVector.getScratchIntervalDayTime()); outputColVector.setFromScratchIntervalDayTime(i); } System.arraycopy(inputIsNull, 0, outputIsNull, 0, n); } } NullUtil.setNullOutputEntriesColScalar(outputColVector, batch.selectedInUse, sel, n); }