/*
* Copyright 2013 LinkedIn, Inc
*
* 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 voldemort.client;
import java.io.IOException;
import java.util.Arrays;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Properties;
import junit.framework.TestCase;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.junit.runners.Parameterized.Parameters;
import voldemort.ServerTestUtils;
import voldemort.TestUtils;
import voldemort.VoldemortException;
import voldemort.client.protocol.admin.AdminClient;
import voldemort.cluster.Cluster;
import voldemort.cluster.Node;
import voldemort.server.AbstractSocketService;
import voldemort.server.StoreRepository;
import voldemort.server.protocol.SocketRequestHandlerFactory;
import voldemort.store.RandomlyFailingDelegatingStore;
import voldemort.store.StorageEngine;
import voldemort.store.StoreDefinition;
import voldemort.store.UnreachableStoreException;
import voldemort.store.memory.InMemoryStorageEngine;
import voldemort.utils.ByteArray;
import voldemort.utils.Pair;
import voldemort.versioning.Versioned;
import com.google.common.collect.AbstractIterator;
/**
* Test to check streaming behavior under failure.
*
*
*/
@RunWith(Parameterized.class)
public class AdminServiceFailureTest extends TestCase {
private static int TEST_KEYS = 10000;
private static double FAIL_PROBABILITY = 0.60;
private AdminClient adminClient;
private Cluster cluster;
private AbstractSocketService adminServer;
StorageEngine<ByteArray, byte[], byte[]> failingStorageEngine;
private enum StreamOperations {
FETCH_ENTRIES,
FETCH_KEYS,
DELETE_PARTITIONS,
UPDATE_ENTRIES,
TRUNCATE_ENTRIES
}
private final boolean useNio;
public AdminServiceFailureTest(boolean useNio) {
this.useNio = useNio;
}
@Parameters
public static Collection<Object[]> configs() {
return Arrays.asList(new Object[][] { { true }, { false } });
}
@Override
@Before
public void setUp() throws IOException {
cluster = ServerTestUtils.getLocalCluster(2, new int[][] { { 0, 1, 2, 3 }, { 4, 5, 6, 7 } });
List<StoreDefinition> storeDefs = ServerTestUtils.getStoreDefs(1);
failingStorageEngine = new RandomlyFailingDelegatingStore<ByteArray, byte[], byte[]>(new InMemoryStorageEngine<ByteArray, byte[], byte[]>(storeDefs.get(0)
.getName()));
adminServer = getAdminServer(cluster.getNodeById(0),
cluster,
storeDefs,
failingStorageEngine);
adminClient = ServerTestUtils.getAdminClient(cluster);
adminServer.start();
}
private AbstractSocketService getAdminServer(Node node,
Cluster cluster,
List<StoreDefinition> storeDefs,
StorageEngine<ByteArray, byte[], byte[]> storageEngine)
throws IOException {
StoreRepository storeRepository = new StoreRepository();
storeRepository.addStorageEngine(storageEngine);
storeRepository.addLocalStore(storageEngine);
SocketRequestHandlerFactory requestHandlerFactory = new SocketRequestHandlerFactory(null,
storeRepository,
ServerTestUtils.createMetadataStore(cluster,
storeDefs),
ServerTestUtils.createServerConfig(useNio,
0,
TestUtils.createTempDir()
.getAbsolutePath(),
null,
null,
new Properties()),
null,
null,
null);
return ServerTestUtils.getSocketService(useNio,
requestHandlerFactory,
node.getAdminPort(),
2,
2,
10000);
}
@Override
@After
public void tearDown() throws IOException {
try {
adminServer.stop();
adminClient.close();
} catch(Exception e) {
// ignore
}
}
protected AdminClient getAdminClient() {
return adminClient;
}
@Test
public void testWithStartFailure() {
// put some entries in store
for(StreamOperations operation: StreamOperations.values()) {
adminServer.stop();
try {
doOperation(operation, 0, failingStorageEngine.getName(), Arrays.asList(0, 1));
fail();
} catch(UnreachableStoreException e) {
// ignore
}
}
}
// client side should get exceptions from servers
@Test
public void testFailures() {
for(StreamOperations operation: StreamOperations.values()) {
try {
doOperation(operation, 0, failingStorageEngine.getName(), Arrays.asList(0, 1));
fail("Unit test should fail for " + operation);
} catch(Exception e) {
// ignore
}
}
}
private void doOperation(StreamOperations e,
int nodeId,
String storeName,
List<Integer> partitionList) {
switch(e) {
case DELETE_PARTITIONS:
putAlltoStore();
getAdminClient().storeMntOps.deletePartitions(nodeId,
storeName,
partitionList,
null);
return;
case FETCH_ENTRIES:
putAlltoStore();
consumeIterator(getAdminClient().bulkFetchOps.fetchEntries(nodeId,
storeName,
partitionList,
null,
false));
return;
case FETCH_KEYS:
putAlltoStore();
consumeIterator(getAdminClient().bulkFetchOps.fetchKeys(nodeId,
storeName,
partitionList,
null,
false));
return;
case UPDATE_ENTRIES:
getAdminClient().streamingOps.updateEntries(nodeId,
storeName,
getRandomlyFailingIterator(ServerTestUtils.createRandomKeyValuePairs(TEST_KEYS)),
null);
return;
case TRUNCATE_ENTRIES:
putAlltoStore();
getAdminClient().storeMntOps.truncate(nodeId, storeName);
return;
default:
throw new RuntimeException("Unknown operation");
}
}
private <K> void consumeIterator(Iterator<K> iterator) {
while(iterator.hasNext())
iterator.next();
}
private void putAlltoStore() {
for(Entry<ByteArray, byte[]> entry: ServerTestUtils.createRandomKeyValuePairs(TEST_KEYS)
.entrySet()) {
try {
failingStorageEngine.put(entry.getKey(),
new Versioned<byte[]>(entry.getValue()),
null);
} catch(Exception e) {
// ignore
}
}
}
private Iterator<Pair<ByteArray, Versioned<byte[]>>> getRandomlyFailingIterator(final Map<ByteArray, byte[]> entryMap) {
return new AbstractIterator<Pair<ByteArray, Versioned<byte[]>>>() {
private final Iterator<Entry<ByteArray, byte[]>> innerIterator = entryMap.entrySet()
.iterator();
@Override
protected Pair<ByteArray, Versioned<byte[]>> computeNext() {
if(Math.random() > FAIL_PROBABILITY) {
throw new VoldemortException("Failing Iterator.");
}
if(!innerIterator.hasNext())
return endOfData();
Entry<ByteArray, byte[]> entry = innerIterator.next();
return new Pair<ByteArray, Versioned<byte[]>>(entry.getKey(),
new Versioned<byte[]>(entry.getValue()));
}
};
}
}