Java tutorial
/** * Copyright (C) 2012-2015 52North Initiative for Geospatial Open Source * Software GmbH * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License version 2 as published * by the Free Software Foundation. * * If the program is linked with libraries which are licensed under one of * the following licenses, the combination of the program with the linked * library is not considered a "derivative work" of the program: * * - Apache License, version 2.0 * - Apache Software License, version 1.0 * - GNU Lesser General Public License, version 3 * - Mozilla Public License, versions 1.0, 1.1 and 2.0 * - Common Development and Distribution License (CDDL), version 1.0 * * Therefore the distribution of the program linked with libraries licensed * under the aforementioned licenses, is permitted by the copyright holders * if the distribution is compliant with both the GNU General Public * License version 2 and the aforementioned licenses. * * This program 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 General * Public License for more details. */ package org.n52.sos; import java.io.BufferedReader; import java.io.File; import java.io.IOException; import java.io.InputStreamReader; import java.util.List; import java.util.Properties; import org.hibernate.cfg.Configuration; import org.hibernate.dialect.Dialect; import org.hibernate.spatial.dialect.h2geodb.GeoDBDialect; import org.hibernate.spatial.dialect.mysql.MySQLSpatial5InnoDBDialect; import org.hibernate.spatial.dialect.oracle.OracleSpatial10gDialect; import org.hibernate.spatial.dialect.postgis.PostgisDialect; import org.hibernate.spatial.dialect.sqlserver.SqlServer2008SpatialDialect; import com.google.common.collect.Lists; /** * Class to generate the create and drop scripts for different databases. * Currently supported spatial databases to create scripts - PostgreSQL/PostGIS * - Oracle - H2/GeoDB * * @author Carsten Hollmann <c.hollmann@52north.org> * @since 4.0.0 * */ public class SQLScriptGenerator { private SQLScriptGenerator() { } private Dialect getDialect(int selection) throws Exception { switch (selection) { case 1: return new PostgisDialect(); case 2: try { return new OracleSpatial10gDialect(); } catch (ExceptionInInitializerError eiie) { printToScreen("The Oracle JDBC driver is missing!"); printToScreen( "To execute the SQL script generator for Oracle you have to uncomment the dependency in the pom.xml."); printToScreen("If the Oracle JDBC driver is not installed in your local Maven repository, "); printToScreen("follow the first steps describes here: "); printToScreen( "https://wiki.52north.org/bin/view/SensorWeb/SensorObservationServiceIVDocumentation#Oracle_support."); throw new MissingDriverException(); } case 3: return new GeoDBDialect(); case 4: return new MySQLSpatial5InnoDBDialect(); case 5: return new SqlServer2008SpatialDialect(); default: throw new Exception("The entered value is invalid!"); } } private void setDirectoriesForModelSelection(int selection, boolean oldConcept, Configuration configuration) throws Exception { switch (selection) { case 1: configuration.addDirectory(new File(SQLScriptGenerator.class.getResource("/mapping/core").toURI())); if (oldConcept) { configuration.addDirectory( new File(SQLScriptGenerator.class.getResource("/mapping/old/observation").toURI())); } else { configuration.addDirectory( new File(SQLScriptGenerator.class.getResource("/mapping/series/observation").toURI())); } break; case 2: configuration.addDirectory(new File(SQLScriptGenerator.class.getResource("/mapping/core").toURI())); if (oldConcept) { configuration.addDirectory( new File(SQLScriptGenerator.class.getResource("/mapping/old/observation").toURI())); } else { configuration.addDirectory( new File(SQLScriptGenerator.class.getResource("/mapping/series/observation").toURI())); } configuration .addDirectory(new File(SQLScriptGenerator.class.getResource("/mapping/transactional").toURI())); break; case 3: configuration.addDirectory(new File(SQLScriptGenerator.class.getResource("/mapping/core").toURI())); if (oldConcept) { configuration.addDirectory( new File(SQLScriptGenerator.class.getResource("/mapping/old/observation").toURI())); configuration.addDirectory(new File( SQLScriptGenerator.class.getResource("/mapping/old/spatialFilteringProfile").toURI())); } else { configuration.addDirectory( new File(SQLScriptGenerator.class.getResource("/mapping/series/observation").toURI())); configuration.addDirectory(new File( SQLScriptGenerator.class.getResource("/mapping/series/spatialFilteringProfile").toURI())); } break; case 4: configuration.addDirectory(new File(SQLScriptGenerator.class.getResource("/mapping/core").toURI())); configuration .addDirectory(new File(SQLScriptGenerator.class.getResource("/mapping/transactional").toURI())); if (oldConcept) { configuration.addDirectory( new File(SQLScriptGenerator.class.getResource("/mapping/old/observation").toURI())); configuration.addDirectory(new File( SQLScriptGenerator.class.getResource("/mapping/old/spatialFilteringProfile").toURI())); } else { configuration.addDirectory( new File(SQLScriptGenerator.class.getResource("/mapping/series/observation").toURI())); configuration.addDirectory(new File( SQLScriptGenerator.class.getResource("/mapping/series/spatialFilteringProfile").toURI())); } break; case 5: configuration.addDirectory(new File(SQLScriptGenerator.class.getResource("/mapping/core").toURI())); configuration .addDirectory(new File(SQLScriptGenerator.class.getResource("/mapping/transactional").toURI())); configuration.addDirectory(new File(SQLScriptGenerator.class.getResource("/mapping/i18n").toURI())); if (oldConcept) { configuration.addDirectory( new File(SQLScriptGenerator.class.getResource("/mapping/old/observation").toURI())); configuration.addDirectory(new File( SQLScriptGenerator.class.getResource("/mapping/old/spatialFilteringProfile").toURI())); } else { configuration.addDirectory( new File(SQLScriptGenerator.class.getResource("/mapping/series/observation").toURI())); configuration.addDirectory(new File( SQLScriptGenerator.class.getResource("/mapping/series/spatialFilteringProfile").toURI())); } break; default: throw new Exception("The entered value is invalid!"); } } private int getDialectSelection() throws IOException { printToScreen("This SQL script generator supports:"); printToScreen("1 PostgreSQL/PostGIS"); printToScreen("2 Oracle"); printToScreen("3 H2/GeoDB"); printToScreen("4 MySQL"); printToScreen("5 SQL Server"); printToScreen(""); printToScreen("Enter your selection: "); BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); String selection = null; selection = br.readLine(); return Integer.parseInt(selection); } private int getModelSelection() throws IOException { printToScreen("Which database model should be created:"); printToScreen("1 Core"); printToScreen("2 Transcational"); printToScreen("3 Spatial Filtering Profile"); printToScreen("4 Core/Transactional/Spatial Filtering Profile"); printToScreen("5 Core/Transactional/Spatial Filtering Profile/Multi Language"); printToScreen(""); printToScreen("Enter your selection: "); BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); String selection = null; selection = br.readLine(); return Integer.parseInt(selection); } private int getConceptSelection() throws IOException { printToScreen("Which observation concept should be created:"); printToScreen("1 old"); printToScreen("2 series"); printToScreen(""); printToScreen("Enter your selection: "); BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); String selection = null; selection = br.readLine(); return Integer.parseInt(selection); } private boolean isOldConcept(int selection) throws Exception { switch (selection) { case 1: return true; case 2: return false; default: throw new Exception("The entered value is invalid!"); } } private String getSchema() throws IOException { printToScreen("For which schema should the database model be created?"); printToScreen("No schema is also valid!"); printToScreen(""); printToScreen("Enter your selection: "); BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); String selection = null; selection = br.readLine(); return selection; } public static void printToScreen(String lineToPrint) { System.out.println(lineToPrint); } private List<String> checkSchema(Dialect dia, String[] create) { String hexStringToCheck = new StringBuilder("FK") .append(Integer.toHexString("observationHasOffering".hashCode()).toUpperCase()).toString(); boolean duplicate = false; List<String> checkedSchema = Lists.newLinkedList(); for (String string : create) { if (string.contains(hexStringToCheck)) { if (!duplicate) { checkedSchema.add(string); duplicate = true; } } else { checkedSchema.add(string); } } return checkedSchema; } public static void main(String[] args) { try { SQLScriptGenerator sqlScriptGenerator = new SQLScriptGenerator(); Configuration configuration = new Configuration().configure("/sos-hibernate.cfg.xml"); int dialectSelection = sqlScriptGenerator.getDialectSelection(); Dialect dia = sqlScriptGenerator.getDialect(dialectSelection); int modelSelection = sqlScriptGenerator.getModelSelection(); boolean oldConcept = sqlScriptGenerator.isOldConcept(sqlScriptGenerator.getConceptSelection()); String schema = sqlScriptGenerator.getSchema(); if (schema != null && !schema.isEmpty()) { Properties p = new Properties(); p.put("hibernate.default_schema", schema); configuration.addProperties(p); } sqlScriptGenerator.setDirectoriesForModelSelection(modelSelection, oldConcept, configuration); // create script String[] create = configuration.generateSchemaCreationScript(dia); List<String> checkedSchema = sqlScriptGenerator.checkSchema(dia, create); printToScreen("Scripts are created for: " + dia.toString()); printToScreen(""); printToScreen("#######################################"); printToScreen("## Create-Script ##"); printToScreen("#######################################"); printToScreen(""); for (String t : checkedSchema) { printToScreen(t + ";"); } // drop script String[] drop = configuration.generateDropSchemaScript(dia); List<String> checkedDrop = sqlScriptGenerator.checkSchema(dia, drop); printToScreen(""); printToScreen("#######################################"); printToScreen("## Drop-Script ##"); printToScreen("#######################################"); printToScreen(""); for (String t : checkedDrop) { printToScreen(t + ";"); } printToScreen(""); printToScreen("#######################################"); } catch (IOException ioe) { printToScreen("ERROR: IO error trying to read your input!"); System.exit(1); } catch (MissingDriverException mde) { System.exit(1); } catch (Exception e) { printToScreen("ERROR: " + e.getMessage()); System.exit(1); } } private class MissingDriverException extends Exception { private static final long serialVersionUID = -5681526838468633998L; public MissingDriverException() { super(); } } }