com.github.molcikas.photon.sqlbuilders.UpdateSqlBuilderService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of photon-core Show documentation
null
The newest version!
package com.github.molcikas.photon.sqlbuilders;

import com.github.molcikas.photon.blueprints.table.ColumnBlueprint;
import com.github.molcikas.photon.blueprints.entity.EntityBlueprint;
import com.github.molcikas.photon.blueprints.table.TableBlueprint;
import com.github.molcikas.photon.options.PhotonOptions;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.*;
import java.util.stream.Collectors;

public final class UpdateSqlBuilderService
{
    private static final Logger log = LoggerFactory.getLogger(UpdateSqlBuilderService.class);

    public static void buildUpdateSqlTemplates(EntityBlueprint aggregateRootEntityBlueprint, PhotonOptions photonOptions)
    {
        buildUpdateSqlRecursive(aggregateRootEntityBlueprint, photonOptions);
    }

    public static String buildSetClauseSql(TableBlueprint tableBlueprint, Set columnNames, PhotonOptions photonOptions)
    {
        StringBuilder sqlBuilder = new StringBuilder();

        Collection columnBlueprints = tableBlueprint
            .getColumns()
            .stream()
            .filter(c -> !c.isPrimaryKeyColumn() && columnNames.contains(c.getColumnName()))
            .sorted(Comparator.comparingInt(ColumnBlueprint::getColumnIndex))
            .collect(Collectors.toList());

        if(columnBlueprints.isEmpty())
        {
            // UPDATE statements cannot have an empty SET list, but we need to run an update to see if the row
            // exists in the table, so set the id equal to itself as a "dummy" set.
            sqlBuilder.append(String.format("[%s].[%s] = [%s].[%s]",
                tableBlueprint.getTableName(),
                tableBlueprint.getPrimaryKeyColumnName(),
                tableBlueprint.getTableName(),
                tableBlueprint.getPrimaryKeyColumnName()
            ));
        }

        int collectionIndex = 0;

        for(ColumnBlueprint columnBlueprint : columnBlueprints)
        {
            sqlBuilder.append(String.format("[%s].[%s] = ?%s",
                tableBlueprint.getTableName(),
                columnBlueprint.getColumnName(),
                collectionIndex < columnBlueprints.size() - 1 ? ", " : ""
            ));
            collectionIndex++;
        }

        return SqlBuilderApplyOptionsService.applyPhotonOptionsToSql(sqlBuilder.toString(), photonOptions);
    }

    private static void buildUpdateSqlRecursive(
        EntityBlueprint entityBlueprint,
        PhotonOptions photonOptions)
    {
        buildUpdateSqlForTableBlueprint(entityBlueprint, entityBlueprint.getTableBlueprint(), photonOptions);
        for(TableBlueprint joinedTableBlueprint : entityBlueprint.getJoinedTableBlueprints())
        {
            buildUpdateSqlForTableBlueprint(entityBlueprint, joinedTableBlueprint, photonOptions);
        }

        entityBlueprint
            .getFieldsWithChildEntities()
            .forEach(entityField -> buildUpdateSqlRecursive(entityField.getChildEntityBlueprint(), photonOptions));
    }

    private static void buildUpdateSqlForTableBlueprint(EntityBlueprint entityBlueprint, TableBlueprint tableBlueprint, PhotonOptions photonOptions)
    {
        int initialCapacity = tableBlueprint.getColumns().size() * 16 + 64;
        StringBuilder sqlBuilder = new StringBuilder(initialCapacity);

        buildUpdateClauseSql(sqlBuilder, tableBlueprint);
        sqlBuilder.append("\nSET %s");
        buildWhereClauseSql(sqlBuilder, tableBlueprint, entityBlueprint);

        String updateSql = SqlBuilderApplyOptionsService.applyPhotonOptionsToSql(sqlBuilder.toString(), photonOptions);
        log.debug("Update Sql for {}:\n{}", tableBlueprint.getTableName(), updateSql);
        tableBlueprint.setUpdateSql(updateSql);
    }

    private static void buildUpdateClauseSql(StringBuilder sqlBuilder, TableBlueprint tableBlueprint)
    {
        sqlBuilder.append(String.format("UPDATE [%s]", tableBlueprint.getTableName()));
    }

    private static void buildWhereClauseSql(StringBuilder sqlBuilder, TableBlueprint tableBlueprint, EntityBlueprint entityBlueprint)
    {
        sqlBuilder.append(String.format("%nWHERE [%s].[%s] = ?",
            tableBlueprint.getTableName(),
            tableBlueprint.getPrimaryKeyColumnName()
        ));

        ColumnBlueprint versionColumn = tableBlueprint.getVersionColumn(entityBlueprint);
        if(versionColumn == null)
        {
            return;
        }

        sqlBuilder.append(String.format(" AND [%s].[%s] = ?",
            tableBlueprint.getTableName(),
            versionColumn.getColumnName()
        ));
    }
}