package com.fr.data;

import com.fr.base.FRContext;

import com.fr.file.DatasourceManager;

import com.fr.stable.ParameterProvider;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.ResultSet;

import java.sql.ResultSetMetaData;

import java.sql.Statement;

import java.util.ArrayList;

//import java.util.*;

/**

 * 带参数的程序数据集Demo

 *

 * @author fanruan

 */

 @SuppressWarnings("unchecked")

public class DataDemo extends AbstractTableData {

    /**

     * 列名数组，保存程序数据集所有列名

     */

    private String[] columnNames;

    /**

     * 定义程序数据集的列数量

     */

    private int columnNum = 10;

    /**

     * 保存查询表的实际列数量

     */

    private int colNum = 0;

    /**

     * 保存查询得到列值

     */

    private ArrayList valueList = null;

    /**

     * 构造函数，定义表结构，该表有10个数据列，列名为column#0，column#1，。。。。。。column#9

     */

    public DataDemo() {

        columnNames = new String[columnNum];

        for (int i = 0; i < columnNum; i++) {

            columnNames[i] = "column#" + String.valueOf(i);

        }

    }

    /**

     * 实现其他四个方法

     *

     * @return columnNum

     */

    @Override

    public int getColumnCount() {

        return columnNum;

    }

    @Override

    public String getColumnName(int columnIndex) {

        return columnNames[columnIndex];

    }

    @Override

    public int getRowCount() {

        init();

        return valueList.size();

    }

    @Override

    public Object getValueAt(int rowIndex, int columnIndex) {

        init();

        if (columnIndex >= colNum) {

            return null;

        }

        return ((Object[]) valueList.get(rowIndex))[columnIndex];

    }

    /**

     * 准备数据

     */

    private void init() {

        // 确保只被执行一次

        if (valueList != null) {

            return;

        }

        // 保存得到的数据库表名

       \\String tableName = ((ParameterProvider) (parameters.get().toArray())[0]).getValue().toString();

        // 构造SQL语句,并打印出来

        String sql = "select * from unit";

        FRContext.getLogger().info("Query SQL of ParamTableDataDemo: \n" + sql);

        // 保存得到的结果集

        valueList = new ArrayList();

        // 下面开始建立数据库连接，按照刚才的SQL语句进行查询

        com.fr.data.impl.Connection conn = DatasourceManager.getInstance().getConnection("JDBC");

        try {

            Connection con = conn.createConnection();

            Statement stmt = con.createStatement();

            ResultSet rs = stmt.executeQuery(sql);

            // 获得记录的详细信息，然后获得总列数

            ResultSetMetaData rsmd = rs.getMetaData();

            colNum = rsmd.getColumnCount();

            // 用对象保存数据

            Object[] objArray = null;

            while (rs.next()) {

                objArray = new Object[colNum];

                for (int i = 0; i < colNum; i++) {

                    objArray[i] = rs.getObject(i + 1);

                }

                // 在valueList中加入这一行数据

                valueList.add(objArray);

            }

            // 释放数据库资源

            rs.close();

            stmt.close();

            con.close();

            // 打印一共取到的数据行数量

        } catch (Exception e) {

            e.printStackTrace();

        }

    }

    /**

     * 获取数据库连接 driverName和 url 可以换成您需要的

     *

     * @return Connection

     */

    public Connection getConnection() {

        String driverName = "oracle.jdbc.OracleDriver";

        String url = "jdbc:oracle:thin:@172.224.404.146:1521:orcl";

        String username = "sms2";

        String password = "w12333";

        Connection con;

        try {

            Class.forName(driverName);

            con = DriverManager.getConnection(url, username, password);

        } catch (Exception e) {

            e.printStackTrace();

            return null;

        }

        return con;

    }

    /**

     * 释放一些资源，因为可能会有重复调用，所以需释放valueList，将上次查询的结果释放掉

     *

     * @throws Exception e

     */

    @Override

    public void release() throws Exception {

        super.release();

        this.valueList = null;

    }

}