From 3def52daf5f58387bb8281cd802ee132b5772c7c Mon Sep 17 00:00:00 2001
From: pkuweblab <pkuweblab@163.com>
Date: Thu, 16 Nov 2023 14:41:37 +0800
Subject: [PATCH] add AlluxioCosException

---
 .../underfs/cos/AlluxioCosException.java      | 100 ++++++++++++++++++
 .../underfs/cos/COSUnderFileSystem.java       |   5 +-
 2 files changed, 103 insertions(+), 2 deletions(-)
 create mode 100644 underfs/cos/src/main/java/alluxio/underfs/cos/AlluxioCosException.java

diff --git a/underfs/cos/src/main/java/alluxio/underfs/cos/AlluxioCosException.java b/underfs/cos/src/main/java/alluxio/underfs/cos/AlluxioCosException.java
new file mode 100644
index 000000000000..dd9187b3b151
--- /dev/null
+++ b/underfs/cos/src/main/java/alluxio/underfs/cos/AlluxioCosException.java
@@ -0,0 +1,100 @@
+/*
+ * The Alluxio Open Foundation licenses this work under the Apache License, version 2.0
+ * (the "License"). You may not use this work except in compliance with the License, which is
+ * available at www.apache.org/licenses/LICENSE-2.0
+ *
+ * This software is distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
+ * either express or implied, as more fully set forth in the License.
+ *
+ * See the NOTICE file distributed with this work for information regarding copyright ownership.
+ */
+
+package alluxio.underfs.cos;
+
+import alluxio.exception.runtime.AlluxioRuntimeException;
+import alluxio.grpc.ErrorType;
+
+import com.qcloud.cos.exception.CosClientException;
+import com.qcloud.cos.exception.CosServiceException;
+import io.grpc.Status;
+
+import java.net.HttpURLConnection;
+
+/**
+ * Alluxio exception for cos.
+ */
+public class AlluxioCosException extends AlluxioRuntimeException {
+  private static final ErrorType ERROR_TYPE = ErrorType.External;
+
+  /**
+   * Converts an AmazonClientException to a corresponding AlluxioCosException.
+   * @param cause cos exception
+   * @return alluxio cos exception
+   */
+  public static AlluxioCosException from(CosClientException cause) {
+    return from(null, cause);
+  }
+
+  /**
+   * Converts an CosClientException with errormessage to a corresponding AlluxioCosException.
+   * @param errorMessage error message
+   * @param cause cos exception
+   * @return alluxio cos exception
+   */
+  public static AlluxioCosException from(String errorMessage, CosClientException cause) {
+    Status status = Status.UNKNOWN;
+    String errorDescription = "ClientException:" + cause.getMessage();
+    if (cause instanceof CosServiceException) {
+      CosServiceException exception = (CosServiceException) cause;
+      status = httpStatusToGrpcStatus(exception.getStatusCode());
+      errorDescription = exception.getErrorCode() + ":" + exception.getErrorMessage();
+    }
+    if (errorMessage == null) {
+      errorMessage = errorDescription;
+    }
+    return new AlluxioCosException(status, errorMessage, cause, cause.isRetryable());
+  }
+
+  private AlluxioCosException(Status status, String message, Throwable cause, boolean isRetryAble) {
+    super(status, message, cause, ERROR_TYPE, isRetryAble);
+  }
+
+  private static Status httpStatusToGrpcStatus(int httpStatusCode) {
+    if (httpStatusCode >= 100 && httpStatusCode < 200) {
+      // 1xx. These headers should have been ignored.
+      return Status.INTERNAL;
+    }
+    switch (httpStatusCode) {
+      case HttpURLConnection.HTTP_BAD_REQUEST:  // 400
+        return Status.INVALID_ARGUMENT;
+      case HttpURLConnection.HTTP_UNAUTHORIZED:  // 401
+        return Status.UNAUTHENTICATED;
+      case HttpURLConnection.HTTP_FORBIDDEN:  // 403
+        return Status.PERMISSION_DENIED;
+      case HttpURLConnection.HTTP_NOT_FOUND:  // 404
+        return Status.NOT_FOUND;
+      case HttpURLConnection.HTTP_BAD_METHOD: // 405
+      case HttpURLConnection.HTTP_NOT_IMPLEMENTED: // 501
+        return Status.UNIMPLEMENTED;
+      case HttpURLConnection.HTTP_CONFLICT:  // 409
+        return Status.ABORTED;
+      case HttpURLConnection.HTTP_LENGTH_REQUIRED: // 411
+      case HttpURLConnection.HTTP_PRECON_FAILED: // 412
+        return Status.FAILED_PRECONDITION;
+      case 416: // Requested Range Not Satisfiable
+        return Status.OUT_OF_RANGE;
+      case HttpURLConnection.HTTP_INTERNAL_ERROR: //500
+        return Status.INTERNAL;
+      case HttpURLConnection.HTTP_MOVED_PERM:  // 301
+      case HttpURLConnection.HTTP_NOT_MODIFIED: //304
+      case 307: // Moved Temporarily
+      case HttpURLConnection.HTTP_BAD_GATEWAY:  // 502
+      case HttpURLConnection.HTTP_UNAVAILABLE:  // 503
+        return Status.UNAVAILABLE;
+      case HttpURLConnection.HTTP_GATEWAY_TIMEOUT:  // 504
+        return Status.DEADLINE_EXCEEDED;
+      default:
+        return Status.UNKNOWN;
+    }
+  }
+}
diff --git a/underfs/cos/src/main/java/alluxio/underfs/cos/COSUnderFileSystem.java b/underfs/cos/src/main/java/alluxio/underfs/cos/COSUnderFileSystem.java
index 14c965c0ef0b..7f589cfd082e 100644
--- a/underfs/cos/src/main/java/alluxio/underfs/cos/COSUnderFileSystem.java
+++ b/underfs/cos/src/main/java/alluxio/underfs/cos/COSUnderFileSystem.java
@@ -180,7 +180,8 @@ protected List<String> deleteObjects(List<String> keys) throws IOException {
           .map(DeleteObjectsResult.DeletedObject::getKey)
           .collect(Collectors.toList());
     } catch (CosClientException e) {
-      throw new IOException("failed to delete objects", e);
+      LOG.warn("failed to delete objects");
+      throw AlluxioCosException.from(e);
     }
   }
 
@@ -324,7 +325,7 @@ protected InputStream openObject(String key, OpenOptions options,
       return new COSInputStream(mBucketNameInternal, key, mClient, options.getOffset(), retryPolicy,
           mUfsConf.getBytes(PropertyKey.UNDERFS_OBJECT_STORE_MULTI_RANGE_CHUNK_SIZE));
     } catch (CosClientException e) {
-      throw new IOException(e.getMessage());
+      throw AlluxioCosException.from(e);
     }
   }
 }