diff --git a/src/main/java/netneutrality/db/models/MRowModel.java b/src/main/java/netneutrality/db/models/MRowModel.java index bec0ddf..4469bc1 100644 --- a/src/main/java/netneutrality/db/models/MRowModel.java +++ b/src/main/java/netneutrality/db/models/MRowModel.java @@ -30,6 +30,14 @@ public class MRowModel { return row; } + public Object getItem(int index) { + try { + return row.get(index); + } catch (Exception e) { + return null; + } + } + @Override public String toString() { return row.toString(); diff --git a/src/main/java/netneutrality/db/models/MTableModel.java b/src/main/java/netneutrality/db/models/MTableModel.java index b1f38be..520d10a 100644 --- a/src/main/java/netneutrality/db/models/MTableModel.java +++ b/src/main/java/netneutrality/db/models/MTableModel.java @@ -14,7 +14,7 @@ import java.util.List; public class MTableModel { private List column; private MRowsModel rows; - + private String tableName; public MTableModel() { @@ -22,6 +22,12 @@ public class MTableModel { this.rows = new MRowsModel(); } + public MTableModel(String tableName) { + this.column = new LinkedList<>(); + this.rows = new MRowsModel(); + this.tableName = tableName; + } + public MTableModel(String... columns) { this.column = new LinkedList<>(); this.rows = new MRowsModel(); @@ -44,7 +50,17 @@ public class MTableModel { } public List getRow(int index) { - return rows.getRowModel(index).getRow(); + if (getNbRows() != 0) + return rows.getRowModel(index).getRow(); + + return null; + } + + public MRowModel getRowModel(int rowIndex) { + if (getNbRows() != 0) + return rows.getRowModel(rowIndex); + + return new MRowModel(); } public List getColumns() {