wtf.metio.yosql.codegen.files.DefaultMethodResultRowConverterConfigurer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yosql-codegen Show documentation
There is a newer version: 2023.5.3
/*
 * This file is part of yosql. It is subject to the license terms in the LICENSE file found in the top-level
 * directory of this distribution and at https://creativecommons.org/publicdomain/zero/1.0/. No part of yosql,
 * including this file, may be copied, modified, propagated, or distributed except according to the terms contained
 * in the LICENSE file.
 */

package wtf.metio.yosql.codegen.files;

import wtf.metio.yosql.models.configuration.ResultRowConverter;
import wtf.metio.yosql.models.immutables.ConverterConfiguration;
import wtf.metio.yosql.models.immutables.SqlConfiguration;

import java.util.Optional;
import java.util.function.Function;
import java.util.function.Predicate;

public final class DefaultMethodResultRowConverterConfigurer implements MethodResultRowConverterConfigurer {

    private final ConverterConfiguration converters;

    public DefaultMethodResultRowConverterConfigurer(final ConverterConfiguration converters) {
        this.converters = converters;
    }

    @Override
    public SqlConfiguration configureResultRowConverter(final SqlConfiguration configuration) {
        return SqlConfiguration.copyOf(configuration)
                .withResultRowConverter(configuration.resultRowConverter()
                        .map(this::setMissingValuesFromRuntimeConfig)
                        .filter(DefaultMethodResultRowConverterConfigurer::ensureConverterIsFullyConfigured)
                        .or(this::getDefaultRowConverter));
    }

    private static boolean ensureConverterIsFullyConfigured(final ResultRowConverter converter) {
        return converter.alias().isPresent()
                && converter.converterType().isPresent()
                && converter.methodName().isPresent()
                && converter.resultType().isPresent();
    }

    private ResultRowConverter setMissingValuesFromRuntimeConfig(final ResultRowConverter original) {
        return ResultRowConverter.builder()
                .setAlias(original.alias().or(() -> getAliasFromRuntimeConfig(original)))
                .setConverterType(original.converterType().or(() -> getConverterTypeFromRuntimeConfig(original)))
                .setMethodName(original.methodName().or(() -> getMethodNameFromRuntimeConfig(original)))
                .setResultType(original.resultType().or(() -> getResultTypeFromRuntimeConfig(original)))
                .build();
    }

    private Optional getDefaultRowConverter() {
        final var defaultConverter = converters.defaultConverter();
        return converters.rowConverters().stream()
                .filter(converter -> defaultConverter.isEmpty() || defaultConverter.get().equals(converter))
                .findFirst()
                .or(() -> defaultConverter);
    }

    private Optional getAliasFromRuntimeConfig(final ResultRowConverter resultConverter) {
        return getConverterFieldFromRuntimeConfig(
                converter -> converterTypeMatches(resultConverter, converter),
                ResultRowConverter::alias);
    }

    private Optional getConverterTypeFromRuntimeConfig(final ResultRowConverter resultConverter) {
        return getConverterFieldFromRuntimeConfig(
                converter -> aliasMatches(resultConverter, converter),
                ResultRowConverter::converterType);
    }

    private Optional getResultTypeFromRuntimeConfig(final ResultRowConverter resultConverter) {
        return getConverterFieldFromRuntimeConfig(
                converter -> aliasMatches(resultConverter, converter)
                        || converterTypeMatches(resultConverter, converter),
                ResultRowConverter::resultType);
    }

    private Optional getMethodNameFromRuntimeConfig(final ResultRowConverter resultConverter) {
        return getConverterFieldFromRuntimeConfig(
                converter -> aliasMatches(resultConverter, converter)
                        || converterTypeMatches(resultConverter, converter),
                ResultRowConverter::methodName);
    }

    private Optional getConverterFieldFromRuntimeConfig(
            final Predicate predicate,
            final Function> mapper) {
        return converters.rowConverters().stream()
                .filter(predicate)
                .map(mapper)
                .flatMap(Optional::stream)
                .findFirst();
    }

    private static boolean aliasMatches(
            final ResultRowConverter resultConverter,
            final ResultRowConverter converter) {
        return converter.alias().equals(resultConverter.alias());
    }

    private static boolean converterTypeMatches(
            final ResultRowConverter resultConverter,
            final ResultRowConverter converter) {
        return converter.converterType().equals(resultConverter.converterType());
    }

}