public class TableTypeTestSuite {
    protected static String TABLE =
            "TABLE EMP_SCHEMA.EMPLOYEE (\n" +
                    "\tID\tVARCHAR\n" +
                    "\tNAME\tVARCHAR2 (NOT NULL)\n" +
                    "\tDEPT\tNUMERIC\n" +
                    "\tPRIMARY KEY (ID)\n" +
                ")";
 
    @Test
    public void testTableType() {
        // setup TableType
        TableType table = new TableType("EMPLOYEE");
        table.setSchema("EMP_SCHEMA");
 
        FieldType col = new FieldType("ID");
        col.setDataType(new VarCharType());
        col.setPk();
        table.addCompositeType(col);
        col = new FieldType("NAME");
        col.setDataType(new VarChar2Type());
        col.setNotNull();
        table.addCompositeType(col);
        col = new FieldType("DEPT");
        col.setDataType(new NumericType());
        table.addCompositeType(col);
 
        // visit
        TableTypeVisitor visitor = new TableTypeVisitor();
        table.accept(visitor);
        assertEquals("TableTypeVisitor test failed:\n", visitor.toString(), TABLE);
    }
}

public class TableTypeVisitor extends BaseDatabaseTypeVisitor {
    public String tableName;
    public String schema;
    public List<String> columnData = new ArrayList<String>();
    public List<String> pkColumns = new ArrayList<String>();
 
    public void beginVisit(TableType databaseType) {
        tableName = databaseType.getTableName();
        schema = databaseType.getSchema();
    }
 
    public void beginVisit(FieldType databaseType) {
        if (databaseType.notNull()) {
            columnData.add(databaseType.getFieldName() + "\t" + databaseType.getTypeName() + " (NOT NULL)");
        } else {
            columnData.add(databaseType.getFieldName() + "\t" + databaseType.getTypeName());
        }
        if (databaseType.pk()) {
            pkColumns.add(databaseType.getFieldName());
        }
    }
 
    public String toString() {
        StringBuilder sb = new StringBuilder("TABLE ");
        if (schema != null) {
            sb.append(schema);
            sb.append(".");
        }
        sb.append(tableName);
        sb.append(" (\n");
        for (String col : columnData) {
            sb.append("\t");
            sb.append(col);
            sb.append("\n");
        }
        if (pkColumns.size() > 0) {
            sb.append("\t");
            sb.append("PRIMARY KEY (");
            for (int i = 0; i < pkColumns.size();) {
                sb.append(pkColumns.get(i));
                if (++i < pkColumns.size()) {
                    sb.append(',');
                }
            }
            sb.append(")\n");
        }
        sb.append(")");
        return sb.toString();
    }
}