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.aggregation.state;
 
 
 import static com.google.common.base.Preconditions.checkNotNull;
 
         implements AccumulatorStateFactory<KeyValuePairsState>
 {
     private final Type keyType;
     private final Type valueType;
 
     public KeyValuePairsStateFactory(Type keyTypeType valueType)
     {
         this. = keyType;
         this. = valueType;
     }
 
     @Override
     {
         return new SingleState();
     }
 
     @Override
     public Class<? extends KeyValuePairsStategetSingleStateClass()
     {
         return SingleState.class;
     }
 
     @Override
     {
         return new GroupedState();
     }
 
     @Override
     public Class<? extends KeyValuePairsStategetGroupedStateClass()
     {
         return GroupedState.class;
     }
 
     public static class GroupedState
             extends AbstractGroupedAccumulatorState
             implements KeyValuePairsState
     {
         private final Type keyType;
         private final Type valueType;
         private final ObjectBigArray<KeyValuePairspairs = new ObjectBigArray<>();
         private long size;
 
         public GroupedState(Type keyTypeType valueType)
         {
             this. = keyType;
             this. = valueType;
         }
 
         @Override
         public void ensureCapacity(long size)
         {
             .ensureCapacity(size);
         }
 
         @Override
         public KeyValuePairs get()
         {
             return .get(getGroupId());
         }
 
         @Override
         public void set(KeyValuePairs value)
         {
             checkNotNull(value"value is null");
 
             KeyValuePairs previous = get();
             if (previous != null) {
                  -= previous.estimatedInMemorySize();
             }
 
             .set(getGroupId(), value);
              += value.estimatedInMemorySize();
         }
 
         @Override
        public void addMemoryUsage(long memory)
        {
             += memory;
        }
        @Override
        public Type getKeyType()
        {
            return ;
        }
        @Override
        public Type getValueType()
        {
            return ;
        }
        @Override
        public long getEstimatedSize()
        {
            return  + .sizeOf();
        }
    }
    public static class SingleState
            implements KeyValuePairsState
    {
        private final Type keyType;
        private final Type valueType;
        private KeyValuePairs pair;
        public SingleState(Type keyTypeType valueType)
        {
            this. = keyType;
            this. = valueType;
        }
        @Override
        public KeyValuePairs get()
        {
            return ;
        }
        @Override
        public void set(KeyValuePairs value)
        {
             = value;
        }
        @Override
        public void addMemoryUsage(long memory)
        {
        }
        @Override
        public Type getKeyType()
        {
            return ;
        }
        @Override
        public Type getValueType()
        {
            return ;
        }
        @Override
        public long getEstimatedSize()
        {
            if ( == null) {
                return 0;
            }
            return .estimatedInMemorySize();
        }
    }
New to GrepCode? Check out our FAQ X