UpgradeGroup.java

/**
 * 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.portal.upgrade.v6_1_0;

import com.liferay.portal.kernel.upgrade.UpgradeProcess;
import com.liferay.portal.kernel.util.LoggingTimer;
import com.liferay.portal.kernel.util.StringBundler;
import com.liferay.portal.upgrade.v6_1_0.util.GroupTable;

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

/**
 * @author Hugo Huijser
 * @author Jorge Ferrer
 */
public class UpgradeGroup extends UpgradeProcess {

	@Override
	protected void doUpgrade() throws Exception {
		alter(
			GroupTable.class, new AlterColumnType("name", "VARCHAR(150) null"));

		updateName();
		updateSite();
	}

	protected long getClassNameId(String className) throws Exception {
		try (PreparedStatement ps = connection.prepareStatement(
				"select classNameId from ClassName_ where value = ?")) {

			ps.setString(1, className);

			try (ResultSet rs = ps.executeQuery()) {
				if (rs.next()) {
					return rs.getLong("classNameId");
				}

				return 0;
			}
		}
	}

	protected void updateName() throws Exception {
		try (LoggingTimer loggingTimer = new LoggingTimer()) {
			long organizationClassNameId = getClassNameId(
				"com.liferay.portal.model.Organization");

			StringBundler sb = new StringBundler(4);

			sb.append("select Group_.groupId, Group_.classPK, ");
			sb.append("Organization_.name from Group_ inner join ");
			sb.append("Organization_ on Organization_.organizationId = ");
			sb.append("Group_.classPK where classNameId = ?");

			try (PreparedStatement ps = connection.prepareStatement(
					sb.toString())) {

				ps.setLong(1, organizationClassNameId);

				try (ResultSet rs = ps.executeQuery()) {
					while (rs.next()) {
						long groupId = rs.getLong("groupId");
						long classPK = rs.getLong("classPK");
						String name = rs.getString("name");

						updateName(groupId, classPK, name);
					}
				}
			}
		}
	}

	protected void updateName(long groupId, long classPK, String name)
		throws Exception {

		try (PreparedStatement ps = connection.prepareStatement(
				"update Group_ set name = ? where groupId = ?")) {

			StringBundler sb = new StringBundler(3);

			sb.append(classPK);
			sb.append(_ORGANIZATION_NAME_DELIMETER);
			sb.append(name);

			ps.setString(1, sb.toString());

			ps.setLong(2, groupId);

			ps.executeUpdate();
		}
	}

	protected void updateSite() throws Exception {
		try (LoggingTimer loggingTimer = new LoggingTimer()) {
			long groupClassNameId = getClassNameId(
				"com.liferay.portal.model.Group");

			runSQL(
				"update Group_ set site = TRUE where classNameId = " +
					groupClassNameId);

			long organizationClassNameId = getClassNameId(
				"com.liferay.portal.model.Organization");

			try (PreparedStatement ps = connection.prepareStatement(
					"select distinct Group_.groupId from Group_ inner join " +
						"Layout on Layout.groupId = Group_.groupId where " +
							"classNameId = ?")) {

				ps.setLong(1, organizationClassNameId);

				try (ResultSet rs = ps.executeQuery()) {
					while (rs.next()) {
						long groupId = rs.getLong("groupId");

						runSQL(
							"update Group_ set site = TRUE where groupId = " +
								groupId);
					}
				}
			}
		}
	}

	private static final String _ORGANIZATION_NAME_DELIMETER =
		" LFR_ORGANIZATION ";

}