/*
* Copyright 2012 Paul Merlin.
*
* 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.entitystore.redis;
import java.io.IOException;
import java.io.Reader;
import java.io.StringReader;
import java.io.StringWriter;
import java.io.Writer;
import java.util.Set;
import org.qi4j.api.configuration.Configuration;
import org.qi4j.api.entity.EntityDescriptor;
import org.qi4j.api.entity.EntityReference;
import org.qi4j.api.injection.scope.This;
import org.qi4j.api.service.ServiceActivation;
import org.qi4j.io.Input;
import org.qi4j.io.Output;
import org.qi4j.io.Receiver;
import org.qi4j.io.Sender;
import org.qi4j.spi.entitystore.EntityAlreadyExistsException;
import org.qi4j.spi.entitystore.EntityNotFoundException;
import org.qi4j.spi.entitystore.EntityStoreException;
import org.qi4j.spi.entitystore.helpers.MapEntityStore;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;
import redis.clients.jedis.Protocol;
/**
* Redis implementation of MapEntityStore.
*/
public class RedisMapEntityStoreMixin
implements ServiceActivation, RedisAccessors, MapEntityStore
{
private static final String DEFAULT_HOST = "127.0.0.1";
private static final String NIL = "nil";
@This
private Configuration<RedisEntityStoreConfiguration> configuration;
private JedisPool pool;
@Override
public void activateService()
throws Exception
{
configuration.refresh();
RedisEntityStoreConfiguration config = configuration.get();
String host = config.host().get() == null ? DEFAULT_HOST : config.host().get();
int port = config.port().get() == null ? Protocol.DEFAULT_PORT : config.port().get();
int timeout = config.timeout().get() == null ? Protocol.DEFAULT_TIMEOUT : config.timeout().get();
String password = config.password().get();
int database = config.database().get() == null ? Protocol.DEFAULT_DATABASE : config.database().get();
pool = new JedisPool( new JedisPoolConfig(), host, port, timeout, password, database );
}
@Override
public void passivateService()
throws Exception
{
pool.destroy();
pool = null;
}
@Override
public JedisPool jedisPool()
{
return pool;
}
@Override
public Reader get( EntityReference entityReference )
throws EntityStoreException
{
Jedis jedis = pool.getResource();
try
{
String jsonState = jedis.get( entityReference.identity() );
if( notFound( jsonState ) )
{
throw new EntityNotFoundException( entityReference );
}
return new StringReader( jsonState );
}
finally
{
pool.returnResource( jedis );
}
}
@Override
public void applyChanges( MapChanges changes )
throws IOException
{
final Jedis jedis = pool.getResource();
try
{
changes.visitMap( new MapChanger()
{
@Override
public Writer newEntity( final EntityReference ref, EntityDescriptor entityDescriptor )
throws IOException
{
return new StringWriter( 1000 )
{
@Override
public void close()
throws IOException
{
super.close();
String statusCode = jedis.set( ref.identity(), toString(), "NX" );
if( !"OK".equals( statusCode ) )
{
throw new EntityAlreadyExistsException( ref );
}
}
};
}
@Override
public Writer updateEntity( final EntityReference ref, EntityDescriptor entityDescriptor )
throws IOException
{
return new StringWriter( 1000 )
{
@Override
public void close()
throws IOException
{
super.close();
String statusCode = jedis.set( ref.identity(), toString(), "XX" );
if( !"OK".equals( statusCode ) )
{
throw new EntityNotFoundException( ref );
}
}
};
}
@Override
public void removeEntity( EntityReference ref, EntityDescriptor entityDescriptor )
throws EntityNotFoundException
{
String jsonState = jedis.get( ref.identity() );
if( notFound( jsonState ) )
{
throw new EntityNotFoundException( ref );
}
jedis.del( ref.identity() );
}
} );
}
finally
{
pool.returnResource( jedis );
}
}
@Override
public Input<Reader, IOException> entityStates()
{
return new Input<Reader, IOException>()
{
@Override
public <ReceiverThrowableType extends Throwable> void transferTo( Output<? super Reader, ReceiverThrowableType> output )
throws IOException, ReceiverThrowableType
{
output.receiveFrom( new Sender<Reader, IOException>()
{
@Override
public <ReceiverThrowableType extends Throwable> void sendTo( Receiver<? super Reader, ReceiverThrowableType> receiver )
throws ReceiverThrowableType, IOException
{
Jedis jedis = pool.getResource();
try
{
Set<String> keys = jedis.keys( "*" );
for( String key : keys )
{
String jsonState = jedis.get( key );
receiver.receive( new StringReader( jsonState ) );
}
}
finally
{
pool.returnResource( jedis );
}
}
} );
}
};
}
private static boolean notFound( String jsonState )
{
return jsonState == null || NIL.equals( jsonState );
}
}