wtf.metio.yosql.codegen.api.AbstractMethodsGenerator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yosql-codegen-api Show documentation
The newest version!
/*
 * 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 http://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.api;

import com.squareup.javapoet.MethodSpec;
import wtf.metio.yosql.models.immutables.SqlConfiguration;
import wtf.metio.yosql.models.immutables.SqlStatement;

import java.util.ArrayList;
import java.util.List;

/**
 * Abstract implementation of a {@link MethodsGenerator}.
 */
public abstract class AbstractMethodsGenerator implements MethodsGenerator {

    @Override
    public final Iterable asMethods(final List statements) {
        final var methods = new ArrayList(statements.size());

        methods.add(constructor(statements));
        statements.stream()
                .filter(SqlStatement::shouldGenerateStandardReadAPI)
                .collect(SqlStatement.groupByName())
                .forEach((methodName, statementsInRepository) -> methods
                        .add(standardReadMethod(SqlConfiguration.fromStatements(statementsInRepository), statementsInRepository)));
        statements.stream()
                .filter(SqlStatement::shouldGenerateStandardWriteAPI)
                .collect(SqlStatement.groupByName())
                .forEach((methodName, statementsInRepository) -> methods
                        .add(standardWriteMethod(SqlConfiguration.fromStatements(statementsInRepository), statementsInRepository)));
        statements.stream()
                .filter(SqlStatement::shouldGenerateStandardCallAPI)
                .collect(SqlStatement.groupByName())
                .forEach((methodName, statementsInRepository) -> methods
                        .add(standardCallMethod(SqlConfiguration.fromStatements(statementsInRepository), statementsInRepository)));
        statements.stream()
                .filter(SqlStatement::shouldGenerateBatchAPI)
                .collect(SqlStatement.groupByName())
                .forEach((methodName, statementsInRepository) -> methods
                        .add(batchWriteMethod(SqlConfiguration.fromStatements(statementsInRepository), statementsInRepository)));
        statements.stream()
                .filter(SqlStatement::shouldGenerateStreamEagerAPI)
                .collect(SqlStatement.groupByName())
                .forEach((methodName, statementsInRepository) -> methods
                        .add(streamEagerReadMethod(SqlConfiguration.fromStatements(statementsInRepository), statementsInRepository)));
        statements.stream()
                .filter(SqlStatement::shouldGenerateStreamLazyAPI)
                .collect(SqlStatement.groupByName())
                .forEach((methodName, statementsInRepository) -> methods
                        .add(streamLazyReadMethod(SqlConfiguration.fromStatements(statementsInRepository), statementsInRepository)));
        statements.stream()
                .filter(SqlStatement::shouldGenerateRxJavaAPI)
                .collect(SqlStatement.groupByName())
                .forEach((methodName, statementsInRepository) -> methods
                        .add(rxJavaReadMethod(SqlConfiguration.fromStatements(statementsInRepository), statementsInRepository)));

        return methods;
    }

    public abstract MethodSpec constructor(List sqlStatementsInRepository);

    public abstract MethodSpec standardReadMethod(
            SqlConfiguration configuration,
            List statements);

    public abstract MethodSpec standardWriteMethod(
            SqlConfiguration configuration,
            List statements);

    public abstract MethodSpec standardCallMethod(
            SqlConfiguration configuration,
            List statements);

    public abstract MethodSpec batchWriteMethod(
            SqlConfiguration configuration,
            List statements);

    public abstract MethodSpec streamEagerReadMethod(
            SqlConfiguration configuration,
            List statements);

    public abstract MethodSpec streamLazyReadMethod(
            SqlConfiguration configuration,
            List statements);

    public abstract MethodSpec rxJavaReadMethod(
            SqlConfiguration configuration,
            List statements);

}