net.sf.aguacate.context.spi.sql.impl.SentenceSqlSelectMultipleRow Maven / Gradle / Ivy

Go to download
package net.sf.aguacate.context.spi.sql.impl;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

public class SentenceSqlSelectMultipleRow extends AbstractSentenceSqlSelect {

	private static final Logger LOGGER = LogManager.getLogger(SentenceSqlSelectMultipleRow.class);

	public SentenceSqlSelectMultipleRow(String name, String table, List methods, List required,
			List optional, String outputName) {
		super(name, table, methods, required, optional, outputName);
	}

	@Override
	Object extract(String[] optional, ResultSet resultSet) throws SQLException {
		List> list;
		if (resultSet.next()) {
			Map first = toMap(optional, resultSet);
			if (resultSet.next()) {
				LOGGER.debug("multiple results");
				list = new ArrayList<>();
				list.add(first);
				do {
					list.add(toMap(optional, resultSet));
				} while (resultSet.next());
			} else {
				LOGGER.debug("single result");
				list = Collections.singletonList(first);
			}
		} else {
			LOGGER.info("no results");
			list = null;
		}
		return list;
	}

	private Map toMap(String[] optional, ResultSet resultSet) throws SQLException {
		Map data = new HashMap<>();
		for (String parameter : optional) {
			data.put(parameter, resultSet.getObject(parameter));
		}
		return data;
	}

}