net.ontopia.persistence.rdbms.DDLWriter.java Source code

Introduction

Here is the source code for net.ontopia.persistence.rdbms.DDLWriter.java

Source

/*
 * #!
 * Ontopia Engine
 * #-
 * Copyright (C) 2001 - 2013 The Ontopia Project
 * #-
 * 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
 * 
 *      http://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 net.ontopia.persistence.rdbms;

import java.io.FileWriter;
import java.io.Writer;
import net.ontopia.utils.CmdlineOptions;
import net.ontopia.utils.CmdlineUtils;
import org.apache.commons.lang3.StringUtils;

/** 
 * INTERNAL: Command line tool for producing DDL files for creating
 * and dropping database schemas.
 */

public class DDLWriter {

    public static void main(String[] argv) throws Exception {

        // Initialize logging
        CmdlineUtils.initializeLogging();

        // Initialize command line option parser and listeners
        CmdlineOptions options = new CmdlineOptions("DDLWriter", argv);

        // Register logging options
        CmdlineUtils.registerLoggingOptions(options);

        // Parse command line options
        try {
            options.parse();
        } catch (CmdlineOptions.OptionsException e) {
            System.err.println("Error: " + e.getMessage());
            System.exit(1);
        }

        // Get command line arguments
        String[] args = options.getArguments();
        if (args.length < 3) {
            System.err.println("Error: need exactly two files as arguments.");
            usage();
            System.exit(1);
        }

        String schema = args[0];
        String dbtype = args[1];
        String[] platforms = StringUtils.split(args[2], ",");
        String createfile = args[3];
        String dropfile = args[4];

        Project project = DatabaseProjectReader.loadProject(schema);

        GenericSQLProducer producer;
        switch (dbtype) {
        case "postgresql":
            producer = new PostgreSQLProducer(project, platforms);
            break;
        case "oracle":
            producer = new OracleSQLProducer(project, platforms);
            break;
        case "sqlserver":
            producer = new SQLServerSQLProducer(project, platforms);
            break;
        case "mysql":
            producer = new MySqlSQLProducer(project, platforms);
            break;
        case "db2":
            producer = new DB2SQLProducer(project, platforms);
            break;
        case "firebird":
            producer = new FirebirdSQLProducer(project, platforms);
            break;
        default:
            producer = new GenericSQLProducer(project, platforms);
            break;
        }

        // Generate create file
        Writer cwriter = new FileWriter(createfile);
        producer.writeCreate(cwriter);
        cwriter.close();

        // Generate create file
        Writer dwriter = new FileWriter(dropfile);
        producer.writeDrop(dwriter);
        dwriter.close();
    }

    protected static void usage() {
        System.out.println("java DDLWriter [options] <schema> <dbtype> <platforms> <createfile> <dropfile>");
        System.out.println("");
        System.out.println("  Creates DDL files for creating and dropping database schemas.");
        System.out.println("");
        System.out.println("  Options:");
        CmdlineUtils.printLoggingOptionsUsage(System.out);
        System.out.println("");
        System.out.println("  <schema>: schema description file");
        System.out.println("  <dbtype>: the database type");
        System.out.println("  <platforms>: the database platforms (comma separated, no whitespace)");
        System.out.println("  <createfile>: filename of create file");
        System.out.println("  <dropfile>:   filename of drop file");
    }

}