com.liferay.change.tracking.internal.reference.builder.BaseTableReferenceInfoBuilder Maven / Gradle / Ivy

Go to download
/**
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 */

package com.liferay.change.tracking.internal.reference.builder;

import com.liferay.change.tracking.internal.reference.TableJoinHolder;
import com.liferay.change.tracking.internal.reference.TableJoinHolderFactory;
import com.liferay.change.tracking.spi.reference.TableReferenceDefinition;
import com.liferay.petra.sql.dsl.Column;
import com.liferay.petra.sql.dsl.Table;
import com.liferay.petra.sql.dsl.query.FromStep;
import com.liferay.petra.sql.dsl.query.JoinStep;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Function;

/**
 * @author Preston Crary
 */
public abstract class BaseTableReferenceInfoBuilder> {

	public BaseTableReferenceInfoBuilder(
		TableReferenceDefinition tableReferenceDefinition,
		Column primaryKeyColumn, boolean parent) {

		_tableReferenceDefinition = tableReferenceDefinition;
		_primaryKeyColumn = primaryKeyColumn;
		_parent = parent;
	}

	public Map, List> getTableJoinHoldersMap() {
		return _tableJoinHoldersMap;
	}

	protected void applyReferenceInnerJoin(
		Function joinFunction) {

		TableJoinHolder tableJoinHolder = TableJoinHolderFactory.create(
			joinFunction, _parent, _primaryKeyColumn,
			_tableReferenceDefinition);

		Column parentPKColumn = tableJoinHolder.getParentPKColumn();

		List tableJoinHolders =
			_tableJoinHoldersMap.computeIfAbsent(
				parentPKColumn.getTable(), table -> new ArrayList<>());

		tableJoinHolders.add(tableJoinHolder);
	}

	private final boolean _parent;
	private final Column _primaryKeyColumn;
	private final Map, List> _tableJoinHoldersMap =
		new HashMap<>();
	private final TableReferenceDefinition _tableReferenceDefinition;

}