genepi.db.GroupedListHandler.java Source code

Introduction

Here is the source code for genepi.db.GroupedListHandler.java

Source

/*******************************************************************************
 * Copyright (C) 2009-2016 Lukas Forer and Sebastian Schnherr
 *  
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by 
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *  
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *  
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see <http://www.gnu.org/licenses/>.
 ******************************************************************************/

package genepi.db;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Vector;

import org.apache.commons.dbutils.ResultSetHandler;

public class GroupedListHandler implements ResultSetHandler<Object> {

    private IRowMapMapper mapper;

    public GroupedListHandler(IRowMapMapper rowMapper) {
        this.mapper = rowMapper;
    }

    public Map<Object, List<Object>> toBeanList(ResultSet rs) throws SQLException {
        Map<Object, List<Object>> result = new HashMap<Object, List<Object>>();

        int row = 0;
        while (rs.next()) {
            Object key = mapper.getRowKey(rs, row);
            Object value = mapper.getRowValue(rs, row);
            if (value != null) {

                List<Object> list = result.get(key);
                if (list == null) {
                    list = new Vector<Object>();
                    result.put(key, list);
                }
                list.add(value);
            }
            row++;
        }

        return result;
    }

    @Override
    public Object handle(ResultSet rs) throws SQLException {
        return toBeanList(rs);
    }

}