|
| 1 | +package org.uniqueck.asciidoctorj.liquibase; |
| 2 | + |
| 3 | +import lombok.AccessLevel; |
| 4 | +import lombok.Getter; |
| 5 | +import org.jdom2.Document; |
| 6 | +import org.jdom2.Element; |
| 7 | +import org.jdom2.JDOMException; |
| 8 | +import org.jdom2.Namespace; |
| 9 | +import org.jdom2.input.SAXBuilder; |
| 10 | +import org.uniqueck.asciidoctorj.liquibase.model.Column; |
| 11 | +import org.uniqueck.asciidoctorj.liquibase.model.Table; |
| 12 | + |
| 13 | +import java.io.File; |
| 14 | +import java.io.IOException; |
| 15 | +import java.util.HashMap; |
| 16 | +import java.util.List; |
| 17 | +import java.util.Map; |
| 18 | + |
| 19 | +@Getter(AccessLevel.PROTECTED) |
| 20 | +public class LiquibaseChangesetParser { |
| 21 | + |
| 22 | + |
| 23 | + final private SAXBuilder saxBuilder; |
| 24 | + |
| 25 | + public LiquibaseChangesetParser() { |
| 26 | + this.saxBuilder = new SAXBuilder(); |
| 27 | + } |
| 28 | + |
| 29 | + public Map<String, Table> parse(File masterFile) { |
| 30 | + Map<String, Table> parsedTables = new HashMap<>(); |
| 31 | + try { |
| 32 | + Document rootDocument = getSaxBuilder().build(masterFile); |
| 33 | + Namespace namespace = getNamespace4Liquibase(rootDocument.getRootElement()); |
| 34 | + |
| 35 | + List<Element> changeSets = rootDocument.getRootElement().getChildren("changeSet", namespace); |
| 36 | + for (Element changeSet : changeSets) { |
| 37 | + Element documentTable = changeSet.getChild("createTable", namespace); |
| 38 | + if (documentTable != null) { |
| 39 | + Table table = extractTable(namespace, documentTable); |
| 40 | + parsedTables.put(table.getName(), table); |
| 41 | + } |
| 42 | + } |
| 43 | + } catch (JDOMException e) { |
| 44 | + e.printStackTrace(); |
| 45 | + } catch (IOException e) { |
| 46 | + e.printStackTrace(); |
| 47 | + } |
| 48 | + return parsedTables; |
| 49 | + } |
| 50 | + |
| 51 | + |
| 52 | + protected Table extractTable(Namespace namespace, Element tableElement) { |
| 53 | + String tableName = tableElement.getAttributeValue("tableName"); |
| 54 | + Table table = new Table(tableName); |
| 55 | + List<Element> columns = tableElement.getChildren("column", namespace); |
| 56 | + for (Element columnElement : columns) { |
| 57 | + extractColumn(table, namespace, columnElement); |
| 58 | + } |
| 59 | + return table; |
| 60 | + } |
| 61 | + |
| 62 | + protected Column extractColumn(Table table, Namespace namespace, Element columnElement) { |
| 63 | + String name = columnElement.getAttributeValue("name"); |
| 64 | + String type = columnElement.getAttributeValue("type"); |
| 65 | + return table.add(name, type); |
| 66 | + } |
| 67 | + |
| 68 | + protected Namespace getNamespace4Liquibase(Element rootElement) { |
| 69 | + return rootElement.getNamespace(); |
| 70 | + } |
| 71 | + |
| 72 | +} |
0 commit comments