it.firegloves.mempoi.dao.impl.DBMempoiDAO Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mempoi Show documentation

A library to simplify export from database to Excel files using Apache POI

There is a newer version: 1.10.1

package it.firegloves.mempoi.dao.impl;


import it.firegloves.mempoi.domain.MempoiColumn;
import it.firegloves.mempoi.exception.MempoiRuntimeException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.slf4j.Marker;
import org.slf4j.MarkerFactory;

import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class DBMempoiDAO implements it.firegloves.mempoi.dao.MempoiDAO {

    private static final Logger logger = LoggerFactory.getLogger(DBMempoiDAO.class);
    Marker marker = MarkerFactory.getMarker("test");

    private static DBMempoiDAO instance = new DBMempoiDAO();

    private DBMempoiDAO() {
    }

    public static DBMempoiDAO getInstance() {
        return instance;
    }


    @Override
    public ResultSet executeExportQuery(PreparedStatement prepStmt) {
        try {
            logger.debug(marker, "EXECUTING EXPORT QUERY: {}", prepStmt);
            return prepStmt.executeQuery();
        } catch (SQLException e) {
            throw new MempoiRuntimeException(e);
        }
    }



    @Override
    public List readMetadata(ResultSet rs) {

        if (null == rs) {
            throw new MempoiRuntimeException("NULL ResultSet!");
        }

        try {
            ResultSetMetaData rsmd = rs.getMetaData();
            List columnList = new ArrayList<>();

            int columnsNumber = rsmd.getColumnCount();
            for (int i = 1; i <= columnsNumber; i++) {
                columnList.add(new MempoiColumn(rsmd.getColumnType(i), rsmd.getColumnLabel(i)));
            }

            return columnList;

        } catch (SQLException e) {
            throw new MempoiRuntimeException(e);
        }
    }

}