/**
* Licensed to Apereo under one or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information regarding copyright ownership. Apereo
* licenses this file to you under the Apache License, Version 2.0 (the "License"); you may not use
* this file except in compliance with the License. You may obtain a copy of the License at the
* following location:
*
* <p>http://www.apache.org/licenses/LICENSE-2.0
*
* <p>Unless required by applicable law or agreed to in writing, software distributed under the
* License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
* express or implied. See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apereo.portal.io.xml.crn;
import com.google.common.base.Function;
import java.sql.ResultSet;
import java.sql.SQLException;
import javax.sql.DataSource;
import org.apereo.portal.io.xml.IPortalData;
import org.apereo.portal.io.xml.IPortalDataType;
import org.apereo.portal.io.xml.SimpleStringPortalData;
import org.springframework.beans.factory.annotation.Required;
import org.springframework.jdbc.core.JdbcOperations;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;
/**
* Simple Function for use in {@link CernunnosDataExporter#setPortalDataRetriever(Function)} that
* uses a parameter-less sql query to build a list of IPortalData objects
*
*/
public class SqlPortalDataFunction
implements Function<IPortalDataType, Iterable<? extends IPortalData>> {
private JdbcOperations jdbcOperations;
private String query;
private String idColumn;
private String nameColumn;
private String descColumn;
/** @param query Query to run */
@Required
public void setQuery(String query) {
this.query = query;
}
/** @param id Column column name that contains the id, if null the first column is used */
public void setIdColumn(String idColumn) {
this.idColumn = idColumn;
}
/** @param nameColumn Column column name that contains the name, if null no name is set */
public void setNameColumn(String nameColumn) {
this.nameColumn = nameColumn;
}
/**
* @param nameColumn Column column name that contains the description, if null no description is
* set
*/
public void setDescColumn(String descColumn) {
this.descColumn = descColumn;
}
@Required
public void setDataSource(DataSource dataSource) {
final JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);
jdbcTemplate.afterPropertiesSet();
this.jdbcOperations = jdbcTemplate;
}
@Override
public Iterable<? extends IPortalData> apply(IPortalDataType input) {
return this.jdbcOperations.query(
this.query,
new RowMapper<IPortalData>() {
@Override
public IPortalData mapRow(ResultSet rs, int rowNum) throws SQLException {
final String id;
if (idColumn != null) {
id = rs.getString(idColumn);
} else {
id = rs.getString(1);
}
final String name = getValue(rs, nameColumn);
final String description = getValue(rs, descColumn);
return new SimpleStringPortalData(id, name, description);
}
protected String getValue(ResultSet rs, String column) throws SQLException {
if (column != null) {
return rs.getString(column);
}
return null;
}
});
}
}