org.hibernate.testing.schema.SchemaUpdateHelper Maven / Gradle / Ivy

Go to download
/*
 * Hibernate, Relational Persistence for Idiomatic Java
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later.
 * See the lgpl.txt file in the root directory or .
 */
package org.hibernate.testing.schema;

import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.Map;

import org.hibernate.boot.spi.MetadataImplementor;
import org.hibernate.cfg.AvailableSettings;
import org.hibernate.engine.config.spi.ConfigurationService;
import org.hibernate.internal.build.AllowSysOut;
import org.hibernate.service.ServiceRegistry;
import org.hibernate.tool.schema.Action;
import org.hibernate.tool.schema.internal.Helper;
import org.hibernate.tool.schema.spi.SchemaManagementToolCoordinator;

/**
 * @author Steve Ebersole
 */
@SuppressWarnings({"WeakerAccess", "unused"})
public class SchemaUpdateHelper {
	public static void update(MetadataImplementor metadata) {
		update( metadata, metadata.getMetadataBuildingOptions().getServiceRegistry() );
	}

	@SuppressWarnings("unchecked")
	public static void update(MetadataImplementor metadata, ServiceRegistry serviceRegistry) {
		final Map settings = serviceRegistry.getService( ConfigurationService.class ).getSettings();
		settings.put( AvailableSettings.HBM2DDL_DATABASE_ACTION, Action.UPDATE );

		SchemaManagementToolCoordinator.process(
				Helper.buildDatabaseModel( metadata ),
				serviceRegistry,
				action -> {}
		);
	}

	@AllowSysOut
	public static void toStdout(MetadataImplementor metadata) {
		toWriter( metadata, new OutputStreamWriter( System.out ) );
	}

	@SuppressWarnings("unchecked")
	public static void toWriter(MetadataImplementor metadata, Writer writer) {
		final ServiceRegistry serviceRegistry = metadata.getMetadataBuildingOptions().getServiceRegistry();
		final Map settings = serviceRegistry.getService( ConfigurationService.class ).getSettings();
		settings.put( AvailableSettings.HBM2DDL_SCRIPTS_ACTION, Action.UPDATE );
		// atm we reuse the CREATE scripts setting
		settings.put( AvailableSettings.HBM2DDL_SCRIPTS_CREATE_TARGET, writer );

		SchemaManagementToolCoordinator.process(
				Helper.buildDatabaseModel( metadata ),
				serviceRegistry,
				action -> {}
		);
	}
}