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.FunctionRegistry.operatorInfo;
import static com.facebook.presto.metadata.Signature.comparableTypeParameter;
import static com.facebook.presto.metadata.Signature.typeParameter;
import static com.facebook.presto.spi.StandardErrorCode.INVALID_CAST_ARGUMENT;
import static com.facebook.presto.type.MapType.toStackRepresentation;
import static com.facebook.presto.type.TypeJsonUtils.canCastFromJson;
import static com.facebook.presto.type.TypeJsonUtils.stackRepresentationToObject;
import static com.facebook.presto.type.TypeUtils.parameterizedTypeName;
import static com.facebook.presto.util.Failures.checkCondition;
import static com.facebook.presto.util.Reflection.methodHandle;
import static com.google.common.base.Preconditions.checkArgument;
public class JsonToMapCast
        extends ParametricOperator
    public static final JsonToMapCast JSON_TO_MAP = new JsonToMapCast();
    private static final MethodHandle METHOD_HANDLE = methodHandle(JsonToMapCast.class"toMap"Type.classConnectorSession.classSlice.class);
    private JsonToMapCast()
    {
        super(., ImmutableList.of(comparableTypeParameter("K"), typeParameter("V")), "map<K,V>", ImmutableList.of(.));
    }
    @Override
    public FunctionInfo specialize(Map<StringTypetypesint arityTypeManager typeManagerFunctionRegistry functionRegistry)
    {
        checkArgument(arity == 1, "Expected arity to be 1");
        Type keyType = types.get("K");
        Type valueType = types.get("V");
        Type mapType = typeManager.getParameterizedType(., ImmutableList.of(keyType.getTypeSignature(), valueType.getTypeSignature()), ImmutableList.of());
        checkCondition(canCastFromJson(mapType), "Cannot cast JSON to %s"mapType);
        MethodHandle methodHandle = .bindTo(mapType);
        return operatorInfo(.mapType.getTypeSignature(), ImmutableList.of(parameterizedTypeName(.)), methodHandletrue, ImmutableList.of(false));
    }
    public static Slice toMap(Type mapTypeConnectorSession connectorSessionSlice json)
    {
        try {
            Object map = stackRepresentationToObject(connectorSessionjsonmapType);
            if (map == null) {
                return null;
            }
            return toStackRepresentation((Map<?, ?>) map, ((MapTypemapType).getKeyType(), ((MapTypemapType).getValueType());
        }
        catch (RuntimeException e) {
            throw new PrestoException("Value cannot be cast to " + mapTypee);
        }
    }
New to GrepCode? Check out our FAQ X