All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • win.doyto.query.service.AssociativeSqlBuilder Maven / Gradle / Ivy

    package win.doyto.query.service;

import win.doyto.query.core.SqlAndArgs;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

/**
 * AssociativeSqlBuilder
 *
 * @author f0rb on 2019-06-13
 */
class AssociativeSqlBuilder {

    public static final String WHERE = " WHERE ";
    public static final String IN_S = " IN (%s)";
    public static final String DELETE_FROM = "DELETE FROM ";

    private final String deallocateFormat;
    private final String count;
    private final String insert;
    private final String insertPlaceholders;
    private final boolean insertCreateUser;

    final String getByLeftId;
    final String deleteByLeftId;
    final String getByRightId;
    final String deleteByRightId;

    public AssociativeSqlBuilder(String table, String left, String right, String createUserColumn) {
        deallocateFormat = DELETE_FROM + table + WHERE + left + IN_S + " AND " + right + IN_S;
        count = "SELECT COUNT(*) FROM " + table + WHERE + left + IN_S + " AND " + right + IN_S;
        if (createUserColumn == null) {
            insert = "INSERT INTO " + table + " (" + left + ", " + right + ") values (?, ?)";
            insertPlaceholders = ", (?, ?)";
            insertCreateUser = false;
        } else {
            insert = "INSERT INTO " + table + " (" + left + ", " + right + ", " + createUserColumn  + ") values (?, ?, ?)";
            insertPlaceholders = ", (?, ?, ?)";
            insertCreateUser = true;
        }

        getByLeftId = "SELECT " + right + " FROM " + table + WHERE + left + " = ?";
        deleteByLeftId = DELETE_FROM + table + WHERE + left + " = ?";
        getByRightId = "SELECT " + left + " FROM " + table + WHERE + right + " = ?";
        deleteByRightId = DELETE_FROM + table + WHERE + right + " = ?";
    }

    private String generateReplaceHolders(int size) {
        return "?" + IntStream.range(1, size).mapToObj(i -> ", ?").collect(Collectors.joining());
    }

    private List unionArgs(Object[] leftIds, Object[] rightIds) {
        List argList = new ArrayList<>(leftIds.length + rightIds.length);
        Collections.addAll(argList, leftIds);
        Collections.addAll(argList, rightIds);
        return argList;
    }

    public SqlAndArgs buildDeallocate(Object[] leftIds, Object[] rightIds) {
        String sql = String.format(deallocateFormat, generateReplaceHolders(leftIds.length), generateReplaceHolders(rightIds.length));
        return new SqlAndArgs(sql, unionArgs(leftIds, rightIds));
    }

    public SqlAndArgs buildCount(Object[] leftIds, Object[] rightIds) {
        String sql = String.format(count, generateReplaceHolders(leftIds.length), generateReplaceHolders(rightIds.length));
        return new SqlAndArgs(sql, unionArgs(leftIds, rightIds));
    }

    public SqlAndArgs buildAllocate(Collection leftIds, Collection rightIds, Long userId) {
        StringBuilder sql = new StringBuilder(insert);
        int totalSize = leftIds.size() * rightIds.size();
        for (int i = 0; i < totalSize - 1; i++) {
            sql.append(insertPlaceholders);
        }

        List args = new ArrayList<>(totalSize * (insertCreateUser ? 3 : 2));
        for (Object leftId : leftIds) {
            for (Object rightId : rightIds) {
                args.add(leftId);
                args.add(rightId);
                if (insertCreateUser) {
                    args.add(userId);
                }
            }
        }
        return new SqlAndArgs(sql.toString(), args);
    }

}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy