jpa/eclipselink.jpa.wdf.test/src/org/eclipse/persistence/testing/framework/wdf/customizer/AdjustArrayTypeCustomizer.java

/*******************************************************************************
 * Copyright (c) 2005, 2009 SAP. All rights reserved.
 * This program and the accompanying materials are made available under the 
 * terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0 
 * which accompanies this distribution. 
 * The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
 * and the Eclipse Distribution License is available at 
 * http://www.eclipse.org/org/documents/edl-v10.php.
 *
 * Contributors:
 *     SAP - initial API and implementation
 ******************************************************************************/

package org.eclipse.persistence.testing.framework.wdf.customizer;

import org.eclipse.persistence.config.DescriptorCustomizer;
import org.eclipse.persistence.descriptors.ClassDescriptor;
import org.eclipse.persistence.internal.helper.DatabaseField;
import org.eclipse.persistence.mappings.DatabaseMapping;
import org.eclipse.persistence.mappings.ManyToManyMapping;
import org.eclipse.persistence.mappings.OneToOneMapping;
import org.eclipse.persistence.platform.database.DatabasePlatform;
import org.eclipse.persistence.platform.database.DerbyPlatform;
import org.eclipse.persistence.platform.database.MaxDBPlatform;
import org.eclipse.persistence.platform.database.MySQLPlatform;
import org.eclipse.persistence.platform.database.OraclePlatform;

public class AdjustArrayTypeCustomizer implements DescriptorCustomizer {

	private static final String VARCHAR = "VARCHAR";
	private static final String BINARY = "BINARY";
	private static Class<? extends DatabasePlatform> databasePlatformClass;

	public static void setDatabasePlatformClass(
			Class<? extends DatabasePlatform> databasePlatformClass) {
		AdjustArrayTypeCustomizer.databasePlatformClass = databasePlatformClass;
	}

	@Override
	public void customize(ClassDescriptor descriptor) throws Exception {

		Iterable<DatabaseMapping> mappings = descriptor.getMappings();
		for (DatabaseMapping mapping : mappings) {
			if (mapping.isDirectToFieldMapping()) {
				adjustDatabaseField(mapping.getField());
			}
			if (mapping.isOneToOneMapping()) {
				OneToOneMapping oneToOneMapping = (OneToOneMapping) mapping;
				for (DatabaseField field : oneToOneMapping
						.getForeignKeyFields()) {
					adjustDatabaseField(field);
				}
			} else if (mapping.isManyToManyMapping()) {
				ManyToManyMapping manyToManyMapping = (ManyToManyMapping) mapping;
				for (DatabaseField field : manyToManyMapping
						.getSourceRelationKeyFields()) {
					adjustDatabaseField(field);
				}
				for (DatabaseField field : manyToManyMapping
						.getTargetRelationKeyFields()) {
					adjustDatabaseField(field);
				}
			}
		}
	}

	private void adjustDatabaseField(DatabaseField field) {
		String columnDefintion = field.getColumnDefinition();
		if (columnDefintion != null) {
			if (columnDefintion.startsWith(BINARY)) {
				adjustBinaryType(field, columnDefintion);
			} else if (columnDefintion.startsWith(VARCHAR)) {
				adjustVarcharType(field, columnDefintion);
			}

		}
	}

	private void adjustBinaryType(DatabaseField field, String columnDefintion) {
		final String newDefinition;
		if (OraclePlatform.class.isAssignableFrom(databasePlatformClass)) {
			newDefinition = columnDefintion.replace(BINARY, "RAW");
		} else if (MySQLPlatform.class.isAssignableFrom(databasePlatformClass)) {
			newDefinition = columnDefintion.replace(BINARY, "binary");
		} else if (MaxDBPlatform.class.isAssignableFrom(databasePlatformClass)) {
			newDefinition = columnDefintion.replace(BINARY, "CHAR") + " BYTE";
		} else if (DerbyPlatform.class.isAssignableFrom(databasePlatformClass)) {
            newDefinition = columnDefintion.replace(BINARY, "CHAR") + " FOR BIT DATA";
		} else {
		    return;
		}
		field.setColumnDefinition(newDefinition);
	}

	private void adjustVarcharType(DatabaseField field, String columnDefintion) {
		final String newDefinition;
		if (OraclePlatform.class.isAssignableFrom(databasePlatformClass)) {
			newDefinition = columnDefintion.replace(VARCHAR, "VARCHAR2");
		} else if (MaxDBPlatform.class.isAssignableFrom(databasePlatformClass)) {
			newDefinition = columnDefintion + " UNICODE";
		} else {
			return;
		}
		field.setColumnDefinition(newDefinition);
	}

}