Search in sources :

Example 1 with LockCountNotMatchException

use of dvoraka.avservice.storage.LockCountNotMatchException in project av-service by dvoraka.

the class DefaultReplicationService method saveFile.

@Override
public void saveFile(FileMessage message) throws FileServiceException {
    log.debug("Save {}: {}", idString, message);
    // depends on an efficiency of the sending algorithm and
    // it still must be different for "bigger" (~10 MB+) files
    final int sizeTimeRatio = 2_000;
    final int maxSaveTime = message.getData().length / sizeTimeRatio;
    log.debug("Setting max save time to {} {}", maxSaveTime, idString);
    if (localCopyExists(message)) {
        throw new ExistingFileException();
    }
    int neighbours = neighbourCount();
    try {
        if (remoteLock.lockForFile(message.getFilename(), message.getOwner(), neighbours)) {
            if (!exists(message)) {
                log.debug("Saving locally {}...", idString);
                fileService.saveFile(message);
                log.debug("Saving remotely {}...", idString);
                sendSaveMessage(message);
                Optional<ReplicationMessageList> responses = responseClient.getResponseWaitSize(message.getId(), MAX_RESPONSE_TIME + maxSaveTime, getReplicationCount() - 1);
                ReplicationMessageList messages = responses.orElseGet(ReplicationMessageList::new);
                long successCount = messages.stream().filter(msg -> msg.getCommand() == Command.SAVE).filter(msg -> msg.getReplicationStatus() == ReplicationStatus.OK).count();
                if (successCount != getReplicationCount() - 1) {
                    // rollback transaction
                    log.debug("Expected {}, got {} {}", getReplicationCount() - 1, successCount, idString);
                    throw new LockCountNotMatchException();
                }
                log.debug("Save success {}.", idString);
            } else {
                throw new ExistingFileException();
            }
        } else {
            log.warn("Save lock problem for {}: {}", idString, message);
            throw new CannotAcquireLockException();
        }
    } catch (InterruptedException e) {
        log.warn("Locking interrupted!", e);
        Thread.currentThread().interrupt();
    } finally {
        remoteLock.unlockForFile(message.getFilename(), message.getOwner(), neighbours);
    }
}
Also used : ReplicationMessage(dvoraka.avservice.common.data.ReplicationMessage) ReplicationHelper(dvoraka.avservice.common.replication.ReplicationHelper) ExistingFileException(dvoraka.avservice.storage.ExistingFileException) ReplicationStatus(dvoraka.avservice.common.data.ReplicationStatus) Autowired(org.springframework.beans.factory.annotation.Autowired) MessageRouting(dvoraka.avservice.common.data.MessageRouting) PreDestroy(javax.annotation.PreDestroy) ReplicationResponseClient(dvoraka.avservice.client.service.response.ReplicationResponseClient) FileService(dvoraka.avservice.storage.service.FileService) Service(org.springframework.stereotype.Service) Objects.requireNonNull(java.util.Objects.requireNonNull) ScheduledExecutorService(java.util.concurrent.ScheduledExecutorService) FileServiceException(dvoraka.avservice.storage.FileServiceException) LockCountNotMatchException(dvoraka.avservice.storage.LockCountNotMatchException) ReplicationServiceClient(dvoraka.avservice.client.service.ReplicationServiceClient) CannotAcquireLockException(dvoraka.avservice.storage.CannotAcquireLockException) ReplicationMessageList(dvoraka.avservice.client.service.response.ReplicationMessageList) Command(dvoraka.avservice.common.data.Command) Set(java.util.Set) CopyOnWriteArraySet(java.util.concurrent.CopyOnWriteArraySet) Collectors(java.util.stream.Collectors) Executors(java.util.concurrent.Executors) FileMessage(dvoraka.avservice.common.data.FileMessage) TimeUnit(java.util.concurrent.TimeUnit) FileNotFoundException(dvoraka.avservice.storage.FileNotFoundException) Logger(org.apache.logging.log4j.Logger) PostConstruct(javax.annotation.PostConstruct) Optional(java.util.Optional) LogManager(org.apache.logging.log4j.LogManager) ExistingFileException(dvoraka.avservice.storage.ExistingFileException) CannotAcquireLockException(dvoraka.avservice.storage.CannotAcquireLockException) ReplicationMessageList(dvoraka.avservice.client.service.response.ReplicationMessageList) LockCountNotMatchException(dvoraka.avservice.storage.LockCountNotMatchException)

Aggregations

ReplicationServiceClient (dvoraka.avservice.client.service.ReplicationServiceClient)1 ReplicationMessageList (dvoraka.avservice.client.service.response.ReplicationMessageList)1 ReplicationResponseClient (dvoraka.avservice.client.service.response.ReplicationResponseClient)1 Command (dvoraka.avservice.common.data.Command)1 FileMessage (dvoraka.avservice.common.data.FileMessage)1 MessageRouting (dvoraka.avservice.common.data.MessageRouting)1 ReplicationMessage (dvoraka.avservice.common.data.ReplicationMessage)1 ReplicationStatus (dvoraka.avservice.common.data.ReplicationStatus)1 ReplicationHelper (dvoraka.avservice.common.replication.ReplicationHelper)1 CannotAcquireLockException (dvoraka.avservice.storage.CannotAcquireLockException)1 ExistingFileException (dvoraka.avservice.storage.ExistingFileException)1 FileNotFoundException (dvoraka.avservice.storage.FileNotFoundException)1 FileServiceException (dvoraka.avservice.storage.FileServiceException)1 LockCountNotMatchException (dvoraka.avservice.storage.LockCountNotMatchException)1 FileService (dvoraka.avservice.storage.service.FileService)1 Objects.requireNonNull (java.util.Objects.requireNonNull)1 Optional (java.util.Optional)1 Set (java.util.Set)1 CopyOnWriteArraySet (java.util.concurrent.CopyOnWriteArraySet)1 Executors (java.util.concurrent.Executors)1