com.liferay.change.tracking.internal.spi.reference.ExpandoTableReferenceAppender Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of com.liferay.change.tracking.service
Liferay Change Tracking Service
There is a newer version: 3.0.108
/**
 * 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.spi.reference;

import com.liferay.change.tracking.internal.reference.TableReferenceAppender;
import com.liferay.change.tracking.spi.reference.builder.ChildTableReferenceInfoBuilder;
import com.liferay.change.tracking.spi.reference.builder.ParentTableReferenceInfoBuilder;
import com.liferay.expando.kernel.model.ExpandoColumnTable;
import com.liferay.expando.kernel.model.ExpandoRowTable;
import com.liferay.expando.kernel.model.ExpandoTableTable;
import com.liferay.expando.kernel.model.ExpandoValueTable;
import com.liferay.petra.sql.dsl.Column;
import com.liferay.petra.sql.dsl.Table;
import com.liferay.petra.sql.dsl.expression.Predicate;
import com.liferay.portal.kernel.model.BaseModel;
import com.liferay.portal.kernel.model.ClassNameTable;

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

/**
 * @author Preston Crary
 */
public class ExpandoTableReferenceAppender implements TableReferenceAppender {

	@Override
	public > void appendChildTableReferences(
		Class> modelClass,
		Column primaryKeyColumn,
		ChildTableReferenceInfoBuilder childTableReferenceInfoBuilder) {

		T table = primaryKeyColumn.getTable();

		if (_excludeTables.contains(table)) {
			return;
		}

		childTableReferenceInfoBuilder.referenceInnerJoin(
			fromStep -> {
				Predicate predicate = primaryKeyColumn.eq(
					ExpandoRowTable.INSTANCE.classPK);

				Column companyIdColumn = table.getColumn(
					"companyId", Long.class);

				if (companyIdColumn != null) {
					predicate = predicate.and(
						companyIdColumn.eq(ExpandoRowTable.INSTANCE.companyId));
				}

				return fromStep.from(
					ExpandoRowTable.INSTANCE
				).innerJoinON(
					table, predicate
				).innerJoinON(
					ExpandoTableTable.INSTANCE,
					ExpandoTableTable.INSTANCE.tableId.eq(
						ExpandoRowTable.INSTANCE.tableId)
				).innerJoinON(
					ClassNameTable.INSTANCE,
					ClassNameTable.INSTANCE.classNameId.eq(
						ExpandoTableTable.INSTANCE.classNameId
					).and(
						ClassNameTable.INSTANCE.value.eq(modelClass.getName())
					)
				);
			});
	}

	@Override
	public > void appendParentTableReferences(
		Class> modelClass,
		Column primaryKeyColumn,
		ParentTableReferenceInfoBuilder parentTableReferenceInfoBuilder) {
	}

	private static final Set> _excludeTables = new HashSet<>(
		Arrays.asList(
			ExpandoRowTable.INSTANCE, ExpandoTableTable.INSTANCE,
			ExpandoColumnTable.INSTANCE, ExpandoValueTable.INSTANCE));

}