com.liferay.notification.internal.upgrade.v3_6_0.NotificationQueueEntryUpgradeProcess Maven / Gradle / Ivy

Go to download
/**
 * SPDX-FileCopyrightText: (c) 2000 Liferay, Inc. https://liferay.com
 * SPDX-License-Identifier: LGPL-2.1-or-later OR LicenseRef-Liferay-DXP-EULA-2.0.0-2023-06
 */

package com.liferay.notification.internal.upgrade.v3_6_0;

import com.liferay.notification.model.NotificationQueueEntry;
import com.liferay.petra.string.StringBundler;
import com.liferay.portal.kernel.dao.jdbc.AutoBatchPreparedStatementUtil;
import com.liferay.portal.kernel.model.ClassName;
import com.liferay.portal.kernel.service.ClassNameLocalService;
import com.liferay.portal.kernel.service.ResourceLocalService;
import com.liferay.portal.kernel.upgrade.UpgradeProcess;

import java.sql.PreparedStatement;
import java.sql.ResultSet;

/**
 * @author Pedro Leite
 */
public class NotificationQueueEntryUpgradeProcess extends UpgradeProcess {

	public NotificationQueueEntryUpgradeProcess(
		ClassNameLocalService classNameLocalService,
		ResourceLocalService resourceLocalService) {

		_classNameLocalService = classNameLocalService;
		_resourceLocalService = resourceLocalService;
	}

	@Override
	protected void doUpgrade() throws Exception {
		try (PreparedStatement preparedStatement1 = connection.prepareStatement(
				StringBundler.concat(
					"select NotificationQueueEntry.notificationQueueEntryId, ",
					"NotificationQueueEntry.userId, ",
					"NotificationQueueEntry.notificationTemplateId, ",
					"NotificationQueueEntry.classNameId, ",
					"NotificationQueueEntry.classPK, ",
					"NotificationTemplate.companyId from ",
					"NotificationQueueEntry inner join NotificationTemplate ",
					"on NotificationQueueEntry.notificationTemplateId = ",
					"NotificationTemplate.notificationTemplateId where ",
					"notificationQueueEntryId not in (select primKeyId from ",
					"ResourcePermission where name = ?)"));
			PreparedStatement preparedStatement2 = connection.prepareStatement(
				StringBundler.concat(
					"select ObjectDefinition.className from ObjectDefinition ",
					"where ObjectDefinition.objectDefinitionId = (select ",
					"ObjectEntry.objectDefinitionId from ObjectEntry where ",
					"objectEntryId = ?)"));
			PreparedStatement preparedStatement3 =
				AutoBatchPreparedStatementUtil.concurrentAutoBatch(
					connection,
					"update NotificationQueueEntry set classNameId = ? where " +
						"notificationQueueEntryId = ?")) {

			preparedStatement1.setString(
				1, NotificationQueueEntry.class.getName());

			try (ResultSet resultSet1 = preparedStatement1.executeQuery()) {
				while (resultSet1.next()) {
					_resourceLocalService.addResources(
						resultSet1.getLong("companyId"), 0,
						resultSet1.getLong("userId"),
						NotificationQueueEntry.class.getName(),
						resultSet1.getLong("notificationQueueEntryId"), false,
						true, true);

					String classNameString = null;

					preparedStatement2.setLong(
						1, resultSet1.getLong("classPK"));

					try (ResultSet resultSet2 =
							preparedStatement2.executeQuery()) {

						if (resultSet2.next()) {
							classNameString = resultSet2.getString("className");
						}
					}

					if (classNameString == null) {
						continue;
					}

					ClassName className = _classNameLocalService.fetchClassName(
						classNameString);

					if (className == null) {
						continue;
					}

					long classNameId = className.getClassNameId();

					if ((classNameId == 0) ||
						(classNameId == resultSet1.getLong("classNameId"))) {

						continue;
					}

					preparedStatement3.setLong(1, classNameId);
					preparedStatement3.setLong(
						2, resultSet1.getLong("notificationQueueEntryId"));

					preparedStatement3.addBatch();
				}
			}

			preparedStatement3.executeBatch();
		}
	}

	private final ClassNameLocalService _classNameLocalService;
	private final ResourceLocalService _resourceLocalService;

}