org.dbtools.schema.schemafile.DatabaseSchema Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dbtools-gen Show documentation
DBTools ORM Class Generator.
The newest version!
package org.dbtools.schema.schemafile;

import org.simpleframework.xml.Attribute;
import org.simpleframework.xml.ElementList;
import org.simpleframework.xml.Root;
import org.simpleframework.xml.Serializer;
import org.simpleframework.xml.core.Persister;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

@Root
public class DatabaseSchema {
    @Attribute(required = false)
    private String schemaLocation;

    @ElementList(entry = "database", inline = true)
    private List databases = new ArrayList<>();

    public List getDatabases() {
        return databases;
    }

    public void setDatabases(List databases) {
        this.databases = databases;
    }

    public SchemaDatabase getDatabase(String name) {
        for (SchemaDatabase database : databases) {
            if (database.getName(false).equals(name)) {
                return database;
            }
        }

        return null;
    }

    public void validate() {
        for (SchemaDatabase database : databases) {
            database.validate();
        }
    }

    public static DatabaseSchema readXMLSchema(String path)  {
        DatabaseSchema schema = null;

        // read schema xml file
        try {
            Serializer serializer = new Persister();
            File source = new File(path);
            schema = serializer.read(DatabaseSchema.class, source);
        } catch (Exception e) {
            e.printStackTrace();
            throw new IllegalStateException("Failure reading schema file [" + new File(path).getAbsolutePath() + "] Error: [" + e.getMessage() + "]");
        }

        // validate
        if (schema != null) {
            schema.validate();
        }

        return schema;
    }

    public static void main(String[] args) {
//        javaToXml();
        xmlToJava();
    }

    public static void javaToXml() {
        try {
            DatabaseSchema schema = new DatabaseSchema();

            List databases = new ArrayList();

            SchemaDatabase db1 = new SchemaDatabase("main");
            SchemaTable t1 = new SchemaTable("table1");

            SchemaTableField f1 = new SchemaTableField("ID", SchemaFieldType.BIGINT);
            f1.setIncrement(true);

            t1.getFields().add(f1);


            List db1Tables = new ArrayList<>();
            db1Tables.add(t1);
            db1Tables.add(new SchemaTable("table2"));
            db1.setTables(db1Tables);
            databases.add(db1);
            schema.setDatabases(databases);


            Serializer serializer = new Persister();
            File result = new File("/home/jcampbell/Desktop/example.xml");

            serializer.write(schema, result);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void xmlToJava() {
        try {
            Serializer serializer = new Persister();
            File source = new File("/home/jcampbell/src/ldschurch/android/ldstools/LDSToolsAndroid/src/main/database/schema.xml");
            DatabaseSchema schema = serializer.read(DatabaseSchema.class, source);

            if (schema != null) {
                System.out.println("SUCCESS");
            }

        } catch (Exception e) {
            e.printStackTrace();
        }

    }
}