/*
* 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.http.HTTPClientConfig;
import com.basho.riak.client.raw.http.HTTPClusterConfig;
import java.util.List;
import org.qi4j.api.configuration.Configuration;
import org.qi4j.api.injection.scope.This;
/**
* Riak Http implementation of MapEntityStore.
*/
public class RiakHttpMapEntityStoreMixin
extends AbstractRiakMapEntityStore
{
private static final String DEFAULT_URL = "http://127.0.0.1:8098/riak";
@This
private Configuration<RiakHttpEntityStoreConfiguration> configuration;
@Override
public void activateService()
throws Exception
{
configuration.refresh();
RiakHttpEntityStoreConfiguration config = configuration.get();
int maxConnections = config.maxConnections().get() == null ? DEFAULT_MAX_CONNECTIONS : config.maxConnections().get();
int timeoutMillis = config.timeout().get();
List<String> urls = config.urls().get();
if( urls.isEmpty() )
{
urls.add( DEFAULT_URL );
}
bucketKey = config.bucket().get() == null ? DEFAULT_BUCKET_KEY : config.bucket().get();
HTTPClusterConfig httpClusterConfig = new HTTPClusterConfig( maxConnections );
for( String url : urls )
{
HTTPClientConfig clientConfig = new HTTPClientConfig.Builder().withTimeout( timeoutMillis ).withUrl( url ).build();
httpClusterConfig.addClient( clientConfig );
}
riakClient = RiakFactory.newClient( httpClusterConfig );
if( !riakClient.listBuckets().contains( bucketKey ) )
{
riakClient.createBucket( bucketKey ).execute();
}
}
}