io.deephaven.server.table.ops.MultiJoinGrpcImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of deephaven-server Show documentation
The 'server' project
The newest version!
//
// Copyright (c) 2016-2024 Deephaven Data Labs and Patent Pending
//
package io.deephaven.server.table.ops;

import com.google.rpc.Code;
import io.deephaven.auth.codegen.impl.TableServiceContextualAuthWiring;
import io.deephaven.engine.table.MultiJoinFactory;
import io.deephaven.engine.table.Table;
import io.deephaven.proto.backplane.grpc.BatchTableRequest;
import io.deephaven.proto.backplane.grpc.MultiJoinInput;
import io.deephaven.proto.backplane.grpc.MultiJoinTablesRequest;
import io.deephaven.proto.util.Exceptions;
import io.deephaven.server.grpc.Common;
import io.deephaven.server.grpc.GrpcErrorHelper;
import io.deephaven.server.session.SessionState;
import io.grpc.StatusRuntimeException;

import javax.inject.Inject;
import javax.inject.Singleton;
import java.util.List;
import java.util.stream.Collectors;

@Singleton
public class MultiJoinGrpcImpl extends GrpcTableOperation {

    @Inject
    public MultiJoinGrpcImpl(
            final TableServiceContextualAuthWiring authWiring) {
        super(authWiring::checkPermissionMultiJoinTables,
                BatchTableRequest.Operation::getMultiJoin,
                MultiJoinTablesRequest::getResultId,
                (MultiDependencyFunction) request -> request.getMultiJoinInputsList().stream()
                        .map(MultiJoinInput::getSourceId).collect(Collectors.toList()));
    }

    @Override
    public void validateRequest(final MultiJoinTablesRequest request) throws StatusRuntimeException {
        GrpcErrorHelper.checkHasNoUnknownFields(request);

        if (request.getMultiJoinInputsList().isEmpty()) {
            throw Exceptions.statusRuntimeException(Code.INVALID_ARGUMENT, "Cannot join zero source tables.");
        }

        // Verify well-formed MultiJoinInput.
        request.getMultiJoinInputsList().forEach(input -> {
            GrpcErrorHelper.checkHasNoUnknownFields(input);
            // Columns to match must be non-empty.
            GrpcErrorHelper.checkRepeatedFieldNonEmpty(input, MultiJoinInput.COLUMNS_TO_MATCH_FIELD_NUMBER);
            // Validate the source table id.
            Common.validate(input.getSourceId());
        });
    }

    @Override
    public Table create(final MultiJoinTablesRequest request,
            final List> sourceTables) {

        final Table firstTable = sourceTables.get(0).get();
        final Table[] allTables = sourceTables.stream().map(SessionState.ExportObject::get).toArray(Table[]::new);

        // Build the multiJoinInput array.
        final io.deephaven.engine.table.MultiJoinInput[] multiJoinInputs =
                new io.deephaven.engine.table.MultiJoinInput[request.getMultiJoinInputsCount()];

        for (int i = 0; i < request.getMultiJoinInputsCount(); i++) {
            final Table table = sourceTables.get(i).get();

            final MultiJoinInput mjInput = request.getMultiJoinInputs(i);
            final String[] columnsToMatch = mjInput.getColumnsToMatchList().toArray(new String[0]);
            final String[] columnsToAdd = mjInput.getColumnsToAddList().toArray(new String[0]);

            multiJoinInputs[i] = io.deephaven.engine.table.MultiJoinInput.of(table, columnsToMatch, columnsToAdd);
        }
        return firstTable.getUpdateGraph(allTables).sharedLock().computeLocked(
                () -> MultiJoinFactory.of(multiJoinInputs).table());
    }
}