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.scalar;
import static com.facebook.presto.metadata.Signature.typeParameter;
import static com.facebook.presto.type.TypeUtils.buildStructuralSlice;
import static com.facebook.presto.type.TypeUtils.parameterizedTypeName;
import static com.facebook.presto.type.TypeUtils.readStructuralBlock;
import static com.facebook.presto.util.Reflection.methodHandle;
import static com.google.common.base.Preconditions.checkArgument;
public class MapKeys
        extends ParametricScalar
    public static final MapKeys MAP_KEYS = new MapKeys();
    private static final Signature SIGNATURE = new Signature("map_keys", ImmutableList.of(typeParameter("K"), typeParameter("V")), "array<K>", ImmutableList.of("map<K,V>"), falsefalse);
    private static final MethodHandle METHOD_HANDLE = methodHandle(MapKeys.class"getKeys"Type.classSlice.class);
    @Override
    public Signature getSignature()
    {
        return ;
    }
    @Override
    public boolean isHidden()
    {
        return false;
    }
    @Override
    public boolean isDeterministic()
    {
        return true;
    }
    @Override
    public String getDescription()
    {
        return "Returns the keys of the given map<K,V> as an array";
    }
    @Override
    public FunctionInfo specialize(Map<StringTypetypesint arityTypeManager typeManagerFunctionRegistry functionRegistry)
    {
        checkArgument(arity == 1, "map_keys expects only one argument");
        Type keyType = types.get("K");
        Type valueType = types.get("V");
        MethodHandle methodHandle = .bindTo(keyType);
        Signature signature = new Signature("map_keys",
                parameterizedTypeName("array"keyType.getTypeSignature()),
                parameterizedTypeName("map"keyType.getTypeSignature(), valueType.getTypeSignature()));
        return new FunctionInfo(signaturegetDescription(), isHidden(), methodHandleisDeterministic(), true, ImmutableList.of(false));
    }
    public static Slice getKeys(Type keyTypeSlice map)
    {
        Block block = readStructuralBlock(map);
        BlockBuilder blockBuilder = new VariableWidthBlockBuilder(new BlockBuilderStatus(), block.getSizeInBytes());
        for (int i = 0; i < block.getPositionCount(); i += 2) {
            keyType.appendTo(blockiblockBuilder);
        }
        return buildStructuralSlice(blockBuilder);
    }
New to GrepCode? Check out our FAQ X