/*
 * Copyright 2019 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.springframework.cloud.dataflow.server.db.migration.postgresql;

import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Types;
import java.util.List;

import org.postgresql.core.SqlCommand;

import org.springframework.cloud.dataflow.server.db.migration.AbstractMigrateUriRegistrySqlCommand;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.support.SqlLobValue;
import org.springframework.jdbc.support.lob.DefaultLobHandler;

/**
 * {@code postgres} related {@link SqlCommand} for migrating data from
 * {@code URI_REGISTRY} into {@code app_registration}.
 *
 * @author Janne Valkealahti
 *
 */
public class PostgresMigrateUriRegistrySqlCommand extends AbstractMigrateUriRegistrySqlCommand {

	@Override
	public void handle(JdbcTemplate jdbcTemplate, Connection connection) {
		// we need to disable connection autocommit for this operation
		// as with postgres a CLOB cannot be inserted with autocommit enabled.

		// TODO: should think if this same autocommit logic could get extracted to
		//       to some sort of base impl as it's used in PostgresMigrateTaskDefinitionsSqlCommand
		//       and PostgresMigrateStreamDefinitionsSqlCommand in a same way

		Boolean autoCommit = null;
		try {
			autoCommit = connection.getAutoCommit();
		} catch (SQLException e) {
			throw new RuntimeException("cannot access connection autocommit setting", e);
		}
		if (autoCommit != null) {
			try {
				connection.setAutoCommit(false);
			} catch (SQLException e) {
				throw new RuntimeException("cannot access connection autocommit setting", e);
			}
		}
		super.handle(jdbcTemplate, connection);
		if (autoCommit != null) {
			try {
				connection.setAutoCommit(autoCommit);
			} catch (SQLException e) {
				throw new RuntimeException("cannot access connection autocommit setting", e);
			}
		}
	}

	@Override
	protected void updateAppRegistration(JdbcTemplate jdbcTemplate, List<AppRegistrationMigrationData> data) {
		DefaultLobHandler lobHandler = new DefaultLobHandler();
		lobHandler.setWrapAsLob(true);
		for (AppRegistrationMigrationData d : data) {
			Long nextVal = jdbcTemplate.queryForObject("select nextval('hibernate_sequence')", Long.class);
			jdbcTemplate.update(
					"insert into app_registration (id, object_version, default_version, metadata_uri, name, type, uri, version) values (?,?,?,?,?,?,?,?)",
					new Object[] { nextVal, 0, d.isDefaultVersion(), new SqlLobValue(d.getMetadataUri(), lobHandler),
							d.getName(), d.getType(), new SqlLobValue(d.getUri(), lobHandler), d.getVersion() },
					new int[] { Types.BIGINT, Types.BIGINT, Types.BOOLEAN, Types.CLOB, Types.VARCHAR, Types.INTEGER,
							Types.CLOB, Types.VARCHAR });
		}
	}
}