Start line:  
End line:  

Snippet Preview

Snippet HTML Code

Stack Overflow Questions
  /*
   * Licensed 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 com.facebook.presto.operator.window;
 
 
 import java.util.List;
 
 import static com.facebook.presto.spi.StandardErrorCode.INVALID_FUNCTION_ARGUMENT;
 import static com.facebook.presto.spi.type.BigintType.BIGINT;
 import static com.facebook.presto.spi.type.BooleanType.BOOLEAN;
 import static com.facebook.presto.spi.type.DoubleType.DOUBLE;
 import static com.facebook.presto.spi.type.VarcharType.VARCHAR;
 import static com.facebook.presto.util.Failures.checkCondition;
 
 public class LeadFunction
         extends ValueWindowFunction
 {
     public static class BigintLeadFunction
             extends LeadFunction
     {
         public BigintLeadFunction(List<IntegerargumentChannels)
         {
             super(argumentChannels);
         }
     }
 
     public static class BooleanLeadFunction
             extends LeadFunction
     {
         public BooleanLeadFunction(List<IntegerargumentChannels)
         {
             super(argumentChannels);
         }
     }
 
     public static class DoubleLeadFunction
             extends LeadFunction
     {
         public DoubleLeadFunction(List<IntegerargumentChannels)
         {
             super(argumentChannels);
         }
     }
 
     public static class VarcharLeadFunction
             extends LeadFunction
     {
         public VarcharLeadFunction(List<IntegerargumentChannels)
         {
             super(argumentChannels);
         }
     }
 
     private final Type type;
     private final int valueChannel;
     private final int offsetChannel;
     private final int defaultChannel;
 
     protected LeadFunction(Type typeList<IntegerargumentChannels)
     {
         this. = type;
         this. = argumentChannels.get(0);
         this. = (argumentChannels.size() > 1) ? argumentChannels.get(1) : -1;
         this. = (argumentChannels.size() > 2) ? argumentChannels.get(2) : -1;
     }
 
     @Override
     public Type getType()
     {
         return ;
     }
 
     @Override
     public void processRow(BlockBuilder outputint frameStartint frameEndint currentPosition)
     {
         if (( >= 0) && .isNull(currentPosition)) {
             output.appendNull();
         }
         else {
             long offset = ( < 0) ? 1 : .getLong(currentPosition);
             checkCondition(offset >= 0, "Offset must be at least 0");
 
             long valuePosition = currentPosition + offset;
 
             if ((valuePosition >= 0) && (valuePosition < .size())) {
                .appendTo(, Ints.checkedCast(valuePosition), output);
            }
            else if ( >= 0) {
                .appendTo(currentPositionoutput);
            }
            else {
                output.appendNull();
            }
        }
    }
New to GrepCode? Check out our FAQ X