/*
* Copyright 2008-2009 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.store.invalidmetadata;
import java.util.List;
import java.util.Map;
import voldemort.VoldemortException;
import voldemort.cluster.Node;
import voldemort.routing.RoutingStrategy;
import voldemort.store.DelegatingStore;
import voldemort.store.InvalidMetadataException;
import voldemort.store.Store;
import voldemort.store.StoreUtils;
import voldemort.store.metadata.MetadataStore;
import voldemort.utils.ByteArray;
import voldemort.versioning.Version;
import voldemort.versioning.Versioned;
/**
* An InvalidMetadataCheckingStore store is a store wrapper that delegates to an
* inner store, and throws {@link InvalidMetadataException} if a client requests
* a partition which is not or should not be available at this node.
*
*
*/
public class InvalidMetadataCheckingStore extends DelegatingStore<ByteArray, byte[], byte[]> {
private final int nodeId;
private final MetadataStore metadata;
/**
* Create a store which delegates its operations to its inner store and
* throws {@link InvalidMetadataException} if the partition for key
* requested should not lie with this node.
*
* @param nodeId The id of the destination node
* @param innerStore The store which we delegate write operations to if
* everything is good
* @param metadata The metadata store
*/
public InvalidMetadataCheckingStore(int nodeId,
Store<ByteArray, byte[], byte[]> innerStore,
MetadataStore metadata) {
super(innerStore);
this.metadata = metadata;
this.nodeId = nodeId;
}
@Override
public boolean delete(ByteArray key, Version version) throws VoldemortException {
StoreUtils.assertValidKey(key);
StoreUtils.assertValidNode(metadata, nodeId);
StoreUtils.assertValidMetadata(key,
metadata.getRoutingStrategy(getName()),
metadata.getCluster().getNodeById(nodeId));
return getInnerStore().delete(key, version);
}
@Override
public Map<ByteArray, List<Versioned<byte[]>>> getAll(Iterable<ByteArray> keys,
Map<ByteArray, byte[]> transforms)
throws VoldemortException {
StoreUtils.assertValidKeys(keys);
StoreUtils.assertValidNode(metadata, nodeId);
RoutingStrategy routingStrategy = metadata.getRoutingStrategy(getName());
Node node = metadata.getCluster().getNodeById(nodeId);
for(ByteArray key: keys)
StoreUtils.assertValidMetadata(key, routingStrategy, node);
return getInnerStore().getAll(keys, transforms);
}
@Override
public void put(ByteArray key, Versioned<byte[]> value, byte[] transforms)
throws VoldemortException {
StoreUtils.assertValidKey(key);
StoreUtils.assertValidNode(metadata, nodeId);
StoreUtils.assertValidMetadata(key,
metadata.getRoutingStrategy(getName()),
metadata.getCluster().getNodeById(nodeId));
getInnerStore().put(key, value, transforms);
}
@Override
public List<Versioned<byte[]>> get(ByteArray key, byte[] transforms) throws VoldemortException {
StoreUtils.assertValidKey(key);
StoreUtils.assertValidNode(metadata, nodeId);
StoreUtils.assertValidMetadata(key,
metadata.getRoutingStrategy(getName()),
metadata.getCluster().getNodeById(nodeId));
return getInnerStore().get(key, transforms);
}
}