From 5e88126776e6d682a48f737d8ab1ad0e04d3e767 Mon Sep 17 00:00:00 2001 From: Xiaoyu Yao Date: Mon, 21 May 2018 16:09:24 -0700 Subject: [PATCH] HDDS-82. Merge ContainerData and ContainerStatus classes. Contributed by Bharat Viswanadham. --- .../proto/DatanodeContainerProtocol.proto | 1 + .../common/helpers/ContainerData.java | 163 ++++++++++++- .../common/impl/ContainerManagerImpl.java | 144 ++++++------ .../common/impl/ContainerStatus.java | 217 ------------------ ...RandomContainerDeletionChoosingPolicy.java | 10 +- ...rderedContainerDeletionChoosingPolicy.java | 20 +- .../ContainerDeletionChoosingPolicy.java | 3 +- .../common/impl/TestContainerPersistence.java | 19 +- 8 files changed, 257 insertions(+), 320 deletions(-) delete mode 100644 hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/ContainerStatus.java diff --git a/hadoop-hdds/common/src/main/proto/DatanodeContainerProtocol.proto b/hadoop-hdds/common/src/main/proto/DatanodeContainerProtocol.proto index e7e5b2b692..95b7cbba1b 100644 --- a/hadoop-hdds/common/src/main/proto/DatanodeContainerProtocol.proto +++ b/hadoop-hdds/common/src/main/proto/DatanodeContainerProtocol.proto @@ -152,6 +152,7 @@ enum ContainerLifeCycleState { OPEN = 1; CLOSING = 2; CLOSED = 3; + INVALID = 4; } message ContainerCommandRequestProto { diff --git a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/helpers/ContainerData.java b/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/helpers/ContainerData.java index 2a079b0af0..14ee33a32f 100644 --- a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/helpers/ContainerData.java +++ b/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/helpers/ContainerData.java @@ -52,6 +52,17 @@ public class ContainerData { private ContainerType containerType; private String containerDBType; + + /** + * Number of pending deletion blocks in container. + */ + private int numPendingDeletionBlocks; + private AtomicLong readBytes; + private AtomicLong writeBytes; + private AtomicLong readCount; + private AtomicLong writeCount; + + /** * Constructs a ContainerData Object. * @@ -66,6 +77,34 @@ public ContainerData(long containerID, this.bytesUsed = new AtomicLong(0L); this.containerID = containerID; this.state = ContainerLifeCycleState.OPEN; + this.numPendingDeletionBlocks = 0; + this.readCount = new AtomicLong(0L); + this.readBytes = new AtomicLong(0L); + this.writeCount = new AtomicLong(0L); + this.writeBytes = new AtomicLong(0L); + } + + /** + * Constructs a ContainerData Object. + * + * @param containerID - ID + * @param conf - Configuration + * @param state - ContainerLifeCycleState + * @param + */ + public ContainerData(long containerID, Configuration conf, + ContainerLifeCycleState state) { + this.metadata = new TreeMap<>(); + this.maxSize = conf.getLong(ScmConfigKeys.SCM_CONTAINER_CLIENT_MAX_SIZE_KEY, + ScmConfigKeys.SCM_CONTAINER_CLIENT_MAX_SIZE_DEFAULT) * OzoneConsts.GB; + this.bytesUsed = new AtomicLong(0L); + this.containerID = containerID; + this.state = state; + this.numPendingDeletionBlocks = 0; + this.readCount = new AtomicLong(0L); + this.readBytes = new AtomicLong(0L); + this.writeCount = new AtomicLong(0L); + this.writeBytes = new AtomicLong(0L); } /** @@ -292,6 +331,14 @@ public synchronized boolean isOpen() { return ContainerLifeCycleState.OPEN == state; } + /** + * checks if the container is invalid. + * @return - boolean + */ + public boolean isValid() { + return !(ContainerLifeCycleState.INVALID == state); + } + /** * Marks this container as closed. */ @@ -317,11 +364,119 @@ public void setBytesUsed(long used) { this.bytesUsed.set(used); } - public long addBytesUsed(long delta) { - return this.bytesUsed.addAndGet(delta); - } - + /** + * Get the number of bytes used by the container. + * @return the number of bytes used by the container. + */ public long getBytesUsed() { return bytesUsed.get(); } + + /** + * Increase the number of bytes used by the container. + * @param used number of bytes used by the container. + * @return the current number of bytes used by the container afert increase. + */ + public long incrBytesUsed(long used) { + return this.bytesUsed.addAndGet(used); + } + + + /** + * Decrease the number of bytes used by the container. + * @param reclaimed the number of bytes reclaimed from the container. + * @return the current number of bytes used by the container after decrease. + */ + public long decrBytesUsed(long reclaimed) { + return this.bytesUsed.addAndGet(-1L * reclaimed); + } + + /** + * Increase the count of pending deletion blocks. + * + * @param numBlocks increment number + */ + public void incrPendingDeletionBlocks(int numBlocks) { + this.numPendingDeletionBlocks += numBlocks; + } + + /** + * Decrease the count of pending deletion blocks. + * + * @param numBlocks decrement number + */ + public void decrPendingDeletionBlocks(int numBlocks) { + this.numPendingDeletionBlocks -= numBlocks; + } + + /** + * Get the number of pending deletion blocks. + */ + public int getNumPendingDeletionBlocks() { + return this.numPendingDeletionBlocks; + } + + /** + * Get the number of bytes read from the container. + * @return the number of bytes read from the container. + */ + public long getReadBytes() { + return readBytes.get(); + } + + /** + * Increase the number of bytes read from the container. + * @param bytes number of bytes read. + */ + public void incrReadBytes(long bytes) { + this.readBytes.addAndGet(bytes); + } + + /** + * Get the number of times the container is read. + * @return the number of times the container is read. + */ + public long getReadCount() { + return readCount.get(); + } + + /** + * Increase the number of container read count by 1. + */ + public void incrReadCount() { + this.readCount.incrementAndGet(); + } + + /** + * Get the number of bytes write into the container. + * @return the number of bytes write into the container. + */ + public long getWriteBytes() { + return writeBytes.get(); + } + + /** + * Increase the number of bytes write into the container. + * @param bytes the number of bytes write into the container. + */ + public void incrWriteBytes(long bytes) { + this.writeBytes.addAndGet(bytes); + } + + /** + * Get the number of writes into the container. + * @return the number of writes into the container. + */ + public long getWriteCount() { + return writeCount.get(); + } + + /** + * Increase the number of writes into the container by 1. + */ + public void incrWriteCount() { + this.writeCount.incrementAndGet(); + } + + } diff --git a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/ContainerManagerImpl.java b/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/ContainerManagerImpl.java index c443ace085..3a78c70d66 100644 --- a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/ContainerManagerImpl.java +++ b/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/ContainerManagerImpl.java @@ -22,13 +22,14 @@ import com.google.common.base.Preconditions; import org.apache.commons.io.FileUtils; import org.apache.hadoop.conf.Configuration; -import org.apache.hadoop.fs.StorageType; import org.apache.hadoop.hdds.scm.ScmConfigKeys; import org.apache.hadoop.hdds.scm.container.common.helpers .StorageContainerException; import org.apache.hadoop.hdfs.server.datanode.StorageLocation; import org.apache.hadoop.hdds.protocol.DatanodeDetails; import org.apache.hadoop.hdds.protocol.datanode.proto.ContainerProtos; +import org.apache.hadoop.hdds.protocol.datanode.proto.ContainerProtos + .ContainerLifeCycleState; import org.apache.hadoop.hdds.protocol.proto .StorageContainerDatanodeProtocolProtos; import org.apache.hadoop.hdds.protocol.proto @@ -39,8 +40,6 @@ .StorageContainerDatanodeProtocolProtos.SCMNodeReport; import org.apache.hadoop.hdds.protocol.proto .StorageContainerDatanodeProtocolProtos.SCMStorageReport; -import org.apache.hadoop.hdds.protocol.proto - .StorageContainerDatanodeProtocolProtos.StorageTypeProto; import org.apache.hadoop.io.IOUtils; import org.apache.hadoop.ozone.OzoneConfigKeys; import org.apache.hadoop.ozone.OzoneConsts; @@ -116,7 +115,7 @@ public class ContainerManagerImpl implements ContainerManager { // TODO: consider primitive collection like eclipse-collections // to avoid autoboxing overhead - private final ConcurrentSkipListMap + private final ConcurrentSkipListMap containerMap = new ConcurrentSkipListMap<>(); // Use a non-fair RW lock for better throughput, we may revisit this decision @@ -246,19 +245,20 @@ private void readContainerInfo(String containerName) // when loading the info we get a null, this often means last time // SCM was ending up at some middle phase causing that the metadata // was not populated. Such containers are marked as inactive. - containerMap.put(containerID, new ContainerStatus(null)); + ContainerData cData = new ContainerData(containerID, conf, + ContainerLifeCycleState.INVALID); + containerMap.put(containerID, cData); return; } containerData = ContainerData.getFromProtBuf(containerDataProto, conf); - ContainerStatus containerStatus = new ContainerStatus(containerData); // Initialize pending deletion blocks count in in-memory // container status. MetadataStore metadata = KeyUtils.getDB(containerData, conf); List> underDeletionBlocks = metadata .getSequentialRangeKVs(null, Integer.MAX_VALUE, MetadataKeyFilters.getDeletingKeyFilter()); - containerStatus.incrPendingDeletionBlocks(underDeletionBlocks.size()); + containerData.incrPendingDeletionBlocks(underDeletionBlocks.size()); List> liveKeys = metadata .getRangeKVs(null, Integer.MAX_VALUE, @@ -277,9 +277,9 @@ private void readContainerInfo(String containerName) return 0L; } }).sum(); - containerStatus.setBytesUsed(bytesUsed); + containerData.setBytesUsed(bytesUsed); - containerMap.put(containerID, containerStatus); + containerMap.put(containerID, containerData); } catch (IOException ex) { LOG.error("read failed for file: {} ex: {}", containerName, ex.getMessage()); @@ -287,7 +287,9 @@ private void readContainerInfo(String containerName) // TODO : Add this file to a recovery Queue. // Remember that this container is busted and we cannot use it. - containerMap.put(containerID, new ContainerStatus(null)); + ContainerData cData = new ContainerData(containerID, conf, + ContainerLifeCycleState.INVALID); + containerMap.put(containerID, cData); throw new StorageContainerException("Unable to read container info", UNABLE_TO_READ_METADATA_DB); } finally { @@ -456,18 +458,19 @@ public void deleteContainer(long containerID, UNCLOSED_CONTAINER_IO); } - ContainerStatus status = containerMap.get(containerID); - if (status == null) { + ContainerData containerData = containerMap.get(containerID); + if (containerData == null) { LOG.debug("No such container. ID: {}", containerID); throw new StorageContainerException("No such container. ID : " + containerID, CONTAINER_NOT_FOUND); } - if (status.getContainer() == null) { + + if(!containerData.isValid()) { LOG.debug("Invalid container data. ID: {}", containerID); throw new StorageContainerException("Invalid container data. Name : " + containerID, CONTAINER_NOT_FOUND); } - ContainerUtils.removeContainer(status.getContainer(), conf, forceDelete); + ContainerUtils.removeContainer(containerData, conf, forceDelete); containerMap.remove(containerID); } catch (StorageContainerException e) { throw e; @@ -509,7 +512,7 @@ public void listContainer(long startContainerID, long count, readLock(); try { - ConcurrentNavigableMap map; + ConcurrentNavigableMap map; if (startContainerID == 0) { map = containerMap.tailMap(containerMap.firstKey(), true); } else { @@ -517,9 +520,9 @@ public void listContainer(long startContainerID, long count, } int currentCount = 0; - for (ContainerStatus entry : map.values()) { + for (ContainerData entry : map.values()) { if (currentCount < count) { - data.add(entry.getContainer()); + data.add(entry); currentCount++; } else { return; @@ -546,7 +549,7 @@ public ContainerData readContainer(long containerID) throw new StorageContainerException("Unable to find the container. ID: " + containerID, CONTAINER_NOT_FOUND); } - ContainerData cData = containerMap.get(containerID).getContainer(); + ContainerData cData = containerMap.get(containerID); if (cData == null) { throw new StorageContainerException("Invalid container data. ID: " + containerID, CONTAINER_INTERNAL_ERROR); @@ -584,8 +587,7 @@ public void closeContainer(long containerID) // I/O failure, this allows us to take quick action in case of container // issues. - ContainerStatus status = new ContainerStatus(containerData); - containerMap.put(containerID, status); + containerMap.put(containerID, containerData); } @Override @@ -614,7 +616,7 @@ public void updateContainer(long containerID, ContainerData data, try { Path location = locationManager.getContainerPath(); - ContainerData orgData = containerMap.get(containerID).getContainer(); + ContainerData orgData = containerMap.get(containerID); if (orgData == null) { // updating a invalid container throw new StorageContainerException("Update a container with invalid" + @@ -652,8 +654,7 @@ public void updateContainer(long containerID, ContainerData data, } // Update the in-memory map - ContainerStatus newStatus = new ContainerStatus(data); - containerMap.replace(containerID, newStatus); + containerMap.replace(containerID, data); } catch (IOException e) { // Restore the container file from backup if(containerFileBK != null && containerFileBK.exists() && deleted) { @@ -699,17 +700,12 @@ protected File getContainerFile(ContainerData data) throws IOException { */ @Override public boolean isOpen(long containerID) throws StorageContainerException { - final ContainerStatus status = containerMap.get(containerID); - if (status == null) { - throw new StorageContainerException( - "Container status not found: " + containerID, CONTAINER_NOT_FOUND); - } - final ContainerData cData = status.getContainer(); - if (cData == null) { + final ContainerData containerData = containerMap.get(containerID); + if (containerData == null) { throw new StorageContainerException( "Container not found: " + containerID, CONTAINER_NOT_FOUND); } - return cData.isOpen(); + return containerData.isOpen(); } /** @@ -727,7 +723,7 @@ public void shutdown() throws IOException { @VisibleForTesting - public ConcurrentSkipListMap getContainerMap() { + public ConcurrentSkipListMap getContainerMap() { return containerMap; } @@ -847,9 +843,9 @@ public List getContainerReports() throws IOException { // And we can never get the exact state since close might happen // after we iterate a point. return containerMap.entrySet().stream() - .filter(containerStatus -> - !containerStatus.getValue().getContainer().isOpen()) - .map(containerStatus -> containerStatus.getValue().getContainer()) + .filter(containerData -> + !containerData.getValue().isOpen()) + .map(containerData -> containerData.getValue()) .collect(Collectors.toList()); } @@ -865,7 +861,7 @@ public ContainerReportsRequestProto getContainerReport() throws IOException { // No need for locking since containerMap is a ConcurrentSkipListMap // And we can never get the exact state since close might happen // after we iterate a point. - List containers = containerMap.values().stream() + List containers = containerMap.values().stream() .collect(Collectors.toList()); ContainerReportsRequestProto.Builder crBuilder = @@ -875,18 +871,17 @@ public ContainerReportsRequestProto getContainerReport() throws IOException { crBuilder.setDatanodeDetails(datanodeDetails.getProtoBufMessage()) .setType(ContainerReportsRequestProto.reportType.fullReport); - for (ContainerStatus container: containers) { + for (ContainerData container: containers) { StorageContainerDatanodeProtocolProtos.ContainerInfo.Builder ciBuilder = StorageContainerDatanodeProtocolProtos.ContainerInfo.newBuilder(); - ciBuilder.setContainerID(container.getContainer().getContainerID()) - .setSize(container.getContainer().getMaxSize()) - .setUsed(container.getContainer().getBytesUsed()) - .setKeyCount(container.getContainer().getKeyCount()) + ciBuilder.setContainerID(container.getContainerID()) + .setSize(container.getMaxSize()) + .setUsed(container.getBytesUsed()) + .setKeyCount(container.getKeyCount()) .setReadCount(container.getReadCount()) .setWriteCount(container.getWriteCount()) .setReadBytes(container.getReadBytes()) - .setWriteBytes(container.getWriteBytes()) - .setContainerID(container.getContainer().getContainerID()); + .setWriteBytes(container.getWriteBytes()); crBuilder.addReports(ciBuilder.build()); } @@ -943,8 +938,8 @@ public ContainerDeletionChoosingPolicy getContainerDeletionChooser() { public void incrPendingDeletionBlocks(int numBlocks, long containerId) { writeLock(); try { - ContainerStatus status = containerMap.get(containerId); - status.incrPendingDeletionBlocks(numBlocks); + ContainerData cData = containerMap.get(containerId); + cData.incrPendingDeletionBlocks(numBlocks); } finally { writeUnlock(); } @@ -954,8 +949,8 @@ public void incrPendingDeletionBlocks(int numBlocks, long containerId) { public void decrPendingDeletionBlocks(int numBlocks, long containerId) { writeLock(); try { - ContainerStatus status = containerMap.get(containerId); - status.decrPendingDeletionBlocks(numBlocks); + ContainerData cData = containerMap.get(containerId); + cData.decrPendingDeletionBlocks(numBlocks); } finally { writeUnlock(); } @@ -968,32 +963,37 @@ public void decrPendingDeletionBlocks(int numBlocks, long containerId) { */ @Override public void incrReadCount(long containerId) { - ContainerStatus status = containerMap.get(containerId); - status.incrReadCount(); + ContainerData cData = containerMap.get(containerId); + cData.incrReadCount(); } public long getReadCount(long containerId) { - ContainerStatus status = containerMap.get(containerId); - return status.getReadCount(); + ContainerData cData = containerMap.get(containerId); + return cData.getReadCount(); } /** - * Increse the read counter for bytes read from the container. + * Increase the read counter for bytes read from the container. * * @param containerId - ID of the container. * @param readBytes - bytes read from the container. */ @Override public void incrReadBytes(long containerId, long readBytes) { - ContainerStatus status = containerMap.get(containerId); - status.incrReadBytes(readBytes); + ContainerData cData = containerMap.get(containerId); + cData.incrReadBytes(readBytes); } + /** + * Returns number of bytes read from the container. + * @param containerId + * @return + */ public long getReadBytes(long containerId) { readLock(); try { - ContainerStatus status = containerMap.get(containerId); - return status.getReadBytes(); + ContainerData cData = containerMap.get(containerId); + return cData.getReadBytes(); } finally { readUnlock(); } @@ -1006,13 +1006,13 @@ public long getReadBytes(long containerId) { */ @Override public void incrWriteCount(long containerId) { - ContainerStatus status = containerMap.get(containerId); - status.incrWriteCount(); + ContainerData cData = containerMap.get(containerId); + cData.incrWriteCount(); } public long getWriteCount(long containerId) { - ContainerStatus status = containerMap.get(containerId); - return status.getWriteCount(); + ContainerData cData = containerMap.get(containerId); + return cData.getWriteCount(); } /** @@ -1023,13 +1023,13 @@ public long getWriteCount(long containerId) { */ @Override public void incrWriteBytes(long containerId, long writeBytes) { - ContainerStatus status = containerMap.get(containerId); - status.incrWriteBytes(writeBytes); + ContainerData cData = containerMap.get(containerId); + cData.incrWriteBytes(writeBytes); } public long getWriteBytes(long containerId) { - ContainerStatus status = containerMap.get(containerId); - return status.getWriteBytes(); + ContainerData cData = containerMap.get(containerId); + return cData.getWriteBytes(); } /** @@ -1041,8 +1041,8 @@ public long getWriteBytes(long containerId) { */ @Override public long incrBytesUsed(long containerId, long used) { - ContainerStatus status = containerMap.get(containerId); - return status.incrBytesUsed(used); + ContainerData cData = containerMap.get(containerId); + return cData.incrBytesUsed(used); } /** @@ -1054,13 +1054,13 @@ public long incrBytesUsed(long containerId, long used) { */ @Override public long decrBytesUsed(long containerId, long used) { - ContainerStatus status = containerMap.get(containerId); - return status.decrBytesUsed(used); + ContainerData cData = containerMap.get(containerId); + return cData.decrBytesUsed(used); } public long getBytesUsed(long containerId) { - ContainerStatus status = containerMap.get(containerId); - return status.getBytesUsed(); + ContainerData cData = containerMap.get(containerId); + return cData.getBytesUsed(); } /** @@ -1071,8 +1071,8 @@ public long getBytesUsed(long containerId) { */ @Override public long getNumKeys(long containerId) { - ContainerStatus status = containerMap.get(containerId); - return status.getNumKeys(); } + ContainerData cData = containerMap.get(containerId); + return cData.getKeyCount(); } /** * Get the container report state to send via HB to SCM. diff --git a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/ContainerStatus.java b/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/ContainerStatus.java deleted file mode 100644 index 5577323a10..0000000000 --- a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/ContainerStatus.java +++ /dev/null @@ -1,217 +0,0 @@ -/** - * Licensed to the Apache Software Foundation (ASF) under one - * or more contributor license agreements. See the NOTICE file - * distributed with this work for additional information - * regarding copyright ownership. The ASF licenses this file - * to you 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.apache.hadoop.ozone.container.common.impl; - -import org.apache.hadoop.ozone.container.common.helpers.ContainerData; - -import java.util.concurrent.atomic.AtomicLong; - -/** - * This class represents the state of a container. if the - * container reading encountered an error when we boot up we will post that - * info to a recovery queue and keep the info in the containerMap. - *

- * if and when the issue is fixed, the expectation is that this entry will be - * deleted by the recovery thread from the containerMap and will insert entry - * instead of modifying this class. - */ -public class ContainerStatus { - private final ContainerData containerData; - - /** - * Number of pending deletion blocks in container. - */ - private int numPendingDeletionBlocks; - - private AtomicLong readBytes; - - private AtomicLong writeBytes; - - private AtomicLong readCount; - - private AtomicLong writeCount; - - /** - * Creates a Container Status class. - * - * @param containerData - ContainerData. - */ - ContainerStatus(ContainerData containerData) { - this.numPendingDeletionBlocks = 0; - this.containerData = containerData; - this.readCount = new AtomicLong(0L); - this.readBytes = new AtomicLong(0L); - this.writeCount = new AtomicLong(0L); - this.writeBytes = new AtomicLong(0L); - } - - /** - * Returns container if it is active. It is not active if we have had an - * error and we are waiting for the background threads to fix the issue. - * - * @return ContainerData. - */ - public ContainerData getContainer() { - return containerData; - } - - /** - * Increase the count of pending deletion blocks. - * - * @param numBlocks increment number - */ - public void incrPendingDeletionBlocks(int numBlocks) { - this.numPendingDeletionBlocks += numBlocks; - } - - /** - * Decrease the count of pending deletion blocks. - * - * @param numBlocks decrement number - */ - public void decrPendingDeletionBlocks(int numBlocks) { - this.numPendingDeletionBlocks -= numBlocks; - } - - /** - * Get the number of pending deletion blocks. - */ - public int getNumPendingDeletionBlocks() { - return this.numPendingDeletionBlocks; - } - - /** - * Get the number of bytes read from the container. - * @return the number of bytes read from the container. - */ - public long getReadBytes() { - return readBytes.get(); - } - - /** - * Increase the number of bytes read from the container. - * @param bytes number of bytes read. - */ - public void incrReadBytes(long bytes) { - this.readBytes.addAndGet(bytes); - } - - /** - * Get the number of times the container is read. - * @return the number of times the container is read. - */ - public long getReadCount() { - return readCount.get(); - } - - /** - * Increase the number of container read count by 1. - */ - public void incrReadCount() { - this.readCount.incrementAndGet(); - } - - /** - * Get the number of bytes write into the container. - * @return the number of bytes write into the container. - */ - public long getWriteBytes() { - return writeBytes.get(); - } - - /** - * Increase the number of bytes write into the container. - * @param bytes the number of bytes write into the container. - */ - public void incrWriteBytes(long bytes) { - this.writeBytes.addAndGet(bytes); - } - - /** - * Get the number of writes into the container. - * @return the number of writes into the container. - */ - public long getWriteCount() { - return writeCount.get(); - } - - /** - * Increase the number of writes into the container by 1. - */ - public void incrWriteCount() { - this.writeCount.incrementAndGet(); - } - - /** - * Get the number of bytes used by the container. - * @return the number of bytes used by the container. - */ - public long getBytesUsed() { - return containerData.getBytesUsed(); - } - - /** - * Increase the number of bytes used by the container. - * @param used number of bytes used by the container. - * @return the current number of bytes used by the container afert increase. - */ - public long incrBytesUsed(long used) { - return containerData.addBytesUsed(used); - } - - /** - * Set the number of bytes used by the container. - * @param used the number of bytes used by the container. - */ - public void setBytesUsed(long used) { - containerData.setBytesUsed(used); - } - - /** - * Decrease the number of bytes used by the container. - * @param reclaimed the number of bytes reclaimed from the container. - * @return the current number of bytes used by the container after decrease. - */ - public long decrBytesUsed(long reclaimed) { - return this.containerData.addBytesUsed(-1L * reclaimed); - } - - /** - * Get the maximum container size. - * @return the maximum container size. - */ - public long getMaxSize() { - return containerData.getMaxSize(); - } - - /** - * Set the maximum container size. - * @param size the maximum container size. - */ - public void setMaxSize(long size) { - this.containerData.setMaxSize(size); - } - - /** - * Get the number of keys in the container. - * @return the number of keys in the container. - */ - public long getNumKeys() { - return containerData.getKeyCount(); - } -} \ No newline at end of file diff --git a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/RandomContainerDeletionChoosingPolicy.java b/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/RandomContainerDeletionChoosingPolicy.java index 06177cb332..97fdb9ecfc 100644 --- a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/RandomContainerDeletionChoosingPolicy.java +++ b/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/RandomContainerDeletionChoosingPolicy.java @@ -41,24 +41,24 @@ public class RandomContainerDeletionChoosingPolicy @Override public List chooseContainerForBlockDeletion(int count, - Map candidateContainers) + Map candidateContainers) throws StorageContainerException { Preconditions.checkNotNull(candidateContainers, "Internal assertion: candidate containers cannot be null"); int currentCount = 0; List result = new LinkedList<>(); - ContainerStatus[] values = new ContainerStatus[candidateContainers.size()]; + ContainerData[] values = new ContainerData[candidateContainers.size()]; // to get a shuffle list - for (ContainerStatus entry : DFSUtil.shuffle( + for (ContainerData entry : DFSUtil.shuffle( candidateContainers.values().toArray(values))) { if (currentCount < count) { - result.add(entry.getContainer()); + result.add(entry); currentCount++; LOG.debug("Select container {} for block deletion, " + "pending deletion blocks num: {}.", - entry.getContainer().getContainerID(), + entry.getContainerID(), entry.getNumPendingDeletionBlocks()); } else { break; diff --git a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/TopNOrderedContainerDeletionChoosingPolicy.java b/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/TopNOrderedContainerDeletionChoosingPolicy.java index 246342672f..9a109e8c99 100644 --- a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/TopNOrderedContainerDeletionChoosingPolicy.java +++ b/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/impl/TopNOrderedContainerDeletionChoosingPolicy.java @@ -41,11 +41,11 @@ public class TopNOrderedContainerDeletionChoosingPolicy private static final Logger LOG = LoggerFactory.getLogger(TopNOrderedContainerDeletionChoosingPolicy.class); - /** customized comparator used to compare differentiate container status. **/ - private static final Comparator CONTAINER_STATUS_COMPARATOR - = new Comparator() { + /** customized comparator used to compare differentiate container data. **/ + private static final Comparator CONTAINER_DATA_COMPARATOR + = new Comparator() { @Override - public int compare(ContainerStatus c1, ContainerStatus c2) { + public int compare(ContainerData c1, ContainerData c2) { return Integer.compare(c2.getNumPendingDeletionBlocks(), c1.getNumPendingDeletionBlocks()); } @@ -53,28 +53,28 @@ public int compare(ContainerStatus c1, ContainerStatus c2) { @Override public List chooseContainerForBlockDeletion(int count, - Map candidateContainers) + Map candidateContainers) throws StorageContainerException { Preconditions.checkNotNull(candidateContainers, "Internal assertion: candidate containers cannot be null"); List result = new LinkedList<>(); - List orderedList = new LinkedList<>(); + List orderedList = new LinkedList<>(); orderedList.addAll(candidateContainers.values()); - Collections.sort(orderedList, CONTAINER_STATUS_COMPARATOR); + Collections.sort(orderedList, CONTAINER_DATA_COMPARATOR); // get top N list ordered by pending deletion blocks' number int currentCount = 0; - for (ContainerStatus entry : orderedList) { + for (ContainerData entry : orderedList) { if (currentCount < count) { if (entry.getNumPendingDeletionBlocks() > 0) { - result.add(entry.getContainer()); + result.add(entry); currentCount++; LOG.debug( "Select container {} for block deletion, " + "pending deletion blocks num: {}.", - entry.getContainer().getContainerID(), + entry.getContainerID(), entry.getNumPendingDeletionBlocks()); } else { LOG.debug("Stop looking for next container, there is no" diff --git a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/interfaces/ContainerDeletionChoosingPolicy.java b/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/interfaces/ContainerDeletionChoosingPolicy.java index 6b60c52719..1ed50fb03b 100644 --- a/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/interfaces/ContainerDeletionChoosingPolicy.java +++ b/hadoop-hdds/container-service/src/main/java/org/apache/hadoop/ozone/container/common/interfaces/ContainerDeletionChoosingPolicy.java @@ -20,7 +20,6 @@ import org.apache.hadoop.hdds.scm.container.common.helpers .StorageContainerException; import org.apache.hadoop.ozone.container.common.helpers.ContainerData; -import org.apache.hadoop.ozone.container.common.impl.ContainerStatus; import java.util.List; import java.util.Map; @@ -41,6 +40,6 @@ public interface ContainerDeletionChoosingPolicy { * @throws StorageContainerException */ List chooseContainerForBlockDeletion(int count, - Map candidateContainers) + Map candidateContainers) throws StorageContainerException; } diff --git a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/container/common/impl/TestContainerPersistence.java b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/container/common/impl/TestContainerPersistence.java index a7cab4e783..89ee6739ec 100644 --- a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/container/common/impl/TestContainerPersistence.java +++ b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/container/common/impl/TestContainerPersistence.java @@ -65,7 +65,6 @@ import java.util.List; import java.util.Map; import java.util.ArrayList; -import java.util.UUID; import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_DATANODE_DATA_DIR_KEY; import static org.apache.hadoop.ozone.OzoneConsts.CONTAINER_ROOT_PREFIX; @@ -191,23 +190,23 @@ public void testCreateContainer() throws Exception { containerManager.createContainer(data); Assert.assertTrue(containerManager.getContainerMap() .containsKey(testContainerID)); - ContainerStatus status = containerManager + ContainerData cData = containerManager .getContainerMap().get(testContainerID); - Assert.assertNotNull(status.getContainer()); - Assert.assertNotNull(status.getContainer().getContainerPath()); - Assert.assertNotNull(status.getContainer().getDBPath()); + Assert.assertNotNull(cData); + Assert.assertNotNull(cData.getContainerPath()); + Assert.assertNotNull(cData.getDBPath()); - Assert.assertTrue(new File(status.getContainer().getContainerPath()) + Assert.assertTrue(new File(cData.getContainerPath()) .exists()); - Path meta = Paths.get(status.getContainer().getDBPath()).getParent(); + Path meta = Paths.get(cData.getDBPath()).getParent(); Assert.assertTrue(meta != null && Files.exists(meta)); MetadataStore store = null; try { - store = KeyUtils.getDB(status.getContainer(), conf); + store = KeyUtils.getDB(cData, conf); Assert.assertNotNull(store); } finally { if (store != null) { @@ -762,7 +761,7 @@ public void testUpdateContainer() throws IOException { // Verify in-memory map ContainerData actualNewData = containerManager.getContainerMap() - .get(testContainerID).getContainer(); + .get(testContainerID); Assert.assertEquals("shire_new", actualNewData.getAllMetadata().get("VOLUME")); Assert.assertEquals("bilbo_new", @@ -805,7 +804,7 @@ public void testUpdateContainer() throws IOException { // Verify in-memory map actualNewData = containerManager.getContainerMap() - .get(testContainerID).getContainer(); + .get(testContainerID); Assert.assertEquals("shire_new_1", actualNewData.getAllMetadata().get("VOLUME")); Assert.assertEquals("bilbo_new_1",