com.speedment.generator.standard.internal.util.GenerateMethodBodyUtil Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of tool-deploy Show documentation
A Speedment bundle that shades all dependencies into one jar. This is useful when deploying an application on a server.
The newest version!
/*
 *
 * Copyright (c) 2006-2019, Speedment, Inc. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); You may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at:
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.speedment.generator.standard.internal.util;

import com.speedment.common.codegen.constant.DefaultType;
import com.speedment.common.codegen.constant.SimpleParameterizedType;
import com.speedment.common.codegen.model.File;
import com.speedment.common.codegen.model.Import;
import com.speedment.common.codegen.model.Method;
import com.speedment.generator.translator.TranslatorSupport;
import com.speedment.runtime.config.Column;
import com.speedment.runtime.config.Table;
import com.speedment.runtime.config.trait.HasEnabled;

import java.sql.SQLException;
import java.util.LinkedList;
import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Supplier;
import java.util.stream.Stream;

import static com.speedment.common.codegen.constant.DefaultAnnotationUsage.OVERRIDE;
import static com.speedment.common.codegen.util.Formatting.*;
import static java.util.stream.Collectors.joining;

/**
 *
 * @author Emil Forslund
 * @since 3.0.0
 */
public final class GenerateMethodBodyUtil {

    public static Method generateFields(
        TranslatorSupport support,
        File file,
        String methodName,
        Supplier> columnsSupplier) {

        return Method.of(methodName,
            DefaultType.stream(
                SimpleParameterizedType.create(
                    com.speedment.runtime.field.Field.class,
                    support.entityType()
                )
            )
        )
            .public_().add(OVERRIDE)
            .add(generateFieldsBody(support, file, columnsSupplier));
    }

    public static String[] generateFieldsBody(
        TranslatorSupport support,
        File file,
        Supplier> columnsSupplier) {

        file.add(Import.of(Stream.class));

        // If there are no matching columns:
        if (columnsSupplier.get().noneMatch($ -> true)) {
            return new String[]{"return Stream.empty();"};
        } else {
            final List rows = new LinkedList<>();

            rows.add("return Stream.of(");
            rows.add(indent(columnsSupplier.get()
                .filter(HasEnabled::isEnabled)
                .map(Column::getJavaName)
                .map(support.namer()::javaStaticFieldName)
                .map(field -> support.typeName() + "." + field)
                .collect(joining("," + nl()))
            ));
            rows.add(");");

            return rows.toArray(new String[rows.size()]);
        }
    }

    @FunctionalInterface
    public interface ReadFromResultSet {

        String readFromResultSet(File file, Column c, AtomicInteger position);
    }

    public static String[] generateApplyResultSetBody(
        ReadFromResultSet readFromResultSet,
        TranslatorSupport support,
        File file,
        Supplier> columnsSupplier) {

        file.add(Import.of(SQLException.class));

        final List rows = new LinkedList<>();
        rows.add("return createEntity()");

        final Stream.Builder streamBuilder = Stream.builder();

        final AtomicInteger position = new AtomicInteger(1);
        columnsSupplier.get()
            .filter(HasEnabled::isEnabled)
            .forEachOrdered(c
                -> streamBuilder.add(indent(".set"
                + support.namer().javaTypeName(c.getJavaName()) + "(\t "
                + readFromResultSet.readFromResultSet(file, c, position)
                + ")"))
            );
        streamBuilder.add(indent(";"));
        streamBuilder.build().forEachOrdered(rows::add);
        return rows.toArray(new String[rows.size()]);
    }

    private GenerateMethodBodyUtil() {
        throw new UnsupportedOperationException();
    }
}
 













© 2015 - 2024 Weber Informatics LLC | Privacy Policy