com.fastchar.database.FastResultSet Maven / Gradle / Ivy

package com.fastchar.database;

import com.fastchar.core.FastEntity;

import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

class FastResultSet {

    private ResultSet resultSet;

    public FastResultSet(ResultSet resultSet) {
        this.resultSet = resultSet;
    }

    public FastEntity getFirstResult() {
        if (resultSet != null) {
            try {
                ResultSetMetaData resultSetMetaData = resultSet.getMetaData();
                final String tableName = resultSetMetaData.getTableName(1);
                if (resultSet.first()) {
                    FastRecord fastEntity = new FastRecord();
                    fastEntity.setTableName(tableName);
                    int columnCount = resultSetMetaData.getColumnCount();
                    for (int i = 1; i <= columnCount; i++) {
                        try {
                            String key = resultSetMetaData.getColumnLabel(i);
                            Object value = resultSet.getObject(key);
                            fastEntity.put(key, value);
                        } catch (Exception ignored) {
                        }
                    }
                    return fastEntity;
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return null;
    }


    public FastEntity getLastResult() {
        if (resultSet != null) {
            try {
                ResultSetMetaData resultSetMetaData = resultSet.getMetaData();
                final String tableName = resultSetMetaData.getTableName(1);
                if (resultSet.last()) {
                    FastRecord fastEntity = new FastRecord();
                    fastEntity.setTableName(tableName);
                    int columnCount = resultSetMetaData.getColumnCount();
                    for (int i = 1; i <= columnCount; i++) {
                        try {
                            String key = resultSetMetaData.getColumnLabel(i);
                            Object value = resultSet.getObject(key);
                            fastEntity.put(key, value);
                        } catch (Exception ignored) {
                        }
                    }
                    return fastEntity;
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return null;
    }


    public List> getListResult() {
        if (resultSet != null) {
            try {
                List> list = new ArrayList<>();
                ResultSetMetaData resultSetMetaData = resultSet.getMetaData();
                final String tableName = resultSetMetaData.getTableName(1);
                int columnCount = resultSetMetaData.getColumnCount();
                while (resultSet.next()) {
                    FastRecord fastEntity = new FastRecord();
                    fastEntity.setTableName(tableName);
                    for (int i = 1; i <= columnCount; i++) {
                        try {
                            String key = resultSetMetaData.getColumnLabel(i);
                            Object value = resultSet.getObject(key);
                            fastEntity.put(key, value);
                        } catch (Exception ignored) {
                        }
                    }
                    list.add(fastEntity);
                }
                return list;
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return null;
    }


    public Map getMap() {
        if (resultSet != null) {
            try {
                Map map = new HashMap<>();
                ResultSetMetaData resultSetMetaData = resultSet.getMetaData();
                if (resultSet.last()) {
                    int columnCount = resultSetMetaData.getColumnCount();
                    for (int i = 1; i <= columnCount; i++) {
                        try {
                            String key = resultSetMetaData.getColumnLabel(i);
                            Object value = resultSet.getObject(key);
                            map.put(key, value);
                        } catch (Exception ignored) {
                        }
                    }
                }
                return map;
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return null;
    }




}