summaryrefslogtreecommitdiff
path: root/cpp/src/IceGrid/SessionServantManager.h
blob: 817f4c117149ea5f5d6fe5b8ffe1f99af3ab8cab (plain)
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
//
// Copyright (c) ZeroC, Inc. All rights reserved.
//

#ifndef ICE_GRID_SESSIONSERVANTLOCATOR_H
#define ICE_GRID_SESSIONSERVANTLOCATOR_H

#include <IceGrid/AdminCallbackRouter.h>
#include <Ice/ServantLocator.h>

#include <Glacier2/Session.h>

#include <set>

namespace IceGrid
{

class SessionServantManager final : public Ice::ServantLocator
{
public:

    SessionServantManager(const std::shared_ptr<Ice::ObjectAdapter>&, const std::string&, bool,
                          const std::string&, const std::shared_ptr<Ice::Object>&,
                          const std::string&, const std::shared_ptr<Ice::Object>&,
                          const std::string&, const std::shared_ptr<Ice::Object>&,
                          const std::shared_ptr<AdminCallbackRouter>&);

    std::shared_ptr<Ice::Object> locate(const Ice::Current&, std::shared_ptr<void>&);
    void finished(const Ice::Current&, const std::shared_ptr<Ice::Object>&, const std::shared_ptr<void>&);
    void deactivate(const std::string&);

    std::shared_ptr<Ice::ObjectPrx> addSession(const std::shared_ptr<Ice::Object>&,
                                               const std::shared_ptr<Ice::Connection>&,
                                               const std::string&);
    void setSessionControl(const std::shared_ptr<Ice::Object>&,
                           const std::shared_ptr<Glacier2::SessionControlPrx>&, const Ice::IdentitySeq&);
    std::shared_ptr<Glacier2::IdentitySetPrx> getGlacier2IdentitySet(const std::shared_ptr<Ice::Object>&);
    std::shared_ptr<Glacier2::StringSetPrx> getGlacier2AdapterIdSet(const std::shared_ptr<Ice::Object>&);
    void removeSession(const std::shared_ptr<Ice::Object>&);

    std::shared_ptr<Ice::ObjectPrx> add(const std::shared_ptr<Ice::Object>&, const std::shared_ptr<Ice::Object>&);
    void remove(const Ice::Identity&);

    void connectionHeartbeat(const std::shared_ptr<Ice::Connection>&);
    void connectionClosed(const std::shared_ptr<Ice::Connection>&);

private:

    std::shared_ptr<Ice::ObjectPrx> addImpl(const std::shared_ptr<Ice::Object>&, const std::shared_ptr<Ice::Object>&);

    struct ServantInfo
    {
        ServantInfo(const std::shared_ptr<Ice::Object>& s,
                    const std::shared_ptr<Ice::Connection>& con,
                    const std::shared_ptr<Ice::Object>& ss) :
            servant(s), connection(con), session(ss)
        {
        }

        const std::shared_ptr<Ice::Object> servant;
        const std::shared_ptr<Ice::Connection> connection;
        const std::shared_ptr<Ice::Object> session;
    };

    struct SessionInfo
    {
        SessionInfo(const std::shared_ptr<Ice::Connection>& c, const std::string& cat) : connection(c), category(cat)
        {
        }

        const std::shared_ptr<Ice::Connection> connection;
        const std::string category;
        std::shared_ptr<Glacier2::SessionControlPrx> sessionControl;
        std::shared_ptr<Glacier2::IdentitySetPrx> identitySet;
        std::shared_ptr<Glacier2::StringSetPrx> adapterIdSet;
        std::set<Ice::Identity> identities;
    };

    const std::shared_ptr<Ice::ObjectAdapter> _adapter;
    const std::string _instanceName;
    const bool _checkConnection;
    const std::string _serverAdminCategory;
    const std::shared_ptr<Ice::Object> _serverAdminRouter;
    const std::string _nodeAdminCategory;
    const std::shared_ptr<Ice::Object> _nodeAdminRouter;
    const std::string _replicaAdminCategory;
    const std::shared_ptr<Ice::Object> _replicaAdminRouter;
    const std::shared_ptr<AdminCallbackRouter> _adminCallbackRouter;

    std::map<Ice::Identity, ServantInfo> _servants;
    std::map<std::shared_ptr<Ice::Object>, SessionInfo> _sessions;
    std::multiset<std::shared_ptr<Ice::Connection>> _adminConnections;

    std::mutex _mutex;

};

};

#endif