/**
*
* Copyright 2009-2011 Rickard Öberg AB
*
* Licensed 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
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* 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.qi4j.library.rest.server.restlet;
import java.util.ArrayList;
import java.util.List;
import java.util.ResourceBundle;
import org.qi4j.api.injection.scope.Service;
import org.qi4j.api.injection.scope.Structure;
import org.qi4j.api.mixin.InitializationException;
import org.qi4j.api.service.ServiceReference;
import org.qi4j.api.structure.Module;
import org.qi4j.library.rest.server.spi.ResponseWriter;
import org.restlet.Response;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Delegates to a list of potential writers. Register writers on startup.
*/
public class ResponseWriterDelegator
implements ResponseWriter
{
List<ResponseWriter> responseWriters = new ArrayList<ResponseWriter>();
@Structure
Module module;
public void init( @Service Iterable<ServiceReference<ResponseWriter>> resultWriters )
throws InitializationException
{
Logger logger = LoggerFactory.getLogger( getClass() );
// Add custom writers first
for( ServiceReference<ResponseWriter> resultWriter : resultWriters )
{
if( !resultWriter.identity().equals( "responsewriterdelegator" ) )
{
logger.info( "Registered result writer:" + resultWriter.identity() );
registerResultWriter( resultWriter.get() );
}
}
// Add defaults
ResourceBundle defaultResultWriters = ResourceBundle.getBundle( "org.qi4j.library.rest.server.rest-server" );
String resultWriterClasses = defaultResultWriters.getString( "responsewriters" );
logger.info( "Using response writers:" + resultWriterClasses );
for( String className : resultWriterClasses.split( "," ) )
{
try
{
Class writerClass = module.classLoader().loadClass( className.trim() );
ResponseWriter writer = (ResponseWriter) module.newObject( writerClass );
registerResultWriter( writer );
}
catch( ClassNotFoundException e )
{
logger.warn( "Could not register response writer " + className, e );
}
}
}
public void registerResultWriter( ResponseWriter writer )
{
responseWriters.add( writer );
}
@Override
public boolean writeResponse( Object result, Response response )
{
for( ResponseWriter responseWriter : responseWriters )
{
if( responseWriter.writeResponse( result, response ) )
{
return true;
}
}
return false;
}
}