package cn.com.lzt.message.data.dto;

import net.sf.json.JSONObject;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class MessageSimpleRowViewer extends MessageViewer {
    private List<String> columns;
    private List<Object[]> data;

    @Override
    public String viewName() {
        return "simpleRow";
    }

    @Override
    public String html() {
        JSONObject obj = new JSONObject();
        obj.put("columns",columns);
        obj.put("data",data);
        return obj.toString();
    }

    public void addColumn(String titleString){
        if(columns==null){
            columns = new ArrayList<>();
        }
        columns.add(titleString);
    }

    public void addRow(Object[] row){
        if(data==null){
            data = new ArrayList<>();
        }
        data.add(row);
    }

    public List<Object[]> getData() {
        return data;
    }

    public void setData(List<Object[]> data) {
        this.data = data;
    }

    public List<String> getColumns() {
        return columns;
    }

    public void setColumns(List<String> columns) {
        this.columns = columns;
    }
}