More work on allocation.

1 files changed, 6 insertions, 2 deletions

diff --git a/cpp/src/IceGrid/AdapterCache.h b/cpp/src/IceGrid/AdapterCache.h
index c6568819785..82d5397289a 100644
--- a/cpp/src/IceGrid/AdapterCache.h
+++ b/cpp/src/IceGrid/AdapterCache.h
@@ -15,6 +15,7 @@
 #include <IceGrid/Cache.h>
 #include <IceGrid/Query.h>
 #include <IceGrid/Internal.h>
+#include <IceGrid/Allocatable.h>
 
 namespace IceGrid
 {
@@ -28,7 +29,7 @@ typedef std::vector<ServerEntryPtr> ServerEntrySeq;
 class AdapterEntry;
 typedef IceUtil::Handle<AdapterEntry> AdapterEntryPtr;
 
-class AdapterEntry : public IceUtil::Shared, public IceUtil::Mutex
+class AdapterEntry : public Allocatable, public IceUtil::Mutex
 {
 public:
     
@@ -57,11 +58,14 @@ public:
     virtual float getLeastLoadedNodeLoad(LoadSample) const;
     virtual std::string getApplication() const;
 
-    void set(const ServerEntryPtr&, const std::string&);
+    void set(const ServerEntryPtr&, const std::string&, bool);
     void destroy();
 
     AdapterPrx getProxy(const std::string& = std::string()) const;
 
+    virtual void allocated(const SessionIPtr&);
+    virtual void released(const SessionIPtr&);
+
 private:
     
     ServerEntryPtr getServer() const;