Switching A DataSource Based On A User

MyCustomDataSourceService.Java

package com.example.jr.datasource;

import java.sql.Connection;

import java.sql.SQLException;

import java.util.Map;

import net.sf.jasperreports.engine.JRParameter;

import net.sf.jasperreports.engine.JRResultSetDataSource;

import org.springframework.security.context.SecurityContextHolder;

import com.jaspersoft.jasperserver.api.common.domain.ExecutionContext;

import com.jaspersoft.jasperserver.api.common.service.BeanForInterfaceImplementationFactory;

import com.jaspersoft.jasperserver.api.engine.jasperreports.service.impl.JdbcDataSourceService;

import com.jaspersoft.jasperserver.api.engine.jasperreports.service.impl.JdbcReportDataSourceServiceFactory;

import com.jaspersoft.jasperserver.api.metadata.common.service.RepositoryService;

import com.jaspersoft.jasperserver.api.metadata.jasperreports.domain.ReportDataSource;

import com.jaspersoft.jasperserver.api.metadata.jasperreports.service.ReportDataSourceService;

import com.jaspersoft.jasperserver.api.metadata.user.domain.impl.client.MetadataUserDetails;

import com.jaspersoft.jasperserver.war.common.JasperServerUtil;

public class MyCustomDataSourceService implements ReportDataSourceService {

// private properties

private JdbcDataSourceService connection;

private Connection conn;

// Bean properties

private Map propertyMap;

private RepositoryService repository;

private BeanForInterfaceImplementationFactory dataSourceServiceFactory;

/**

* Default constructor

*/

public MyCustomDataSourceService() {

super();

}

public Map getPropertyMap() {

return propertyMap;

}

public void setPropertyMap(Map propertyMap) {

this.propertyMap = propertyMap;

}

public RepositoryService getRepository() {

return repository;

}

public void setRepository(RepositoryService repository) {

this.repository = repository;

}

public BeanForInterfaceImplementationFactory getDataSourceServiceFactory() {

return dataSourceServiceFactory;

}

public void setDataSourceServiceFactory(

BeanForInterfaceImplementationFactory dataSourceServiceFactory) {

this.dataSourceServiceFactory = dataSourceServiceFactory;

}

@Override

public void closeConnection() {

if (conn != null) {

try {

conn.close();

} catch (SQLException e) {

}

}

if (connection != null) {

connection.closeConnection();

}

}

@Override

public void setReportParameterValues(Map parameterValues) {

MetadataUserDetails userDetails = (MetadataUserDetails) SecurityContextHolder

.getContext().getAuthentication().getPrincipal();

String userName = userDetails.getUsername();

String dataSourceURI = "/datasources/" + userName;

connection = getRepositoryDatasource(dataSourceURI);

try {

conn = connection.getDataSource().getConnection();

parameterValues.put(JRParameter.REPORT_CONNECTION, conn);

} catch (SQLException sqle) {

//

}

}

private JdbcDataSourceService getRepositoryDatasource(String repositoryURI) {

try {

ExecutionContext context = JasperServerUtil.getExecutionContext();

ReportDataSource datasource = (ReportDataSource) repository

.getResource(context, repositoryURI);

JdbcReportDataSourceServiceFactory factory = (JdbcReportDataSourceServiceFactory) dataSourceServiceFactory

.getBean(datasource.getClass());

JdbcDataSourceService DSservice = (JdbcDataSourceService) factory

.createService(datasource);

return DSservice;

} catch (Exception ex) {

return null;

}

}

}

applicationContext-MyCustomDS.xml

<?xml version="1.0" encoding="UTF-8"?>

<beans xmlns="http://www.springframework.org/schema/beans"

xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"

xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans-3.1.xsd">

<!-- define a custom data source -->

<bean id="CustomDataSource" class="com.jaspersoft.jasperserver.api.engine.jasperreports.util.CustomDataSourceDefinition">

<!-- this prop is always the same; it registers the custom ds -->

<property name="factory" ref="customDataSourceServiceFactory"/>

<!-- name used in message catalog and elsewhere -->

<property name="name" value="CustomDataSource"/>

<!-- class name of implementation -->

<property name="serviceClassName" value="com.example.jr.datasource.MyCustomDataSourceService"/>

<!-- definitions of properties you can edit -->

<property name="propertyDefinitions">

<list>

<map>

<entry key="name" value="repository"/>

<entry key="hidden" value="true"/>

<entry key="default" value-ref="repositoryService"/>

</map>

<map>

<entry key="name" value="dataSourceServiceFactory"/>

<entry key="hidden" value="true"/>

<entry key="default" value-ref="dataSourceServiceFactories"/>

</map>

</list>

</property>

</bean>

</beans>