blob: db9704430846332a493a8dbf95429abd72a327d2 (
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
73
74
75
76
77
78
79
80
81
82
83
|
// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
#include <zencore/compactbinary.h>
#include <zencore/logbase.h>
#include <zencore/thread.h>
#include <zencore/uid.h>
ZEN_THIRD_PARTY_INCLUDES_START
#include <tsl/robin_map.h>
ZEN_THIRD_PARTY_INCLUDES_END
#include <optional>
#include <string>
#include <vector>
namespace zen {
class SessionLogStore;
class SessionLog;
/** Session tracker
*
* Acts as a log and session info concentrator when dealing with multiple
* servers and external processes acting as a group.
*/
class SessionsService
{
public:
struct SessionInfo
{
Oid Id;
std::string AppName;
Oid JobId;
CbObject Metadata;
DateTime CreatedAt;
DateTime UpdatedAt;
};
class Session : public TRefCounted<Session>
{
public:
Session(const SessionInfo& Info);
~Session();
Session(Session&&) = delete;
Session& operator=(Session&&) = delete;
const SessionInfo& Info() const { return m_Info; }
void UpdateMetadata(CbObjectView Metadata)
{
// Should this be additive rather than replacing the whole thing? We'll see.
m_Info.Metadata = CbObject::Clone(Metadata);
m_Info.UpdatedAt = DateTime::Now();
}
private:
SessionInfo m_Info;
Ref<SessionLog> m_Log;
};
SessionsService();
~SessionsService();
bool RegisterSession(const Oid& SessionId, std::string AppName, const Oid& JobId, CbObjectView Metadata);
bool UpdateSession(const Oid& SessionId, CbObjectView Metadata);
Ref<Session> GetSession(const Oid& SessionId) const;
std::vector<Ref<Session>> GetSessions() const;
bool RemoveSession(const Oid& SessionId);
uint64_t GetSessionCount() const;
private:
LoggerRef& Log() { return m_Log; }
LoggerRef m_Log;
mutable RwLock m_Lock;
tsl::robin_map<Oid, Ref<Session>, Oid::Hasher> m_Sessions;
std::unique_ptr<SessionLogStore> m_SessionLogs;
};
} // namespace zen
|