io.deephaven.server.table.ops.WhereInGrpcImpl 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 io.deephaven.api.JoinMatch;
import io.deephaven.auth.codegen.impl.TableServiceContextualAuthWiring;
import io.deephaven.base.verify.Assert;
import io.deephaven.engine.table.Table;
import io.deephaven.proto.backplane.grpc.BatchTableRequest;
import io.deephaven.proto.backplane.grpc.TableReference;
import io.deephaven.proto.backplane.grpc.WhereInRequest;
import io.deephaven.server.grpc.Common;
import io.deephaven.server.grpc.GrpcErrorHelper;
import io.deephaven.server.session.SessionState;
import io.deephaven.util.SafeCloseable;
import io.grpc.StatusRuntimeException;

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

@Singleton
public class WhereInGrpcImpl extends GrpcTableOperation {

    private static List refs(WhereInRequest request) {
        return List.of(request.getLeftId(), request.getRightId());
    }

    @Inject
    public WhereInGrpcImpl(final TableServiceContextualAuthWiring authWiring) {
        super(authWiring::checkPermissionWhereIn, BatchTableRequest.Operation::getWhereIn,
                WhereInRequest::getResultId, WhereInGrpcImpl::refs);
    }

    @Override
    public final void validateRequest(WhereInRequest request) throws StatusRuntimeException {
        GrpcErrorHelper.checkHasField(request, WhereInRequest.LEFT_ID_FIELD_NUMBER);
        GrpcErrorHelper.checkHasField(request, WhereInRequest.RIGHT_ID_FIELD_NUMBER);
        GrpcErrorHelper.checkRepeatedFieldNonEmpty(request, WhereInRequest.COLUMNS_TO_MATCH_FIELD_NUMBER);
        GrpcErrorHelper.checkHasNoUnknownFields(request);
        Common.validate(request.getLeftId());
        Common.validate(request.getRightId());
    }

    @Override
    public final Table create(final WhereInRequest request, final List> sourceTables) {
        Assert.eq(sourceTables.size(), "sourceTables.size()", 2);
        Assert.gtZero(request.getColumnsToMatchCount(), "request.getColumnsToMatchCount()");
        final Table left = sourceTables.get(0).get();
        final Table right = sourceTables.get(1).get();
        final List columnsToMatch = JoinMatch.from(request.getColumnsToMatchList());
        try (final SafeCloseable ignored = lock(left, right)) {
            return request.getInverted() ? left.whereNotIn(right, columnsToMatch) : left.whereIn(right, columnsToMatch);
        }
    }

    private SafeCloseable lock(Table left, Table right) {
        if (left.isRefreshing()) {
            return left.getUpdateGraph().sharedLock().lockCloseable();
        } else if (right.isRefreshing()) {
            return right.getUpdateGraph().sharedLock().lockCloseable();
        }
        return null;
    }
}