EvictorBase.cpp
2.41 KB
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
#include <EvictorBase.h>
using namespace std;
EvictorBase::EvictorBase(int size) :
_size(size)
{
if (_size < 0)
{
_size = 1000;
}
}
shared_ptr<Ice::Object>
EvictorBase::locate(const Ice::Current& c, shared_ptr<void>& cookie)
{
lock_guard<mutex> lk(_mutex);
//
// Check if we have a servant in the map already.
//
shared_ptr<EvictorEntry> entry;
auto i = _map.find(c.id);
if(i != _map.end())
{
//
// Got an entry already, dequeue the entry from its current position.
//
entry = i->second;
_queue.erase(entry->queuePos);
}
else
{
//
// We do not have an entry. Ask the derived class to
// instantiate a servant and add a new entry to the map.
//
entry = make_shared<EvictorEntry>();
entry->servant = add(c, entry->userCookie); // Down-call
if(!entry->servant)
{
return 0;
}
entry->useCount = 0;
i = _map.insert(make_pair(c.id, entry)).first;
}
//
// Increment the use count of the servant and enqueue
// the entry at the front, so we get LRU order.
//
++(entry->useCount);
entry->queuePos = _queue.insert(_queue.begin(), i);
cookie = entry;
return entry->servant;
}
void
EvictorBase::finished(const Ice::Current&, const shared_ptr<Ice::Object>&, const shared_ptr<void>& cookie)
{
lock_guard<mutex> lk(_mutex);
auto entry = static_pointer_cast<EvictorEntry>(cookie);
//
// Decrement use count and check if there is something to evict.
//
--(entry->useCount);
evictServants();
}
void
EvictorBase::deactivate(const string&)
{
lock_guard<mutex> lk(_mutex);
_size = 0;
evictServants();
}
void
EvictorBase::evictServants()
{
//
// If the evictor queue has grown larger than the limit,
// look at the excess elements to see whether any of them
// can be evicted.
//
auto p = _queue.rbegin();
auto excessEntries = static_cast<int>(_map.size() - _size);
for(int i = 0; i < excessEntries; ++i)
{
auto mapPos = *p;
if(mapPos->second->useCount == 0)
{
evict(mapPos->second->servant, mapPos->second->userCookie); // Down-call
p = EvictorQueue::reverse_iterator(_queue.erase(mapPos->second->queuePos));
_map.erase(mapPos);
}
else
{
++p;
}
}
}