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;
 
 
 import java.util.List;
 
 import static com.google.common.base.Preconditions.checkArgument;
 import static com.google.common.base.Preconditions.checkNotNull;
 import static com.google.common.base.Preconditions.checkState;
 
 public class LimitOperator
         implements Operator
 {
     public static class LimitOperatorFactory
             implements OperatorFactory
     {
         private final int operatorId;
         private final List<Typetypes;
         private final long limit;
         private boolean closed;
 
         public LimitOperatorFactory(int operatorIdList<? extends Typetypeslong limit)
         {
             this. = operatorId;
             this. = ImmutableList.copyOf(types);
             this. = limit;
         }
 
         @Override
         public List<TypegetTypes()
         {
             return ;
         }
 
         @Override
         public Operator createOperator(DriverContext driverContext)
         {
             checkState(!"Factory is already closed");
             OperatorContext operatorContext = driverContext.addOperatorContext(LimitOperator.class.getSimpleName());
             return new LimitOperator(operatorContext);
         }
 
         @Override
         public void close()
         {
              = true;
         }
     }
 
     private final OperatorContext operatorContext;
     private final List<Typetypes;
     private Page nextPage;
     private long remainingLimit;
 
     public LimitOperator(OperatorContext operatorContextList<Typetypeslong limit)
     {
         this. = checkNotNull(operatorContext"operatorContext is null");
         this. = checkNotNull(types"types is null");
 
         checkArgument(limit >= 0, "limit must be at least zero");
         this. = limit;
     }
 
     @Override
     {
         return ;
     }
 
     @Override
     public List<TypegetTypes()
     {
         return ;
     }
 
     @Override
     public void finish()
     {
          = 0;
     }
 
     @Override
     public boolean isFinished()
    {
        return  == 0 &&  == null;
    }
    @Override
    public boolean needsInput()
    {
        return  > 0 &&  == null;
    }
    @Override
    public void addInput(Page page)
    {
        checkState(needsInput());
        if (page.getPositionCount() <= ) {
             -= page.getPositionCount();
             = page;
        }
        else {
            Block[] blocks = new Block[page.getChannelCount()];
            for (int channel = 0; channel < page.getChannelCount(); channel++) {
                Block block = page.getBlock(channel);
                blocks[channel] = block.getRegion(0, (int);
            }
             = new Page((intblocks);
             = 0;
        }
    }
    @Override
    public Page getOutput()
    {
        Page page = ;
         = null;
        return page;
    }
New to GrepCode? Check out our FAQ X