tech.ydb.jdbc.query.YdbQueryOptions Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ydb-jdbc-driver Show documentation
JDBC Driver over YDB Java SDK
There is a newer version: 2.3.5
package tech.ydb.jdbc.query;

import java.util.Map;

import com.google.common.annotations.VisibleForTesting;

import tech.ydb.jdbc.settings.ParsedProperty;
import tech.ydb.jdbc.settings.YdbOperationProperties;
import tech.ydb.jdbc.settings.YdbOperationProperty;


/**
 *
 * @author Aleksandr Gorshenin
 */
public class YdbQueryOptions {
    private final boolean isEnforceSyntaxV1;

    private final boolean isDetectQueryType;
    private final boolean isDetectJdbcParameters;
    private final boolean isDeclareJdbcParameters;

    private final boolean isPrepareDataQueries;
    private final boolean isDetectBatchQueries;

    @VisibleForTesting
    YdbQueryOptions(
            boolean enforceV1,
            boolean detectQueryType,
            boolean detectJbdcParams,
            boolean declareJdbcParams,
            boolean prepareDataQuery,
            boolean detectBatchQuery
    ) {
        this.isEnforceSyntaxV1 = enforceV1;

        this.isDetectQueryType = detectQueryType;
        this.isDetectJdbcParameters = detectJbdcParams;
        this.isDeclareJdbcParameters = declareJdbcParams;

        this.isPrepareDataQueries = prepareDataQuery;
        this.isDetectBatchQueries = detectBatchQuery;
    }

    public boolean isEnforceSyntaxV1() {
        return isEnforceSyntaxV1;
    }

    public boolean isDetectQueryType() {
        return isDetectQueryType;
    }

    public boolean isDetectJdbcParameters() {
        return isDetectJdbcParameters;
    }

    public boolean isDeclareJdbcParameters() {
        return isDeclareJdbcParameters;
    }

    public boolean iPrepareDataQueries() {
        return isPrepareDataQueries;
    }

    public boolean isDetectBatchQueries() {
        return isDetectBatchQueries;
    }

    public static YdbQueryOptions createFrom(YdbOperationProperties props) {
        int level = props.getJdbcSupportLevel();

        boolean enforceV1 = level > 5;
        boolean declareJdbcParams = level > 4;
        boolean detectJbdcParams = level > 3;
        boolean detectBatchQuery = level > 2;
        boolean prepareDataQuery = level > 1;
        boolean detectQueryType = level > 0;

        // forced properies
        Map, ParsedProperty> params = props.getParams();
        if (params.containsKey(YdbOperationProperty.ENFORCE_SQL_V1)) {
            enforceV1 = params.get(YdbOperationProperty.ENFORCE_SQL_V1).getParsedValue();
        }

        if (params.containsKey(YdbOperationProperty.DISABLE_AUTO_PREPARED_BATCHES)) {
            boolean v = params.get(YdbOperationProperty.DISABLE_AUTO_PREPARED_BATCHES).getParsedValue();
            detectBatchQuery = !v;
        }

        if (params.containsKey(YdbOperationProperty.DISABLE_PREPARE_DATAQUERY)) {
            boolean v = params.get(YdbOperationProperty.DISABLE_PREPARE_DATAQUERY).getParsedValue();
            prepareDataQuery = !v;
            detectBatchQuery = detectBatchQuery && prepareDataQuery;
        }

        if (params.containsKey(YdbOperationProperty.DISABLE_JDBC_PARAMETERS_DECLARE)) {
            boolean v = params.get(YdbOperationProperty.DISABLE_JDBC_PARAMETERS_DECLARE).getParsedValue();
            declareJdbcParams = !v;
        }

        if (params.containsKey(YdbOperationProperty.DISABLE_JDBC_PARAMETERS)) {
            boolean v = params.get(YdbOperationProperty.DISABLE_JDBC_PARAMETERS).getParsedValue();
            detectJbdcParams = !v;
            declareJdbcParams = declareJdbcParams && detectJbdcParams;
        }

        if (params.containsKey(YdbOperationProperty.DISABLE_DETECT_SQL_OPERATIONS)) {
            boolean v = params.get(YdbOperationProperty.DISABLE_DETECT_SQL_OPERATIONS).getParsedValue();
            detectQueryType = !v;
            detectJbdcParams = detectJbdcParams && detectQueryType;
            declareJdbcParams = declareJdbcParams && detectJbdcParams;
        }

        return new YdbQueryOptions(
                enforceV1,
                detectQueryType,
                detectJbdcParams,
                declareJdbcParams,
                prepareDataQuery,
                detectBatchQuery
        );
    }
}