/*
* 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.riak;
import com.basho.riak.client.RiakFactory;
import com.basho.riak.client.raw.pbc.PBClientConfig;
import com.basho.riak.client.raw.pbc.PBClusterConfig;
import java.util.List;
import org.qi4j.api.configuration.Configuration;
import org.qi4j.api.injection.scope.This;
/**
* Riak Protobuf implementation of MapEntityStore.
*/
public class RiakProtobufMapEntityStoreMixin
extends AbstractRiakMapEntityStore
{
private static final int DEFAULT_CONNECTION_TIMEOUT = 1000;
private static final int DEFAULT_IDLE_CONNECTION_TTL = 1000;
private static final int DEFAULT_SOCKET_BUFFER_SIZE_KB = 16;
private static final String DEFAULT_HOST = "127.0.0.1";
private static final int DEFAULT_PORT = 8087;
@This
private Configuration<RiakProtobufEntityStoreConfiguration> configuration;
@Override
public void activateService()
throws Exception
{
configuration.refresh();
RiakProtobufEntityStoreConfiguration config = configuration.get();
int maxConnections = config.maxConnections().get() == null ? DEFAULT_MAX_CONNECTIONS : config.maxConnections().get();
int connectionTimeout = config.connectionTimeout().get() == null ? DEFAULT_CONNECTION_TIMEOUT : config.connectionTimeout().get();
int idleConnectionTTL = config.idleConnectionTTL().get() == null ? DEFAULT_IDLE_CONNECTION_TTL : config.idleConnectionTTL().get();
int maxPoolSize = config.maxPoolSize().get();
int initialPoolSize = config.initialPoolSize().get();
int socketBufferSize = config.socketBufferSizeKb().get() == null ? DEFAULT_SOCKET_BUFFER_SIZE_KB : config.socketBufferSizeKb().get();
List<String> hosts = config.hosts().get();
if( hosts.isEmpty() )
{
hosts.add( DEFAULT_HOST );
}
bucketKey = config.bucket().get() == null ? DEFAULT_BUCKET_KEY : config.bucket().get();
PBClusterConfig pbClusterConfig = new PBClusterConfig( maxConnections );
for( String host : hosts )
{
String[] splitted = host.split( ":" );
int port = DEFAULT_PORT;
if( splitted.length > 1 )
{
host = splitted[0];
port = Integer.valueOf( splitted[1] );
}
PBClientConfig clientConfig = new PBClientConfig.Builder().withConnectionTimeoutMillis( connectionTimeout ).
withIdleConnectionTTLMillis( idleConnectionTTL ).
withPoolSize( maxPoolSize ).
withInitialPoolSize( initialPoolSize ).
withSocketBufferSizeKb( socketBufferSize ).
withHost( host ).withPort( port ).build();
pbClusterConfig.addClient( clientConfig );
}
riakClient = RiakFactory.newClient( pbClusterConfig );
if( !riakClient.listBuckets().contains( bucketKey ) )
{
riakClient.createBucket( bucketKey ).execute();
}
}
}