aboutsummaryrefslogtreecommitdiffstats
path: root/storage/src/vespa/storage/persistence/asynchandler.h
blob: 4f5c242570cf5ad8a8eced1c9928f35c6c14b599 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once

#include "types.h"
#include "messages.h"
#include <vespa/storageapi/message/persistence.h>

namespace document { class BucketIdFactory; }
namespace vespalib { class ISequencedTaskExecutor; }
namespace storage {

namespace spi {
    struct PersistenceProvider;
    class Context;
}
class PersistenceUtil;
class BucketOwnershipNotifier;

/**
 * Handle async operations that uses a sequenced executor.
 * It is stateless and thread safe.
 */
class AsyncHandler : public Types {
public:
    AsyncHandler(const PersistenceUtil&, spi::PersistenceProvider&, BucketOwnershipNotifier  &,
                 vespalib::ISequencedTaskExecutor & executor, const document::BucketIdFactory & bucketIdFactory);
    MessageTrackerUP handlePut(api::PutCommand& cmd, MessageTrackerUP tracker) const;
    MessageTrackerUP handleRemove(api::RemoveCommand& cmd, MessageTrackerUP tracker) const;
    MessageTrackerUP handleUpdate(api::UpdateCommand& cmd, MessageTrackerUP tracker) const;
    MessageTrackerUP handleRunTask(RunTaskCommand & cmd, MessageTrackerUP tracker) const;
    MessageTrackerUP handleSetBucketState(api::SetBucketStateCommand& cmd, MessageTrackerUP tracker) const;
    MessageTrackerUP handleDeleteBucket(api::DeleteBucketCommand& cmd, MessageTrackerUP tracker) const;
    static bool is_async_message(api::MessageType::Id type_id) noexcept;
private:
    bool checkProviderBucketInfoMatches(const spi::Bucket&, const api::BucketInfo&) const;
    static bool tasConditionExists(const api::TestAndSetCommand & cmd);
    bool tasConditionMatches(const api::TestAndSetCommand & cmd, MessageTracker & tracker,
                             spi::Context & context, bool missingDocumentImpliesMatch = false) const;
    const PersistenceUtil            & _env;
    spi::PersistenceProvider         & _spi;
    BucketOwnershipNotifier          & _bucketOwnershipNotifier;
    vespalib::ISequencedTaskExecutor & _sequencedExecutor;
    const document::BucketIdFactory  & _bucketIdFactory;
};

} // storage