All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • com.hazelcast.jet.sql.impl.expression.ToRowFunction Maven / Gradle / Ivy

    There is a newer version: 5.5.0
    Show newest version
    /*
 * Copyright 2021 Hazelcast Inc.
 *
 * Licensed under the Hazelcast Community License (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://hazelcast.com/hazelcast-community-license
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.hazelcast.jet.sql.impl.expression;

import com.hazelcast.jet.impl.util.ReflectionUtils;
import com.hazelcast.jet.sql.impl.JetSqlSerializerHook;
import com.hazelcast.nio.serialization.IdentifiedDataSerializable;
import com.hazelcast.nio.serialization.genericrecord.GenericRecord;
import com.hazelcast.sql.impl.QueryException;
import com.hazelcast.sql.impl.SqlErrorCode;
import com.hazelcast.sql.impl.expression.Expression;
import com.hazelcast.sql.impl.expression.ExpressionEvalContext;
import com.hazelcast.sql.impl.expression.RowValue;
import com.hazelcast.sql.impl.expression.UniExpressionWithType;
import com.hazelcast.sql.impl.row.Row;
import com.hazelcast.sql.impl.type.QueryDataType;

import java.util.ArrayList;
import java.util.IdentityHashMap;
import java.util.List;
import java.util.Set;

import static java.util.Collections.newSetFromMap;

public class ToRowFunction extends UniExpressionWithType implements IdentifiedDataSerializable {

    public ToRowFunction() { }

    private ToRowFunction(Expression operand) {
        super(operand, QueryDataType.ROW);
    }

    public static ToRowFunction create(Expression operand) {
        return new ToRowFunction(operand);
    }

    @Override
    public RowValue eval(final Row row, final ExpressionEvalContext context) {
        final Object object = this.operand.eval(row, context);
        final QueryDataType queryDataType = operand.getType();

        return convert(object, queryDataType, newSetFromMap(new IdentityHashMap<>()));
    }

    private RowValue convert(final Object obj, final QueryDataType dataType, final Set seenObjects) {
        if (obj instanceof GenericRecord) {
            throw QueryException.error("TO_ROW function is only supported for Java types");
        }

        if (!seenObjects.add(obj)) {
            throw QueryException.error(SqlErrorCode.DATA_EXCEPTION, "Cycle detected in row value");
        }

        final List fieldValues = new ArrayList<>();
        for (final QueryDataType.QueryDataTypeField field : dataType.getObjectFields()) {
            final Object fieldValue = ReflectionUtils.getFieldValue(field.getName(), obj);
            if (!field.getDataType().isCustomType() || fieldValue == null) {
                fieldValues.add(fieldValue);
            } else {
                fieldValues.add(convert(fieldValue, field.getDataType(), seenObjects));
            }
        }
        return new RowValue(fieldValues);
    }

    @Override
    public QueryDataType getType() {
        return QueryDataType.ROW;
    }

    @Override
    public int getFactoryId() {
        return JetSqlSerializerHook.F_ID;
    }

    @Override
    public int getClassId() {
        return JetSqlSerializerHook.TO_ROW;
    }
}
    
    
    
    

    




    
    
    © 2015 - 2024 Weber Informatics LLC | Privacy Policy