blob: c0ebe0e717ce14fb41376d1131121e0f2cc5d088 (
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include <vespa/documentapi/common.h>
#include <vespa/messagebus/routing/hop.h>
#include <vespa/messagebus/routing/iroutingpolicy.h>
#include <map>
#include <mutex>
namespace documentapi {
/**
* This policy implements the logic to select a subset of services that matches a slobrok pattern.
*
* @author Simon Thoresen Hult
* @version $Id$
*/
class SubsetServicePolicy : public mbus::IRoutingPolicy {
private:
struct CacheEntry {
uint32_t _offset;
uint32_t _generation;
std::vector<mbus::Hop> _recipients;
CacheEntry();
};
std::mutex _lock;
uint32_t _subsetSize;
std::map<string, CacheEntry> _cache;
/**
* Returns the appropriate recipient hop for the given routing context. This method provides synchronized access to
* the internal cache.
*
* @param ctx The routing context.
* @return The recipient hop to use.
*/
mbus::Hop getRecipient(mbus::RoutingContext &ctx);
/**
* Updates and returns the cache entry for the given routing context. This method assumes that synchronization is
* handled outside of it.
*
* @param ctx The routing context.
* @return The updated cache entry.
*/
CacheEntry &update(mbus::RoutingContext &ctx);
/**
* Returns a cache key for this instance of the policy. Because behaviour is based on the hop in which the policy
* occurs, the cache key is the hop string itself.
*
* @param ctx The routing context.
* @return The cache key.
*/
string getCacheKey(const mbus::RoutingContext &ctx) const;
public:
/**
* Creates an instance of a subset service policy. The parameter string is parsed as an integer number that is the
* number of services to include in the set to choose from.
*
* @param param The number of services to include in the set.
*/
SubsetServicePolicy(const string ¶m);
~SubsetServicePolicy();
void select(mbus::RoutingContext &context) override;
void merge(mbus::RoutingContext &context) override;
};
}
|