001 /*
002 * SPDX-License-Identifier: Apache-2.0
003 *
004 * Copyright 2019-2022 Andres Almiray.
005 *
006 * Licensed under the Apache License, Version 2.0 (the "License");
007 * you may not use this file except in compliance with the License.
008 * You may obtain a copy of the License at
009 *
010 * http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018 package org.kordamp.gradle.plugin.oci.tasks.object
019
020 import com.oracle.bmc.objectstorage.ObjectStorageAsyncClient
021 import com.oracle.bmc.objectstorage.ObjectStorageClient
022 import com.oracle.bmc.objectstorage.model.Bucket
023 import com.oracle.bmc.objectstorage.model.ObjectSummary
024 import com.oracle.bmc.objectstorage.requests.DeleteObjectRequest
025 import com.oracle.bmc.objectstorage.requests.GetBucketRequest
026 import com.oracle.bmc.objectstorage.requests.ListObjectsRequest
027 import com.oracle.bmc.objectstorage.responses.DeleteObjectResponse
028 import com.oracle.bmc.responses.AsyncHandler
029 import groovy.transform.CompileStatic
030 import org.kordamp.gradle.plugin.oci.tasks.AbstractOCITask
031 import org.kordamp.gradle.plugin.oci.tasks.interfaces.OCITask
032 import org.kordamp.gradle.plugin.oci.tasks.traits.AsyncAwareTrait
033 import org.kordamp.gradle.plugin.oci.tasks.traits.BucketNameAwareTrait
034 import org.kordamp.gradle.plugin.oci.tasks.traits.NamespaceNameAwareTrait
035 import org.kordamp.jipsy.annotations.TypeProviderFor
036
037 /**
038 * @author Andres Almiray
039 * @since 0.3.0
040 */
041 @CompileStatic
042 @TypeProviderFor(OCITask)
043 class ClearBucketTask extends AbstractOCITask implements NamespaceNameAwareTrait,
044 BucketNameAwareTrait,
045 AsyncAwareTrait {
046 static final String TASK_DESCRIPTION = 'Clears a Bucket.'
047
048 @Override
049 protected void doExecuteTask() {
050 validateNamespaceName()
051 validateBucketName()
052
053 ObjectStorageClient client = createObjectStorageClient()
054
055 String bucketName = getResolvedBucketName().get()
056 String namespaceName = getResolvedNamespaceName().get()
057
058 List<GetBucketRequest.Fields> fields = new ArrayList<>(2)
059 fields.add(GetBucketRequest.Fields.ApproximateCount)
060 fields.add(GetBucketRequest.Fields.ApproximateSize)
061
062 Bucket bucket = client.getBucket(GetBucketRequest.builder()
063 .namespaceName(namespaceName)
064 .bucketName(bucketName)
065 .fields(fields)
066 .build())
067 .bucket
068
069 println("Bucket ${bucketName} has ${console.cyan(String.valueOf(bucket.approximateCount))} elements")
070 if (bucket.approximateCount < 1) {
071 return
072 }
073
074 ListObjectsRequest.Builder builder = ListObjectsRequest.builder()
075 .namespaceName(namespaceName)
076 .bucketName(bucketName)
077
078 if (getResolvedAsync().get()) {
079 ObjectStorageAsyncClient asyncClient = createObjectStorageAsyncClient()
080 AsyncHandler<DeleteObjectRequest, DeleteObjectResponse> handler = new AsyncHandler<DeleteObjectRequest, DeleteObjectResponse>() {
081 @Override
082 void onSuccess(DeleteObjectRequest req, DeleteObjectResponse rep) {
083 println("Object ${console.yellow(req.objectName)} deleted ")
084 }
085
086 @Override
087 void onError(DeleteObjectRequest req, Throwable error) {
088 println("Could not delete object ${console.yellow(req.objectName)}: ${error}")
089 }
090 }
091
092 for (ObjectSummary objectSummary : client.listObjects(builder.build()).listObjects.objects) {
093 asyncClient.deleteObject(DeleteObjectRequest.builder()
094 .namespaceName(namespaceName)
095 .bucketName(bucketName)
096 .objectName(objectSummary.name)
097 .build(),
098 handler)
099 }
100 } else {
101 for (ObjectSummary objectSummary : client.listObjects(builder.build()).listObjects.objects) {
102 try {
103 client.deleteObject(DeleteObjectRequest.builder()
104 .namespaceName(namespaceName)
105 .bucketName(bucketName)
106 .objectName(objectSummary.name)
107 .build())
108 println("Object ${console.yellow(objectSummary.name)} deleted ")
109 } catch (Exception e) {
110 println("Could not delete object ${console.yellow(objectSummary.name)}: ${e}")
111 }
112 }
113 }
114 }
115 }
|