searchcore/src/vespa/searchcore/proton/test/mock_shared_threading_service.h


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

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once

#include "transport_helper.h"
#include <vespa/persistence/dummyimpl/dummy_bucket_executor.h>
#include <vespa/searchcore/proton/server/i_shared_threading_service.h>
#include <vespa/vespalib/util/clock.h>
#include <vespa/vespalib/util/invokeserviceimpl.h>

namespace proton {

class MockSharedThreadingService : public ISharedThreadingService {
private:
    using ThreadExecutor = vespalib::ThreadExecutor;
    ThreadExecutor              & _warmup;
    ThreadExecutor              & _shared;
    std::unique_ptr<vespalib::ISequencedTaskExecutor> _field_writer;
    vespalib::InvokeServiceImpl   _invokeService;
    Transport                     _transport;
    storage::spi::dummy::DummyBucketExecutor _bucket_executor;
    vespalib::Clock               _clock;
public:
    MockSharedThreadingService(ThreadExecutor& warmup_in,
                               ThreadExecutor& shared_in,
                               size_t num_bucket_executors = 2);
    ~MockSharedThreadingService() override;
    ThreadExecutor& warmup() override { return _warmup; }
    ThreadExecutor& shared() override { return _shared; }
    vespalib::ISequencedTaskExecutor& field_writer() override { return *_field_writer; }
    vespalib::InvokeService & invokeService() override { return _invokeService; }
    FNET_Transport & transport() override { return _transport.transport(); }
    storage::spi::BucketExecutor& bucket_executor() override { return _bucket_executor; }
    const vespalib::Clock & clock() const override { return _clock; }
};

}