HDFS-12145. Ozone: OzoneFileSystem: Ozone & KSM should support / delimited key names. Contributed by Mukul Kumar Singh.
(cherry picked from commit fe0a957e92d506330ba005abba18f665eb5ac1cc)
This commit is contained in:
parent
8cf0c864c4
commit
a4b3160efb
@ -28,6 +28,7 @@
|
||||
|
||||
import com.google.protobuf.ByteString;
|
||||
|
||||
import org.apache.commons.codec.digest.DigestUtils;
|
||||
import org.apache.hadoop.hdfs.ozone.protocol.proto.ContainerProtos.ChunkInfo;
|
||||
import org.apache.hadoop.hdfs.ozone.protocol.proto.ContainerProtos.KeyData;
|
||||
import org.apache.hadoop.ozone.protocol.proto.OzoneProtos.KeyValue;
|
||||
@ -209,7 +210,8 @@ private synchronized void writeChunkToContainer() throws IOException {
|
||||
ChunkInfo chunk = ChunkInfo
|
||||
.newBuilder()
|
||||
.setChunkName(
|
||||
key + "_stream_" + streamId + "_chunk_" + ++chunkIndex)
|
||||
DigestUtils.md5Hex(key) + "_stream_"
|
||||
+ streamId + "_chunk_" + ++chunkIndex)
|
||||
.setOffset(0)
|
||||
.setLen(data.size())
|
||||
.build();
|
||||
|
@ -40,7 +40,7 @@
|
||||
* This interface defines operations permitted on a key.
|
||||
*/
|
||||
|
||||
@Path("/{volume}/{bucket}/{keys}")
|
||||
@Path("/{volume}/{bucket}/{keys:.*}")
|
||||
public interface Keys {
|
||||
|
||||
/**
|
||||
|
@ -18,6 +18,8 @@
|
||||
package org.apache.hadoop.ozone.web.client;
|
||||
|
||||
import org.apache.commons.codec.digest.DigestUtils;
|
||||
import org.apache.commons.lang.RandomStringUtils;
|
||||
import org.apache.commons.lang.math.RandomUtils;
|
||||
import org.apache.hadoop.fs.StorageType;
|
||||
import org.apache.hadoop.hdfs.server.datanode.DataNode;
|
||||
import org.apache.hadoop.ozone.MiniOzoneCluster;
|
||||
@ -28,6 +30,7 @@
|
||||
import org.apache.hadoop.ozone.web.exceptions.OzoneException;
|
||||
import org.apache.hadoop.ozone.web.utils.OzoneUtils;
|
||||
import org.apache.hadoop.test.GenericTestUtils;
|
||||
import org.apache.hadoop.util.StringUtils;
|
||||
import org.apache.log4j.Level;
|
||||
import org.apache.log4j.Logger;
|
||||
import org.junit.AfterClass;
|
||||
@ -122,17 +125,41 @@ static File createRandomDataFile(String dir, String fileName, long size) {
|
||||
return tmpFile;
|
||||
}
|
||||
|
||||
/**
|
||||
* This function generates multi part key which are delimited by a certain
|
||||
* delimiter. Different parts of key are random string of random length
|
||||
* between 0 - 4. Number of parts of the keys are between 0 and 5.
|
||||
*
|
||||
* @param delimiter delimiter used to delimit parts of string
|
||||
* @return Key composed of multiple parts delimited by "/"
|
||||
*/
|
||||
static String getMultiPartKey(String delimiter) {
|
||||
int numParts = RandomUtils.nextInt(5) + 1;
|
||||
String[] nameParts = new String[numParts];
|
||||
for (int i = 0; i < numParts; i++) {
|
||||
int stringLength = numParts == 1 ? 5 : RandomUtils.nextInt(5);
|
||||
nameParts[i] = RandomStringUtils.randomAlphanumeric(stringLength);
|
||||
}
|
||||
return StringUtils.join(delimiter, nameParts);
|
||||
}
|
||||
|
||||
static class PutHelper {
|
||||
private final OzoneRestClient client;
|
||||
private final String dir;
|
||||
private final String keyName;
|
||||
|
||||
OzoneVolume vol;
|
||||
OzoneBucket bucket;
|
||||
File file;
|
||||
|
||||
PutHelper(OzoneRestClient client, String dir) {
|
||||
this(client, dir, OzoneUtils.getRequestID().toLowerCase());
|
||||
}
|
||||
|
||||
PutHelper(OzoneRestClient client, String dir, String key) {
|
||||
this.client = client;
|
||||
this.dir = dir;
|
||||
this.keyName = key;
|
||||
}
|
||||
|
||||
public OzoneVolume getVol() {
|
||||
@ -163,8 +190,8 @@ private String putKey() throws
|
||||
String bucketName = OzoneUtils.getRequestID().toLowerCase();
|
||||
bucket = vol.createBucket(bucketName, acls, StorageType.DEFAULT);
|
||||
|
||||
String keyName = OzoneUtils.getRequestID().toLowerCase();
|
||||
file = createRandomDataFile(dir, keyName, 1024);
|
||||
String fileName = OzoneUtils.getRequestID().toLowerCase();
|
||||
file = createRandomDataFile(dir, fileName, 1024);
|
||||
|
||||
bucket.putKey(keyName, file);
|
||||
return keyName;
|
||||
@ -174,7 +201,12 @@ private String putKey() throws
|
||||
|
||||
@Test
|
||||
public void testPutKey() throws OzoneException {
|
||||
// Test non-delimited keys
|
||||
runTestPutKey(new PutHelper(ozoneRestClient, path));
|
||||
// Test key delimited by a random delimiter
|
||||
String delimiter = RandomStringUtils.randomAscii(1);
|
||||
runTestPutKey(new PutHelper(ozoneRestClient, path,
|
||||
getMultiPartKey(delimiter)));
|
||||
}
|
||||
|
||||
static void runTestPutKey(PutHelper helper) throws OzoneException {
|
||||
@ -229,6 +261,10 @@ public void testPutAndGetKeyWithDnRestart()
|
||||
throws OzoneException, IOException, URISyntaxException {
|
||||
runTestPutAndGetKeyWithDnRestart(
|
||||
new PutHelper(ozoneRestClient, path), ozoneCluster);
|
||||
String delimiter = RandomStringUtils.randomAscii(1);
|
||||
runTestPutAndGetKeyWithDnRestart(
|
||||
new PutHelper(ozoneRestClient, path,
|
||||
getMultiPartKey(delimiter)), ozoneCluster);
|
||||
}
|
||||
|
||||
static void runTestPutAndGetKeyWithDnRestart(
|
||||
@ -263,6 +299,9 @@ static void runTestPutAndGetKeyWithDnRestart(
|
||||
@Test
|
||||
public void testPutAndGetKey() throws OzoneException, IOException {
|
||||
runTestPutAndGetKey(new PutHelper(ozoneRestClient, path));
|
||||
String delimiter = RandomStringUtils.randomAscii(1);
|
||||
runTestPutAndGetKey(new PutHelper(ozoneRestClient, path,
|
||||
getMultiPartKey(delimiter)));
|
||||
}
|
||||
|
||||
static void runTestPutAndGetKey(PutHelper helper)
|
||||
@ -326,6 +365,9 @@ static void runTestPutAndGetKey(PutHelper helper)
|
||||
@Test
|
||||
public void testPutAndDeleteKey() throws OzoneException, IOException {
|
||||
runTestPutAndDeleteKey(new PutHelper(ozoneRestClient, path));
|
||||
String delimiter = RandomStringUtils.randomAscii(1);
|
||||
runTestPutAndDeleteKey(new PutHelper(ozoneRestClient, path,
|
||||
getMultiPartKey(delimiter)));
|
||||
}
|
||||
|
||||
static void runTestPutAndDeleteKey(PutHelper helper)
|
||||
@ -347,6 +389,9 @@ static void runTestPutAndDeleteKey(PutHelper helper)
|
||||
@Test
|
||||
public void testPutAndListKey() throws OzoneException, IOException {
|
||||
runTestPutAndListKey(new PutHelper(ozoneRestClient, path));
|
||||
String delimiter = RandomStringUtils.randomAscii(1);
|
||||
runTestPutAndListKey(new PutHelper(ozoneRestClient, path,
|
||||
getMultiPartKey(delimiter)));
|
||||
}
|
||||
|
||||
static void runTestPutAndListKey(PutHelper helper)
|
||||
@ -378,9 +423,9 @@ static void runTestPutAndListKey(PutHelper helper)
|
||||
Assert.assertEquals(keyList2.size(), 1);
|
||||
|
||||
// test startKey parameter of list keys
|
||||
keyList1 = helper.getBucket().listKeys("100", "list-key4", null);
|
||||
keyList1 = helper.getBucket().listKeys("100", "list-key4", "list-key");
|
||||
keyList2 = client.listKeys(helper.getVol().getVolumeName(),
|
||||
helper.getBucket().getBucketName(), "100", "list-key4", null);
|
||||
helper.getBucket().getBucketName(), "100", "list-key4", "list-key");
|
||||
Assert.assertEquals(keyList1.size(), 5);
|
||||
Assert.assertEquals(keyList2.size(), 5);
|
||||
|
||||
@ -416,6 +461,9 @@ static void runTestPutAndListKey(PutHelper helper)
|
||||
@Test
|
||||
public void testGetKeyInfo() throws OzoneException, IOException {
|
||||
runTestGetKeyInfo(new PutHelper(ozoneRestClient, path));
|
||||
String delimiter = RandomStringUtils.randomAscii(1);
|
||||
runTestGetKeyInfo(new PutHelper(ozoneRestClient, path,
|
||||
getMultiPartKey(delimiter)));
|
||||
}
|
||||
|
||||
static void runTestGetKeyInfo(PutHelper helper) throws OzoneException {
|
||||
|
@ -17,6 +17,7 @@
|
||||
*/
|
||||
package org.apache.hadoop.ozone.web.client;
|
||||
|
||||
import org.apache.commons.lang.RandomStringUtils;
|
||||
import org.apache.hadoop.ozone.OzoneConfigKeys;
|
||||
import org.apache.hadoop.ozone.RatisTestHelper;
|
||||
import org.apache.hadoop.ozone.web.exceptions.OzoneException;
|
||||
@ -57,6 +58,9 @@ public static void shutdown() {
|
||||
@Test
|
||||
public void testPutKey() throws OzoneException {
|
||||
runTestPutKey(new PutHelper(ozoneRestClient, path));
|
||||
String delimiter = RandomStringUtils.randomAlphanumeric(1);
|
||||
runTestPutKey(new PutHelper(ozoneRestClient, path,
|
||||
getMultiPartKey(delimiter)));
|
||||
}
|
||||
|
||||
@Test
|
||||
@ -64,25 +68,41 @@ public void testPutAndGetKeyWithDnRestart()
|
||||
throws OzoneException, IOException, URISyntaxException {
|
||||
runTestPutAndGetKeyWithDnRestart(
|
||||
new PutHelper(ozoneRestClient, path), suite.getCluster());
|
||||
String delimiter = RandomStringUtils.randomAlphanumeric(1);
|
||||
runTestPutAndGetKeyWithDnRestart(
|
||||
new PutHelper(ozoneRestClient, path, getMultiPartKey(delimiter)),
|
||||
suite.getCluster());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testPutAndGetKey() throws OzoneException, IOException {
|
||||
runTestPutAndGetKey(new PutHelper(ozoneRestClient, path));
|
||||
String delimiter = RandomStringUtils.randomAlphanumeric(1);
|
||||
runTestPutAndGetKey(new PutHelper(ozoneRestClient, path,
|
||||
getMultiPartKey(delimiter)));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testPutAndDeleteKey() throws OzoneException, IOException {
|
||||
runTestPutAndDeleteKey(new PutHelper(ozoneRestClient, path));
|
||||
String delimiter = RandomStringUtils.randomAlphanumeric(1);
|
||||
runTestPutAndDeleteKey(new PutHelper(ozoneRestClient, path,
|
||||
getMultiPartKey(delimiter)));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testPutAndListKey() throws OzoneException, IOException {
|
||||
runTestPutAndListKey(new PutHelper(ozoneRestClient, path));
|
||||
String delimiter = RandomStringUtils.randomAlphanumeric(1);
|
||||
runTestPutAndListKey(new PutHelper(ozoneRestClient, path,
|
||||
getMultiPartKey(delimiter)));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetKeyInfo() throws OzoneException, IOException {
|
||||
runTestGetKeyInfo(new PutHelper(ozoneRestClient, path));
|
||||
String delimiter = RandomStringUtils.randomAlphanumeric(1);
|
||||
runTestGetKeyInfo(new PutHelper(ozoneRestClient, path,
|
||||
getMultiPartKey(delimiter)));
|
||||
}
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user