package fucoin.supervisor;

import javax.swing.table.DefaultTableModel;
import java.util.Vector;

public class AmountTableModel extends DefaultTableModel {

    public AmountTableModel() {
        super(new Object[]{"Address", "Name", "Amount"}, 0);
    }

    public void clear() {
        while (getRowCount() > 0) {
            removeRow(0);
        }
    }

    public void updateTable(String address, String name, int amount) {

        Vector<Object> rows = this.getDataVector();
        for (int i = 0; i < rows.size(); i++) {
           if (rows.get(i) instanceof Vector){
               Vector<Object> row = (Vector<Object>) rows.get(i);
               if(row.get(0).equals(address)){
                   setValueAt(amount, i, 2);
                   return;
               }
           }
        }

        this.addRow(new Object[]{address, name, amount});
    }
}