Skip to content

Commit

Permalink
ISSUE-1307 Allow to specify secondary bucket suffix
Browse files Browse the repository at this point in the history
  • Loading branch information
@quantranhong1999
quantranhong1999 committed Nov 15, 2024
1 parent 6f18936 commit e73bc02
Show file tree
Hide file tree
Showing 9 changed files with 120 additions and 82 deletions.
2 changes: 2 additions & 0 deletions docs/modules/ROOT/pages/tmail-backend/configure/index.adoc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -209,6 +209,8 @@ Specified to TMail backend, we can configure the following configurations in the
| S3 service endpoint
| objectstorage.s3.secondary.region
| S3 region
| objectstorage.s3.secondary.bucket.suffix
| Required (if secondary blob store is enabled). The suffix of bucket names for the secondary blob store. e.g. "-copy".
| objectstorage.s3.secondary.accessKeyId
| https://docs.aws.amazon.com/general/latest/gr/aws-sec-cred-types.html#access-keys-and-secret-access-keys[S3 access key id]
| objectstorage.s3.secondary.secretKey
Expand Down
80 changes: 45 additions & 35 deletions ...src/main/java/com/linagora/tmail/blob/secondaryblobstore/FailedBlobOperationListener.java
View file
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
package com.linagora.tmail.blob.secondaryblobstore;

import static com.linagora.tmail.blob.secondaryblobstore.ObjectStorageIdentity.PRIMARY;
import static com.linagora.tmail.blob.secondaryblobstore.ObjectStorageIdentity.SECONDARY;
import static com.linagora.tmail.blob.secondaryblobstore.SecondaryBlobStoreDAO.withSuffix;

import org.apache.james.blob.api.BlobId;
import org.apache.james.blob.api.BlobStoreDAO;
Expand All @@ -21,32 +20,24 @@ public static class FailedBlobOperationListenerGroup extends Group {

private final BlobStoreDAO primaryBlobStoreDAO;
private final BlobStoreDAO secondaryBlobStoreDAO;
private final String secondaryBucketSuffix;

public FailedBlobOperationListener(BlobStoreDAO primaryBlobStoreDAO,
BlobStoreDAO secondaryBlobStoreDAO) {
BlobStoreDAO secondaryBlobStoreDAO,
String secondaryBucketSuffix) {
this.primaryBlobStoreDAO = primaryBlobStoreDAO;
this.secondaryBlobStoreDAO = secondaryBlobStoreDAO;
this.secondaryBucketSuffix = secondaryBucketSuffix;
}

@Override
public Publisher<Void> reactiveEvent(Event event) {
FailedBlobEvents.BlobEvent blobEvent = (FailedBlobEvents.BlobEvent) event;
ObjectStorageIdentity failedObjectStorage = blobEvent.getFailedObjectStorage();
BlobStoreDAO successfulBlobStoreDAO = successfulBlobStoreDAO(failedObjectStorage);
BlobStoreDAO failedBlobStoreDAO = failedBlobStoreDAO(failedObjectStorage);

if (blobEvent instanceof FailedBlobEvents.BlobAddition blobAdditionEvent) {
return Mono.from(copyBlob(blobAdditionEvent.bucketName(), blobAdditionEvent.blobId(), successfulBlobStoreDAO, failedBlobStoreDAO));
}

if (blobEvent instanceof FailedBlobEvents.BlobsDeletion blobsDeletionEvent) {
return Mono.from(failedBlobStoreDAO.delete(blobsDeletionEvent.bucketName(), blobsDeletionEvent.blobIds()));
}

if (blobEvent instanceof FailedBlobEvents.BucketDeletion bucketDeletionEvent) {
return Mono.from(failedBlobStoreDAO.deleteBucket(bucketDeletionEvent.bucketName()));
}
return Mono.empty();
return switch (event) {
case FailedBlobEvents.BlobAddition blobAdditionEvent -> handleFailedBlobsAdditionEvent(blobAdditionEvent);
case FailedBlobEvents.BlobsDeletion blobsDeletionEvent -> handleFailedBlobsDeletionEvent(blobsDeletionEvent);
case FailedBlobEvents.BucketDeletion bucketDeletionEvent -> handleFailedBucketDeletionEvent(bucketDeletionEvent);
default -> Mono.empty();
};
}

@Override
Expand All @@ -61,25 +52,44 @@ public Group getDefaultGroup() {
return new FailedBlobOperationListenerGroup();
}

private Mono<Void> copyBlob(BucketName bucketName, BlobId blobId, BlobStoreDAO fromBlobStore, BlobStoreDAO toBlobStore) {
return Mono.from(fromBlobStore.readReactive(bucketName, blobId))
.flatMap(inputStream -> Mono.from(toBlobStore.save(bucketName, blobId, inputStream)))
private Mono<Void> handleFailedBlobsAdditionEvent(FailedBlobEvents.BlobAddition blobAdditionEvent) {
return switch (blobAdditionEvent.failedObjectStorage()) {
case PRIMARY -> readFromSecondaryAndSaveToPrimary(blobAdditionEvent.bucketName(), blobAdditionEvent.blobId());
case SECONDARY -> readFromPrimaryAndSaveToSecondary(blobAdditionEvent.bucketName(), blobAdditionEvent.blobId());
};
}

private Mono<Void> readFromSecondaryAndSaveToPrimary(BucketName bucketName, BlobId blobId) {
return Mono.from(secondaryBlobStoreDAO.readReactive(withSuffix(bucketName, secondaryBucketSuffix), blobId))
.flatMap(inputStream -> Mono.from(primaryBlobStoreDAO.save(bucketName, blobId, inputStream)))
.subscribeOn(Schedulers.boundedElastic());
}

private Mono<Void> readFromPrimaryAndSaveToSecondary(BucketName bucketName, BlobId blobId) {
return Mono.from(primaryBlobStoreDAO.readReactive(bucketName, blobId))
.flatMap(inputStream -> Mono.from(secondaryBlobStoreDAO.save(withSuffix(bucketName, secondaryBucketSuffix), blobId, inputStream)))
.subscribeOn(Schedulers.boundedElastic());
}

private BlobStoreDAO successfulBlobStoreDAO(ObjectStorageIdentity failedObjectStorage) {
if (SECONDARY.equals(failedObjectStorage)) {
return primaryBlobStoreDAO;
} else {
return secondaryBlobStoreDAO;
}
private Mono<Void> handleFailedBlobsDeletionEvent(FailedBlobEvents.BlobsDeletion blobsDeletionEvent) {
return switch (blobsDeletionEvent.failedObjectStorage()) {
case PRIMARY -> deleteBlobsFromPrimaryBucket(blobsDeletionEvent);
case SECONDARY -> deleteBlobsFromSecondaryBucket(blobsDeletionEvent);
};
}

private Mono<Void> deleteBlobsFromPrimaryBucket(FailedBlobEvents.BlobsDeletion blobsDeletionEvent) {
return Mono.from(primaryBlobStoreDAO.delete(blobsDeletionEvent.bucketName(), blobsDeletionEvent.blobIds()));
}

private Mono<Void> deleteBlobsFromSecondaryBucket(FailedBlobEvents.BlobsDeletion blobsDeletionEvent) {
return Mono.from(secondaryBlobStoreDAO.delete(withSuffix(blobsDeletionEvent.bucketName(), secondaryBucketSuffix), blobsDeletionEvent.blobIds()));
}

private BlobStoreDAO failedBlobStoreDAO(ObjectStorageIdentity failedObjectStorage) {
if (PRIMARY.equals(failedObjectStorage)) {
return primaryBlobStoreDAO;
} else {
return secondaryBlobStoreDAO;
}
private Publisher<Void> handleFailedBucketDeletionEvent(FailedBlobEvents.BucketDeletion bucketDeletionEvent) {
return switch (bucketDeletionEvent.failedObjectStorage()) {
case PRIMARY -> primaryBlobStoreDAO.deleteBucket(bucketDeletionEvent.bucketName());
case SECONDARY -> secondaryBlobStoreDAO.deleteBucket(withSuffix(bucketDeletionEvent.bucketName(), secondaryBucketSuffix));
};
}
}
25 changes: 16 additions & 9 deletions ...store/src/main/java/com/linagora/tmail/blob/secondaryblobstore/SecondaryBlobStoreDAO.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,10 @@ public long size() {
}
}

public static BucketName withSuffix(BucketName bucketName, String suffix) {
return BucketName.of(bucketName.asString() + suffix);
}

record SavingStatus(Optional<Throwable> e, ObjectStorageIdentity objectStorageIdentity) {
public static SavingStatus success(ObjectStorageIdentity objectStorageIdentity) {
return new SavingStatus(Optional.empty(), objectStorageIdentity);
Expand All @@ -82,13 +86,16 @@ public boolean isSuccess() {

private final BlobStoreDAO primaryBlobStoreDAO;
private final BlobStoreDAO secondaryBlobStoreDAO;
private final String secondaryBucketSuffix;
private final EventBus eventBus;

public SecondaryBlobStoreDAO(BlobStoreDAO primaryBlobStoreDAO,
BlobStoreDAO secondaryBlobStoreDAO,
String secondaryBucketSuffix,
EventBus eventBus) {
this.primaryBlobStoreDAO = primaryBlobStoreDAO;
this.secondaryBlobStoreDAO = secondaryBlobStoreDAO;
this.secondaryBucketSuffix = secondaryBucketSuffix;
this.eventBus = eventBus;
}

Expand All @@ -98,7 +105,7 @@ public InputStream read(BucketName bucketName, BlobId blobId) throws ObjectStore
return primaryBlobStoreDAO.read(bucketName, blobId);
} catch (Exception ex) {
LOGGER.warn("Fail to read from the first blob store with bucket name {} and blobId {}. Use second blob store", bucketName.asString(), blobId.asString(), ex);
return secondaryBlobStoreDAO.read(bucketName, blobId);
return secondaryBlobStoreDAO.read(withSuffix(bucketName, secondaryBucketSuffix), blobId);
}
}

Expand All @@ -107,7 +114,7 @@ public Mono<InputStream> readReactive(BucketName bucketName, BlobId blobId) {
return Mono.from(primaryBlobStoreDAO.readReactive(bucketName, blobId))
.onErrorResume(ex -> {
LOGGER.warn("Fail to read from the first blob store with bucket name {} and blobId {}. Use second blob store", bucketName.asString(), blobId.asString(), ex);
return Mono.from(secondaryBlobStoreDAO.readReactive(bucketName, blobId));
return Mono.from(secondaryBlobStoreDAO.readReactive(withSuffix(bucketName, secondaryBucketSuffix), blobId));
});
}

Expand All @@ -116,14 +123,14 @@ public Mono<byte[]> readBytes(BucketName bucketName, BlobId blobId) {
return Mono.from(primaryBlobStoreDAO.readBytes(bucketName, blobId))
.onErrorResume(ex -> {
LOGGER.warn("Fail to read from the first blob store with bucket name {} and blobId {}. Use second blob store", bucketName.asString(), blobId.asString(), ex);
return Mono.from(secondaryBlobStoreDAO.readBytes(bucketName, blobId));
return Mono.from(secondaryBlobStoreDAO.readBytes(withSuffix(bucketName, secondaryBucketSuffix), blobId));
});
}

@Override
public Mono<Void> save(BucketName bucketName, BlobId blobId, byte[] data) {
return Flux.merge(asSavingStatus(primaryBlobStoreDAO.save(bucketName, blobId, data), ObjectStorageIdentity.PRIMARY),
asSavingStatus(secondaryBlobStoreDAO.save(bucketName, blobId, data), ObjectStorageIdentity.SECONDARY))
asSavingStatus(secondaryBlobStoreDAO.save(withSuffix(bucketName, secondaryBucketSuffix), blobId, data), ObjectStorageIdentity.SECONDARY))
.collectList()
.flatMap(savingStatuses -> merge(savingStatuses,
failedObjectStorage -> eventBus.dispatch(new FailedBlobEvents.BlobAddition(Event.EventId.random(), bucketName, blobId, failedObjectStorage), NO_REGISTRATION_KEYS)));
Expand All @@ -136,7 +143,7 @@ public Mono<Void> save(BucketName bucketName, BlobId blobId, InputStream inputSt
fileBackedOutputStream -> Mono.fromCallable(() -> IOUtils.copy(inputStream, fileBackedOutputStream))
.flatMap(size -> Flux.merge(
asSavingStatus(primaryBlobStoreDAO.save(bucketName, blobId, new FileBackedOutputStreamByteSource(fileBackedOutputStream, size)), ObjectStorageIdentity.PRIMARY),
asSavingStatus(secondaryBlobStoreDAO.save(bucketName, blobId, new FileBackedOutputStreamByteSource(fileBackedOutputStream, size)), ObjectStorageIdentity.SECONDARY))
asSavingStatus(secondaryBlobStoreDAO.save(withSuffix(bucketName, secondaryBucketSuffix), blobId, new FileBackedOutputStreamByteSource(fileBackedOutputStream, size)), ObjectStorageIdentity.SECONDARY))
.collectList()
.flatMap(savingStatuses -> merge(savingStatuses,
failedObjectStorage -> eventBus.dispatch(
Expand All @@ -149,7 +156,7 @@ public Mono<Void> save(BucketName bucketName, BlobId blobId, InputStream inputSt
@Override
public Mono<Void> save(BucketName bucketName, BlobId blobId, ByteSource content) {
return Flux.merge(asSavingStatus(primaryBlobStoreDAO.save(bucketName, blobId, content), ObjectStorageIdentity.PRIMARY),
asSavingStatus(secondaryBlobStoreDAO.save(bucketName, blobId, content), ObjectStorageIdentity.SECONDARY))
asSavingStatus(secondaryBlobStoreDAO.save(withSuffix(bucketName, secondaryBucketSuffix), blobId, content), ObjectStorageIdentity.SECONDARY))
.collectList()
.flatMap(savingStatuses -> merge(savingStatuses,
failedObjectStorage -> eventBus.dispatch(new FailedBlobEvents.BlobAddition(Event.EventId.random(), bucketName, blobId, failedObjectStorage), NO_REGISTRATION_KEYS)));
Expand All @@ -158,7 +165,7 @@ public Mono<Void> save(BucketName bucketName, BlobId blobId, ByteSource content)
@Override
public Mono<Void> delete(BucketName bucketName, BlobId blobId) {
return Flux.merge(asSavingStatus(primaryBlobStoreDAO.delete(bucketName, blobId), ObjectStorageIdentity.PRIMARY),
asSavingStatus(secondaryBlobStoreDAO.delete(bucketName, blobId), ObjectStorageIdentity.SECONDARY))
asSavingStatus(secondaryBlobStoreDAO.delete(withSuffix(bucketName, secondaryBucketSuffix), blobId), ObjectStorageIdentity.SECONDARY))
.collectList()
.flatMap(savingStatuses -> merge(savingStatuses,
failedObjectStorage -> eventBus.dispatch(new FailedBlobEvents.BlobsDeletion(Event.EventId.random(), bucketName, ImmutableList.of(blobId), failedObjectStorage), NO_REGISTRATION_KEYS)));
Expand All @@ -167,7 +174,7 @@ public Mono<Void> delete(BucketName bucketName, BlobId blobId) {
@Override
public Mono<Void> delete(BucketName bucketName, Collection<BlobId> blobIds) {
return Flux.merge(asSavingStatus(primaryBlobStoreDAO.delete(bucketName, blobIds), ObjectStorageIdentity.PRIMARY),
asSavingStatus(secondaryBlobStoreDAO.delete(bucketName, blobIds), ObjectStorageIdentity.SECONDARY))
asSavingStatus(secondaryBlobStoreDAO.delete(withSuffix(bucketName, secondaryBucketSuffix), blobIds), ObjectStorageIdentity.SECONDARY))
.collectList()
.flatMap(savingStatuses -> merge(savingStatuses,
failedObjectStorage -> eventBus.dispatch(new FailedBlobEvents.BlobsDeletion(Event.EventId.random(), bucketName, blobIds, failedObjectStorage), NO_REGISTRATION_KEYS)));
Expand All @@ -176,7 +183,7 @@ public Mono<Void> delete(BucketName bucketName, Collection<BlobId> blobIds) {
@Override
public Mono<Void> deleteBucket(BucketName bucketName) {
return Flux.merge(asSavingStatus(primaryBlobStoreDAO.deleteBucket(bucketName), ObjectStorageIdentity.PRIMARY),
asSavingStatus(secondaryBlobStoreDAO.deleteBucket(bucketName), ObjectStorageIdentity.SECONDARY))
asSavingStatus(secondaryBlobStoreDAO.deleteBucket(withSuffix(bucketName, secondaryBucketSuffix)), ObjectStorageIdentity.SECONDARY))
.collectList()
.flatMap(savingStatuses -> merge(savingStatuses,
failedObjectStorage -> eventBus.dispatch(new FailedBlobEvents.BucketDeletion(Event.EventId.random(), bucketName, failedObjectStorage), NO_REGISTRATION_KEYS)));
Expand Down
Loading

0 comments on commit e73bc02

Please sign in to comment.