|
@@ -0,0 +1,173 @@
|
|
|
+/**
|
|
|
+ * 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
|
|
|
+ * <p>
|
|
|
+ * http://www.apache.org/licenses/LICENSE-2.0
|
|
|
+ * <p>
|
|
|
+ * 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.om.request.s3.multipart;
|
|
|
+
|
|
|
+import java.io.IOException;
|
|
|
+
|
|
|
+import com.google.common.base.Optional;
|
|
|
+import org.slf4j.Logger;
|
|
|
+import org.slf4j.LoggerFactory;
|
|
|
+
|
|
|
+import org.apache.hadoop.ozone.audit.OMAction;
|
|
|
+import org.apache.hadoop.ozone.om.OMMetadataManager;
|
|
|
+import org.apache.hadoop.ozone.om.OzoneManager;
|
|
|
+import org.apache.hadoop.ozone.om.exceptions.OMException;
|
|
|
+import org.apache.hadoop.ozone.om.helpers.OmKeyInfo;
|
|
|
+import org.apache.hadoop.ozone.om.helpers.OmMultipartKeyInfo;
|
|
|
+import org.apache.hadoop.ozone.om.request.key.OMKeyRequest;
|
|
|
+import org.apache.hadoop.ozone.om.response.OMClientResponse;
|
|
|
+import org.apache.hadoop.ozone.om.response.s3.multipart
|
|
|
+ .S3MultipartUploadAbortResponse;
|
|
|
+import org.apache.hadoop.ozone.protocol.proto.OzoneManagerProtocolProtos;
|
|
|
+import org.apache.hadoop.ozone.protocol.proto.OzoneManagerProtocolProtos
|
|
|
+ .KeyArgs;
|
|
|
+import org.apache.hadoop.ozone.protocol.proto.OzoneManagerProtocolProtos
|
|
|
+ .MultipartUploadAbortResponse;
|
|
|
+import org.apache.hadoop.ozone.protocol.proto.OzoneManagerProtocolProtos
|
|
|
+ .OMRequest;
|
|
|
+import org.apache.hadoop.ozone.protocol.proto.OzoneManagerProtocolProtos
|
|
|
+ .OMResponse;
|
|
|
+import org.apache.hadoop.ozone.security.acl.IAccessAuthorizer;
|
|
|
+import org.apache.hadoop.ozone.security.acl.OzoneObj;
|
|
|
+import org.apache.hadoop.util.Time;
|
|
|
+import org.apache.hadoop.utils.db.cache.CacheKey;
|
|
|
+import org.apache.hadoop.utils.db.cache.CacheValue;
|
|
|
+
|
|
|
+import static org.apache.hadoop.ozone.om.lock.OzoneManagerLock.Resource.BUCKET_LOCK;
|
|
|
+
|
|
|
+/**
|
|
|
+ * Handles Abort of multipart upload request.
|
|
|
+ */
|
|
|
+public class S3MultipartUploadAbortRequest extends OMKeyRequest {
|
|
|
+
|
|
|
+ private static final Logger LOG =
|
|
|
+ LoggerFactory.getLogger(S3MultipartUploadAbortRequest.class);
|
|
|
+
|
|
|
+ public S3MultipartUploadAbortRequest(OMRequest omRequest) {
|
|
|
+ super(omRequest);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public OMRequest preExecute(OzoneManager ozoneManager) throws IOException {
|
|
|
+ KeyArgs keyArgs =
|
|
|
+ getOmRequest().getAbortMultiPartUploadRequest().getKeyArgs();
|
|
|
+
|
|
|
+ return getOmRequest().toBuilder().setAbortMultiPartUploadRequest(
|
|
|
+ getOmRequest().getAbortMultiPartUploadRequest().toBuilder()
|
|
|
+ .setKeyArgs(keyArgs.toBuilder().setModificationTime(Time.now())))
|
|
|
+ .setUserInfo(getUserInfo()).build();
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public OMClientResponse validateAndUpdateCache(OzoneManager ozoneManager,
|
|
|
+ long transactionLogIndex) {
|
|
|
+ OzoneManagerProtocolProtos.KeyArgs keyArgs =
|
|
|
+ getOmRequest().getAbortMultiPartUploadRequest().getKeyArgs();
|
|
|
+
|
|
|
+ String volumeName = keyArgs.getVolumeName();
|
|
|
+ String bucketName = keyArgs.getBucketName();
|
|
|
+ String keyName = keyArgs.getKeyName();
|
|
|
+
|
|
|
+ OMMetadataManager omMetadataManager = ozoneManager.getMetadataManager();
|
|
|
+ boolean acquiredLock = false;
|
|
|
+ IOException exception = null;
|
|
|
+ OmMultipartKeyInfo multipartKeyInfo = null;
|
|
|
+ String multipartKey = null;
|
|
|
+ try {
|
|
|
+ // check Acl
|
|
|
+ if (ozoneManager.getAclsEnabled()) {
|
|
|
+ checkAcls(ozoneManager, OzoneObj.ResourceType.KEY,
|
|
|
+ OzoneObj.StoreType.OZONE, IAccessAuthorizer.ACLType.WRITE,
|
|
|
+ volumeName, bucketName, keyName);
|
|
|
+ }
|
|
|
+
|
|
|
+ acquiredLock =
|
|
|
+ omMetadataManager.getLock().acquireLock(BUCKET_LOCK, volumeName,
|
|
|
+ bucketName);
|
|
|
+
|
|
|
+ validateBucketAndVolume(omMetadataManager, volumeName, bucketName);
|
|
|
+
|
|
|
+ multipartKey = omMetadataManager.getMultipartKey(volumeName,
|
|
|
+ bucketName, keyName, keyArgs.getMultipartUploadID());
|
|
|
+
|
|
|
+ OmKeyInfo omKeyInfo =
|
|
|
+ omMetadataManager.getOpenKeyTable().get(multipartKey);
|
|
|
+
|
|
|
+ // If there is no entry in openKeyTable, then there is no multipart
|
|
|
+ // upload initiated for this key.
|
|
|
+ if (omKeyInfo == null) {
|
|
|
+ throw new OMException("Abort Multipart Upload Failed: volume: " +
|
|
|
+ volumeName + "bucket: " + bucketName + "key: " + keyName,
|
|
|
+ OMException.ResultCodes.NO_SUCH_MULTIPART_UPLOAD_ERROR);
|
|
|
+ } else {
|
|
|
+ multipartKeyInfo = omMetadataManager
|
|
|
+ .getMultipartInfoTable().get(multipartKey);
|
|
|
+
|
|
|
+
|
|
|
+ // Update cache of openKeyTable and multipartInfo table.
|
|
|
+ // No need to add the cache entries to delete table, as the entries
|
|
|
+ // in delete table are not used by any read/write operations.
|
|
|
+ omMetadataManager.getOpenKeyTable().addCacheEntry(
|
|
|
+ new CacheKey<>(multipartKey),
|
|
|
+ new CacheValue<>(Optional.absent(), transactionLogIndex));
|
|
|
+ omMetadataManager.getMultipartInfoTable().addCacheEntry(
|
|
|
+ new CacheKey<>(multipartKey),
|
|
|
+ new CacheValue<>(Optional.absent(), transactionLogIndex));
|
|
|
+ }
|
|
|
+
|
|
|
+ } catch (IOException ex) {
|
|
|
+ exception = ex;
|
|
|
+ } finally {
|
|
|
+ if (acquiredLock) {
|
|
|
+ omMetadataManager.getLock().releaseLock(BUCKET_LOCK, volumeName,
|
|
|
+ bucketName);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // audit log
|
|
|
+ auditLog(ozoneManager.getAuditLogger(), buildAuditMessage(
|
|
|
+ OMAction.ABORT_MULTIPART_UPLOAD, buildKeyArgsAuditMap(keyArgs),
|
|
|
+ exception, getOmRequest().getUserInfo()));
|
|
|
+
|
|
|
+ OMResponse.Builder omResponse = OMResponse.newBuilder()
|
|
|
+ .setCmdType(OzoneManagerProtocolProtos.Type.AbortMultiPartUpload)
|
|
|
+ .setStatus(OzoneManagerProtocolProtos.Status.OK)
|
|
|
+ .setSuccess(true);
|
|
|
+
|
|
|
+
|
|
|
+ if (exception == null) {
|
|
|
+ LOG.debug("Abort Multipart request is successfully completed for " +
|
|
|
+ "KeyName {} in VolumeName/Bucket {}/{}", keyName, volumeName,
|
|
|
+ bucketName);
|
|
|
+ return new S3MultipartUploadAbortResponse(multipartKey,
|
|
|
+ keyArgs.getModificationTime(), multipartKeyInfo,
|
|
|
+ omResponse.setAbortMultiPartUploadResponse(
|
|
|
+ MultipartUploadAbortResponse.newBuilder()).build());
|
|
|
+ } else {
|
|
|
+ LOG.error("Abort Multipart request is failed for " +
|
|
|
+ "KeyName {} in VolumeName/Bucket {}/{}", keyName, volumeName,
|
|
|
+ bucketName, exception);
|
|
|
+ return new S3MultipartUploadAbortResponse(multipartKey,
|
|
|
+ keyArgs.getModificationTime(), multipartKeyInfo,
|
|
|
+ createErrorOMResponse(omResponse, exception));
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+}
|